DiCE: The Infinitely Differentiable Monte-Carlo Estimator
auto - encoding variational bayes .