27c0a8c002
* add llms subdir + update README * nits * use same pre-commit as mlx * update readmes a bit * format |
||
|---|---|---|
| .. | ||
| datasets.py | ||
| jax_main.py | ||
| main.py | ||
| README.md | ||
| tf_main.py | ||
| torch_main.py | ||
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.