mlx/python
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
..
mlx MLX in C++ example (#1736) 2025-01-02 19:09:04 -08:00
src Dynamic broadcasting for shapeless compile/export (#1722) 2025-01-09 11:04:24 -08:00
tests Dynamic broadcasting for shapeless compile/export (#1722) 2025-01-09 11:04:24 -08:00