Saved in:
Bibliographic Details
Main Authors: Yin, Junqi, Palash, Mijanur, Laiu, M. Paul, Meena, Muralikrishnan Gopalakrishnan, Gounley, John, Kops, Stephen M. de Bruyn, Wang, Feiyi, Sankaran, Ramanan, Zhang, Pei
Format: Preprint
Published: 2025
Subjects:
Online Access:https://arxiv.org/abs/2507.16697
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866911070612881408
author Yin, Junqi
Palash, Mijanur
Laiu, M. Paul
Meena, Muralikrishnan Gopalakrishnan
Gounley, John
Kops, Stephen M. de Bruyn
Wang, Feiyi
Sankaran, Ramanan
Zhang, Pei
author_facet Yin, Junqi
Palash, Mijanur
Laiu, M. Paul
Meena, Muralikrishnan Gopalakrishnan
Gounley, John
Kops, Stephen M. de Bruyn
Wang, Feiyi
Sankaran, Ramanan
Zhang, Pei
contents Turbulence plays a crucial role in multiphysics applications, including aerodynamics, fusion, and combustion. Accurately capturing turbulence's multiscale characteristics is essential for reliable predictions of multiphysics interactions, but remains a grand challenge even for exascale supercomputers and advanced deep learning models. The extreme-resolution data required to represent turbulence, ranging from billions to trillions of grid points, pose prohibitive computational costs for models based on architectures like vision transformers. To address this challenge, we introduce a multiscale hierarchical Turbulence Transformer that reduces sequence length from billions to a few millions and a novel RingX sequence parallelism approach that enables scalable long-context learning. We perform scaling and science runs on the Frontier supercomputer. Our approach demonstrates excellent performance up to 1.1 EFLOPS on 32,768 AMD GPUs, with a scaling efficiency of 94%. To our knowledge, this is the first AI model for turbulence that can capture small-scale eddies down to the dissipative range.
format Preprint
id arxiv_https___arxiv_org_abs_2507_16697
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Pixel-Resolved Long-Context Learning for Turbulence at Exascale: Resolving Small-scale Eddies Toward the Viscous Limit
Yin, Junqi
Palash, Mijanur
Laiu, M. Paul
Meena, Muralikrishnan Gopalakrishnan
Gounley, John
Kops, Stephen M. de Bruyn
Wang, Feiyi
Sankaran, Ramanan
Zhang, Pei
Fluid Dynamics
Machine Learning
Turbulence plays a crucial role in multiphysics applications, including aerodynamics, fusion, and combustion. Accurately capturing turbulence's multiscale characteristics is essential for reliable predictions of multiphysics interactions, but remains a grand challenge even for exascale supercomputers and advanced deep learning models. The extreme-resolution data required to represent turbulence, ranging from billions to trillions of grid points, pose prohibitive computational costs for models based on architectures like vision transformers. To address this challenge, we introduce a multiscale hierarchical Turbulence Transformer that reduces sequence length from billions to a few millions and a novel RingX sequence parallelism approach that enables scalable long-context learning. We perform scaling and science runs on the Frontier supercomputer. Our approach demonstrates excellent performance up to 1.1 EFLOPS on 32,768 AMD GPUs, with a scaling efficiency of 94%. To our knowledge, this is the first AI model for turbulence that can capture small-scale eddies down to the dissipative range.
title Pixel-Resolved Long-Context Learning for Turbulence at Exascale: Resolving Small-scale Eddies Toward the Viscous Limit
topic Fluid Dynamics
Machine Learning
url https://arxiv.org/abs/2507.16697