SOURCE-LINKED INTELLIGENCE

GRADSOLVE: fast exact gradients for ODE ensembles on GPUs

arXiv · AI, language, vision and robotics · article · Sep 2, 2026 · UTC

Ordinary differential equations (ODEs) underlie models in science and engineering, and many applications need derivatives of their solutions with respect to parameters. Ensembles of independent trajectories suit graphics processing units (GPUs), but current GPU software forces a trade-off: the fastest ensemble solvers cannot be differentiated in reverse mode at the speed they solve, and the solvers built for differentiation solve more slowly. No single tool has yet offered a reverse-mode gradient at the speed of a fused-kernel solve. We present GRADSOLVE, an open-source JAX library for solving

Read original source ↗ Open in workspace

recordType

paper

region

Global