mirror of https://github.com/ml-explore/mlx-examples.git synced 2025-09-23 23:28:12 +08:00

Files

Angelos Katharopoulos e9b32747b4 Add grad checkpointing and PE in the transformer example (#387 )

* Add grad checkpointing and PE in the transformer example

* Remove other frameworks from LM example

* Remove the other frameworks from MNIST example

* Improve the transformer LM example

* Fix black and change LR

2024-02-01 13:04:03 -08:00

main.py

Refactor activation function and loss calculation (#325 )

2024-01-16 13:42:56 -08:00

mnist.py

fashion-mnist example (#180 )

2023-12-23 07:34:45 -08:00

README.md

updated README (#184 )

2023-12-24 06:19:53 -08:00

requirements.txt

Adding Requirements.txt

2023-12-11 20:45:39 -06: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

For a full list of options run:

python main.py --help

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