Reproducing Figure 3b-d from: Heimberg et al. (2024). "A cell atlas foundation model for scalable search of similar human cells." Nature 638.