JAX-CFD

JAX-CFD 用 JAX 实现不可压 Navier-Stokes 求解，使整个时间推进循环可微、 可在加速器上 JIT 编译。

适用场景：让学习型修正项穿过求解器参与训练的混合建模；基于梯度的湍流封闭建模；湍流模型发现。