ranganathkrishnan/bayesian-torch

Bayesian-Torch is a library of neural network layers and utilities extending the core of PyTorch to enable the user to perform stochastic variational inference in Bayesian deep neural networks

PythonBSD-3-Clause