BayesFlow

BayesFlow is a Python library for efficient Bayesian inference with deep learning. It provides users with:

A user-friendly API for amortized Bayesian workflows
A rich collection of generative models, from diffusion to consistency models
Multi-backend support via Keras3: You can use PyTorch, TensorFlow, or JAX

Conceptual Overview

With BayesFlow, you can easily train neural networks for tasks like parameter estimation, model comparison, and validation. It works for both complex simulators that cannot be expressed as parametric models (i.e., simulation-based inference) as well as traditional statistical models. BayesFlow provides a streamlined workflow layer for inference, especially in situations where conventional methods are unavailable or inefficient.

Install

We currently support Python 3.11 to 3.13. You can install the latest stable version from PyPI using:

pip install "bayesflow>=2.0"

If you want the latest features, you can install from source:

pip install git+https://github.com/bayesflow-org/bayesflow.git@dev

If you encounter problems with this or require more control, please refer to the instructions to install from source below.

Backend

To use BayesFlow, you will also need to install one of the following machine learning backends. Note that BayesFlow will not run without a backend.

If you don't know which backend to use, we recommend JAX as it is currently the fastest backend.

As of version 2.0.7, the backend will be set automatically. If you have multiple backends, you can manually set the backend environment variable as described by keras. For example, inside your Python script write:

import os
os.environ["KERAS_BACKEND"] = "jax"
import bayesflow

If you use conda, you can alternatively set this individually for each environment in your terminal. For example:

conda env config vars set KERAS_BACKEND=jax

Or just plainly set the environment variable in your shell:

export KERAS_BACKEND=jax

Getting Started

Using the high-level interface is easy, as demonstrated by the minimal working example below:

import bayesflow as bf

workflow = bf.BasicWorkflow(
    inference_network=bf.networks.FlowMatching(),
    inference_variables=["parameters"],
    inference_conditions=["observables"],
    simulator=bf.simulators.SIR()
)

history = workflow.fit_online(epochs=20, batch_size=32, num_batches_per_epoch=200)

diagnostics = workflow.plot_default_diagnostics(test_data=300)

For an in-depth exposition, check out our expanding list of resources below.

Books

Many examples from Bayesian Cognitive Modeling: A Practical Course by Lee & Wagenmakers (2013) in BayesFlow.

Videos

A few video tutorial videos are available as part of the Learning Bayesian Statistics podcast:

Marvin Schmitt on Amortized Bayesian Inference with Neural Networks
Jonas Arruda on Diffusion Models for Simulation-Based Inference

Tutorial notebooks

Diffusion starter - A small tutorial on the power of diffusion models for SBI.
Linear regression - Fit your first Bayesian regression with varying sample size.
Image data - Learn parameters from or generate image data.
Bayes estimators - From simple point estimates to fully Bayesian inference.
Model comparison - Learn Bayes factors using probabilistic classification.
From ABC to BayesFlow - Upgrade from sequential to amortized inference.
SIR - Model infectuous diseases through an end-to-end Bayesian workflow.
Bayesian experimental design - Perform adaptive sequential experiments.
Estimating likelihoods - Learn synthetic likelihood functions.
Multimodal data - Fuse different data types for more informative inference.
Ensembles - Train different networks at the same time and combine inferences.
Ratio estimation - Learn neural ratios for downstream MCMC sampling.

Tutorial papers

Arruda, J., Bracher, N., Köthe, U., Hasenauer, J., & Radev, S. T. (2025). Diffusion Models in Simulation-Based Inference: A Tutorial Review. arXiv preprint arXiv:2512.20685. Project page. Paper

More tutorials are always welcome! Please consider making a pull request if you have a cool application that you want to contribute.

Contributing

If you want to contribute to BayesFlow, we recommend installing it from source, see CONTRIBUTING.md for more details.

Reporting Issues

If you encounter any issues, please don't hesitate to open an issue here on Github or ask questions on our Discourse Forums.

Documentation & Help

Documentation is available at https://bayesflow.org. Please use the BayesFlow Forums for any BayesFlow-related questions and discussions, and GitHub Issues for bug reports and feature requests.

Citing BayesFlow

If you are using the new multi-backend version of BayesFlow, we recommend citing our new software paper (Kühmichel et al., 2026). For uses of the legacy version, you can still reference Radev et al., (2023).

BibTeX:

@article{kuhmichel2026bayesflow,
  title={{BayesFlow} 2: Multi-backend amortized {B}ayesian inference in Python},
  author={Kühmichel, Lars and Huang, Jerry M and Pratz, Valentin and Arruda, Jonas and Olischläger, Hans and Habermann, Daniel and Kucharsky, Simon and Elsemüller, Lasse and Mishra, Aayush and Bracher, Niels and Jedhoff, Svenja and Schmitt, Marvin and Bürkner, Paul-Christian and Radev, Stefan T},
  journal={arXiv preprint arXiv:2602.07098},
  year={2026}
}

@article{bayesflow_2023_software,
  title = {{BayesFlow}: Amortized {B}ayesian workflows with neural networks},
  author = {Radev, Stefan T and Schmitt, Marvin and Schumacher, Lukas and Elsemüller, Lasse and Pratz, Valentin and Schälte, Yannik and Köthe, Ullrich and Bürkner, Paul-Christian},
  journal = {Journal of Open Source Software},
  volume = {8},
  number = {89},
  pages = {5702},
  year = {2023}
}

FAQ

Question: I am starting with Bayesflow, which backend should I use?

Answer: We recommend JAX as it is currently the fastest backend.

Question: I am getting ModuleNotFoundError: No module named 'tensorflow' when I try to import BayesFlow.

Answer: One of these applies:

You want to use tensorflow as your backend, but you have not installed it. See here.
You want to use a backend other than tensorflow, but have not set the environment variable correctly. See here.
You have set the environment variable, but it is not being picked up by Python. This can happen silently in some development environments (e.g., VSCode or PyCharm). Try setting the backend as shown here in your Python script via os.environ.

Question: What is the difference between Bayesflow 2 and previous versions?

Answer: BayesFlow 2.0+ is a complete rewrite of the library. It shares the same overall goals with previous versions, but has much better modularity and extensibility. What

Bayesflow

Install / Use

README