learn-jax topic
List
learn-jax repositories
get-started-with-JAX
578
Stars
92
Forks
Watchers
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...
