mirror of
https://github.com/ml-explore/mlx-examples.git
synced 2025-06-26 18:51:18 +08:00
15 lines
398 B
Markdown
15 lines
398 B
Markdown
# 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](https://paperswithcode.com/dataset/penn-treebank). Choose a different dataset with the `--dataset` option.
|
|
|
|
To run the PyTorch, Jax or TensorFlow examples install the respective framework.
|