Symbolic Implementation of Extensions of the $\text{PyCosmo}$ Boltzmann Solver

Abstract

$\text{PyCosmo}$ is a Python-based framework for the fast computation of cosmological model predictions. One of its core features is the symbolic representation of the Einstein-Boltzmann system of equations. Efficient $\text{C/C++}$ code is generated from the $\text{SymPy}$ symbolic expressions making use of the $\text{sympy2c}$ package. This enables easy extensions of the equation system for the implementation of new cosmological models. We illustrate this with three extensions of the $\text{PyCosmo}$ Boltzmann solver to include a dark energy component with a constant equation of state, massive neutrinos and a radiation streaming approximation. We describe the $\text{PyCosmo}$ framework, highlighting new features, and the symbolic implementation of the new models. We compare the $\text{PyCosmo}$ predictions for the $\Lambda$CDM model extensions with $\text{CLASS}$, both in terms of accuracy and computational speed. We find a good agreement, to better than 0.1% when using high-precision settings and a comparable computational speed. Links to the Python Package Index (PyPI) page of the code release and to the PyCosmo Hub, an online platform where the package is installed, are available at: https://cosmology.ethz.ch/research/software-lab/PyCosmo.html.

arXiv comment35 pages including 5 figures and 3 tables. Link to $\text{PyCosmo}$ package: https://cosmology.ethz.ch/research/software-lab/PyCosmo.html

References in corpus (39)

• SciPy 1.0--Fundamental Algorithms for Scientific Computing in Python
• The NumPy array: a structure for efficient numerical computation
• Evidence for oscillation of atmospheric neutrinos
• Efficient Computation of CMB anisotropies in closed FRW models
• Direct Evidence for Neutrino Flavor Transformation from Neutral-Current Interactions in the Sudbury Neutrino Observatory
• The Complete Light-curve Sample of Spectroscopically Confirmed Type Ia Supernovae from Pan-STARRS1 and Cosmological Constraints from The Combined Pantheon Sample
• First Results from KamLAND: Evidence for Reactor Anti-Neutrino Disappearance
• The Cosmic Linear Anisotropy Solving System (CLASS) II: Approximation schemes
• Baryonic Features in the Matter Transfer Function
• A Line of Sight Approach to Cosmic Microwave Background Anisotropies
• Cosmological Perturbation Theory in the Synchronous and Conformal Newtonian Gauges
• Cosmological Perturbation Theory in the Synchronous vs. Conformal Newtonian Gauge
• Dark Energy Survey Year 3 Results: Cosmological Constraints from Galaxy Clustering and Weak Lensing
• Massive neutrinos and cosmology
• Early Dark Energy Can Resolve The Hubble Tension
• 2020 Global reassessment of the neutrino oscillation picture
• The linear power spectrum of observed source number counts
• The Cosmic Linear Anisotropy Solving System (CLASS) I: Overview
• What galaxy surveys really measure
• Wave Dark Matter
• The Cosmic Linear Anisotropy Solving System (CLASS) IV: Efficient implementation of non-cold relics
• Early Dark Energy Does Not Restore Cosmological Concordance
• New Perspective on Galaxy Clustering as a Cosmological Probe: General Relativistic Effects
• Core Cosmology Library: Precision Cosmological Predictions for LSST
• General Relativistic Description of the Observed Galaxy Power Spectrum: Do We Understand What We Measure?
• Effective Field Theory of Cosmic Acceleration: an implementation in CAMB
• The General Primordial Cosmic Perturbation
• hi_class: Horndeski in the Cosmic Linear Anisotropy Solving System
• The CLASSgal code for Relativistic Cosmological Large Scale Structure
• The evolution of galaxy clustering
• Constraining Neutrino Masses by CMB Experiments Alone
• Dark energy with non-adiabatic sound speed: initial conditions and detectability
• Dark energy equation of state parameter and its variation at low redshifts
• iCosmo: an Interactive Cosmology Package
• The Cosmic Linear Anisotropy Solving System (CLASS) III: Comparision with CAMB for LambdaCDM
• Photons and Baryons before Atoms: Improving the Tight-Coupling Approximation
• Speeding Up Cosmological Boltzmann Codes
• The Einstein-Boltzmann equations revisited
• Optimal Boltzmann hierarchies with non-vanishing spatial curvature

Cited by in corpus (6)

• The CosmoVerse White Paper: Addressing observational tensions in cosmology with systematics and fundamental physics
• Redshift requirements for cosmic shear with intrinsic alignment
• Fast Simulation of Cosmological Neutral Hydrogen based on the Halo Model
• Assessing theoretical uncertainties for cosmological constraints from weak lensing surveys
• SHAM-OT: Rapid Subhalo Abundance Matching with Optimal Transport
• galsbi: A Python package for the GalSBI galaxy population model