产品
diffrax
diffrax是JAX生态中的常微分方程求解库，提供多种隐式与显式求解器，并支持通过伴随方法高效计算梯度。
时间轴 (近 90 天)
10月28日
KPPax调用diffrax库的隐式求解器及前向、反向模式伴随，对刚性化学系统进行积分。
待验证50%
10月28日
KPPax利用diffrax内置的伴随积分，无需手动推导和编写伴随方程代码即可获得精确梯度，避免自动微分直接穿透刚性求解器时的内存爆炸与数值不稳定。
待验证50%
10月8日
torchdiffeq和diffrax等框架内置了刚性求解器选项，如diffrax中的Kvaerno5和torchdiffeq中的dopri5
待验证50%