candl: Cosmic Microwave Background Analysis with a Differentiable Likelihood
arXiv:2401.13433 · doi:10.1051/0004-6361/202449432
Abstract
We present candl, an automatically differentiable python likelihood for analysing cosmic microwave background (CMB) power spectrum measurements. candl is powered by JAX, which makes it fast and easy to calculate derivatives of the likelihood. This facilitates, for example, robust Fisher matrices without finite-difference methods. We show the benefits of candl through a series of example calculations, covering forecasting, robustness tests, and gradient-based Markov chain Monte Carlo sampling. These also include optimising the band power bin width to minimise parameter errors of a realistic mock data set. Moreover, we calculate the correlation of parameter constraints from correlated and partially overlapping subsets of the SPT-3G 2018 data release. In a traditional analysis framework, these tasks are slow and require careful fine-tuning to obtain stable results. As such, a fully differentiable pipeline allows for a higher level of scrutiny; we argue that this is the paradigm shift required to leverage incoming data from ground-based experiments, which will significantly improve the cosmological parameter constraints from the Planck mission. candl comes with the latest primary and lensing power spectrum data from the South Pole Telescope and Atacama Cosmology Telescope collaborations and will be used as part of the upcoming SPT-3G and data releases. Along with the core code, we release a series of auxiliary tools, which simplify common analysis tasks and interface the likelihood with other cosmological software. candl is pip-installable and publicly available at https://github.com/Lbalkenhol/candl .
8 pages, 3 figures
References in corpus (15)
- The NumPy array: a structure for efficient numerical computation
- PolyChord: nested sampling for cosmology
- Composable Effects for Flexible and Accelerated Probabilistic Programming in NumPyro
- CMB Polarization can constrain cosmology better than CMB temperature
- A Measurement of Gravitational Lensing of the Cosmic Microwave Background Using SPT-3G 2018 Data
- Massive data compression for parameter-dependent covariance matrices
- Efficient Cosmological Parameter Estimation with Hamiltonian Monte Carlo
- JAX-COSMO: An End-to-End Differentiable and GPU Accelerated Cosmology Library
- Likelihood methods for the combined analysis of CMB temperature and polarisation power spectra
- CosmoPower-JAX: high-dimensional Bayesian inference with differentiable cosmological emulators
- Capse.jl: efficient and auto-differentiable CMB power spectra emulation
- LimberJack.jl: auto-differentiable methods for angular power spectra analyses
- Uncertainty-aware and Data-efficient Cosmological Emulation using Gaussian Processes and PCA
- Making maps of cosmological parameters
- The Cosmic Microwave Background and Its Polarization
Cited by in corpus (10)
- The CosmoVerse White Paper: Addressing observational tensions in cosmology with systematics and fundamental physics
- SPT-3G D1: CMB temperature and polarization power spectra and cosmology from 2019 and 2020 observations of the SPT-3G Main field
- LINX: A Fast, Differentiable, and Extensible Big Bang Nucleosynthesis Package
- Reducing nuisance prior sensitivity via non-linear reparameterization, with application to EFT analyses of large-scale structure
- jaxspec : a fast and robust Python library for X-ray spectral fitting
- Constraining the primordial power spectrum using a differentiable likelihood
- KiDS-Legacy: Constraining dark energy, neutrino mass, and curvature
- Beyond thermal approximations: Precise cosmological bounds on Axion-Like Particles
- Genetic algorithm demystified for cosmological parameter estimation
- ABCMB: A Python+JAX Package for the Cosmic Microwave Background Power Spectrum