{
  "model_name": "GenCast",
  "model_type": "gencast",
  "architectures": [
    "GenCast"
  ],
  "framework": "JAX with Haiku",
  "domain": "atmosphere",
  "task": "probabilistic-global-weather-forecasting",
  "implementation": {
    "entry_point": "model/gencast.py",
    "scope": "official-equivalent model API with a configurable mini random-initialization profile"
  },
  "architecture": {
    "family": "graph-based denoising diffusion model",
    "mesh_refinement_level": 4,
    "mesh_nodes": 2562,
    "latent_size": 512,
    "mlp_hidden_layers": 1,
    "sparse_transformer_layers": 16,
    "attention_heads": 4,
    "feed_forward_size": 2048,
    "attention_k_hop": 16,
    "attention_type": "triblockdiag_mha"
  },
  "data": {
    "dataset": "ERA5",
    "input_duration_hours": 24,
    "forecast_step_hours": 12,
    "target_channels": 84,
    "pressure_levels_hpa": [
      50,
      100,
      150,
      200,
      250,
      300,
      400,
      500,
      600,
      700,
      850,
      925,
      1000
    ],
    "pressure_level_variables": [
      "geopotential",
      "specific_humidity",
      "temperature",
      "u_component_of_wind",
      "v_component_of_wind",
      "vertical_velocity"
    ],
    "surface_target_variables": [
      "2m_temperature",
      "10m_u_component_of_wind",
      "10m_v_component_of_wind",
      "mean_sea_level_pressure",
      "sea_surface_temperature",
      "total_precipitation_12hr"
    ]
  },
  "sampling": {
    "noise_levels": 20,
    "minimum_noise_level": 0.03,
    "maximum_noise_level": 80.0,
    "rho": 7.0
  },
  "configuration_sources": [
    "conf/config.yaml",
    "model/gencast.py",
    "model/graphcast/gencast.py",
    "model/data_loader.py"
  ]
}