mlx/python
Awni Hannun b7c9f1d38f
scatter axis + gather axis primitives (#1813)
* scatter axis + gather axis primitives

* add transforms

* comment
2025-01-31 20:48:08 -08:00
..
mlx Small distributed launch helper (#1810) 2025-01-29 17:55:04 -08:00
src Start to cleanup/unify accelerate and common back-ends (Part 1/N) (#1777) 2025-01-29 14:34:49 -08:00
tests scatter axis + gather axis primitives (#1813) 2025-01-31 20:48:08 -08:00