Bayesian Neural Networks Python Packages

uncertainty-toolbox

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

probflow

A Python package for building Bayesian models with TensorFlow or PyTorch

pybandits

Python library for Multi-Armed Bandits

bayesian-torch

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

supernnova

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

deepgp-torch

Deep Gaussian processes based on GPyTorch — doubly-stochastic variational DGPs: modular SVGP layers, a configurable deep-GP builder, ELBO training, and calibrated prediction

blitz-bayesian-pytorch

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

multimodal-auv

Leveraging Bayesian Neural Networks for multimodal AUV data fusion, enabling precise and uncertainty-aware mapping of underwater environments.

bayestorch

Lightweight Bayesian deep learning library for fast prototyping based on PyTorch

neurobayes

Fully and Partially Bayesian Neural Nets

causalbgm

bayesgm: An AI-powered versatile Bayesian Generative Modeling Framework

bayesgm

bayesgm: An AI-powered versatile Bayesian Generative Modeling Framework

bnnsurv

Official TensorFlow implementation of Uncertainty Estimation in Deep Bayesian Survival Models (BHI 2023)

pytorch-ard

Pytorch implementation of Variational Dropout Sparsifies Deep Neural Networks

torchbayesian

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

pybaobab

Generates the training data for hierarchical inference of strongly-lensed systems with Bayesian neural networks

safeai

Reusable, Easy-to-use Uncertainty module package built with Tensorflow, Keras