GRADSOLVE：GPU上ODE集合的快速精确梯度计算

GRADSOLVE是一个开源JAX库，专为在NVIDIA GPU上求解和反向微分低维ODE集合而设计。该库记录自适应求解器接受的步骤，并对固定步长重放进行微分，返回精确的离散伴随梯度。在三种GPU代际上，GRADSOLVE的梯度计算速度比Diffrax的检查点伴随方法快5.6-14.1倍，前向求解速度比DiffEqGPU.jl快2.8倍。