Simple VAE model in pytorch for generating new images - View it on GitHub