LINX: A Fast, Differentiable, and Extensible Big Bang Nucleosynthesis Package

Abstract

We introduce LINX (Light Isotope Nucleosynthesis with JAX), a new differentiable public Big Bang Nucleosynthesis (BBN) 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 (CMB) power spectrum emulators, LINX can be used for joint CMB and BBN analyses without requiring extensive computational resources, including on personal hardware.

References in corpus (60)

• SciPy 1.0--Fundamental Algorithms for Scientific Computing in Python
• Array Programming with NumPy
• Cosmological parameters from CMB and other data: a Monte-Carlo approach
• MCMC using Hamiltonian dynamics
• The Cosmic Linear Anisotropy Solving System (CLASS) II: Approximation schemes
• One percent determination of the primordial deuterium abundance
• CMB power spectrum parameter degeneracies in the era of precision cosmology
• The effects of He I 10830 on helium abundance determinations
• Precision big bang nucleosynthesis with improved Helium-4 predictions
• Big-Bang Nucleosynthesis After Planck
• The Cosmic Linear Anisotropy Solving System (CLASS) IV: Efficient implementation of non-cold relics
• NACRE II: an update of the NACRE compilation of charged-particle-induced thermonuclear reaction rates for nuclei with mass number $A<16$
• Neutrino decoupling including flavour oscillations and primordial nucleosynthesis
• Primordial Nucleosynthesis for the New Cosmology: Determining Uncertainties and Examining Concordance
• Compilation and R-matrix analysis of Big Bang nuclear reaction rates
• PArthENoPE: Public Algorithm Evaluating the Nucleosynthesis of Primordial Elements
• Nuclear Reaction Network for Primordial Nucleosynthesis: a detailed analysis of rates, uncertainties and light nuclei yields
• Standard Big-Bang Nucleosynthesis up to CNO with an improved extended nuclear network
• Precision Early Universe Thermodynamics made simple: $N_{\mathrm{eff}}$ and Neutrino Decoupling in the Standard Model and beyond
• COSMOPOWER: emulating cosmological power spectra for accelerated Bayesian inference from next-generation surveys
• Refined Bounds on MeV-scale Thermal Dark Sectors from BBN and the CMB
• Neutrino decoupling beyond the Standard Model: CMB constraints on the Dark Matter mass with a fast and precise $N_{\mathrm{eff}}$ evaluation
• Neutron Lifetime and Axial Coupling Connection
• AlterBBN: A program for calculating the BBN abundances of the elements in alternative cosmologies
• PArthENoPE reloaded
• Radiative neutron capture on a proton at BBN energies
• New reaction rates for improved primordial D/H calculation and the cosmic evolution of deuterium
• A new tension in the cosmological model from primordial deuterium?
• Towards a precision calculation of the effective number of neutrinos $N_{\mathrm{eff}}$ in the Standard Model I: The QED equation of state
• Primordial Deuterium after LUNA: concordances and error budget
• Thermal Dark Matter Below an MeV
• The Impact of New d(p,γ)He3 Rates on Big Bang Nucleosynthesis
• Updated BBN constraints on electromagnetic decays of MeV-scale particles
• Bayesian Estimation of Thermonuclear Reaction Rates
• JAX-COSMO: An End-to-End Differentiable and GPU Accelerated Cosmology Library
• Nucleosynthesis Without a Computer
• Towards a precision calculation of $N_{\mathrm{eff}}$ in the Standard Model III: Improved estimate of NLO contributions to the collision integral
• Joint CMB and BBN Constraints on Light Dark Sectors with Dark Radiation
• Indications for a Nonzero Lepton Asymmetry from Extremely Metal-Poor Galaxies
• Data compression in cosmology: A compressed likelihood for Planck data
• Primordial neutrino asymmetry evolution with full mean-field effects and collisions
• Resolving conclusions about the early Universe requires accurate nuclear measurements
• Thermonuclear reaction rates and primordial nucleosynthesis
• Dark radiation constraints on portal interactions with hidden sectors
• CosmoPower-JAX: high-dimensional Bayesian inference with differentiable cosmological emulators
• Thermonuclear fusion rates for tritium + deuterium using Bayesian methods
• Bayesian estimation of thermonuclear reaction rates for deuterium+deuterium reactions
• Addendum: Refined bounds on MeV-scale thermal dark sectors from BBN and the CMB
• Capse.jl: efficient and auto-differentiable CMB power spectra emulation
• Astrophysical S-factors, thermonuclear rates, and electron screening potential for the ${}^3$He(d,p)${}^4$He Big Bang reaction via a hierarchical Bayesian model
• candl: Cosmic Microwave Background Analysis with a Differentiable Likelihood
• What does cosmology tell us about the mass of thermal-relic dark matter?
• Bayesian Estimation of the D(p,$\gamma$)${}^3$He Thermonuclear Reaction Rate
• Hierarchical Bayesian Thermonuclear Rate for the ${}^7$Be(n,p)${}^7$Li Big Bang Nucleosynthesis Reaction
• Differentiable Matrix Elements with MadJax
• DISCO-DJ I: a differentiable Einstein-Boltzmann solver for cosmology
• QED corrections to the big-bang nucleosynthesis reaction rates
• Extragalactic 85Rb/87Rb and 6Li/7Li ratios in the Small Magellanic Cloud
• Neutrino-Dark Sector Equilibration and Primordial Element Abundances
• Cosmological Parameter Estimation with a Joint-Likelihood Analysis of the Cosmic Microwave Background and Big Bang Nucleosynthesis

Cited by in corpus (5)

• Fast and Flexible Neutrino Decoupling Part I: The Standard Model
• Cosmological Parameter Estimation with a Joint-Likelihood Analysis of the Cosmic Microwave Background and Big Bang Nucleosynthesis
• Inside the Black Box of Big Bang Nucleosynthesis: A Comprehensive Sensitivity Atlas for the Precision Era
• ABCMB: A Python+JAX Package for the Cosmic Microwave Background Power Spectrum
• Insights for Early Dark Energy with Big Bang Nucleosynthesis