stein-variational-gradient-descent
Lightweight Bayesian deep learning library for fast prototyping based on PyTorch