MNIST

This example shows how to run some simple models on MNIST.

Install the dependencies:

pip install -r requirements.txt

Run the example with:

python main.py

By default, the example runs on the CPU. To run on the GPU, use:

python main.py --gpu

For a full list of options run:

python main.py --help

To run the PyTorch or JAX examples install the respective framework.