zhangyiss/mlx-examples
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx-examples.git synced 2025-09-01 04:14:38 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Add grad checkpointing and PE in the transformer example (#387)

Browse Source 
* 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
This commit is contained in:
Angelos Katharopoulos
2024-02-01 13:04:03 -08:00
committed by GitHub
parent ec14583c2a
commit e9b32747b4
8 changed files with 36 additions and 946 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
transformer_lm/requirements.txt Normal file
View File

@@ -0,0 +1 @@
mlx >= 0.12