Jul 2026
FlashPDE: A Drop-In Fused Triton Operator Library for Neural PDE Solvers
FlashPDE provides a hardware-efficient execution layer that bridges differentiable PDE solvers and GPU-optimized numerical computation within the PyTorch ecosystem, while maintaining numerical agreement with PyTorch finite-difference references.
Pei-Yu Zang, Bosen Xie, Ruoxi Xu et al.
· arXiv.org · 1 citation