mirror of
https://github.com/ml-explore/mlx.git
synced 2025-12-16 01:49:05 +08:00
111f1e71af
* faster contiguous gather for indices in the first axis * work per thread > 1 * angelos suggestion for scales / biases