Pi Vae Pytorch Beta