easy neural ode:论文“学习易于求解的微分方程”的代码 源码

学习易于求解的微分方程 论文代码: Jacob Kelly *,Jesse Bettencourt *,Matthew James Johnson,David Duvenaud。 “学习易于解决的微分方程。” 神经信息处理系统(2020)。 *平等贡献 包括以下模型的JAX实现: 用于分类的 时间序列的 用于密度估计 包括以下自适应步进数值求解器的JAX实现: Heun-Euler heun (二阶) 费尔伯格(RK1(2)) fehlberg (二阶) Bogacki-Shampine bosh (3阶) Cash-Karp cash_karp (第四订单) Fehlberg rk_fehlberg (四阶) Owrenzen owrenzen (4级) Dormand-Prince dopri (5阶) Owrenzen owrenzen5 (5阶)