FlashPDE: A Drop-In Fused Triton Operator Library for Neural PDE Solvers

Peiyu Zang, Bosen Xie, Ruoxiang Xu +1

Physics-Informed Neural Networks (PINNs) solve PDEs by incorporating physical constraints into neural-network training, but large-scale problems are limited by automatic-differenti…