zhangyiss/mlx
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx.git synced 2025-09-19 19:38:16 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity
Files
d6977f2a57538ab454d2e798d1c6b4b9e77a25f9
mlx/python
History
Awni Hannun d6977f2a57 Add sdpa with sinks (#2558) 
* add sdpa with sinks

* fix 2 pass

* fix matrix sdpa

* fix perf regression

* add to cuda (#2580)
2025-09-10 14:53:00 -07:00
..
mlx
Adding Relu2 (#2582)
2025-09-10 07:24:30 -07:00
scripts
nccl dep + default for cuda (#2526)
2025-08-21 17:57:49 -07:00
src
Add sdpa with sinks (#2558)
2025-09-10 14:53:00 -07:00
tests
Add sdpa with sinks (#2558)
2025-09-10 14:53:00 -07:00