mlx-examples/transformer_lm/README.md
2023-11-29 08:17:26 -08:00

15 lines
397 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 TensorFlowexamples install the respective framework.