#Triton

Meet Flash-KMeans: An IO-Aware, Exact K-Means That Runs Over 200× Faster Than FAISS on GPUs

Flash-KMeans, an open-source IO-aware k-means implementation, achieves over 200× speedup on GPUs compared to FAISS by optimizing distance matrix operations and reducing atomic contention.