1 result found Sort:

gordicaleksa
get-started-with-JAX gordicaleksa
97
630
mit
8
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...
jax lax xla flax haiku numpy optax python jupyter tutorial learn-jax deep-learning machine-learning
Created 2021-10-28
33 commits to main branch, last one 2 years ago