A Collection of Variational Autoencoders (VAE) in PyTorch.

Vector Quantized VAEs - PyTorch Implementation

Implementation of Generating Diverse High-Fidelity Images with VQ-VAE-2 in PyTorch

The Official PyTorch Implementation of "LSGM: Score-based Generative Modeling in Latent Space" (NeurIPS 2021)

Code for the paper "Regularizing Variational Autoencoder with Diversity and Uncertainty Awareness"

Easy generative modeling in PyTorch

Experimental implementation for a sparse-dictionary based version of the VQ-VAE2 paper

Temporal Difference Variational Auto-Encoder (TD-VAE) (Implementation in PyTorch)

Taming Transformers for High-Resolution Image Synthesis

Pytorch implementation of stochastically quantized variational autoencoder (SQ-VAE)

The official implementation of Autoregressive Image Generation using Residual Quantization (CVPR '22)