Articles | Volume 19, issue 3
https://doi.org/10.5194/gmd-19-1103-2026
https://doi.org/10.5194/gmd-19-1103-2026
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

Data sets

LEX v1.6.0 Data Zhu Xingyu et al. https://doi.org/10.5281/zenodo.15730773

LEX v1.6.0 Training Data and Scaling Params (v1.6.0) Zhu Xingyu et al. https://doi.org/10.5281/zenodo.18411486

Model code and software

LEX v1.6.0 Zhu Xingyu et al. https://doi.org/10.5281/zenodo.15486687

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.
Share