jax topic
ivy
The Unified AI Framework
gym
Fully differentiable RL environments, written in Ivy.
mech
Mechanics functions with end-to-end support for deep learning developers, written in Ivy.
vision
3D Vision functions with end-to-end support for deep learning developers, written in Ivy.
annotated-s4
Implementation of https://srush.github.io/annotated-s4
jax-gcn
Graph Convolutional Networks in JAX
big_transfer
Official repository for the "Big Transfer (BiT): General Visual Representation Learning" paper.
einops
Flexible and powerful tensor operations for readable and reliable code (for pytorch, jax, TF and others)
pyprobml
Python code for "Probabilistic Machine learning" book by Kevin Murphy
get-started-with-JAX
The purpose of this repo is to make it easy to get started with JAX, Flax, and Haiku. It contains my "Machine Learning with JAX" series of tutorials (YouTube videos and Jupyter Notebooks) as well as t...