Skip to content

Embed UMAP graphs too large for the GPU from their upper triangle, 200 epochs above 10k cells - #827

Open
Intron7 wants to merge 3 commits into
mainfrom
perf/umap-large-graphs
Open

Intron7 wants to merge 3 commits into
mainfrom
perf/umap-large-graphs

Conversation

@Intron7

@Intron7 Intron7 commented Oct 4, 2026

Copy link
Copy Markdown
Member
  • graphs whose device COO would not fit are cut to their upper triangle on the host (if canonical and symmetric) and embedded with twice the epochs: same quality, half the memory
  • n_epochs defaults to 200 for more than 10,000 cells, like scanpy and umap-learn

…0 epochs above 10k cells

- graphs whose device COO would not fit are cut to their upper triangle on the host
  (if canonical and symmetric) and embedded with twice the epochs: same quality, half the memory
- n_epochs defaults to 200 for more than 10,000 cells, like scanpy and umap-learn

Signed-off-by: Intron7 <sdicks@nvidia.com>
Signed-off-by: Intron7 <sdicks@nvidia.com>
cuML repels only the head of each edge, so with one triangle the repulsion a cell gets
depends on its position (41 of 700 PBMC cells got none). Keep the 200-epoch default.

Signed-off-by: Intron7 <sdicks@nvidia.com>

This branch has not been deployed

No deployments
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.

1 participant