Introducing sapphire : Towards Hybrid Physics-informed, Data-driven Modeling of Galaxy Formation
Abstract Semianalytic models (SAMs) have been treating galaxy populations as dynamical systems for ≳50 yr, but their evolution equations remain poorly constrained. We introduce sapphire , a modular, automatically differentiable, GPU-accelerated SAM written in JAX . For the first time, we compute exact Jacobian and Hessian matrices of a galaxy formation SAM, using the V. Pandya et al. (2023) nonlinear differential equation system as an example. These allow efficient, interpretable local and global sensitivity analyses, which reveal that supernova energy loading is the key astrophysical parameter. We use gradient descent and Hamiltonian Monte Carlo (HMC) to perform comprehensive mock parameter recovery tests. These indicate that the z = 0 stellar-to-halo-mass relation alone does not contain enough information to infer many astrophysical parameters. Using observations of star-forming galaxies from the MaNGA survey and the P. Behroozi et al. (2019) empirical model as one baseline, we derive multiple posteriors assuming different combinations of data, including z = 0 interstellar medium gas fractions and metallicities. The inferred physical parameters suggest that galaxies self-regulate their star formation primarily through preventative rather than ejective feedback, though this remains uncertain due to the lack of satellite galaxies, black holes, and multiphase galactic atmosphere physics. Both Fisher and HMC forecasts demonstrate the potential of sapphire to enable precision inference for galaxy formation and cosmology in a hybrid physics-informed, data-driven way, but more work is needed to expand its library of models and methods. We make sapphire publicly available at https://github.com/virajpandya/sapphire .
Authors
- Richard Stiskalek (ORCID: https://orcid.org/0000-0002-0986-314X)
- Greg L. Bryan (ORCID: https://orcid.org/0000-0003-2630-9228)
- Pablo Lemos (ORCID: https://orcid.org/0000-0002-4728-8473)
- Matthew Ho (ORCID: https://orcid.org/0000-0003-3207-8868)
- Austen Gabrielpillai (ORCID: https://orcid.org/0000-0003-4295-3793)
- Drummond B. Fielding (ORCID: https://orcid.org/0000-0003-3806-8548)
- Bryan A. Terrazas (ORCID: https://orcid.org/0000-0001-5529-7305)
- Rachel S. Somerville (ORCID: https://orcid.org/0000-0002-6748-6821)
- G. Mark Voit (ORCID: https://orcid.org/0000-0002-3514-0383)
- Tjitske Starkenburg (ORCID: https://orcid.org/0000-0003-2539-8206)
- Sophie Koudmani (ORCID: https://orcid.org/0000-0002-1528-5091)
- Kartheik G. Iyer (ORCID: https://orcid.org/0000-0001-9298-3523)
- Lucia A. Perez (ORCID: https://orcid.org/0000-0002-8449-1956)
- Christopher C. Lovell (ORCID: https://orcid.org/0000-0001-7964-5933)
- T. Lucas Makinen (ORCID: https://orcid.org/0000-0002-3795-6933)
- Christian Kragh Jespersen (ORCID: https://orcid.org/0000-0002-8896-6496)
- Lars Hernquist (ORCID: https://orcid.org/0000-0001-6950-1629)
- Viraj Pandya (ORCID: https://orcid.org/0000-0002-2499-9205)
- Christopher Carr (ORCID: https://orcid.org/0000-0002-5840-0424)
- William F. Robinson
- Marta Laska
Institutions
- Northwestern University (US)
- Pennsylvania State University (US)
- University of Hertfordshire (GB)
- Princeton University (US)
- University of Cambridge (GB)
- University of Oxford (GB)
- Center for Astrophysics Harvard & Smithsonian (US)
- Mila - Quebec Artificial Intelligence Institute (CA)
- Flatiron Institute (US)
- NSF-Simons AI Institute for the Sky (US)
- New York University (US)
- Oberlin College (US)
- Université de Montréal (CA)
- Columbia University (US)
- Michigan State University (US)
Publication Details
- Journal
- The Astrophysical Journal
- Published
- 2026-10-26
- DOI
- https://doi.org/10.3847/1538-4357/aea55e
- Primary Topic
- Galaxies: Formation, Evolution, Phenomena
- Type
- article
- Field-Weighted Citation Impact
- 0.00
Funders
- National Aeronautics and Space Administration
- Simons Foundation