相关论文: JAX-SCM v1.0: a modern atmospheric single-column m…
A single-column model (SCM) is constructed in the regional climate model RegCM4. The evolution of a dry convection boundary layer (DCBL) is used to evaluate this SCM and compare four planetary boundary layer (PBL) schemes, the…
Boundary layer turbulence, particularly the vertical fluxes of momentum, shapes the evolution of winds and currents and plays a critical role in weather, climate, and biogeochemical processes. In this work, a unified, data-driven…
In our effort to facilitate machine learning-assisted computational fluid dynamics (CFD), we introduce the second iteration of JAX-Fluids. JAX-Fluids is a Python-based fully-differentiable CFD solver designed for compressible single- and…
We introduce JAX-LaB, a differentiable, Python-based Lattice Boltzmann simulation library designed for modeling multiphase and multiphysics fluid dynamics problems in hydrologic, geologic, and engineered porous media settings. The library…
The rapid rise of scientific machine learning (SciML) has expanded the role of differentiable modeling, surrogate modeling, and data-driven constitutive laws in large-scale simulation. The JAX framework provides an attractive environment…
The group focused on a model problem of idealised moist air convection in a single column of atmosphere. Height, temperature and moisture variables were chosen to simplify the mathematical representation (along the lines of the Boussinesq…
Coordinating growing grid flexibility under uncertainty is becoming increasingly important for efficient and reliable power-system operation. A core computational requirement is the efficient large-scale batched evaluation of AC power flow…
Theoretical analyses of the hurricane boundary layer have traditionally relied on slab models, which provide a limited description of wind profiles. Literature on height-resolving methods is typically based on linear analyses, which may…
This project aims to advance differentiable fluid dynamics for hypersonic coupled flow over porous media, demonstrating the potential of automatic differentiation (AD)-based optimization for end-to-end solutions. Leveraging AD efficiently…
Most state-of-the-art approaches for weather and climate modeling are based on physics-informed numerical models of the atmosphere. These approaches aim to model the non-linear dynamics and complex interactions between multiple variables,…
This paper introduces JAX-FEM, an open-source differentiable finite element method (FEM) library. Constructed on top of Google JAX, a rising machine learning library focusing on high-performance numerical computing, JAX-FEM is implemented…
Shock dynamics and nonlinear wave propagation are fundamental to computational fluid dynamics (CFD) and high-speed flow modeling. In this study, we developed explicit and implicit finite-difference solvers for the one-dimensional Burgers…
Accurate prediction of wall-bounded flows remains central to advancing both theoretical understanding and computational methods in fluid mechanics. In this study, we perform a numerical simulation of channel flow using a complementary…
A single-column model is used to investigate regime transitions within the stable atmospheric boundary layer, focusing on the role of small-scale fluctuations in wind and temperature dynamics and of turbulence intermittency as triggers for…
Sub-grid scale parameterizations in atmospheric models involve numerous uncertain parameters that must be tuned to align simulations with observations. Here, we propose a framework for assessing objective tuning frameworks using the Single…
We study a single column model of moist convection in the atmosphere. We state the conditions for it to represent a stable steady state. We then evolve the column by subjecting it to an upward displacement which can release instability,…
Global climate models (GCMs), typically run at ~100-km resolution, capture large-scale environmental conditions but cannot resolve convection and cloud processes at kilometer scales. Convection-permitting models offer higher-resolution…
We present an implementation of interval analysis and mixed monotone interval reachability analysis as function transforms in Python, fully composable with the computational framework JAX. The resulting toolbox inherits several key features…
Remote observing of exoplanetary atmospheres is now possible, offering us access to circulation regimes unlike any of the familiar Solar System cases. Atmospheric circulation models are being developed to study these new regimes but model…
A single-column model run under the weak temperature gradient approximation, a parameterization of large-scale dynamics appropriate for the tropical atmosphere, is shown to have multiple stable equilibria. Under conditions permitting…