Attention

The vector search and clustering algorithms in RAFT are being migrated to a new library dedicated to vector search called cuVS. We will continue to support the vector search algorithms in RAFT during this move, but will no longer update them after the RAPIDS 24.06 (June) release. We plan to complete the migration by RAPIDS 24.08 (August) release.

C++ API

• Core
  • Resources
    • Vocabulary
    • Device Resources
    • Device Resources Manager
    • Resource Functions
      • Comms
      • cuBLAS Handle
      • cuBLASLt Handle
      • CUDA Stream
      • CUDA Stream Pool
      • cuSolverDn Handle
      • cuSolverSp Handle
      • cuSparse Handle
      • Device ID
      • Device Memory Resource
      • Device Properties
      • Sub Communicators
      • Thrust Exec Policy
      • Custom runtime-shared resources
  • logger
  • Key-Value Pair
  • NVTX
  • Interruptible
  • Operators and Functors
  • Mathematical Functions
  • Bitset
  • Bitmap
• Cluster
  • K-Means
  • K-Means
  • Hierarchical Clustering
  • Spectral Clustering
• Distance
  • Distance Types
    • Pairwise Distance
    • 1-Nearest Neighbors
• Linear Algebra
  • Arithmetic
    • Addition
    • Binary Op
    • Division
    • Multiplication
    • Power
    • Square Root
    • Subtraction
    • Ternary Op
    • Unary Op
  • BLAS Routines
    • axpy
    • dot
    • gemm
    • gemv
  • Mapping and Reduction
    • Coalesced Reduction
    • Map
    • Map Reduce
    • Mean Squared Error
    • Norm
    • Normalize
    • Reduction
    • Reduce Cols By Key
    • Reduce Rows By Key
    • Strided Reduction
  • Matrix Operations
    • Transpose
  • Matrix-Vector Operations
    • Arithmetic
    • Operations
  • Linear Algebra Solvers
    • Eigen Decomposition
    • QR Decomposition
    • Randomized Singular-Value Decomposition
    • Singular-Value Decomposition
    • Least Squares
• Matrix
  • Matrix Arithmetic
    • Line-wise Operation
    • Power
    • Ratio
    • Reciprocal
    • Sign-flip
    • Square Root
  • Matrix Manipulation
    • Initialization
    • Reverse
    • Threshold
  • Matrix Ordering
    • Argmax
    • Argmin
    • Select-K
    • Column-wise Sort
  • Matrix Reductions
    • Matrix Norm
  • Matrix Selection
    • Copy
    • Diagonal
    • Gather
    • Slicing
    • Triangular
• Multi-dimensional Data
  • Multi-dimensional Representation
    • Data Layouts
    • Shapes
    • Accessors
  • mdspan: Multi-dimensional Non-owning View
    • Device Vocabulary
    • Device Factories
    • Managed Vocabulary
    • Managed Factories
    • Host Vocabulary
    • Host Factories
    • Pinned Vocabulary
    • Pinned Factories
    • Validation Routines
  • mdarray: Multi-dimensional Owning Container
    • Device Vocabulary
    • Device Factories
    • Host Vocabulary
    • Host Factories
    • Managed Vocabulary
    • Managed Factories
    • Pinned Vocabulary
    • Pinned Factories
  • span: One-dimensional Non-owning View
  • mdbuffer: Multi-dimensional Maybe-Owning Container
  • memory_type_dispatcher
  • temporary_device_buffer: Temporary raft::device_mdspan Producing Object
    • Factories
• Multi-node Multi-GPU
  • Common Types
  • Comms Interface
  • MPI Comms
  • NCCL+UCX Comms
• Neighbors
  • Brute-Force
  • IVF-Flat
    • Serializer Methods
    • Helper Methods
  • IVF-PQ
    • Serializer Methods
    • Candidate Refinement
    • Helper Methods
  • Epsilon Neighborhood
  • Random Ball Cover
  • CAGRA
    • Serializer Methods
• Random
  • Random State
    • Data Generation
      • make_blobs
      • make_regression
      • rmat
    • Univariate Random Sampling
    • Multi-Variable Random Sampling
      • Multi-Variable Gaussian
    • Sampling Without Replacement
• Solvers
  • Linear Assignment Problem
  • Minimum Spanning Tree
• Sparse
  • Sparse Types
    • COO Matrix
      • Basic Vocabulary
      • Device COO Matrix
      • Host COO Matrix
    • CSR Matrix
      • Basic Vocabulary
      • Device CSR Matrix
      • Host CSR Matrix
  • Sparse Distance
  • Sparse Linear Algebra
  • Sparse Matrix Operations
  • Sparse Neighbors
  • Sparse Solvers
• Stats
  • Summary Statistics
    • Covariance
    • Histogram
    • Mean
    • Mean Center
    • Mean Variance
    • Min/Max
    • Standard Deviation
    • Sum
    • Weighted Average
  • Probability & Information Theory
    • Contingency Matrix
    • Entropy
    • KL-Divergence
    • Mutual Information
  • Regression Model Scoring
    • Information Criterion
    • R2 Score
    • Regression Metrics
  • Classification Model Scoring
    • Accuracy
  • Clustering Model Scoring
    • Adjusted Rand Index
    • Completeness Score
    • Cluster Dispersion
    • Rand Index
    • Silhouette Score
    • V Measure
  • Neighborhood Model Scoring
    • Trustworthiness
    • Neighborhood Recall
• Utilities
  • Memory Pool