JAX 实现可微 Voronoi 动网格流体求解器,梯度可穿透网格运动

CatAstro_Piyush · x · 2026-11-04

PMocz 发布开源项目 differentiable-voronoi-fluids:一个用 JAX 写的 2D 可压缩 Euler 流体求解器,采用 Arepo 风格的移动 Voronoi 网格,reverse-mode 自动微分可穿透整个仿真过程,包括网格运动本身。

技术关键在于把网格拆成两部分处理:连通性(哪些 seed 互为邻居)在位置上是分段常量、无法直接微分,因此放在 host 上通过 jax.purecallback 调 SciPy 的 Delaunay(Qhull)计算,并对梯度做 stop;面几何等连续量则留在 JAX 内保持可微。作者说明这是一个原型:周期盒、无壁面、无自适应加密、仅流体力学。