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