ICLRAug, 2023

使用正向和反向模式自动微分将偏微分方程引入到 JAX

TL;DR使用有限元方法扩展了 JAX 自动微分库,通过 Firedrake 有限元库实现对偏微分方程的符号化表示,从而有效地将有限元解算器与任意可微程序进行组合。