Bayesian Neural Networks Python Packages

uncertainty-toolbox

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

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

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

pytorch-ard

Pytorch implementation of Variational Dropout Sparsifies Deep Neural Networks

probflow

A Python package for building Bayesian models with TensorFlow or PyTorch

multimodal-auv

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

bayesgm

bayesgm: An AI-powered versatile Bayesian Generative Modeling Framework

neurobayes

Fully and Partially Bayesian Neural Nets

causalbgm

CausalBGM: An AI-powered Bayesian generative modeling approach for causal inference in observational studies

bnnsurv

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

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