Variational Autoencoder Implementations ML Frameworks

Implementations and tutorials of Variational Autoencoders (VAEs) and their variants (CVAEs, VQ-VAEs, KAN-AEs) for generative modeling. Does NOT include standard autoencoders for denoising/compression tasks or other generative models like diffusion/GANs.

There are 77 variational autoencoder implementations frameworks tracked. 4 score above 50 (established tier). The highest-rated is mseitzer/pytorch-fid at 67/100 with 3,833 stars and 112,753 monthly downloads.

Get all 77 projects as JSON

curl "https://pt-edge.onrender.com/api/v1/datasets/quality?domain=ml-frameworks&subcategory=variational-autoencoder-implementations&limit=20"

Open to everyone — 100 requests/day, no key needed. Get a free key for 1,000/day.

#	Framework	Score	Tier	Stars	Language
1	mseitzer/pytorch-fid Compute FID scores with PyTorch.	67	Established	3,833	Python
2	Naresh1318/Adversarial_Autoencoder A wizard's guide to Adversarial Autoencoders	56	Established	433	Python
3	ratschlab/aestetik AESTETIK: Convolutional autoencoder for learning spot representations from...	56	Established	20	Python
4	acids-ircam/RAVE Official implementation of the RAVE model: a Realtime Audio Variational autoEncoder	53	Established	1,698	Python
5	AntixK/PyTorch-VAE A Collection of Variational Autoencoders (VAE) in PyTorch.	49	Emerging	7,605	Python
6	jaanli/variational-autoencoder Variational autoencoder implemented in tensorflow and pytorch (including...	44	Emerging	1,183	Python
7	nathanhubens/Autoencoders Implementation of simple autoencoders networks with Keras	43	Emerging	264	Jupyter Notebook
8	ethanluoyc/pytorch-vae A Variational Autoencoder (VAE) implemented in PyTorch	43	Emerging	432	Python
9	YannDubs/disentangling-vae Experiments for understanding disentanglement in VAE latent representations	43	Emerging	841	Python
10	greenelab/tybalt Training and evaluating a variational autoencoder for pan-cancer gene expression data	42	Emerging	174	HTML
11	zalandoresearch/pytorch-vq-vae PyTorch implementation of VQ-VAE by Aäron van den Oord et al.	41	Emerging	602	Jupyter Notebook
12	Kaixhin/Autoencoders Torch implementations of various types of autoencoders	40	Emerging	476	Lua
13	nadavbh12/VQ-VAE Minimalist implementation of VQ-VAE in Pytorch	40	Emerging	559	Python
14	stormraiser/GAN-weight-norm Code for "On the Effects of Batch and Weight Normalization in Generative...	40	Emerging	181	Lua
15	msmbuilder/vde Variational Autoencoder for Dimensionality Reduction of Time-Series	40	Emerging	190	Jupyter Notebook
16	nicola-decao/s-vae-pytorch Pytorch implementation of Hyperspherical Variational Auto-Encoders	39	Emerging	386	Python
17	Nemzy/video_generator This is implementation of convolutional variational autoencoder in...	39	Emerging	73	Jupyter Notebook
18	samsinai/VAE_protein_function Protein function prediction using a variational autoencoder	38	Emerging	93	Jupyter Notebook
19	bramtoula/vdna Pytorch implementation of Visual DNA, an approach to represent and compare images.	38	Emerging	40	Python
20	mikelkou/fava Functional Associations using Variational Autoencoders	37	Emerging	43	Jupyter Notebook
21	timbmg/VAE-CVAE-MNIST Variational Autoencoder and Conditional Variational Autoencoder on MNIST in PyTorch	36	Emerging	658	Python
22	archinetai/cqt-pytorch An invertible and differentiable implementation of the Constant-Q Transform (CQT).	35	Emerging	72	Python
23	sony/sqvae Pytorch implementation of stochastically quantized variational autoencoder (SQ-VAE)	35	Emerging	193	Python
24	ZephirFXEC/VQVDB OpenVDB Fog Volume Compression using VQ-VAE Neural Network	35	Emerging	92	C++
25	ritheshkumar95/pytorch-vqvae Vector Quantized VAEs - PyTorch Implementation	34	Emerging	949	Python
26	Kaixhin/nninit Weight initialisation schemes for Torch7 neural network modules	33	Emerging	100	Lua
27	csinva/gan-vae-pretrained-pytorch Pretrained GANs + VAEs + classifiers for MNIST/CIFAR in pytorch.	33	Emerging	207	Jupyter Notebook
28	petrobras/WPRAutoencoders This is one of Petrobras' open repositories on GitHub. It contains the...	33	Emerging	45	Jupyter Notebook
29	nnormandin/Conditional_VAE conditional variational autoencoder written in Keras [not actively maintained]	32	Emerging	97	Jupyter Notebook
30	zhanghang1989/Torch-Encoding-Layer Deep Texture Encoding Network	30	Emerging	92	Lua
31	xuxu-wei/SUAVE Implementation of a Hybrid Variational Autoencoder (VAE) for label...	30	Emerging	47	Python
32	ggbioing/mcvae Multi-Channel Variational Auto Encoder: A Bayesian Deep Learning Framework...	29	Experimental	31	Python
33	loeweX/ComplexAutoEncoder Code for the paper: Complex-Valued Autoencoders for Object Discovery	29	Experimental	59	Python
34	microsoft/BindVAE Variational Auto Encoders for learning binding signatures of transcription factors	29	Experimental	14	Python
35	niazangels/vae-pokedex The companion code for How to Autoencode your Pokemon	29	Experimental	80	Jupyter Notebook
36	matthewvowels1/Awesome-VAEs A curated list of awesome work on VAEs, disentanglement, representation...	28	Experimental	842	—
37	xuanlinli17/autoregressive_inference Code for "Discovering Non-monotonic Autoregressive Orderings with...	28	Experimental	12	Python
38	despoisj/ConvolutionalAutoencoder Quick and dirty example of the application of convolutional autoencoders in...	28	Experimental	53	Python
39	s-omranpour/X-VAE-keras simple implementations of different kinds of VAE in tf.keras	28	Experimental	12	Jupyter Notebook
40	gd-vae/gd-vae Geometric Dynamic Variational Autoencoders (GD-VAEs) for learning embedding...	28	Experimental	37	TeX
41	rese1f/Awesome-VQVAE A collection of resources and papers on Vector Quantized Variational...	28	Experimental	331	—
42	yukara-ikemiya/minimal-sqvae A minimal Pytorch Implementation of Stochastically Quantized Variational...	27	Experimental	34	Python
43	SerezD/vqvae-vqgan-pytorch-lightning VQ-VAE/GAN implementation in pytorch-lightning	27	Experimental	49	Python
44	ttgump/spaVAE Dependency-aware deep generative models for multitasking analysis of spatial...	27	Experimental	43	Jupyter Notebook
45	julian-8897/conv-vae-pytorch Convolutional Variational Autoencoder in Pytorch benchmarked on CelebA Dataset	26	Experimental	20	Jupyter Notebook
46	cheind/autoregressive :kiwi_fruit: Autoregressive Models in PyTorch.	25	Experimental	81	Python
47	1Konny/VQ-VAE Pytorch Implementation of "Neural Discrete Representation Learning"	25	Experimental	92	Jupyter Notebook
48	TheEngineerProgrammer/my_masters_thesis cVAE, VQ-VAE, VQ-VAE2, cVAE-cGAN, PixelCNN and Gated PixelCNN in tensorflow...	25	Experimental	4	Jupyter Notebook
49	duanzhiihao/qres-vae Authors' PyTorch implementation of "Lossy Image Compression with Quantized...	24	Experimental	16	Jupyter Notebook
50	dhanushkamath/VariationalAutoencoder An IPython notebook explaining the concepts of Variational Autoencoders and...	24	Experimental	21	Jupyter Notebook
51	gulfaraz/balancing-shape-texture-bias-in-imagenet-trained-vgg19 A study on the effect of normalization in predictions by CNN models	24	Experimental	4	Jupyter Notebook
52	SekiroRong/KAN-AutoEncoder KAE : KAN-based AutoEncoder (AE, VAE, VQ-VAE, RVQ, etc.)	22	Experimental	39	Jupyter Notebook
53	MikolajSzawerda/music-sae Sparse Autoencoders (SAEs) for unsupervised music representation learning.	22	Experimental	4	Jupyter Notebook
54	mdhabibi/DeepLearning-VAE Exploring the depths of generative learning with a $\beta$-Variational...	22	Experimental	7	Jupyter Notebook
55	AmirhosseinHonardoust/Teaching-Neural-Networks-to-Imagine-Tables A comprehensive deep dive into how Variational Autoencoders (VAEs) learn to...	21	Experimental	22	—
56	pmarinroig/c-vae A fully functional Convolutional VAE implemented in pure C from scratch.	21	Experimental	23	C
57	codeperfectplus/autoEncoders Deep convolutional autoencoder for image denoising	21	Experimental	8	PureBasic
58	zhiqin1998/RecurrentIPR This is the code repo of our AACL-IJCNLP 2022 on IPR protection of Recurrent...	21	Experimental	8	Python
59	SantoshDhirwani/stick_breaking_vae Implementing Stick-Breaking Variational Auto-encoder in Pytorch & Keras (Python3)	20	Experimental	8	Jupyter Notebook
60	zhihanyang2022/aevb-tutorial Minimal VAE, Conditional VAE (CVAE), Gaussian Mixture VAE (GMVAE) and...	20	Experimental	14	Jupyter Notebook
61	NicklasVraa/VAE_based_segmentation Exploring variational-autoencoder-based semantic segmentation for analyzing CT-scans.	19	Experimental	21	Jupyter Notebook
62	dpernes/vae Customisable Variational Autoencoder in PyTorch	19	Experimental	9	Python
63	13muskanp/Image-Denoising-Using-Autoencoder Building and training an image denoising autoencoder using Keras with...	19	Experimental	4	Jupyter Notebook
64	pierridotite/CIFAR10-CompressAI Convolutional autoencoder for compressing and reconstructing CIFAR-10 images...	18	Experimental	3	Jupyter Notebook
65	andleb/DistributionalAutoencodersScore Official code repository for Distributional Autoencoders Know the Score, NeurIPS 2025	17	Experimental	2	Jupyter Notebook
66	akashsara/fusion-dance Pixel VQ-VAEs for Improved Pixel Art Representation	16	Experimental	17	Jupyter Notebook
67	pi-tau/vae Pytorch implementation of a Variational Autoencoder trained on CIFAR-10. The...	16	Experimental	16	Python
68	aillaud/VQVAE_Flax Implementation of basic autoencodeur, VAE and VQVAE in Flax	15	Experimental	6	Jupyter Notebook
69	khanmhmdi/Cifar-Image-Reconstruction-using-Autoencoder-Models Cifar-10 Image Reconstruction using Auto-encoder Models	13	Experimental	7	Jupyter Notebook
70	sagnik1511/Conv-AE-Tensorflow-Keras Convolutional Auto Encoder in Tensorflow >= 2.2.0	13	Experimental	7	Jupyter Notebook
71	O-Albz/TIF360ProjectD A convolutional conditional variational autoencoder (CVAE) for audio...	13	Experimental	2	Jupyter Notebook
72	Robotmurlock/VariationalAutoEncoder Implementation of the Auto-Encoding Variational Bayes paper in Pytorch with...	13	Experimental	11	Jupyter Notebook
73	hkproj/vae-from-scratch-notes Notes about the video on the Variational Autoencoder	12	Experimental	13	—
74	IParraMartin/Sparse-Autoencoder A PyTorch implementation of a Sparse Auto Encoder (SAE) using MSE loss and...	12	Experimental	27	Python
75	Nehal-Garg/cnn-autoencoder-mnist CNN-based autoencoder for MNIST image reconstruction and representation learning	11	Experimental	—	Jupyter Notebook
76	gabrielelobosco/denoising-autoencoder-r Image Denoising with Autoencoders in R (University Project) Built a...	11	Experimental	—	R
77	isabellahmann/systematic-vae-evaluation Reproducible evaluation of pretrained VAEs + bias/mitigation on low-resource...	11	Experimental	—	Jupyter Notebook

Comparisons in this category

PyTorch-VAE and pytorch-vae (49 vs 43) pytorch-vae and pytorch-vq-vae (43 vs 41)