Cosmology utilities
===================

.. warning::

   These utilities are internal helpers used by all estimators.  The API is
   not yet considered stable and may change.

The :mod:`sum_stat.cosmology` module provides JAX-native implementations of
comoving distance, comoving volume, and critical surface density integrals.
These are used internally by all estimators that require a cosmology;
end users typically pass an :class:`astropy.cosmology.FLRW` object which is
converted automatically via :func:`~sum_stat.cosmology.astropy_to_jax_cosmo`.

.. automodule:: sum_stat.cosmology
   :members: