Commit Graph

7 Commits

Author SHA1 Message Date
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
b1b9b11801 updates + format 2023-12-14 12:09:10 -08:00
Sarthak Yadav
29b7a97342 updated header 2023-12-14 16:28:00 +01:00
Sarthak Yadav
15a6c155a8 simplified ResNet, expanded README with throughput and performance 2023-12-14 09:05:04 +01:00
Sarthak Yadav
2439333a57 fixed doc for ResNet 2023-12-12 19:07:39 +01:00
Sarthak Yadav
f37e777243 added CIFAR10 + ResNet example 2023-12-12 19:01:06 +01:00