Geoscientific Model Development
Geoscientific Model Development
GMD
Articles | Volume 19, issue 3
Articles | Volume 19, issue 3
https://doi.org/10.5194/gmd-19-1103-2026
© Author(s) 2026. This work is distributed under
the Creative Commons Attribution 4.0 License.
https://doi.org/10.5194/gmd-19-1103-2026
© Author(s) 2026. This work is distributed under
the Creative Commons Attribution 4.0 License.
Articles | Volume 19, issue 3
Model description paper
 | 
03 Feb 2026
Model description paper |  | 03 Feb 2026

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

Xingyu Zhu, Yongquan Qu, and Xiaoming Shi

Viewed

Total article views: 8,076 (including HTML, PDF, and XML)
HTML PDF XML Total Supplement BibTeX EndNote
6,886 1,003 187 8,076 245 192 244
  • HTML: 6,886
  • PDF: 1,003
  • XML: 187
  • Total: 8,076
  • Supplement: 245
  • BibTeX: 192
  • EndNote: 244
Views and downloads (calculated since 30 Jun 2025)
Cumulative views and downloads (calculated since 30 Jun 2025)

Viewed (geographical distribution)

Total article views: 8,076 (including HTML, PDF, and XML) Thereof 7,942 with geography defined and 134 with unknown origin.
Country # Views %
  • 1
1
 
 
 
 

Cited

Saved (final revised paper)

Latest update: 23 Sep 2026
Download
Short summary
By using the newly developed Python library JAX, we developed a fast and differentiable large-eddy simulation model, named LEX. Evaluated with a warm bubble case, LEX maintains high accuracy as the Cloud Model 1. With the hardware acceleration and better numerical stability, LEX can be quite faster. To report its differentiability, we further trained a deep learning-based parameterization scheme. The newly trained model can surpass the conventional scheme and get proper forecast results.
Read more
Share