jax_cosmo.power module

jax_cosmo.power.primordial_matter_power(cosmo, k)

Primordial power spectrum Pk = k^n

jax_cosmo.power.linear_matter_power(cosmo, k, a=1.0, transfer_fn=<function Eisenstein_Hu>, **kwargs)

Computes the linear matter power spectrum.

Parameters:

• k (array_like) – Wave number in h Mpc^{-1}
• a (array_like, optional) – Scale factor (def: 1.0)
• transfer_fn (transfer_fn(cosmo, k, **kwargs)) – Transfer function

Returns:

pk – Linear matter power spectrum at the specified scale and scale factor.

Return type:

array_like

jax_cosmo.power.nonlinear_matter_power(cosmo, k, a=1.0, transfer_fn=<function Eisenstein_Hu>, nonlinear_fn=<function halofit>)

Computes the non-linear matter power spectrum.

This function is just a wrapper over several nonlinear power spectra.