Out-of-distribution prediction with disentangled representations for scRNA-seq data