Bayesian Deep Learning Python Packages

uncertainty-toolbox

Uncertainty Toolbox: a Python toolbox for predictive uncertainty quantification, calibration, metrics, and visualization

kfac-jax

Second Order Optimization and Curvature Estimation with K-FAC in JAX.

bayesian-torch

A library for Bayesian neural network layers and uncertainty estimation in Deep Learning extending the core of PyTorch

blitz-bayesian-pytorch

A simple and extensible library to create Bayesian Neural Network layers on PyTorch.

supernnova

Open Source Photometric classification https://supernnova.readthedocs.io

bayestorch

Lightweight Bayesian deep learning library for fast prototyping based on PyTorch

torchbayesian

PyTorch-native Bayesian neural networks for uncertainty quantification in deep learning

bdl-benchmarks

BDL Benchmarks