A clean, mathematically grounded implementation of a Variational Autoencoder (VAE) and $\beta$-VAE in PyTorch, trained on the Fashion-MNIST dataset. This repository focuses on a technically precise ...
NVAE is a deep hierarchical variational autoencoder that enables training SOTA likelihood-based generative models on several image datasets. Above, the images will be ...