bayesian-layers
A library for Bayesian neural network layers and uncertainty estimation in Deep Learning extending the core of PyTorch
A simple and extensible library to create Bayesian Neural Network layers on PyTorch.