NVIDIA/cuvs

解決する課題

cuVS は、近似最近傍探索（ANN）とクラスタリングのための高性能な GPU アクセラレーション実装を提供します。NVIDIA GPU を使用したベクトル類似度検索のプロセスを簡素化し、大規模な非構造化データやセマンティック検索の処理に不可欠です。

仕組み

cuVS は RAPIDS RAFT ライブラリ上に構築され、CAGRA などの最先端アルゴリズムを実装して、密ベクトルに対するインデックス構築と検索を実行します。Python、C++、C、Rust 向けの API を提供し、開発者はこれらの GPU アクセラレーションルーチンをデータベース、機械学習パイプライン、またはスタンドアロンアプリケーションに統合できます。

対象ユーザー

大規模なベクトル検索、生成 AI（RAG）、レコメンダーシステム、コンピュータビジョン、および k-NN グラフ構築やクラスタリング（例：K-means、HDBSCAN）などのデータマイニングタスクに取り組む開発者とデータサイエンティスト。

主な特長

• 多言語サポート：Python、C++、C、Rust 向けのネイティブ API。
• 高性能：高速なインデックス構築、低レイテンシ検索、高スループット向けに最適化。
• 幅広い応用：テキスト、音声、画像、分子データのセマンティック検索をサポート。
• 相互運用性：GPU でインデックスを構築し、CPU でデプロイ可能。
• 自動最適化：新しい NVIDIA アーキテクチャと CUDA バージョン向けにコードを更新。