zhangyiss/mlx-examples
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx-examples.git synced 2025-06-24 09:21:18 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity
848f118ac5
mlx-examples/mnist
History
Awni Hannun 27c0a8c002
Add llms subdir + update README (#145) 
* add llms subdir + update README

* nits

* use same pre-commit as mlx

* update readmes a bit

* format
2023-12-20 10:22:25 -08:00
..
jax_main.py Add llms subdir + update README (#145) 2023-12-20 10:22:25 -08:00
main.py Add llms subdir + update README (#145) 2023-12-20 10:22:25 -08:00
mnist.py Add llms subdir + update README (#145) 2023-12-20 10:22:25 -08:00
README.md Adding Requirements.txt 2023-12-11 20:45:39 -06:00
requirements.txt Adding Requirements.txt 2023-12-11 20:45:39 -06:00
torch_main.py Add llms subdir + update README (#145) 2023-12-20 10:22:25 -08:00

README.md

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

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