rapidsai / raft
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.
☆850Updated this week
Alternatives and similar repositories for raft:
Users that are interested in raft are comparing it to the libraries listed below
- cuVS - a library for vector search and clustering on the GPU☆319Updated this week
- Framework for evaluating ANNS algorithms on billion scale datasets.☆366Updated 2 weeks ago
- RAPIDS Memory Manager☆546Updated this week
- ☆516Updated this week
- Graph-structured Indices for Scalable, Fast, Fresh and Filtered Approximate Nearest Neighbor Search☆1,262Updated this week
- CUDA implementation of Hierarchical Navigable Small World Graph algorithm☆155Updated 3 years ago
- common in-memory tensor structure☆957Updated 2 weeks ago
- Knowhere is an open-source vector search engine, integrating FAISS, HNSW, etc.☆209Updated last year
- CUDA Kernel Benchmarking Library☆582Updated 3 months ago
- CUDA Core Compute Libraries☆1,504Updated this week
- The NVIDIA® Tools Extension SDK (NVTX) is a C-based Application Programming Interface (API) for annotating events, code ranges, and resou…☆352Updated this week
- ☆145Updated this week
- A throughput-oriented high-performance serving framework for LLMs☆748Updated 5 months ago
- GGNN: State of the Art Graph-based GPU Nearest Neighbor Search☆152Updated 3 weeks ago
- Graph Library for Approximate Similarity Search☆104Updated 9 months ago
- A library for accelerating Transformer models on NVIDIA GPUs, including using 8-bit floating point (FP8) precision on Hopper and Ada GPUs…☆2,240Updated this week
- A library of algorithms for approximate nearest neighbor search in high dimensions, along with a set of useful tools for designing such a…☆134Updated last week
- TorchX is a universal job launcher for PyTorch applications. TorchX is designed to have fast iteration time for training/research and sup…☆348Updated this week
- A tool for bandwidth measurements on NVIDIA GPUs.☆380Updated 3 weeks ago
- An open-source efficient deep learning framework/compiler, written in python.☆687Updated last week
- HierarchicalKV is a part of NVIDIA Merlin and provides hierarchical key-value storage to meet RecSys requirements. The key capability of…☆139Updated this week
- A library for building and serving multi-node distributed faiss indices.☆262Updated last year
- Pipeline Parallelism for PyTorch☆754Updated 6 months ago
- A Python-level JIT compiler designed to make unmodified PyTorch programs faster.☆1,033Updated 10 months ago
- The Triton backend for the ONNX Runtime.☆139Updated this week
- Examples demonstrating available options to program multiple GPUs in a single node or a cluster☆628Updated last week
- ☆410Updated last year
- Knowhere is a vector search engine, integrating FAISS, HNSW, DiskANN.☆206Updated this week
- WholeGraph - large scale Graph Neural Networks☆101Updated 3 months ago
- Provides end-to-end model development pipelines for LLMs and Multimodal models that can be launched on-prem or cloud-native.☆491Updated last month