jaxdae：一种用于耦合多物理场中微分代数方程的JAX原生可微求解器

摘要

本文提出了jaxdae，一个基于JAX的微分代数方程（DAE）可微求解器，能够统一前向求解和反向模式灵敏度计算。它通过自适应BDF、Radau和Rosenbrock积分器结合Pantelides指标约简，并利用冻结步长网格上的变步长BDF-2重求解实现可微性。