PyTorch package for the discrete VAE used for DALL·E.

A Collection of Variational Autoencoders (VAE) in PyTorch.

Collection of generative models, e.g. GAN, VAE in Pytorch and Tensorflow.

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

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

Vector Quantized VAEs - PyTorch Implementation

For beginner, this will be the best start for VAEs, GANs, and CVAE-GAN. This contains AE, DAE, VAE, GAN, CGAN, DCGAN, WGAN, WGAN-GP, VAE-GAN, CVAE-GAN. All use PyTorch.

Implementations of various VAE-based semi-supervised and generative models in PyTorch

PyTorch implementation of VQ-VAE by Aäron van den Oord et al.

Pytorch implementation of β-VAE

Minimalist implementation of VQ-VAE in Pytorch

Annotated, understandable, and visually interpretable PyTorch implementations of: VAE, BIRVAE, NSGAN, MMGAN, WGAN, WGANGP, LSGAN, DRAGAN, BEGAN, RaGAN, InfoGAN, fGAN, FisherGAN

A simple tutorial of Variational AutoEncoders with Pytorch

Official Pytorch implementation of the paper "Action-Conditioned 3D Human Motion Synthesis with Transformer VAE", ICCV 2021

A Variational Autoencoder (VAE) implemented in PyTorch

Pytorch implementation of BERT4Rec and Netflix VAE.

Pytorch implementation of Hyperspherical Variational Auto-Encoders

An unofficial implementation of both ViT-VQGAN and RQ-VAE in Pytorch

A CNN Variational Autoencoder (CNN-VAE) implemented in PyTorch

Official implementation of the paper "Generalized Zero- and Few-Shot Learning via Aligned Variational Autoencoders" (CVPR 2019)

PyTorch implementation of VQ-VAE + WaveNet by [Chorowski et al., 2019] and VQ-VAE on speech signals by [van den Oord et al., 2017]

VAE for color images

Pretrained GANs + VAEs + classifiers for MNIST/CIFAR in pytorch.

[CVPR 2021 Oral] Official PyTorch implementation of Soft-IntroVAE from the paper "Soft-IntroVAE: Analyzing and Improving Introspective Variational Autoencoders"

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

VAEGAN from "Autoencoding beyond pixels using a learned similarity metric" implemented in Pytorch. Clean, clear and with comments.

VQ-VAE implementation / pytorch

PyTorch implementation of VQ-VAE-2 from "Generating Diverse High-Fidelity Images with VQ-VAE-2"

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

Variational Autoencoder (VAE) with perception loss implementation in pytorch