DISCO-DJ: differentiable particle-mesh simulation software
- Author(s)
- Florian List, Oliver Hahn, Thomas Flöss, Lukas Winkler
- Abstract
DISCO-DJ performs large-scale structure (LSS) cosmological simulations with full support for automatic differentiation via the JAX ecosystem. It integrates a suite of modern cosmological modelling tools, including a fast particle-mesh (PM) N-body solver enhanced by theory-informed time integrators, such as BullFrog and FastPM, and discreteness-suppression techniques (e.g., de-aliasing, sheet-based resampling, higher-order mass assignment kernels). The code also provides an interface to a differentiable Einstein–Boltzmann solver, high-order Lagrangian perturbation theory (nLPT) recursion relations, and semi-classical propagator perturbation theory (PPT). By utilizing JAX’s forward- and reverse-mode automatic differentiation (including memory-efficient adjoint methods), DISCO-DJ enables cosmological inference scenarios where differentiable pipelines from initial conditions to late-time observables are required.
- Organisation(s)
- Department of Astrophysics, Department of Mathematics
- Publication date
- 10-2025
- Austrian Fields of Science 2012
- 103044 Cosmology
- Portal url
- https://ucrisportal.univie.ac.at/en/publications/4be3cd01-e9c5-4c33-aea2-879d83456c70