Differentiable N-body code for Galactic Dynamics -- Odisseo

Fuente: arXiv
Salvato in:
Dettagli Bibliografici
Autori principali: Viterbo, Giuseppe, Buck, Tobias
Natura: Preprint
Pubblicazione: 2025
Soggetti:
Accesso online:
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866917110379184128
author Viterbo, Giuseppe
Buck, Tobias
author_facet Viterbo, Giuseppe
Buck, Tobias
contents We introduce \textsc{Odisseo} (Optimized Differentiable Integrator for Stellar Systems Evolution of Orbits), a differentiable N-body code designed to constrain the gravitational potential of the Milky Way (MW) through dynamical modeling of accreted structures such as stellar streams. \textsc{Odisseo} is implemented in JAX, enabling just-in-time compilation, automatic differentiation, and hardware acceleration on GPUs and TPUs. The code features efficient, fully vectorized force calculations and exhibits near-linear scaling when distributing a single simulation across multiple GPUs, making it suitable for large scale optimization tasks. As a demonstration, we present a case study using a mock GD-1 stellar stream simulation, where we optimize four physical parameters via gradient descent: the accretion time and progenitor mass, as well as the masses of the host Navarro-Frenk-White (NFW) halo and Miyamoto-Nagai (MN) disk. \textsc{Odisseo} accurately recovers stream morphology and underlying parameters in a differentiable and scalable framework, providing a powerful tool for dynamical studies of the Milky Way and its accreted substructures.
format Preprint
id arxiv_https___arxiv_org_abs_2511_22468
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Differentiable N-body code for Galactic Dynamics -- Odisseo
Viterbo, Giuseppe
Buck, Tobias
Astrophysics of Galaxies
Instrumentation and Methods for Astrophysics
We introduce \textsc{Odisseo} (Optimized Differentiable Integrator for Stellar Systems Evolution of Orbits), a differentiable N-body code designed to constrain the gravitational potential of the Milky Way (MW) through dynamical modeling of accreted structures such as stellar streams. \textsc{Odisseo} is implemented in JAX, enabling just-in-time compilation, automatic differentiation, and hardware acceleration on GPUs and TPUs. The code features efficient, fully vectorized force calculations and exhibits near-linear scaling when distributing a single simulation across multiple GPUs, making it suitable for large scale optimization tasks. As a demonstration, we present a case study using a mock GD-1 stellar stream simulation, where we optimize four physical parameters via gradient descent: the accretion time and progenitor mass, as well as the masses of the host Navarro-Frenk-White (NFW) halo and Miyamoto-Nagai (MN) disk. \textsc{Odisseo} accurately recovers stream morphology and underlying parameters in a differentiable and scalable framework, providing a powerful tool for dynamical studies of the Milky Way and its accreted substructures.
title Differentiable N-body code for Galactic Dynamics -- Odisseo
topic Astrophysics of Galaxies
Instrumentation and Methods for Astrophysics
url https://arxiv.org/abs/2511.22468