zhangyiss/mlx-examples
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx-examples.git synced 2025-12-16 02:08:55 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity
Files
7fbca214b142dbedb73d696329f154d6683e23fb
mlx-examples/transformer_lm/requirements.txt

2 lines
12 B
Plaintext
Raw Normal View History

Add grad checkpointing and PE in the transformer example (#387) * Add grad checkpointing and PE in the transformer example * Remove other frameworks from LM example * Remove the other frameworks from MNIST example * Improve the transformer LM example * Fix black and change LR
2024-02-01 13:04:03 -08:00
mlx >= 0.12
Reference in New Issue Copy Permalink