A differentiable and GPU accelerated cosmology library in JAX.
Checkout the jax-cosmo GitHub page.
Because jax-cosmo is written in plain Python, installing it is trivial:
$ pip install jax-cosmo
That's all there is to it.
notebooks/jax-cosmo-intro.ipynb notebooks/CCL_comparison.ipynb
api/modules
genindexmodindexsearch