cuML

Researchers from UC Berkeley and UT Austin released Flash-KMeans, an IO-aware, exact k-means library that runs over 200× faster than FAISS on GPUs by restructuring data movement. The open-source libra…

FlashLib, a new GPU library for classical machine learning operators, achieves speedups of up to 208× over cuML on Hopper GPUs for algorithms including KMeans, KNN, and PCA. The library is designed to…