JAX-COSMO: une librairie auto-differentiable et executable sur GPU pour la Cosmology.
Extrait de JAX-COSMO: An End-to-End Differentiable and GPU Accelerated Cosmology Library (https://arxiv.org/abs/2302.05163) Nous présentons jax-cosmo, une bibliothèque pour les calculs en cosmologique automatiquement différentiables. jax-cosmo utilise la bibliothèque JAX, qui…