KPPax利用diffrax内置的伴随积分，无需手动推导和编写伴随方程代码即可获得精确梯度，避免自动微分直接穿透刚性求解器时的内存爆炸与数值不稳定。