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

avitai-artifex

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

scvi

Deep probabilistic analysis of single-cell and spatial omics data

numalogic

Collection of operational Machine Learning models and tools.

nichecompass

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

pilotgm

Patient-Level Analysis of Single Cell Disease Atlas with Optimal Transport of Gaussian Mixtures Variational Autoencoders

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

pi-vae-pytorch

A Pytorch implementation of Poisson Identifiable VAE (pi-VAE), a variational auto encoder used to construct latent variable models of neural activity while simultaneously modeling the relation between the latent and task variables.

rectorch

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

nhssynth

Synthetic data generation pipeline leveraging a Differentially Private Variational Auto Encoder assessed using a variety of metrics

move-dl

Multi-omics variational autoencoder

distvae-tabular

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

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.

deepgen

A Collection of modern Deep Generative Models

eve-pytorch

EVE - Pytorch

snpmanifold

Manifold learning for single-cell single-nucleotide genetic variations

pytorch-policy

Reinforcement Learning in Pytorch

pyroved

Invariant representation learning from imaging and spectral data

antipode

Simultaneous integration, differential expression and clustering of single cell RNAseq data using pyro-powered variational inference.

odin-ai

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

claritycore

Next-generation Open Source toolkit for low-level vision. Engineered for state-of-the-art performance in image and video restoration, including Super-Resolution, Denoising, Deblurring, and more.