JaxAutodiffProvider#

class cosmicjellyfish.fishermatrix.jaxcosmo_derivatives.JaxAutodiffProvider[source]#

Bases: _JaxObservableProvider

Exact forward-mode derivatives of the JAX observables (jaxcosmo backend).

compute(request)[source]#

Return {parameter: derivative} for request.freeparams.

Parameters with an analytic derivative in request.special_deriv_function keep it, exactly as the finite-difference engine does. The Ps_ shot-noise parameters are the exception: the legacy closed form differentiates the legacy power spectrum, while this provider must differentiate the JAX one, so they always use the JAX jacobian.

Return type:: dict[str, object]