zhangyiss/mlx-examples
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx-examples.git synced 2025-07-24 21:21:16 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity
Examples in the MLX framework
mlx
2 Commits 10 Branches 0 Tags 80 MiB
Python 82.3%
Jupyter Notebook 17.3%
Swift 0.4%
b243c1d8f4
Go to file
Awni Hannun b243c1d8f4 a few examples
2023-11-29 08:17:26 -08:00
mnist a few examples 2023-11-29 08:17:26 -08:00
transformer_lm a few examples 2023-11-29 08:17:26 -08:00
whisper a few examples 2023-11-29 08:17:26 -08:00
.gitignore a few examples 2023-11-29 08:17:26 -08:00
.pre-commit-config.yaml a few examples 2023-11-29 08:17:26 -08:00
LICENSE Initial commit 2023-11-28 15:37:50 -08:00
README.md a few examples 2023-11-29 08:17:26 -08:00

README.md

Transformer LM

This is an example of a decoder-only Transformer LM. The only dependency is MLX.

Run the example on the GPU with:

python main.py --gpu

By default the dataset is the PTB corpus. Choose a different dataset with the --dataset option.

To run the PyTorch, Jax or TensorFlowexamples install the respective framework.