Skip to content

Use NVIDIA's RAPIDS cuML variant of UMAP #3

@JohnTigue

Description

@JohnTigue

There is a GPU-accelerated version of UMAP provided by the RAPIDS cuML library. The original implemention, umap-learn, is CPU only, says the docs.

(Confirm this. Numba supposedly can use CUDA GPUs, so why not?)

Some docs:

Metadata

Metadata

Assignees

Labels

No labels
No labels

Type

No type

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions