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.
☆838Updated 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☆298Updated this week
- Framework for evaluating ANNS algorithms on billion scale datasets.☆364Updated this week
- RAPIDS Memory Manager☆534Updated this week
- Graph-structured Indices for Scalable, Fast, Fresh and Filtered Approximate Nearest Neighbor Search☆1,245Updated this week
- ☆515Updated this week
- common in-memory tensor structure☆942Updated 2 weeks ago
- CUDA implementation of Hierarchical Navigable Small World Graph algorithm☆154Updated 3 years ago
- Knowhere is an open-source vector search engine, integrating FAISS, HNSW, etc.☆210Updated last year
- Triton Model Analyzer is a CLI tool to help with better understanding of the compute and memory requirements of the Triton Inference Serv…☆454Updated this week
- ☆142Updated this week
- CUDA Kernel Benchmarking Library☆561Updated 3 months ago
- PyTriton is a Flask/FastAPI-like interface that simplifies Triton's deployment in Python environments.☆772Updated last week
- A Python-level JIT compiler designed to make unmodified PyTorch programs faster.☆1,029Updated 10 months ago
- GGNN: State of the Art Graph-based GPU Nearest Neighbor Search☆151Updated last week
- The NVIDIA® Tools Extension SDK (NVTX) is a C-based Application Programming Interface (API) for annotating events, code ranges, and resou…☆347Updated this week
- TorchX is a universal job launcher for PyTorch applications. TorchX is designed to have fast iteration time for training/research and sup…☆345Updated this week
- CUDA Core Compute Libraries☆1,468Updated 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…☆132Updated 2 weeks ago
- Knowhere is a vector search engine, integrating FAISS, HNSW, DiskANN.☆205Updated this week
- KvikIO - High Performance File IO☆182Updated this week
- High performance model preprocessing library on PyTorch☆651Updated 10 months ago
- Graph Library for Approximate Similarity Search☆104Updated 9 months ago
- A Fusion Code Generator for NVIDIA GPUs (commonly known as "nvFuser")☆303Updated this week
- HugeCTR is a high efficiency GPU framework designed for Click-Through-Rate (CTR) estimating training☆971Updated 4 months ago
- HierarchicalKV is a part of NVIDIA Merlin and provides hierarchical key-value storage to meet RecSys requirements. The key capability of…☆138Updated this week
- An open-source efficient deep learning framework/compiler, written in python.☆681Updated last week
- FB (Facebook) + GEMM (General Matrix-Matrix Multiplication) - https://code.fb.com/ml-applications/fbgemm/☆1,260Updated this week
- NVIDIA Merlin is an open source library providing end-to-end GPU-accelerated recommender systems, from feature engineering and preprocess…☆801Updated 2 months ago
- A library to analyze PyTorch traces.☆332Updated last week
- Triton Python, C++ and Java client libraries, and GRPC-generated client examples for go, java and scala.☆600Updated last week