dice : the infinitely differentiable monte .
auto - encoding variational bayes .