A differentiable cosmology library in JAX

This reposotory is a prototype for a differentiable cosmology library written in Jax. For this prototype we are going to keep things simple, and port as much exciting code as possible from a pure Python package: https://github.com/cosmicpy/cosmicpy