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
Dec 11, 2024 - Cuda