UMAP knn build with cuVS all-neighbors

[jinsolp]

Description

Replacing usage of raft NN Descent to cuVS all-neighbors wrapper (which internally uses NN Descent and includes the batching feature). Planning to add some interesting numbers to this PR...

Usage

from cuml.manifold.umap import UMAP as cuUMAP

cuml_model = cuUMAP(
        n_neighbors=10,
        build_algo="nn_descent",
        build_kwds={"n_clusters": 4, "n_nearest_clusters": 2, "nn_descent": {"graph_degree": 15, "max_iterations": 20}},
    )

embedding = cuml_model.fit_transform(data, data_on_host=True)

cuml_model = cuUMAP(
        n_neighbors=10,
        build_algo="ivfpq",
        build_kwds={"n_clusters": 4, "n_nearest_clusters": 2},
    )
embedding = cuml_model.fit_transform(data, data_on_host=True)

[copy-pr-bot]

This pull request requires additional validation before any workflows can run on NVIDIA's runners.

Pull request vetters can view their responsibilities here.

Contributors can view more details about this message here.

[jinsolp]

Needs to be reverted once this PR in cuvs is merged

[jinsolp]

/ok to test

[copy-pr-bot]

Auto-sync is disabled for draft pull requests in this repository. Workflows must be run manually.

Contributors can view more details about this message here.

[review-notebook-app]

Check out this pull request on 

See visual diffs & provide feedback on Jupyter Notebooks.

---

Powered by ReviewNB