Bayesian Neural Networks Python Packages

uncertainty-toolbox

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

supernnova

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

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.

bayestorch

Lightweight Bayesian deep learning library for fast prototyping based on PyTorch

pytorch-ard

Make your PyTorch faster

torchbayesian

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

multimodal-auv

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

probflow

A Python package for building Bayesian models with TensorFlow or 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)

pybaobab

Data generator for hierarchically modeling strongly-lensed systems with Bayesian neural networks

safeai

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