Crate cuvs

cuVS: Rust bindings for Vector Search on the GPU

This crate provides Rust bindings for cuVS, allowing you to run approximate nearest neighbors search on the GPU.

Modules

brute_force

Brute Force KNN

cagra

CAGRA is a graph-based nearest neighbors implementation with state-of-the art query performance for both small- and large-batch sized search.

distance

distance_type

ivf_flat

The IVF-Flat method is an ANN algorithm. It uses an inverted file index (IVF) with unmodified (that is, flat) vectors. This algorithm provides simple knobs to reduce the overall search space and to trade-off accuracy for speed.

ivf_pq

Inverted File Product Quantization

Structs

ManagedTensor

ManagedTensor is a wrapper around a dlpack DLManagedTensor object. This lets you pass matrices in device or host memory into cuvs.

Resources

Resources are objects that are shared between function calls, and includes things like CUDA streams, cuBLAS handles and other resources that are expensive to create.

Enums

Error

Type Aliases

Result