GradRepair-ODE: Certified Gradient Repair for Neural ODE Training

Ziqian Bi, Xin Liang Chia

Neural ordinary differential equations use numerical solvers inside the training loop. The solver determines the forward trajectory and also affects the gradient passed to the opti…