Description
High-performance N-body simulations are crucial for astrophysics, cosmology, and computational physics. Odisseo (https://github.com/vepe99/Odisseo) is a differentiable N-body simulator designed for efficiency, scalability, and long-term usability. Implemented in JAX, it leverages just-in-time (JIT) compilation, automatic differentiation, and GPU/TPU acceleration
to provide fast, vectorized, and distributed pairwise force calculations. Odisseo supports external potentials and enables gradient-based optimization, making it a powerful tool for scientific inference and machine learning applications. Its JAX-based architecture ensures modularity, ease of extension, and long-term maintainability, fostering community-driven development.
Primary authors
Giuseppe Viterbo
(IWR, Heidelberg University)
Dr
Tobias Buck
(IWR, Heidelberg University)
