torchdiffeq
torchdiffeq是基于PyTorch的常微分方程(ODE)求解器库,支持通过伴随方法进行ODE解的反向传播,保持恒定内存开销。该库兼容GPU加速,提供多种求解算法,包括自适应和固定步长方法。支持可微分事件处理功能,适用于深度学习研究。torchdiffeq为研究人员提供了实现和探索基于ODE的机器学习模型的工具。