ResearchBenchmarksTime-dependent PDE approximation on JAX-PI Ginzburg-LandauFollow1.84L2 Errorbeignet1.60923.16714.7256.2829May 22, 2026Evaluation ResultsMethodMethodLinksL2 ErrorTime (sec)beignetParams=855,936 × 5Params=855,936 × 52026.051.841.98JAX-PIParams=331,396 × 5Params=331,396 × 52026.052.072.13PirateNetParams=1,916,553 × 5Params=1,916,553 × 52026.057.616.42