PyTorch-VAE TODO VanillaVAE Conditional VAE Gamma VAE Beta VAE InfoVAE WAE AAE TwoStageVAE MMD-VAE VAE-GAN VAE with Vamp Prior IWAE VLAE