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과 같은 다양한 클러스터링 및 시각화 알고리즘을 통합 지원.