| # FuXi — cube embedding + Swin-V2 U-Transformer (PyTorch) | |
| [← Model zoo](../README.md) · [All model pages](README.md) · **Paper:** *npj Climate and Atmospheric Science* 2023, [doi:10.1038/s41612-023-00512-1](https://doi.org/10.1038/s41612-023-00512-1) | |
| `weatherai.models.fuxi` (adapted from [WeatherLearn](https://github.com/lizhuoq/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](fuxi_ens.md) and [FuXi-S2S](fuxi_s2s.md). | |
| ```python | |
| from weatherai.models import FuXi | |
| m = FuXi() | |
| ``` | |
| Tests: `tests/models/fuxi`. | |