3 Commits

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
Jagadish Krishnamoorthy	5fc117025a	Update README.md ... Fixed typo	2023-12-06 00:30:37 -08:00
Awni Hannun	b243c1d8f4	a few examples	2023-11-29 08:17:26 -08:00