Probabilistic Machine Learning Python Packages

synthefy-nori

Open-source tabular foundation model for fast, training-free regression with predictive distributions.

torchebm

🍓 Simulation-free, GPU-first generative modeling in PyTorch ⚡ Composable primitives for scalable, stable training of modern EBMs, diffusion, flow matching, and Schrödinger bridges.

survshap

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

inferno-torch

Bayesian Deep Learning in PyTorch

fglib2

Clean Factor Graphs in Python

libcirkit

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

neurobayes

Fully and Partially Bayesian Neural Nets

mfpml

Multi-fidelity probability machine learning

synthefy-tabular

Open-source tabular foundation model for fast, training-free regression with predictive distributions.