JAX-COSMO: An End-to-End Differentiable and GPU Accelerated Cosmology Library

Jean-Eric Campagne, François Lanusse, Joe Zuntz +7

We present jax-cosmo, a library for automatically differentiable cosmological theory calculations. It uses the JAX library, which has created a new coding ecosystem, especially in…