zhangyiss/mlx
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx.git synced 2025-06-25 01:41:17 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity
sockets-distributed
mlx/examples/extensions/axpby
History
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
..
axpby.cpp Add gemv masked to JIT plus some fixes (#1310) 2024-08-07 13:38:07 -07:00
axpby.h feat: metal formatting and pre-commit bump (#1038) 2024-04-30 07:18:09 -07:00
axpby.metal feat: metal formatting and pre-commit bump (#1038) 2024-04-30 07:18:09 -07:00