PyTorch Lightning
https://www.pytorchlightning.ai/blog/neural-odes-with-pytorch...Hypersolvers in PyTorch Lightning: Faster Neural Differential Equations. Neural Differential Equations inference is typically slower than comparable discrete neural networks, since these continuous models come with the additional overhead of solving a differential equation. Various approaches have been proposed to alleviate these limitations e ...