/JAX-DNN-MNIST

A demonstration of NN training in JAX (via STAX).

Primary LanguageJupyter Notebook

JAX-DNN-MNIST

Open In Colab