ResearchBenchmarksTime-dependent PDE approximation on JAX-PI Gray-ScottFollow1.31L2 ErrorJAX-PI1.042.86254.6856.5075May 22, 2026Evaluation ResultsMethodMethodLinksL2 ErrorL2 Error Computation Time (s)JAX-PIParams=331,396 × 10Params=331,396 × 102026.051.312.39PirateNetParams=1,916,553 × 10Params=1,916,553 × 102026.052.15.49beignetParams=855,936 × 10Params=855,936 × 102026.058.062