mlx/python
Awni Hannun 005e7efa64
fix mask in sdpa (#1980)
* fix mask in sdpa

* fix attention mask

* Re-enable routing for array mask

---------

Co-authored-by: Jagrit Digani <digani@apple.com>
2025-03-20 14:53:12 -07:00
..
mlx Update smooth_l1_loss in losses.py (#1974) 2025-03-19 20:19:02 -07:00
src Support fused masking in Attention (#1924) 2025-03-20 11:01:32 -07:00
tests fix mask in sdpa (#1980) 2025-03-20 14:53:12 -07:00