mlx/python
Awni Hannun 0c5eea226b
Reduce specializations (#1607)
* start of reduce specializations

* fix all reduce

* fix many dims

* fix

* non-jit tests clear

* cleanup instantiations

* cpu merges

* change dim specializations

* optimize

* fix jit

* fix jit

* use higher precision for integer sum+prod

* fixes
2024-11-21 19:53:00 -08:00
..
mlx Formatting fixes (#1606) 2024-11-20 15:30:36 -08:00
src throw for invalid case and remove test (#1575) 2024-11-08 12:04:03 -08:00
tests Reduce specializations (#1607) 2024-11-21 19:53:00 -08:00