2 Commits

Author	SHA1	Message	Date
Awni Hannun	f45a1ab83c	Update a few examples to use compile (#420) ... * update a few examples to use compile * update mnist * add compile to vae and rename some stuff for simplicity * update reqs * use state in eval * GCN example with RNG + dropout * add a bit of prefetching	2024-02-08 13:00:41 -08:00
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