Astra: efficient probabilistic weather forecasting at 0.25° with two-sided transport attention (Kantorovich Attention). Paper overview and the 0.25° evaluation code for arXiv:2610.35545.
attention-mechanism optimal-transport jax probabilistic-forecasting weather-forecasting crps ensemble-forecasting weatherbench2
-
Updated
Nov 6, 2026 - Python