models/fuxi.md

FuXi — cube embedding + Swin-V2 U-Transformer (PyTorch)

← Model zoo · All model pages · Paper: npj Climate and Atmospheric Science 2023, doi:10.1038/s41612-023-00512-1

weatherai.models.fuxi (adapted from WeatherLearn, MIT): space-time cube embedding of two input steps, a Swin-V2 U-Transformer and a fully connected head. Architecture only — the official FuXi weights are released as ONNX and are not loaded here. The ensemble and sub-seasonal members of the family are separate, verified ports: FuXi-ENS and FuXi-S2S.

from weatherai.models import FuXi
m = FuXi()

Tests: tests/models/fuxi.