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 the content I found useful while learning about the JAX ecosystem.

GitHub
630 stars
8 watching
97 forks
Language: Jupyter Notebook
last commit: 10 months ago
Linked from 1 awesome list

deep-learningflaxhaikujaxjupyterlaxlearn-jaxmachine-learningnumpyoptaxpythontutorialxla

Screenshot of gordicaleksa/get-started-with-JAX website

https://www.youtube.com/c/TheAIEpiphany/

Backlinks from these awesome lists: