bayesflow.wrappers

Optional integrations with third-party libraries.

Submodules are imported lazily — neither mamba nor pymc is loaded when you import bayesflow. A descriptive error is raised if the required dependency is missing.

Submodules

mamba

Mamba summary network backed by the mamba-ssm state-space library. Install with pip install mamba-ssm.

pymc

NeuralDistribution — use a trained RatioApproximator (NRE) or ContinuousApproximator (NLE) as a likelihood term inside a PyMC model. Install with pip install pymc.

Examples

>>> from bayesflow.wrappers.pymc import NeuralDistribution
>>> dist = NeuralDistribution(approximator, param_names=["mu", "sigma"])

Modules

mamba