mlx-examples project
| mnist | ||
| transformer_lm | ||
| whisper | ||
| .gitignore | ||
| .pre-commit-config.yaml | ||
| LICENSE | ||
| 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.