tszpower: Cosmological Parameter Inference with tSZ Power Spectrum

A repository for computing the thermal Sunyaev-Zel'dovich (tSZ) power spectrum and performing cosmological parameter inference using both Markov Chain Monte Carlo (MCMC) and Simulation-Based Inference (SBI). Built with JAX for accelerated computation and automatic differentiation.

Key Features

• tSZ Power Spectrum Computation: Fast calculation of the tSZ power spectrum using JAX.
• Cosmological Parameter Inference:
  • Traditional MCMC sampling with cobaya
  • Modern simulation-based inference (SBI) with neural density estimators
• JAX-powered: Automatic differentiation.

Installation

To install tszpower in editable mode, clone the repository and run:

pip install -e .

Usage

Example usage of the code is provided in the examples folder. In addition, the code is built upon the notebooks found in the notebook folder.