mirror of
https://github.com/ml-explore/mlx-examples.git
synced 2025-06-25 09:51:19 +08:00
398 B
398 B
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.