torchode：面向 PyTorch 的并行 ODE 求解器
机器学习
2023-01-18 v2 数值分析
数值分析
摘要
我们为 PyTorch 生态系统引入了一个 ODE 求解器，它能够并行独立求解多个 ODE，同时实现显著的性能提升。我们的实现分别跟踪每个 ODE 的进度，并针对 GPU 和 PyTorch JIT 编译器的兼容性进行了精心优化。其设计使研究人员能够轻松增强求解器的任何方面，并收集和分析内部求解器统计信息。在我们的实验中，我们的实现每步比其他 ODE 求解器快至多 4.3 倍，并且对批内交互具有鲁棒性，而批内交互会导致其他求解器采取多达 4 倍的步数。代码见 https://github.com/martenlienen/torchode
引用
@article{arxiv.2210.12375,
title = {torchode: A Parallel ODE Solver for PyTorch},
author = {Marten Lienen and Stephan Günnemann},
journal= {arXiv preprint arXiv:2210.12375},
year = {2023}
}
备注
Accepted at The Symbiosis of Deep Learning and Differential Equations Workshop, NeurIPS, 2022