Awni Hannun 657f466402 use sdpa and exportable functions in transformer multi head attention (#1760)

2025-01-09 13:11:55 -08:00

..

mlx

use sdpa and exportable functions in transformer multi head attention (#1760)

2025-01-09 13:11:55 -08:00

src

Dynamic broadcasting for shapeless compile/export (#1722)

2025-01-09 11:04:24 -08:00

tests

use sdpa and exportable functions in transformer multi head attention (#1760)

2025-01-09 13:11:55 -08:00