mirror of
https://github.com/ml-explore/mlx-examples.git
synced 2025-10-31 10:58:07 +08:00
f45a1ab83c
* update a few examples to use compile * update mnist * add compile to vae and rename some stuff for simplicity * update reqs * use state in eval * GCN example with RNG + dropout * add a bit of prefetching
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.