Consider Epistemic Neural Networks

[MaxGhenis]

Deepmind just released a paper on this: https://arxiv.org/abs/2107.08924

Repo: https://github.com/deepmind/enn

This paper also proposes evaluating uncertainty with KL divergence