Flipout: Efficient Pseudo-Independent Weight Perturbations on Mini-Batches
auto - encoding variational bayes .