4 Commits
Author SHA1 Message Date
Awni HannunandGitHub 885e520ecd Some fixes for gemma2 (#99)
* some fixes for gemma2

* format

* fixes

* format
2024-08-01 20:06:11 -07:00
b951b78eb2 phi3 (#54)
* phi3

Co-authored-by: David Koski <dkoski@apple.com>
2024-04-24 09:31:01 -07:00
Awni HannunandGitHub 15b38cd146 Use fast (#38)
* update to latest mlx swift and use fast norms
* gpu usage -> memory usage
2024-03-27 16:37:35 -07:00
Awni HannunandGitHub 4ed4ec69e7 LeNet on MNIST + readme update (#12)
* LeNet on MNIST + readme update

* tanh + remove device toggle

* remove device entirely
2024-03-04 14:16:20 -08:00