Skip to content

feat: improve UMAP performance on browser with WebGPU#194

Merged
donghaoren merged 2 commits intomainfrom
dr/gpu-umap
Apr 27, 2026
Merged

feat: improve UMAP performance on browser with WebGPU#194
donghaoren merged 2 commits intomainfrom
dr/gpu-umap

Conversation

@donghaoren
Copy link
Copy Markdown
Collaborator

@donghaoren donghaoren commented Apr 25, 2026

  • Add GPU-accelerated paths for the two main computational bottlenecks in UMAP: nearest-neighbor search (NNDescent) and embedding optimization.
  • Restructure benchmark code under packages/umap/benchmark.

@donghaoren donghaoren requested a review from domoritz April 25, 2026 01:42
Copy link
Copy Markdown
Member

@domoritz domoritz left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Very cool 🚀

@donghaoren donghaoren merged commit 2e2b4f3 into main Apr 27, 2026
8 checks passed
@donghaoren donghaoren deleted the dr/gpu-umap branch April 27, 2026 16:23
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants