zhangyiss/mlx-examples
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx-examples.git synced 2025-10-31 02:48:07 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity
Files
3cf436b529ea58d6c0c0a29c0dd799908cd4497d
mlx-examples/transformer_lm
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
..
datasets.py
Add llms subdir + update README (#145)
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
README.md
Update README.md
2023-12-06 00:30:37 -08:00
tf_main.py
Add llms subdir + update README (#145)
2023-12-20 10:22:25 -08:00
torch_main.py
Add llms subdir + update README (#145)
2023-12-20 10:22:25 -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 TensorFlow examples install the respective framework.