a lagrangian perspective on latent variable generative models .
Adam: A Method for Stochastic Optimization