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]