File size: 791 Bytes
57e06a2 09af8e7 57e06a2 09af8e7 57e06a2 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 | # 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`.
|