# CERIDWEN: Fast and Flexible GPU-Accelerated Stellar Population Inference

Preprint · Amanda Stoffers, Sandro Tacchella, Benjamin D. Johnson

- Archive id: arxiv:2609.30145
- Published: 2026-09-24
- arXiv: https://arxiv.org/abs/2609.30145
- Source URL: https://arxiv.org/abs/2609.30145
- PDF: https://arxiv.org/pdf/2609.30145
- Provenance: arxiv

## Abstract

JWST has increased both the number of high-redshift galaxies with high-quality spectral energy distributions (SEDs) and their information content. In parallel, wide-area surveys from Euclid, Rubin's LSST, and Roman will increase galaxy samples by orders of magnitude. Analysing these datasets requires stellar-population models that are both flexible and computationally efficient. We present CERIDWEN, a GPU-native SED fitting framework written in JAX, with an end-to-end differentiable forward model spanning stellar populations, nebular emission, dust attenuation and emission, and projection into the observer frame. Its vectorised, compiled architecture lets nested sampling replace a batch of live points in parallel on the GPU, which makes flexible stellar-population models tractable under full Bayesian inference. Automatic differentiation also provides exact gradients for the gradient-based samplers in the package. We jointly infer time-dependent chemical-enrichment histories instead of a single stellar metallicity, and demonstrate non-parametric star-formation histories (SFHs) with $\\sim$120 age bins. Using $α$-enhanced stellar libraries from FSPS, CERIDWEN can sample stellar \[$α$/Fe\] jointly with \[Fe/H\], mass, and SFH, so that the joint posterior represents the \[Fe/H\]-\[$α$/Fe\] degeneracy explicitly. In controlled mocks, CERIDWEN recovers parameters with well-calibrated posterior uncertainties, while fits to real JWST observations reproduce posteriors from the established Prospector framework: on a single GPU, CERIDWEN completes a fit in a median sampling time of $\\sim$4 min, $\\sim$134$\\times$ faster per fit than equivalent CPU-based Prospector runs. CERIDWEN therefore makes full Bayesian inference practical for larger galaxy samples and more flexible stellar-population models, reducing computational constraints on the physical complexity explored in SED fitting.

_Abstract reproduced under CC0-1.0 (arXiv descriptive metadata)._

---
Space Kitty Archive · https://space-kitty.com · API: https://space-kitty.com/api/v1 · MCP: https://space-kitty.com/mcp · Premium: $0.20/request via x402 (Base Sepolia (testnet)). Metadata: arXiv (CC0), OpenAlex (CC0), Europe PMC (CC BY/CC0 works), ClinicalTrials.gov.