amortized proximal optimization .
Stochastic Gradient Descent as Approximate Bayesian Inference