mlx-examples/transformer_lm/README.md

15 lines
398 B
Markdown
Raw Normal View History

2023-11-30 00:17:26 +08:00
# 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.
2023-12-06 16:30:37 +08:00
To run the PyTorch, Jax or TensorFlow examples install the respective framework.