JAX-COSMO: An End-to-End Differentiable and GPU Accelerated Cosmology Library
Extrait de JAX-COSMO: An End-to-End Differentiable and GPU Accelerated Cosmology Library (https://arxiv.org/abs/2302.05163) We present jax-cosmo, a library for automatically differentiable cosmological theory calculations.jax-cosmo uses the JAX library, which has created…