mlx-examples

zhangyiss/mlx-examples

Fork 0

mirror of https://github.com/ml-explore/mlx-examples.git synced 2025-07-31 03:11:13 +08:00

Commit Graph

Author	SHA1	Message	Date
Angelos Katharopoulos	e9b32747b4	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

Author

SHA1

Message

Date

Angelos Katharopoulos

e9b32747b4

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

1 Commits