Proximal Implicit ODE Solvers for Accelerating Learning Neural ODEs

Justin Baker, Hedi Xia, Yiwei Wang +7

Learning neural ODEs often requires solving very stiff ODE systems, primarily using explicit adaptive step size ODE solvers. These solvers are computationally expensive, requiring…