cosmicjellyfish.cosmology.jaxcosmo_provider#
CosmicJellyfish-facing provider of the differentiable jaxcosmo backend.
The heavy lifting is done by the pure-JAX functions of
cosmicjellyfish.cosmology.jaxcosmo_theory. This module is the outer boundary: it
resolves CosmicJellyfish parameter conventions, evaluates the (jitted) theory on the configured
(z, k) grids and wraps the results into the SciPy-spline-like objects expected by
cosmicjellyfish.cosmology.cosmology.cosmo_functions. Gradients do not flow through this
module; the autodiff path uses the theory layer directly.
Supported model: flat LCDM / w0waCDM with massless neutrinos, total-matter power only.
Functions
Map CosmicJellyfish parameters to the |
Classes
Build total-matter background and power-spectrum interpolators with |