PulseExploreJournal ClubDebatesTrendingResearchersJournals
Instagram
HomeExploreJournal ClubTrending
Synapse
⌘+K
Synapse
February 5, 2026Geoscientific model development1 citationsOpen Access

LEX v1.6.0: a new large-eddy simulation model in JAX with GPU acceleration and automatic differentiation

View Full Paper
XZXingyu ZhuHong Kong University of Science and TechnologyYQYongquan QuColumbia UniversityXSXiaoming ShiHong Kong University of Science and Technology

Key Points

  • The study aims to develop a large-eddy simulation framework using JAX to improve turbulence modeling in gray-zone conditions.
  • Developed a new large-eddy simulation model utilizing the JAX library.
  • Applied generalized pseudo-incompressible equations for simulations.
  • Used traditional and deep learning-based subgrid-scale turbulence models for performance comparison.
  • Conducted simulations on a classic warm bubble case with gray-zone resolution.
  • The traditional Smagorinsky model moderately improved the rising speed but failed in structure evolution.
  • The deep learning-based SGS model accurately simulated thermal bubble expansion and rotor development.
  • Gray-zone simulation results were comparable to benchmark large-eddy simulation resolutions.

Abstract

Abstract. Large-eddy simulations (LES) are essential tools for studies on atmospheric turbulence and clouds and play critical roles in the development of turbulence and convection parameterizations. Current numerical weather models have approached kilometer-scale resolution as supercomputing facilities advance. However, this resolution range is in the so-called gray zone, where subgrid-scale (SGS) turbulence actively interacts with resolved motion and significantly influences the large-scale characteristics of simulated weather systems. Thus, a novel LES framework is required to enable the development of new SGS approaches for the gray zone. Here we used the Python library JAX to develop a new LES model. It is based on the generalized pseudo-incompressible equations formulated by Durran (2008). For a classic warm bubble case, the traditional Smagorinsky model fails to reproduce the correct structure evolution of the warm bubble, though it can modestly correct the rising speed in gray-zone resolution simulations. Utilizing the capability of JAX for automatic differentiation, we trained a deep learning-based SGS turbulence model for the same case. The trained deep learning SGS model, based on a simple autoencoder (AE), enables this physics-deep learning hybrid model to accurately simulate the expansion of the thermal bubble and the development of rotors surrounding the center of the bubble at a gray-zone resolution. The gray-zone simulation results are comparable to those of the benchmark LES resolution.

Ask AI
Helpful
Bookmark
Share
View Full Paper

Cite This Study

Zhu et al. (2026) studied this question.

synapsesocial.com/papers/6984346ff1d9ada3c1fb2835https://doi.org/10.5194/gmd-19-1103-2026
Ask AI
Helpful
Bookmark
Share
View Full Paper

Also Consider

Synapse has enriched 5 closely related papers on similar clinical questions. Consider them for comparative context:

  1. 1To exascale and beyond – The Simple Cloud-Resolving E3SM Atmosphere Model (SCREAM), a performance portable global atmosphere model for cloud-resolving scales2024 · 10 citations
  2. 2Neural Network Parameterization of Subgrid‐Scale Physics From a Realistic Geography Global Storm‐Resolving Simulation2024 · 24 citations
  3. 3A Time-Splitting Scheme for the Elastic Equations Incorporating Second-Order Runge–Kutta Time Differencing1998 · 151 citations
  4. 4GENERAL CIRCULATION EXPERIMENTS WITH THE PRIMITIVE EQUATIONS1963 · 13,955 citations
  5. 5Spectral structure of stratified turbulence: direct numerical simulations and predictions by large eddy simulation2012 · 24 citations