flowMC¤

A JAX-based normalizing-flow-enhanced MCMC sampler for probabilistic inference¤

flowMC is a JAX-based package for normalizing-flow-enhanced Markov chain Monte Carlo (MCMC) sampling. By using normalizing flows as a global proposal, flowMC accelerates convergence for multi-modal and high-dimensional posteriors while running natively on GPU with minimal hyperparameter tuning.

Warning

flowMC has not yet reached v1.0.0 and the API may change. Use at your own risk. Consider pinning to a specific version if you need API stability.

Documentation¤

Installation — How to install flowMC
Quick Start — A basic example to get started
Tutorials — Hands-on worked examples
Guides — In-depth coverage of each module
FAQ — Answers to common questions
Citing flowMC — How to cite flowMC
Contributing to flowMC — How to contribute to flowMC