Skip to main content
Ctrl+K

raft 24.04.00 documentation

  • Quick Start
  • Installation
  • C++ API
  • Python API
  • Using The Pre-Compiled Binary
    • Vector Search in C++ Tutorial
    • RAFT ANN Benchmarks
    • RAFT Dask API
    • Using RAFT Comms
    • Developer Guide
    • Contributing
  • GitHub
  • Twitter
  • Quick Start
  • Installation
  • C++ API
  • Python API
  • Using The Pre-Compiled Binary
    • Vector Search in C++ Tutorial
    • RAFT ANN Benchmarks
    • RAFT Dask API
    • Using RAFT Comms
    • Developer Guide
    • Contributing
  • GitHub
  • Twitter

Section Navigation

  • Cluster
  • Common
  • Distance
  • Matrix
  • Neighbors
  • Random
  • Python API

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.

Python API#

  • Cluster
    • KMeans
      • KMeansParams
      • fit()
      • cluster_cost()
      • compute_new_centroids()
  • Common
    • Basic Vocabulary
      • DeviceResources
        • DeviceResources.getHandle()
        • DeviceResources.sync()
      • Stream
        • Stream.get_ptr()
        • Stream.sync()
      • device_ndarray
        • device_ndarray.c_contiguous
        • device_ndarray.copy_to_host()
        • device_ndarray.dtype
        • device_ndarray.empty()
        • device_ndarray.f_contiguous
        • device_ndarray.shape
        • device_ndarray.strides
    • Interruptible
      • cuda_interruptible()
      • synchronize()
      • cuda_yield()
    • CUDA Array Interface Helpers
      • cai_wrapper
  • Distance
    • pairwise_distance()
    • fused_l2_nn_argmin()
  • Matrix
    • select_k()
  • Neighbors
    • Brute Force
      • knn()
    • CAGRA
      • IndexParams
      • build()
      • SearchParams
      • search()
      • Serializer Methods
        • save()
        • load()
    • HNSW
      • SearchParams
      • from_cagra()
      • search()
      • Serializer Methods
        • save()
        • load()
    • IVF-Flat
      • IndexParams
      • build()
      • extend()
      • SearchParams
      • search()
      • Serializer Methods
        • save()
        • load()
    • IVF-PQ
      • IndexParams
      • build()
      • extend()
      • SearchParams
      • search()
      • Serializer Methods
        • save()
        • load()
      • Candidate Refinement
        • refine()
  • Random
    • rmat()

previous

Utilities

next

Cluster

Show Source

© Copyright 2023, NVIDIA Corporation.

Created using Sphinx 6.2.1.

Built with the PyData Sphinx Theme 0.15.2.