atsourav / mnist_jax_w_hparams_gui Goto Github PK
View Code? Open in Web Editor NEWMLP classifier on the MNIST dataset implemented in JAX with a GUI for entering hyperparameters, and a custom visualization of runs on TensorBoard.
License: MIT License