Skip to content

UMAP fast transform - #7754

Draft
viclafargue wants to merge 1 commit into
NVIDIA:mainfrom
viclafargue:umap-fast-transform
Draft

UMAP fast transform#7754
viclafargue wants to merge 1 commit into
NVIDIA:mainfrom
viclafargue:umap-fast-transform

Conversation

@viclafargue

Copy link
Copy Markdown
Contributor

Closes #6215

This PR enables faster transform operations in UMAP.

  • fast_transform=False (default) matches current behavior, the transform function builds the KNN graph with bruteforce search
  • fast_transform=True builds a CAGRA index during fitting and later use it for faster approximate build of the KNN graph during transform

@copy-pr-bot

copy-pr-bot Bot commented Feb 3, 2026

Copy link
Copy Markdown

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.

@github-actions github-actions Bot added Cython / Python Cython or Python issue CUDA/C++ labels Feb 3, 2026
@jinsolp
jinsolp self-requested a review February 14, 2026 01:25
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

CUDA/C++ Cython / Python Cython or Python issue

Projects

None yet

Development

Successfully merging this pull request may close these issues.

[FEA] Implement UMAP transform with batched nn descent

3 participants