JaxPhotoObservable

class cosmicjellyfish.LSSsurvey.jaxcosmo_photo.JaxPhotoObservable(configuration, fiducial, backend_parameters=None, observables=None)

Bases: object

Differentiable photometric angular power spectra.

Parameters:

• configuration (object) – Analysis configuration exposing settings, specs and obs (a AnalysisContext or the legacy cfg module).

• fiducial (mapping) – Merged fiducial dictionary (cosmology + intrinsic alignment + bias + photo-z), the same object that PhotoCov passes as allparsfid.

• backend_parameters (mapping, optional) – Content of jaxcosmo/default.yaml. Defaults to the one carried by configuration.

• observables (sequence of str, optional) – Probes this observable computes. Defaults to configuration.obs; pass the requested subset when the configuration carries probes computed by another observable.

canonical_name(name)

Map a CosmicJellyfish parameter name to the name used inside the traced function.

Return type:

str

as_dict(array)

Convert a (nkeys, nell) array into the CosmicJellyfish C_ell dictionary.

Return type:

dict[str, ndarray]

cls(overrides=None)

Angular power spectra at the fiducial point (optionally with parameter overrides).

Return type:

dict[str, ndarray]

jacobian(names)

Exact derivatives dC_ell/dtheta for each name, as {name: Cl dictionary}.

Cosmological and nuisance parameters are differentiated in two separate forward-mode passes so that the (expensive) theory tables carry tangents only when needed.

Return type:

dict[str, dict[str, ndarray]]