VAE with a VampPrior
Adam: A Method for Stochastic Optimization