Variational Autoencoder Python Packages

scvi-tools

Deep probabilistic analysis of single-cell and spatial omics data

pythae

Unifying Variational Autoencoder (VAE) implementations in Pytorch (NeurIPS 2022)

normflows

PyTorch implementation of normalizing flow models

focal-frequency-loss

[ICCV 2021] Focal Frequency Loss for Image Reconstruction and Synthesis

numalogic

Collection of operational time series ML models and tools

scvi

Deep probabilistic analysis of single-cell and spatial omics data

avitai-artifex

A research-focused modular generative modeling library built on JAX/Flax NNX

nhssynth

Package to accompany P41

nichecompass

End-to-end analysis of spatial multi-omics data

antipode

Variational Inference for Cell Type Evolution

move-dl

Multi-omics variational autoencoder

pi-vae-pytorch

Implementation of Poisson Identifiable VAE (pi-VAE) in Pytorch

rectorch

rectorch: state-of-the-art recsys approaches implemented in pytorch.

distvae-tabular

Official implementation of 'Distributional Learning of Variational AutoEncoder: Application to Synthetic Data Generation' (DistVAE) with pytorch (NeurIPS 2023 accepted paper).

deepgen

A Collection of modern Deep Generative Models

pyroved

Invariant representation learning from imaging and spectral data

ponyo

Software to simulate compendium-wide gene expression data using a VAE.

eve-pytorch

EVE - Pytorch

snpmanifold

Manifold learning for single-cell single-nucleotide genetic variations

free-transformer

Free Transformer Architecture - an extension of the decoder Transformer that conditions its generative process on random latent variables which are learned without supervision

visionengine

DHRL ("VisionEngine"), built on Tensorflow 2.2, is an inference framework for natural image data that is tailored to applications in the natural sciences, integrating analytical, theoretical and empirical approaches using generative modeling.

tfae

TensorFlow extension for building and training different types of autoencoders

odin-ai

Orgainzed Digital Intelligent Network (O.D.I.N)

sisua

SemI-SUpervised generative Autoencoder for single cell data