Fast, differentiable, and extensible big bang nucleosynthesis package
Phys. Rev. D 112, 063531 – Published 17 September, 2025
DOI: https://doi.org/10.1103/f3tj-r882
Abstract
We introduce light isotope nucleosynthesis with jax (linx), a new differentiable public big bang nucleosynthesis code designed for fast parameter estimation. By leveraging jax, linx achieves both speed and differentiability, enabling the use of Bayesian inference, including gradient-based methods. We discuss the formalism used in linx for rapid primordial elemental abundance predictions and give examples of how linx can be used. When combined with differentiable cosmic microwave background power spectrum emulators, linx can be used for joint cosmic microwave background and big bang nucleosynthesis analyses without requiring extensive computational resources, including on personal hardware.