Open Source · MarkTechPost ·
Meet Flash-KMeans: An IO-Aware, Exact K-Means That Runs Over 200× Faster Than FAISS on GPUs
Flash-KMeans is an open-source, IO-aware implementation of exact Lloyd's k-means built in Triton GPU kernels. It avoids distance-matrix materialization and atomic contention, and the authors report up to 17.9× end-to-end speedup on an NVIDIA H200, including over 200× versus FAISS.