Scale batched GPU NMF factorize to millions of cells - #7

Merged

OlivierBakker merged 3 commits into

TrynkaLab:mainfrom

JerryIshihara:fix/scalable-recon-error

Jul 7, 2026

Commits on Jul 7, 2026

Scale batched-fit convergence check to large n (identity recon error)
JerryIshihara
committed
Compute ‖X‖² in row chunks (torch.dot caps vector length at 2³¹)
JerryIshihara
committed
Densify norm_counts once in factorize_gpu (avoid per-batch re-densify)
JerryIshihara
committed