nngen Deep Generative Models VAE [code] PixelCNN [code] GatedPixelCNN [code] VQ-VAE [code] [EMA version code] Diffusion Models DDPM [ddpm_mnist code] DDIM [ddim_mnist code] DDPM+ [improved_ddpm mnist code] Text-to-Image Models Stable Diffusion [code]