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.
machine-learning
information-retrieval
statistics
clustering
gpu
distance
linear-algebra
cuda
primitives
sparse
nearest-neighbors
solvers
vector-similarity
anns
random-sampling
vector-search
building-blocks
llm
vector-store
neighborhood-methods
-
Updated
Jun 28, 2024 - Cuda