zhangyiss/mlx
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx.git synced 2025-06-25 09:51:17 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity
0189ab6ab6
mlx/mlx/backend/metal/kernels/reduction
History
Awni Hannun 0189ab6ab6
More jitting (#1132) 
* docs + circle min size build

* jit scan, arange, softmax

* add sort

* jit reductions

* remove print

* fix deps

* clean includes / nits
2024-05-23 16:23:44 -07:00
..
ops.h More jitting (#1132) 2024-05-23 16:23:44 -07:00
reduce_all.h More jitting (#1132) 2024-05-23 16:23:44 -07:00
reduce_col.h More jitting (#1132) 2024-05-23 16:23:44 -07:00
reduce_init.h More jitting (#1132) 2024-05-23 16:23:44 -07:00
reduce_row.h More jitting (#1132) 2024-05-23 16:23:44 -07:00