4 papers
Differentiable Hybrid Neural-CFD Modelling of Wall-Bounded Turbulence: Coupled Learning of Subgrid-Scale and Wall Closures
Xiantao Fan, Yi Liu, Meng Wang +1
Wall-modelled large-eddy simulation (WMLES) treats the subgrid-scale (SGS) closure, wall closure and numerical discretization as independent components, although their effects are…
JAX-AMG: A GPU-Accelerated Differentiable Sparse Linear Solver Library for JAX
Yi Liu, Xiantao Fan, Jian-Xun Wang
Sparse linear systems from PDE discretizations are central to scientific computing, yet no existing JAX-ecosystem solver simultaneously provides GPU-accelerated algebraic multigrid…
Patched Flow Matching: Generative Wall-Pressure Reconstruction Beyond Training-Domain Scales from Sparse Sensors
Meet Hemant Parikh, Yi Liu, Jian-Xun Wang
Characterizing the complete wall-pressure spectrum in turbulent wall-bounded flows requires simultaneous access to the viscous-scale high-wavenumber content and the outer-layer low…
JAX-BTE: A GPU-Accelerated Differentiable Solver for Phonon Boltzmann Transport Equations
Wenjie Shang, Jiahang Zhou, J. P. Panda +5
This paper introduces JAX-BTE, a GPU-accelerated, differentiable solver for the phonon Boltzmann Transport Equation (BTE) based on differentiable programming. JAX-BTE enables accur…