zhangyiss/mlx
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx.git synced 2025-07-24 02:41:19 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity
6931f84412
mlx/mlx/backend/metal/jit
History
Awni Hannun 4f72c66911
improvements to scatter / gather (#1541)
2024-10-30 19:30:54 -07:00
..
arange.h More jitting (#1132) 2024-05-23 16:23:44 -07:00
gemv_masked.h Add gemv masked to JIT plus some fixes (#1310) 2024-08-07 13:38:07 -07:00
includes.h Add gemv masked to JIT plus some fixes (#1310) 2024-08-07 13:38:07 -07:00
indexing.h improvements to scatter / gather (#1541) 2024-10-30 19:30:54 -07:00
softmax.h More jitting (#1132) 2024-05-23 16:23:44 -07:00
steel_conv.h Option to JIT steel gemm / conv (#1139) 2024-05-23 18:07:34 -07:00
steel_gemm.h Option to JIT steel gemm / conv (#1139) 2024-05-23 18:07:34 -07:00