mlx/mlx/backend/metal/jit
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