get_jax_theory#

cosmicjellyfish.cosmology.jax_theory.get_jax_theory(code, backend_parameters=None, cosmo_model='LCDM')[source]#

Return the differentiable theory implementation selected by options["code"].

Parameters:

code (str) – Value of options["code"]. Currently only "jaxcosmo" is available.
backend_parameters (mapping, optional) – Parsed default_boltzmann_yaml_files/<code>/default.yaml.
cosmo_model (str) – "LCDM" or "w0waCDM"; fixed-w models are rejected by the backend.

Raises:

NotImplementedError – If the selected backend has no differentiable implementation.

Return type: