zhangyiss/mlx
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx.git synced 2025-10-22 02:58:16 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity
Files
17310d91a6ffc606947df3f7a6f7c7fdf376f7d2
mlx/mlx/backend/metal/jit
History
Awni Hannun 111f1e71af Faster contiguous gather for indices in the first axis (#2552) 
* faster contiguous gather for indices in the first axis

* work per thread > 1

* angelos suggestion for scales / biases
2025-08-28 21:26:30 -07:00
..
includes.h
Faster contiguous gather for indices in the first axis (#2552)
2025-08-28 21:26:30 -07:00
indexing.h
Use int64 stride everywhere (#1671)
2024-12-09 11:09:02 -08:00