stochastic-variational-inference
A library for Bayesian neural network layers and uncertainty estimation in Deep Learning extending the core of PyTorch