WaveCastNet: An AI-enabled Wavefield Forecasting Framework for Earthquake Early Warning. - Dongwei Lyu; Rie Nakata; Pu Ren; Michael W. Mahoney; Arben Pitarka; Nori Nakata; N. Benjamin Erichson (CoRR, 2024)

Tuning Frequency Bias of State Space Models. - Annan Yu; Dongwei Lyu; Soon Hoe Lim; Michael W. Mahoney; N. Benjamin Erichson (ICLR, 2025)

FLEX: A Backbone for Diffusion-Based Modeling of Spatio-temporal Physical Systems. - N. Benjamin Erichson; Vinicius Mikuni; Dongwei Lyu; Yang Gao; Omri Azencot; Soon Hoe Lim; Michael W. Mahoney (CoRR, 2025)