gwkokab.cosmology

Cosmology module for GWKokab, adapted from code written by Reed Essick included in the gw-distributions package at source.

Classes

Cosmology

Flat ΛCDM cosmology with Mpc-based units for distances and comoving volumes.

Functions

PLANCK_2013_Cosmology(...)
PLANCK_2015_Cosmology(...)	Cosmology: See Table 4 in arXiv:1502.01589, OmegaMatter from astropy Planck
PLANCK_2018_Cosmology(...)	Cosmology: See Table 1 in arXiv:1807.06209.
default_cosmology(→ gwkokab.cosmology._cosmology.Cosmology)	Returns the default cosmology based on GWKOKAB_DEFAULT_COSMOLOGY.

Package Contents

class gwkokab.cosmology.Cosmology(Ho: jaxtyping.ArrayLike, omega_matter: jaxtyping.ArrayLike, omega_radiation: jaxtyping.ArrayLike, omega_lambda: jaxtyping.ArrayLike, max_z: float = 4.0, dz: float = 0.001)

Bases: equinox.Module

Flat ΛCDM cosmology with Mpc-based units for distances and comoving volumes.

Initialize the cosmology with given parameters.

Parameters:

• Ho (ArrayLike) – Hubble constant in m/s/Mpc.

• omega_matter (ArrayLike) – Matter density parameter (Ω_m).

• omega_radiation (ArrayLike) – Radiation density parameter (Ω_r).

• omega_lambda (ArrayLike) – Dark energy density parameter (Ω_Λ).

• max_z (float, optional) – Maximum redshift to consider, by default 4.0

• dz (float, optional) – Redshift bin width, by default DEFAULT_DZ

DL_to_z(DL: jaxtyping.ArrayLike) → jaxtyping.ArrayLike

Approximate inversion DL -> z using precomputed grid.

dDLdz(z: jaxtyping.ArrayLike) → jaxtyping.ArrayLike

Derivative of luminosity distance with respect to redshift.

dDcdz(z: jaxtyping.ArrayLike) → jaxtyping.ArrayLike

dVcdz(z: jaxtyping.ArrayLike, Dc: jaxtyping.ArrayLike | None = None) → jaxtyping.ArrayLike

logdVcdz(z: jaxtyping.ArrayLike, Dc: jaxtyping.ArrayLike | None = None) → jaxtyping.ArrayLike

z_to_DL(z: jaxtyping.ArrayLike) → jaxtyping.ArrayLike

Luminosity distance in Mpc.

z_to_Dc(z: jaxtyping.ArrayLike) → jaxtyping.ArrayLike

Fast JAX-safe interpolation of comoving distance.

z_to_E(z: jaxtyping.ArrayLike) → jaxtyping.ArrayLike

property DL: jaxtyping.ArrayLike

property Dc: jaxtyping.ArrayLike

Ho: jaxtyping.ArrayLike

OmegaKappa: jaxtyping.ArrayLike

OmegaLambda: jaxtyping.ArrayLike

OmegaMatter: jaxtyping.ArrayLike

OmegaRadiation: jaxtyping.ArrayLike

Vc: jaxtyping.ArrayLike

property z: jaxtyping.ArrayLike

gwkokab.cosmology.PLANCK_2013_Cosmology() → gwkokab.cosmology._cosmology.Cosmology

gwkokab.cosmology.PLANCK_2015_Cosmology() → gwkokab.cosmology._cosmology.Cosmology

Cosmology: See Table 4 in arXiv:1502.01589, OmegaMatter from astropy Planck 2015.

gwkokab.cosmology.PLANCK_2018_Cosmology() → gwkokab.cosmology._cosmology.Cosmology

Cosmology: See Table 1 in arXiv:1807.06209.

gwkokab.cosmology.default_cosmology() → gwkokab.cosmology._cosmology.Cosmology

Returns the default cosmology based on GWKOKAB_DEFAULT_COSMOLOGY.