CMB Analysis With A Differentiable Likelihood

Authors

L. Balkenhol, C. Trendafilova, K. Benabed, S. Galli

Paper

Source

https://github.com/Lbalkenhol/candl

Documentation

candl is a differentiable likelihood framework for analysing CMB power spectrum measurements. Key features are:

• JAX-compatibility, allowing for fast and easy computation of gradients and Hessians of the likelihoods.
• The latest public data releases from the South Pole Telescope and Atacama Cosmology Telescope collaborations.
• Interface tools for work with other popular cosmology software packages (e.g. Cobaya and MontePython).
• Auxiliary tools for common analysis tasks (e.g. generation of mock data).

candl supports the analysis of primary CMB and lensing power spectrum data (TT, TE, EE, BB, ϕϕ, κκ).

Installation

candl can be installed with pip:

pip install candl-like

After installation, we recommend testing by executing the following python code:

import candl.tests
candl.tests.run_all_tests()

This well test all data sets included in candl.

Data Sets

The pip installation of candl currently ships with the following data sets:

• SPT-3G 2018 TT/TE/EE (Balkenhol et al. 2023)
• SPT-3G 2018 Lensing (Pan et al. 2023)
• ACT DR4 TT/TE/EE (Aiola et al. 2020, Choi et al. 2020)
• ACT DR6 Lensing (Madhavacheril et al. 2023, Qu et al. 2023)

Detailed information on these data sets, how to install data sets separately from the likelihood code, and instructions on how you can add your own data sets can be found in the docs.

JAX

JAX is a Google-developed python library. In its own words: "JAX is Autograd and XLA, brought together for high-performance numerical computing."

candl is written in a JAX-friendly way. That means JAX is optional and you can install and run candl without JAX and perform traditional inference tasks such as MCMC sampling with Cobaya. However, if JAX is installed, the likelihood is fully differentiable thanks to automatic differentiation and many functions are jitted for speed.

Documentation

You can find the documentation here.

Citing candl

If you use candl please cite the release paper. Be sure to also cite the relevant papers for any samplers, theory codes, and data sets you use.

---