mlx/benchmarks
Jagrit Digani 6686e61ca4
Reduce update (#783)
* Split reduction files to reduce compile times

* Add small and medium axis size specializations for row reductions

* Add non-row-reduction options for small and med kernels
2024-03-04 19:09:51 -08:00
..
cpp Implement the 'where' primitive for conditional selection (#664) 2024-02-22 15:10:48 -08:00
numpy Add isort pre-commit and run (#68) 2023-12-08 11:31:47 -08:00
python Reduce update (#783) 2024-03-04 19:09:51 -08:00