6 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
Awni Hannun	a5d6d0436c	Support Hugging Face models (#215) ... * support hf direct models	2024-01-03 15:13:26 -08:00
Awni Hannun	27c0a8c002	Add llms subdir + update README (#145) ... * add llms subdir + update README * nits * use same pre-commit as mlx * update readmes a bit * format	2023-12-20 10:22:25 -08:00
Tristan Bilot	b606bfa6a7	fix comments before merge	2023-12-11 23:10:46 +01:00
Tristan Bilot	b95e48e146	use tree_flatten within L2 regularization	2023-12-11 20:15:11 +01:00
Tristan Bilot	ed5a830626	add GCN implementation	2023-12-11 17:48:07 +01:00