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
Markus Enzweiler
2b61d9deb6
Updated CIFAR-10 ResNet example to use BatchNorm instead of LayerNorm ( #257 )
...
* replaced nn.LayerNorm by nn.BatchNorm
* mlx>=0.0.8 required
* updated default to 30 epochs instead of 100
* updated README after adding BatchNorm
* requires mlx>=0.0.9
* updated README.md with results for mlx-0.0.9
2024-01-12 05:43:11 -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
b9439ce74e
typo / nits
2023-12-14 12:14:01 -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
