FlashML-org/flashlib

NVIDIA cuML: GPU-Accelerated Machine Learning

H2Oai GPU Edition

cuVS - a library for vector search and clustering on the GPU

[NeurIPS2026] Fast and memory-efficient exact kmeans

RAFT contains fundamental widely-used algorithms and primitives for machine learning and information retrieval. The algorithms are CUDA-accelerated and form building blocks for more easily writing high performance applications.

Extension for Scikit-learn is a seamless way to speed up your Scikit-learn application

Traditional machine learning on top of Nx

A Quirky Assortment of CuTe Kernels

Benchmarks of approximate nearest neighbor libraries in Python