NVIDIA / raft

git clone https://github.com/NVIDIA/raft.git

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.