ResearchBenchmarksPDE Solving on two-body Allen-Cahn equation 1M DFollow35.82Speed (it/s)SDZE12.721618.718324.71530.7117Mar 25, 2026Evaluation ResultsMethodMethodLinksSpeed (it/s)SDZEDerivative computation...Derivative computation order=Zeroth-order (ZO), Implementation framework=JAX2026.0335.82STDE (FO)Derivative computation...Derivative computation order=First-order (FO), Implementation framework=JAX2026.0313.61