hanxiao/flash-kmeans-mlx

IO-aware batched K-Means for Apple Silicon, ported from Flash-KMeans (Triton/CUDA) to pure MLX. Up to 94x faster than sklearn.

41
/ 100
Emerging

Available on PyPI.

Maintenance 13 / 25
Adoption 10 / 25
Maturity 18 / 25
Community 0 / 25

How are scores calculated?

Stars

11

Forks

Language

Python

License

Apache-2.0

Last pushed

Mar 18, 2026

Monthly downloads

227

Commits (30d)

0

Dependencies

2

Get this data via API

curl "https://pt-edge.onrender.com/api/v1/quality/ml-frameworks/hanxiao/flash-kmeans-mlx"

Open to everyone — 100 requests/day, no key needed. Get a free key for 1,000/day.