Just on that UMAP point, I’ve found a lot success using PaCMAP (https://github.com/YingfanWang/PaCMAP) instead, which AFAIK is the SOTA in dimensionality reduction algorithms.