Abstract:Physical systems are governed by partial differential equations (PDEs). The Navier-Stokes equations describe fluid flows and are representative of nonlinear physical systems with complex spatio-temporal interactions. Fluid flows are omnipresent in nature and engineering applications, and their accurate simulation is essential for providing insights into these processes. While PDEs are typically solved with numerical methods, the recent success of machine learning (ML) has shown that ML methods can provide novel avenues of finding solutions to PDEs. ML is becoming more and more present in computational fluid dynamics (CFD). However, up to this date, there does not exist a general-purpose ML-CFD package which provides 1) powerful state-of-the-art numerical methods, 2) seamless hybridization of ML with CFD, and 3) automatic differentiation (AD) capabilities. AD in particular is essential to ML-CFD research as it provides gradient information and enables optimization of preexisting and novel CFD models. In this work, we propose JAX-FLUIDS: a comprehensive fully-differentiable CFD Python solver for compressible two-phase flows. JAX-FLUIDS allows the simulation of complex fluid dynamics with phenomena like three-dimensional turbulence, compressibility effects, and two-phase flows. Written entirely in JAX, it is straightforward to include existing ML models into the proposed framework. Furthermore, JAX-FLUIDS enables end-to-end optimization. I.e., ML models can be optimized with gradients that are backpropagated through the entire CFD algorithm, and therefore contain not only information of the underlying PDE but also of the applied numerical methods. We believe that a Python package like JAX-FLUIDS is crucial to facilitate research at the intersection of ML and CFD and may pave the way for an era of differentiable fluid dynamics.

A Generalized Framework for Integrating Machine Learning into Computational Fluid Dynamics

In Situ Framework for Coupling Simulation and Machine Learning with Application to CFD

An integrated framework for accelerating reactive flow simulation using GPU and machine learning models

Recent Advances on Machine Learning for Computational Fluid Dynamics: A Survey

Machine learning accelerated computational fluid dynamics

Combining machine learning with computational fluid dynamics using OpenFOAM and SmartSim

Enhancing computational fluid dynamics with machine learning

Emerging Trends in Machine Learning for Computational Fluid Dynamics

Towards a new paradigm in intelligence-driven computational fluid dynamics simulations

JAX-FLUIDS: A fully-differentiable high-order computational fluid dynamics solver for compressible two-phase flows

Machine Learning to Assist Filtered Two‐fluid Model Development for Dense Gas–particle Flows

Machine Learning for Fluid Mechanics

A TensorFlow-based New High-Performance Computational Framework for CFD

Recent Progress of Machine Learning in Flow Modeling and Active Flow Control

Prospective on Applying Machine Learning in Computational Fluid Dynamics (CFD) Simulation of Metallurgical Reactors

An interpretable framework of data-driven turbulence modeling using deep neural networks

A multi-fidelity machine learning based semi-Lagrangian finite volume scheme for linear transport equations and the nonlinear Vlasov-Poisson system

DRLFluent: A Distributed Co-Simulation Framework Coupling Deep Reinforcement Learning with Ansys-Fluent on High-Performance Computing Systems

Machine Learning Based Fluid-Transportation Monitoring and Controlling