Peiyu Zang

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

Peiyu Zang, Bosen Xie, Ruoxiang Xu, Yongqiang Cai

Physics-Informed Neural Networks (PINNs) solve PDEs by incorporating physical constraints into neural-network training, but large-scale problems are limited by automatic-differentiation memory overhead and inefficient execution of grid-based PDE operators. We present FlashPDE, a…Read more