NVIDIA/cuvs

cuVS - a library for vector search and clustering on the GPU

What it solves

cuVS は、GPU を利用した高性能なベクトル類似検索とクラスタリングを簡素化します。非構造化データから生成された大量のマルチメディア埋め込み（密ベクトル）に対し、インデックス構築の高速化と低レイテンシ・高スループット検索のニーズに応え、セマンティック検索やデータマイニングに不可欠です。

How it works

RAPIDS RAFT ライブラリ上に構築された cuVS は、近似最近傍（ANN）およびクラスタリングアルゴリズムの最先端 GPU 加速実装を提供します。ユーザーはインデックス（例: CAGRA アルゴリズム）を構築し、Python、C++、C、Rust など複数の言語で検索を実行できます。

Who it’s for

このライブラリは、生成 AI、検索強化生成（RAG）、レコメンダーシステム、コンピュータビジョン、そして k-NN グラフ構築や可視化（例: UMAP、t-SNE）といったデータマイニングタスクに取り組む開発者やデータサイエンティスト向けに設計されています。

Highlights

  • 高速なインデックス構築と高スループット検索性能。
  • Python、C++、C、Rust をサポートするマルチ言語対応。
  • GPU でインデックスを構築し、CPU でデプロイできる相互運用性。
  • K-means や HDBSCAN など、さまざまなクラスタリング・可視化アルゴリズムを統合的にサポート。