NVIDIA / NVIDIA/cuvs

[Tracker] Pending items for NN Descent

Open
#419 0 comments 0 reactions 1 assignee View on GitHub

@divyegala is already working on this.

Since Oct 28, 2024.

improvement Python vector search
Dominant language
Cuda
Stars
854
Forks
236
Avg merge
3d 3h
Merged PRs (30d)
62

Description

  • Consolidate code in cuVS by migrating over diff of features from RAFT
  • Triage CI wheel test failures
  • Triage batch test failures
  • Add mutual reachability API and triage HDBSCAN numerical instabilities when using NN Descent (https://github.com/rapidsai/cuml/pull/5939)
  • Additional distances
  • For float16 input, make sure data is not copied if it's already on device
  • indices are returned on host and distances on device. Return both in the same location and let user choose

Contributor guide

Open the contributing guide

First steps

  1. Read the whole issue, then the project's contributing guide.
  2. Comment on the issue to say you are picking it up — it saves two people doing the same work.
  3. Fork the repository and make your change on a branch.
  4. Open a pull request that references the issue number.

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.