mlx/mlx/backend/no_cpu
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
..
CMakeLists.txt More fixes for arrays with large sizes (#1405) 2024-09-17 12:46:31 -07:00
primitives.cpp Dynamic broadcasting for shapeless compile/export (#1722) 2025-01-09 11:04:24 -08:00