${\text{SwiftC}}_{\ell }$: fast differentiable angular power spectra beyond Limber

Abstract

The upcoming stage IV wide-field surveys will provide high precision measurements of the large-scale structure (LSS) of the universe. Their interpretation requires fast and accurate theoretical predictions including large scales. For this purpose, we introduce ${\text{SwiftC}}_{\ell }$, a fast, accurate and differentiable $\text{JAX}$-based pipeline for the computation of the angular power spectrum beyond the Limber approximation. It uses a new FFTLog-based method which can reach arbitrary precision and includes interpolation along $k$, allowing for $k$-dependent growth factor and biases. ${\text{SwiftC}}_{\ell }$ includes a wide range of probes and effects such as galaxy clustering, including magnification bias, redshift-space distortions and primordial non-Gaussianity, weak lensing, including intrinsic alignment, cosmic microwave background (CMB) lensing and CMB integrated Sachs-Wolfe effect. We compare our pipeline to the other available beyond-Limber codes within the N5K challenge from the Rubin Observatory Legacy Survey of Space and Time (LSST) Dark Energy Science Collaboration. ${\text{SwiftC}}_{\ell }$ computes the 120 different angular power spectra over 103 $\ell$-multipoles in 5 ms on one GPU core while the computation of the gradient is approximately 4$\times$ slower. Using a pre-calculation, ${\text{SwiftC}}_{\ell }$ is thus about 40$\times$ faster than the winner of the N5K challenge with comparable accuracy. Furthermore, all outputs are auto-differentiable, facilitating gradient-based sampling and robust and accurate Fisher forecasts. We showcase a Markov Chain Monte Carlo, a Hamiltonian Monte Carlo and a Fisher forecast on an LSST-like survey, illustrating ${\text{SwiftC}}_{\ell }$'s differentiability, speed and reliability in measuring cosmological parameters. The code is publicly available at https://cosmo-gitlab.phys.ethz.ch/cosmo_public/swiftcl.

19 pages, 7 figures, accepted to JCAP

References in corpus (21)

• SciPy 1.0--Fundamental Algorithms for Scientific Computing in Python
• Array Programming with NumPy
• emcee: The MCMC Hammer
• The clustering of galaxies in the completed SDSS-III Baryon Oscillation Spectroscopic Survey: cosmological analysis of the DR12 galaxy sample
• Dark Energy Survey Year 1 Results: Cosmological Constraints from Galaxy Clustering and Weak Lensing
• Extended Limber Approximation
• KiDS+VIKING-450: Cosmic shear tomography with optical+infrared data
• The Early Data Release of the Dark Energy Spectroscopic Instrument
• Core Cosmology Library: Precision Cosmological Predictions for LSST
• GetDist: a Python package for analysing Monte Carlo samples
• Beyond Limber: Efficient computation of angular power spectra for galaxy clustering and weak lensing
• Can we actually constrain $f_{\mathrm{NL}}$ using the scale-dependent bias effect? An illustration of the impact of galaxy bias uncertainties using the BOSS DR12 galaxy power spectrum
• JAX-COSMO: An End-to-End Differentiable and GPU Accelerated Cosmology Library
• Limits on primordial non-Gaussianities from BOSS galaxy-clustering data
• Beware of commonly used approximations I: errors in forecasts
• Integrated approach to cosmology: Combining CMB, large-scale structure and weak lensing
• Angpow: a software for the fast computation of accurate tomographic power spectra
• CosmoPower-JAX: high-dimensional Bayesian inference with differentiable cosmological emulators
• The N5K Challenge: Non-Limber Integration for LSST Cosmology
• LimberJack.jl: auto-differentiable methods for angular power spectra analyses
• BLAST: Beyond Limber Angular power Spectra Toolkit. A fast and efficient algorithm for 3x2 pt analysis