mlx/examples
Awni Hannun 30bbea2f08
Add gemv masked to JIT plus some fixes (#1310)
* add gemv masked to JIT plus some fixes

* some cleanup

* add utils

* fix

* fix 2

* more cleaning

* fix

* remove unused mps matmul support

* one more nit

* revert
2024-08-07 13:38:07 -07:00
..
cpp Add docs for the distributed namespace (#1184) 2024-06-06 11:37:00 -07:00
extensions Add gemv masked to JIT plus some fixes (#1310) 2024-08-07 13:38:07 -07:00
python Added MSE message (#500) 2024-01-19 06:27:50 -08:00