BlackJAX Integration

Internal API reference for the BlackJAX backend implementation used by MCMCInference.

BlackJAX implementations for filter-based posterior inference.

run_blackjax_mcmc(mcmc_config: BaseMCMCConfig, rng_key: jnp.ndarray, model: Callable, obs_times: jnp.ndarray, obs_values: jnp.ndarray, ctrl_times: jnp.ndarray | None = None, ctrl_values: jnp.ndarray | None = None, *model_args, **model_kwargs) -> dict

Run BlackJAX-based inference and return posterior samples.