hanxiao / flash-kmeans-mlxView on GitHub
IO-aware batched K-Means for Apple Silicon, ported from Flash-KMeans (Triton/CUDA) to pure MLX. Up to 94x faster than sklearn.
17Mar 22, 2026Updated 3 months ago

Alternatives and similar repositories for flash-kmeans-mlx

Users that are interested in flash-kmeans-mlx are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.

Sorting:

Are these results useful?