jax-cosmo documentation¶
A differentiable and GPU accelerated cosmology library in JAX.
Checkout the jax-cosmo GitHub page.
Install jax-cosmo¶
Because jax-cosmo is written in plain Python, installing it is trivial:
$ pip install jax-cosmo
That’s all there is to it.