jax_cosmo.probes module¶

class jax_cosmo.probes.WeakLensing(redshift_bins, ia_bias=None, multiplicative_bias=0.0, sigma_e=0.26, **kwargs)[source]¶

Bases: jax_cosmo.jax_utils.container

Class representing a weak lensing probe, with a bunch of bins

redshift_bins: list of nzredshift distributions ia_bias: (optional) if provided, IA will be added with the NLA model, either a single bias object or a list of same size as nzs multiplicative_bias: (optional) adds an (1+m) multiplicative bias, either single value or list of same length as redshift bins

sigma_e: intrinsic galaxy ellipticity

kernel(cosmo, z, ell)[source]¶

Compute the radial kernel for all nz bins in this probe.

radial_kernel: shape (nbins, nz)

n_tracers¶: Returns the number of tracers for this probe, i.e. redshift bins

noise()[source]¶: Returns the noise power for all redshifts return: shape [nbins]

zmax¶: Returns the maximum redsfhit probed by this probe

class jax_cosmo.probes.NumberCounts(redshift_bins, bias, has_rsd=False, **kwargs)[source]¶

Bases: jax_cosmo.jax_utils.container

Class representing a galaxy clustering probe, with a bunch of bins

redshift_bins: nzredshift distributions

has_rsd….

kernel(cosmo, z, ell)[source]¶

Compute the radial kernel for all nz bins in this probe.

radial_kernel: shape (nbins, nz)

n_tracers¶: Returns the number of tracers for this probe, i.e. redshift bins

noise()[source]¶: Returns the noise power for all redshifts return: shape [nbins]

zmax¶: Returns the maximum redsfhit probed by this probe