JAX Tutorials for PyTorchers

Name		Name	Last commit message	Last commit date
Latest commit History 14 Commits
LICENSE		LICENSE
README.md		README.md
flax-dropout-batchnorm-example.ipynb		flax-dropout-batchnorm-example.ipynb
flax-mlp-mnist-example.ipynb		flax-mlp-mnist-example.ipynb
jax-numpy-mlp-mnist-example.ipynb		jax-numpy-mlp-mnist-example.ipynb
pmap-mlp-mnist-example.ipynb		pmap-mlp-mnist-example.ipynb

Provide feedback