Mixed Precision Training of Neural ODEs
This paper presents a mixed precision training framework for neural ODEs consisting of explicit ODE solvers and a custom backpropagation scheme and shows their effectiveness in a range of learning tasks and publishes the extendable, open-source PyTorch package rampde, whose syntax resembles that of leading packages to...