Variational Autoencoder Python Packages

scvi-tools

Deep probabilistic analysis of single-cell and spatial omics data

focal-frequency-loss

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

normflows

PyTorch implementation of normalizing flow models

pythae

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

scvi

Deep probabilistic analysis of single-cell and spatial omics data

time-causal-vae

Time-causal financial generative models: refactored TC-VAE baselines with causal VQ/RVQ tokenizers, token priors, S&P500/VIX, Hawkes/SVMHJD, multi-dimensional benchmarks, and path-risk diagnostics.

avitai-artifex

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

numalogic

Collection of operational time series ML models and tools

distvae-tabular

Package for Synthetic Data Generation using Distributional Learninig of VAE

pi-vae-pytorch

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

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

nichecompass

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

nhssynth

Package to accompany P41

rectorch

rectorch is a pytorch-based framework for state-of-the-art top-N recommendation

move-dl

Multi-omics variational autoencoder

deepgen

Collection of modern Deep Generative Models

eve-pytorch

Implementation of evolutionary model of variant effect (EVE), a deep generative model of evolutionary data, in PyTorch.

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.

pytorch-policy

This repository contains model-free deep reinforcement learning algorithms implemented in Pytorch

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.

pyroved

Variational encoder-decoder models in Pyro probabilistic programming language

pilotgm

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

pyraug

Data Augmentation with VAE

rapidae

Rapidae: Python Library for Rapid Creation and Experimentation of Autoencoders