Accelerating massive ensembles of ordinary differential equations

arXiv:2609.35661 · astro-ph.IM, astro-ph.CO, physics.comp-ph · Submitted 2026-09-28 · Read on arXiv

astro-ph.IM, astro-ph.CO, physics.comp-ph

Submitted: 2026-09-28

Updated: 2026-09-28

Comments: 17 pages, 15 figures, 1 table. Comments are welcome!

Code: https://github.com/lawrenceberry/modax

License: http://arxiv.org/licenses/nonexclusive-distrib/1.0/

Terminology

• Benchmark — 19× in this paper · explained in 2 episodes
• Core — 17× in this paper · explained in 2 episodes
• Sparsity — 17× in this paper · explained in 4 episodes
• Abundance — 8× in this paper · explained in 1 episode
• Accuracy — 7× in this paper · explained in 1 episode
• Uncertainty Quantification — 6× in this paper · explained in 9 episodes
• Bispectrum — 5× in this paper · explained in 3 episodes
• Boltzmann Equation — 2× in this paper · explained in 2 episodes
• Non-Gaussianity — 2× in this paper · explained in 2 episodes
• PID Controller — 2× in this paper · explained in 2 episodes
• 21-cm signal — 1× in this paper · explained in 1 episode
• Adjoint Method — 1× in this paper · explained in 2 episodes
• Anisotropy — 1× in this paper · explained in 4 episodes
• Calibration — 1× in this paper · explained in 7 episodes
• Diffusion Model — 1× in this paper · explained in 10 episodes
• Diffusion Models — 1× in this paper · explained in 9 episodes
• Inverse Problem — 1× in this paper · explained in 2 episodes
• Langevin Dynamics — 1× in this paper · explained in 3 episodes
• Ordinary Differential Equations (ODEs) — 1× in this paper · explained in 2 episodes
• Primordial non-Gaussianity — 1× in this paper · explained in 2 episodes
• Robustness — 1× in this paper · explained in 5 episodes
• Trace — 1× in this paper · explained in 2 episodes

Sources

• TASI Lectures on Inflation
• PRyMordial: The First Three Minutes, Within and Beyond the Standard Model
• Neural Ordinary Differential Equations
• The shape of primordial non-Gaussianity and the CMB bispectrum
• General CMB and Primordial Bispectrum Estimation I: Mode Expansion, Map-Making and Measures of f_NL
• LINX: A Fast, Differentiable, and Extensible Big Bang Nucleosynthesis Package
• CMB power spectrum parameter degeneracies in the era of precision cosmology
• On Neural Differential Equations
• The Cosmic Linear Anisotropy Solving System (CLASS) I: Overview
• Efficient Computation of CMB anisotropies in closed FRW models
• PArthENoPE: Public Algorithm Evaluating the Nucleosynthesis of Primordial Elements
• Differentiable astrophysics at scale: solving and differentiating ODE ensembles on the GPU
• GRADSOLVE: fast exact gradients for ODE ensembles on GPUs
• Automated Translation and Accelerated Solving of Differential Equations on Multiple GPU Platforms
• Nested Slice Sampling: Vectorized Nested Sampling for GPU-Accelerated Inference

Related papers

• A signal dedispersion algorithm for imaging-based transient searches
• EXPO: a quantum leap in fast, wide-band X-ray polarimetry for astrophysics
• Understanding cosmic explosions
• High-frequency water megamasers as precision probes of black holes, galaxies and cosmology: A case for a next-generation sub-mm/mm interferometer
• ALMA2040: Mapping the Spatially Resolved Dust and Star Formation Physics Driving High-z Galaxy Assembly
• Direct measures of black hole masses through cosmic time