OhGreat/bayesian_nn_pytorch

Bayesian Neural Networks modules in Pytorch.

Bayesian Neural Networks in Pytorch

Repository under work Bayesian Neural Network modules for experimentation.

References:

local reparametrization trick from: https://arxiv.org/abs/1506.02557

initialization based on: https://arxiv.org/pdf/1810.01279.pdf