# 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.