config.json

{
  "model_name": "FuXi v2.1",
  "model_type": "fuxi_v21",
  "architectures": [
    "FuXi21"
  ],
  "framework": "PyTorch",
  "domain": "atmosphere",
  "task": "global-medium-range-weather-forecasting",
  "implementation": {
    "entry_point": "model/FuXi21.py",
    "scope": "trainable forward-graph reconstruction with full-resolution and smoke profiles"
  },
  "architecture": {
    "family": "windowed Transformer with rotary attention and PixelShuffle decoder",
    "grid_shape": [
      721,
      1440
    ],
    "input_channels": 85,
    "output_channels": 85,
    "static_channels": 6,
    "patch_size": 6,
    "embedding_size": 1536,
    "depth": 30,
    "attention_heads": 24,
    "mlp_size": 4096,
    "window_size": 20,
    "activation": "SiLU/GELU"
  },
  "data": {
    "dataset": "ERA5",
    "spatial_resolution_degrees": 0.25,
    "time_step_hours": 6,
    "input_steps": 2,
    "pressure_levels_hpa": [
      50,
      100,
      150,
      200,
      250,
      300,
      400,
      500,
      600,
      700,
      850,
      925,
      1000
    ],
    "protocol": "non_official_protocol"
  },
  "configuration_sources": [
    "conf/config.yaml",
    "model/FuXi21.py"
  ]
}