Diff-FlowFSI: A GPU-optimized differentiable CFD platform for high-fidelity turbulence and FSI simulations

Abstract

Turbulent flows and fluid-structure interactions (FSI) are ubiquitous in scientific and engineering applications, but their accurate and efficient simulation remains a major challenge due to strong nonlinearities, multiscale interactions, and high computational demands. Traditional CFD solvers, though effective, struggle with scalability and adaptability for tasks such as inverse modeling, optimization, and data assimilation. Recent advances in machine learning (ML) have inspired hybrid modeling approaches that integrate neural networks with physics-based solvers to enhance generality and capture unresolved dynamics. However, realizing this integration requires solvers that are not only physically accurate but also differentiable and GPU-efficient. In this work, we introduce Diff-FlowFSI, a GPU-accelerated, fully differentiable CFD platform designed for high-fidelity turbulence and FSI simulations. Implemented in JAX, Diff-FlowFSI features a vectorized finite volume solver combined with the immersed boundary method to handle complex geometries and fluid-structure coupling. The platform enables GPU-enabled fast forward simulations, supports automatic differentiation for gradient-based inverse problems, and integrates seamlessly with deep learning components for hybrid neural-CFD modeling. We validate Diff-FlowFSI across a series of benchmark turbulence and FSI problems, demonstrating its capability to accelerate scientific computing at the intersection of physics and machine learning.