This repo contains code for the Nova Science coding group.
This online book is quickly becoming the standard text for Deep Learning.
And here is repo with list of projects in JAX.
The mnist.ipynb notebook builds a simple MNIST classifier in JAX. Here is a Google Colab version. And for a classification over 99.5% please consult the Jiucheng branch.
The rng.ipynb notebook explains Pseudo Random Number Generation in JAX. Here is a Google Colab version.
The autodiff.ipynb looks of examples of how autodiff is used in JAX. Here is a Google Colab version.