Excited to announce that our work has been accepted at Bayesian Deep Learning Workshop at NeurIPS 2021. Preprint availble on arxiv: https://arxiv.org/abs/2111.04665