AtSourav/mnist_jax_w_hparams_GUI

MLP classifier on the MNIST dataset implemented in JAX with a GUI for entering hyperparameters, and a custom visualization of runs on TensorBoard.

PythonMIT