PyTorch implementation of bayesian neural network [torchbnn] - View it on GitHub