Probabilistic Machine Learning Python Packages

random-events

A cognitive architecture for the entire stack of robotics

torchebm

🍓 Build and train energy-based and diffusion models in PyTorch ⚡.

inferno-torch

Bayesian Deep Learning in PyTorch

survshap

SurvSHAP(t): Time-dependent explanations of machine learning survival models

libcirkit

a python framework to build, learn and reason about probabilistic circuits and tensor networks

mfpml

Multi-fidelity probability machine learning

fglib2

Clean Factor Graphs in Python

neurobayes

Fully and Partially Bayesian Neural Networks