12:09
官方账号arXiv cs.LG@Alessio Spurio Mancini
精选73°
推荐理由：GRADSOLVE解决了ODE集合在GPU上求解与反向微分速度不匹配的问题，大幅提升计算效率。
微分
共 1 条相关 AI 资讯
9月3日
12:09
GRADSOLVE是一个开源JAX库，专为在NVIDIA GPU上求解和反向微分低维ODE集合而设计。该库记录自适应求解器接受的步骤，并对固定步长重放进行微分，返回精确的离散伴随梯度。在三种GPU代际上，GRADSOLVE的梯度计算速度比Diffrax的检查点伴随方法快5.6-14.1倍，前向求解速度比DiffEqGPU.jl快2.8倍。