Searcharxiv⌕ Search
Home/Archive/arXiv

arXiv · 2609.36381

Suêtes: An end-to-end differentiable non-hydrostatic limited-area dynamical core

Tim Whittaker·Seth Taylor·Elsa Cardoso-Bihlo·Alejandro Di Luca·Alex Bihlo
View original source ↗Read source document ↗Export metadata ↓

Abstract

We present Suêtes, a fully differentiable, non-hydrostatic, limited-area atmospheric dynamical core implemented in JAX. Suêtes includes both a semi-implicit semi-Lagrangian scheme and a Split-Explicit Runge-Kutta scheme with acoustic substepping, enabling applications from coarse regional simulations to convection-permitting dynamics. Its terrain-following geometry, spatial discretization, time integration, initial conditions, lateral boundary treatment, and included physical parameterizations are all compatible with reverse-mode automatic differentiation. A single reverse-mode pass can therefore compute sensitivities of forecast diagnostics to selected initial and boundary fields, physical parameters, topography, and computational geometry. We provide a series of numerical experiments---including a tracer-source inverse problem, a sensitivity analysis of a three-dimensional squall-line, adversarial perturbation construction an ERA5-driven downslope hurricane force wind event, topography optimization, and terrain-following coordinate optimization---demonstrating the versatility of Suêtes as a framework for gradient-based diagnosis and model development at regional and convection-permitting scale.

Explore related subjects

physics.ao-phmath.DSphysics.flu-dyn

Keep this discovery

Explore connections, maps & timelines

BibTeXRIS

Tim Whittaker, Seth Taylor, Elsa Cardoso-Bihlo, Alejandro Di Luca, Alex Bihlo. 2026-09-28. Suêtes: An end-to-end differentiable non-hydrostatic limited-area dynamical core. https://arxiv.org/abs/2609.36381

Cite the original work for its findings. Save a collection to share your selection of sources.

KEEP EXPLORING

Related papers

arXiv2026

A Neural-Network Model-Measurement-Based Observation Operator For Weather Radar Reflectivity Assimilation

In three-dimensional variational data assimilation (3DVar) for numerical weather prediction (NWP), the observation operator $\mathcal{H}$ plays a central role by mapping model state variables to an observation equivalent. For weather radar, however, specifying $\mathcal{H}$ is particularly challenging: reflectivity is a nonlinear, microphysics-dependent diagnostic quantity that only indirectly relates to the model's prognostic variables, making traditional parameterised radar operators complex, regime-dependent and difficult to tune. In this study, we propose a neural-network (NN)-based observation operator for radar reflectivity and apply it within a 3DVar framework. Using five years (2019-2023) of radar reflectivity data from the Lisca radar and 4.4 km-resolution short-range forecasts from ALADIN model over Slovenia, we train a convolutional encoder-decoder neural network to map model temperature, humidity, horizontal wind components and surface pressure fields to radar reflectivity. Across independent test cases spanning clear-sky, stratiform, and convective regimes, the NN-based operator accurately reproduces the spatial structure and intensity of observed reflectivity, relying primarily on the model state near the observation point. In the extreme precipitation case, which caused widespread floods in Slovenia on August 4, 2023, assimilating the full radar disc reduces the domain-averaged reflectivity root-mean-square error from 5.99 dBZ to 3.47 dBZ and improves the alignment between the analysed and observed convective bands. Embedded within 3DVar, the Jacobian of the NN observation operator allows radar reflectivity observations to inform model state variables, producing corresponding analysis increments. The proposed NN radar observation operator offers a flexible alternative to traditional parameterised radar operators for improving convective-storm forecasts.

physics.ao-ph↗
arXiv2026

FAST-ML: A Hybrid Physics-Machine Learning Framework for Tropical Cyclone Intensity Forecasting

Rapid intensification (RI) remains one of the most consequential and difficult aspects of tropical cyclone (TC) forecasting. Although full-physics numerical weather prediction models can represent the processes governing RI, resolving storm-environment interactions remains computationally expensive, while purely data-driven approaches often lack physical interpretability. We present FAST-ML, a hybrid framework that bridges data-driven efficiency with physical constraints. A physically informed dual-stream neural parameterization ingests 3D ERA5 fields to diagnose ventilation controls---environmental wind shear and mid-level entropy deficit. By optimizing these parameters end-to-end through a differentiable FAST intensity model, this architecture establishes a robust new paradigm for observation-driven parameter optimization, ensuring storm evolution remains strictly governed by thermodynamic principles. By better capturing the storm's continuous intensity evolution, FAST-ML improves upon its physical baseline, reducing ensemble CRPS across forecast lead times, with a reduction of approximately 31% at 60 h and nearly halving the RI false alarm ratio without sacrificing detection skill. In a 100-member ensemble configuration, FAST-ML produces intensity forecasts comparable to FNV3 for selected storms under the evaluated input configurations. Furthermore, zero-shot tests on selected Eastern Pacific storms provide encouraging evidence of cross-basin transferability. FAST-ML provides a modular intensity forecasting framework that can be coupled with externally supplied storm tracks and environmental fields. It demonstrates that observation-driven parameter learning within physically constrained dynamics simultaneously enhances accuracy, interpretability, and computational efficiency.

physics.ao-ph↗
arXiv2026

Experimental validation of an open-source low-cost single-camera 6-DOF tracking of floating-body motion in wave tanks

Accurately measuring motion of floating structures in experimental settings is both important and non-trivial. In this paper, we present a single-camera, 6-degree-of-freedom motion-tracking system that is both low-cost and simple to set up for an experimental campaign. The system uses fiducial markers and open-source computer vision tools to estimate the positions and orientations of multi-marker geometries and track them. We validate the accuracy and limitations on a precisely controlled linear actuator with static, regular, and irregular motion. The performance is quantified depending on both the camera-to-marker distance and direction of motion relative to the camera plane. Additionally, we present a practical use case for our system in a wave tank. Although the motion direction perpendicular to the camera plane shows the highest errors, the system achieves sub-millimeter accuracy at short and moderate camera-to-marker distances. For the irregular motion validation, the best cases reproduced the displacements with an RMSE of approximately 0.5 mm. Overall, the results indicate that low-cost single-camera fiducial-marker tracking can provide sufficiently accurate, non-intrusive motion measurements for a range of hydrodynamic laboratory experiments.

physics.ao-ph↗