5–7 juin 2023
IAP
Fuseau horaire Europe/Paris

JAX-LPT: differentiable Lagrangian Perturbation Theory simulator

6 juin 2023, 16:10
20m
IAP

IAP

98bis Boulevard Arago 75014 Paris

Orateur

Axel Lapel (Institut d'Astrophysique de Paris)

Description

This talk introduces JAX-LPT: a novel code implementing fast, GPU-compatible, and differentiable simulations based on first and second-order Lagrangian Perturbation Theory. JAX-LPT can facilitate the swift generation of initial conditions for N-body simulations and integrate effectively within Bayesian hierarchical models for field-level inference, contributing to a community-wide effort to develop a new generation of differentiable numerical tools for complex statistical inference problems (e.g. pmwd, JaxPM). The session will unpack the code's design and implementation, showcase basic utilization, and preview current and potential deployments in a few research projects.

Auteur principal

Axel Lapel (Institut d'Astrophysique de Paris)

Documents de présentation