mlx/mlx/backend/no_metal
Awni Hannun 1ccaf80575
Dynamic broadcasting for shapeless compile/export (#1722)
* working towards dynamic broadcast

* shapeless broadcast

* fix build + nits

* use broadcast arrays in quantize matmul

* some cleanup / consistency

* mend

* some comments

* add vjp, jvp for broadcast axes
2025-01-09 11:04:24 -08:00
..
allocator.cpp Allow querying the allocator for the buffer size (#1404) 2024-09-11 21:02:16 -07:00
CMakeLists.txt Chore: add pre-commit hook for cmake (#1362) 2024-09-16 12:53:01 -07:00
event.cpp Fix array is_available race cases (#1468) 2024-10-07 19:13:50 -07:00
metal.cpp Wired (#1510) 2024-10-25 09:35:33 -07:00
primitives.cpp Dynamic broadcasting for shapeless compile/export (#1722) 2025-01-09 11:04:24 -08:00