Angelos Katharopoulos
e7267d30f8
Distributed support cifar ( #1301 )
2025-03-05 13:33:15 -08:00
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
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
Awni Hannun
b1b9b11801
updates + format
2023-12-14 12:09:10 -08:00
Sarthak Yadav
15a6c155a8
simplified ResNet, expanded README with throughput and performance
2023-12-14 09:05:04 +01:00
Sarthak Yadav
f37e777243
added CIFAR10 + ResNet example
2023-12-12 19:01:06 +01:00