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