Ahmad-Momani/get-started-with-JAX

The purpose of this repo is to make it easy to get started with JAX, Flax, and Haiku.

Jupyter NotebookMIT