/JAX-DNN-MNIST

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

Primary LanguageJupyter Notebook

Stargazers

No one’s star this repository yet.