markov chain monte carlo and variational inference : bridging the gap .
Adam: A Method for Stochastic Optimization