Feb, 2024

JAX-Fluids 2.0:面向可区分 CFD 的高性能计算压缩性两相流

TL;DR我们介绍了第二个版本的基于 Python 的完全可微分计算流体力学求解器 JAX-Fluids,该求解器设计用于可压缩的单相和两相流。我们通过引入能够在 GPU 和 TPU 上高效扩展的 JAX 原始操作并展示稳定的自动微分梯度计算, 实现了高性能计算能力的并行化策略。新版本还提供了增强的两相流模拟能力、正性保持限制器以增加鲁棒性、支持拉伸笛卡尔网格、重构的输入 / 输出处理、全面的后处理例程,以及最新的高阶数值离散方案。我们通过模拟单相和两相流的结果验证了新增的数值模型,包括湍流边界层和通道流、气 - 氦冲击泡交互和气 - 水冲击滴交互。