scvi-tools
scvi-tools 是一个基于 PyTorch 和 PyTorch Lightning 的单细胞基因组学概率模型 Python 框架,使用变分推断提供深度生成模型,用于分析多种单细胞数据模态。它支持单细胞 RNA-seq(降维、批次校正、整合)、scATAC-seq 染色质可及性、多模态数据整合(CITE-seq、multiome、配对/非配对数据集)、空间转录组(去卷积、空间映射)、差异表达分析、细胞类型注释与迁移学习,以及甲基化、流式细胞术、RNA 速率等专门模态,并支持构建自定义概率模型。所有模型遵循统一 API:setup_anndata 注册数据、创建模型、train 训练、提取潜在表示与归一化表达,结果可存回 AnnData 供 scanpy 下游分析。模型要求原始计数数据,支持 GPU 加速和批次校正。