sbi reloaded: A toolkit for simulation-based inference workflows
Jan Boelts*, Michael Deistler*, Manuel Gloeckler, and 30 more authors
Journal of Open Source Software, Apr 2025
* equal contribution
The sbi package provides a unified Python interface for simulation-based inference (SBI) algorithms, enabling researchers to perform Bayesian inference on models with intractable likelihoods across diverse scientific domains. This paper presents sbi reloaded, a comprehensive update that has evolved through extensive community contributions and user feedback. The package now includes multiple state-of-the-art inference algorithms (NPE, NLE, NRE, FMPE, MNLE, and more), enhanced diagnostics and validation tools, improved computational efficiency, and a flexible API supporting custom neural network architectures and proposal distributions. Key improvements include GPU acceleration, distributed training support, comprehensive testing infrastructure, and extensive documentation with tutorials. With over 30 contributors and a growing user base, sbi has become the standard toolkit for neural simulation-based inference, balancing ease of use with the flexibility needed for cutting-edge research.