API Reference

Core VAE Models

The core VAE architectures for single-cell gene expression modeling.

TransformerVAE(encoder, decoder, ...)

Lightning Training Modules

PyTorch Lightning modules for training VAE and diffusion models.

VAE(vae_model, vae_optimizer[, ...])
LatentDiffusion(vae_model, vae_optimizer, ...)

Data Loading

Data module for loading and preprocessing single-cell data.

DataModule(train_adata_path, ...[, ...])