A memory-efficient neural ODE framework based on high-level adjoint differentiation

Hong Zhang, Wenjun Zhao

Neural ordinary differential equations (neural ODEs) have emerged as a novel network architecture that bridges dynamical systems and deep learning. However, the gradient obtained w…