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