REVIEW 11 cited by
CosmoPower-JAX: high-dimensional Bayesian inference with differentiable cosmological emulators
Not yet reviewed by Pith; the record is open.
This paper has not been read by Pith yet. Machine review is queued; the pith claim, tier, and objections will appear here once it completes.
SPECIMEN: schema-true, not a live event
T0 review · schema-true
One-sentence machine reading of the paper's core claim.
pith:XXXXXXXX · record.json · timestamp
CosmoPower-JAX: high-dimensional Bayesian inference with differentiable cosmological emulators
read the original abstract
We present CosmoPower-JAX, a JAX-based implementation of the CosmoPower framework, which accelerates cosmological inference by building neural emulators of cosmological power spectra. We show how, using the automatic differentiation, batch evaluation and just-in-time compilation features of JAX, and running the inference pipeline on graphics processing units (GPUs), parameter estimation can be accelerated by orders of magnitude with advanced gradient-based sampling techniques. These can be used to efficiently explore high-dimensional parameter spaces, such as those needed for the analysis of next-generation cosmological surveys. We showcase the accuracy and computational efficiency of CosmoPower-JAX on two simulated Stage IV configurations. We first consider a single survey performing a cosmic shear analysis totalling 37 model parameters. We validate the contours derived with CosmoPower-JAX and a Hamiltonian Monte Carlo sampler against those derived with a nested sampler and without emulators, obtaining a speed-up factor of $\mathcal{O}(10^3)$. We then consider a combination of three Stage IV surveys, each performing a joint cosmic shear and galaxy clustering (3x2pt) analysis, for a total of 157 model parameters. Even with such a high-dimensional parameter space, CosmoPower-JAX provides converged posterior contours in 3 days, as opposed to the estimated 6 years required by standard methods. CosmoPower-JAX is fully written in Python, and we make it publicly available to help the cosmological community meet the accuracy requirements set by next-generation surveys.
Forward citations
Cited by 11 Pith papers
-
Strongest constraints on dark acoustic oscillations from the Lyman-alpha forest
First constraints from Ly-alpha forest limit DAO participation to at most 30% of dark matter for peaks below 50 h/Mpc at 95% CL using a deep kernel emulator of simulations.
-
Multi-tracers, multi-surveys: a joint Fisher analysis of DESI+PFS
Multi-tracer analysis of the PFS-DESI overlap calibrates EFT priors for DESI, improving σ(fσ8) by 8% and σ(Mν) by 54% via better b1σ8 constraints.
-
Alleviating prior dependencies for DESI DR1 clustering fits through reparameterization
A hybrid Jeffreys+baseline prior removes multi-σ prior-volume projection in DESI DR1 full-shape fits of H0, w0, and wa, yielding late-time expansion constraints consistent with HOD-informed Bayesian and frequentist analyses.
-
Alleviating prior dependencies for DESI DR1 clustering fits through reparameterization
Jeffreys prior over EFTofLSS coefficients mitigates projection effects in DESI DR1 power spectrum multipole fits, recentering posteriors for late-time expansion parameters.
-
Multi-tracers, multi-surveys: a joint Fisher analysis of DESI+PFS
A joint multi-tracer Fisher analysis of DESI and PFS cross-spectra calibrates bias parameters from data, improving forecasted constraints on fσ8 by 33%, neutrino mass by 80%, and Ωm by 49% compared to single-tracer baselines.
-
Using Neural Emulators and Hamiltonian Monte Carlo to constrain the Epoch of Reionization's History with the Ly$\alpha$ Forest Power Spectrum
A JAX pipeline that emulates both the Lyα forest power spectrum and its model-dependent covariance with neural networks, then runs Hamiltonian Monte Carlo, recovers reionization-history parameters from mock observations.
-
SPT-3G D1: CMB temperature and polarization power spectra and cosmology from 2019 and 2020 observations of the SPT-3G Main field
SPT-3G delivers the most precise CMB EE and TE spectra at high multipoles to date, giving LCDM parameters with H0 = 66.66 ± 0.60 km/s/Mpc from ground-based data alone and reaching Planck-level constraints when combine...
-
Symbolic Emulators for Cosmology: Accelerating Cosmological Analyses Without Sacrificing Precision
Symbolic emulators approximate key Lambda CDM functions to 0.001-0.05% accuracy across relevant redshifts and Omega_m values, enabling faster 3x2pt inference with consistent results.
-
halox: Dark matter halo properties and large-scale structure calculations using JAX
A new JAX library provides differentiable, GPU-accelerated implementations of NFW halo profiles, the Tinker et al. halo mass function and halo bias, validated against colossus and Astropy.
-
cloelike: A Python Library for Cosmological Likelihood Inference in the Euclid Era
cloelike is a new open Python package implementing composable Gaussian likelihoods for WL, GCph, GGL, full-shape spectra, and BAO in joint probe combinations for Euclid analyses.
-
cloelib: A Flexible Python Library for Computing Cosmological Observables in the Euclid Era
cloelib is a modular JAX-based Python library for cosmological observables intended as reference infrastructure for Euclid's first data release.
discussion (0)
Sign in with ORCID, Apple, or X to comment. Anyone can read and Pith papers without signing in.