Improving Auto-Design of Neural PDE Solvers with a Domain-Specific Language

Shengxin Kong, Liwen Xu, Jingwen Fu

Neural PDE solver auto-design is fundamentally a search-space representation problem. In the space of unrestricted Python programs, valid solvers form an extremely sparse subset: m…