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

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

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 8 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 1 episode
• Anisotropy — 1× in this paper · explained in 3 episodes
• Calibration — 1× in this paper · explained in 7 episodes
• Diffusion Model — 1× in this paper · explained in 8 episodes
• Diffusion Models — 1× in this paper · explained in 6 episodes
• Inverse Problem — 1× in this paper · explained in 2 episodes
• Langevin Dynamics — 1× in this paper · explained in 2 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
• AVICA: A fully automated CASA pipeline for large volume VLBI data calibration
• Spectral Map Making with SPHEREx
• Long-Integration Magnetar Burst Observatory (LIMBO): Instrument Summary and Early FRB Rate Constraints
• Towards independent event horizon imaging of the supermassive black holes in M87 and the Milky Way
• A PINK update: Improvements to the CELEBI fast radio burst data reduction and analysis pipeline