config.json

{
  "model_name": "NeuralGCM",
  "model_type": "neuralgcm",
  "architectures": [
    "PressureLevelModel"
  ],
  "framework": "JAX with Haiku and Gin",
  "domain": "atmosphere-and-climate",
  "task": "global-weather-forecasting-and-climate-simulation",
  "implementation": {
    "entry_point": "model/NeuralGCM.py",
    "scope": "vendored NeuralGCM pressure-level inference and training facade with differentiable dynamics and neural parameterizations"
  },
  "architecture": {
    "family": "hybrid differentiable general circulation model",
    "dynamical_core": "spectral atmospheric dynamics",
    "learned_components": [
      "encoder",
      "decoder",
      "subgrid physical parameterizations"
    ],
    "input_format": "xarray pressure-level fields",
    "grid_shape_order": [
      "longitude",
      "latitude"
    ],
    "time_step_hours": 6,
    "pressure_level_count": 37,
    "dynamic_variable_count": 7,
    "forcing_variable_count": 2,
    "flattened_input_channels": 261,
    "profiles": {
      "weather_forecast": {
        "type": "deterministic weather forecast",
        "resolution_degrees": 0.7,
        "grid_shape": [
          512,
          256
        ],
        "checkpoint": "weight/models_v1_deterministic_0_7_deg.pkl"
      },
      "climate_scale": {
        "type": "deterministic climate simulation",
        "resolution_degrees": 1.4,
        "grid_shape": [
          256,
          128
        ],
        "checkpoint": "weight/models_v1_deterministic_1_4_deg.pkl"
      },
      "forecast_2_8_deg": {
        "type": "deterministic weather forecast",
        "resolution_degrees": 2.8,
        "grid_shape": [
          128,
          64
        ],
        "checkpoint": "weight/models_v1_deterministic_2_8_deg.pkl"
      },
      "stochastic_1_4_deg": {
        "type": "stochastic weather forecast",
        "resolution_degrees": 1.4,
        "grid_shape": [
          256,
          128
        ],
        "checkpoint": "weight/models_v1_stochastic_1_4_deg.pkl"
      }
    }
  },
  "data": {
    "dataset": "ERA5",
    "source_grid_shape": [
      721,
      1440
    ],
    "time_step_hours": 6,
    "input_steps": 1,
    "pressure_levels_hpa": [
      1,
      2,
      3,
      5,
      7,
      10,
      20,
      30,
      50,
      70,
      100,
      125,
      150,
      175,
      200,
      225,
      250,
      300,
      350,
      400,
      450,
      500,
      550,
      600,
      650,
      700,
      750,
      775,
      800,
      825,
      850,
      875,
      900,
      925,
      950,
      975,
      1000
    ],
    "input_variables": [
      "geopotential",
      "specific_humidity",
      "temperature",
      "u_component_of_wind",
      "v_component_of_wind"
    ],
    "optional_input_variables": [
      "specific_cloud_ice_water_content",
      "specific_cloud_liquid_water_content"
    ],
    "forcing_variables": [
      "sea_ice_cover",
      "sea_surface_temperature"
    ],
    "protocol": "era5_37_pressure_levels_261_channel"
  },
  "weights": {
    "license": "CC BY-SA 4.0",
    "source": "gs://neuralgcm/models/v1/"
  },
  "configuration_sources": [
    "conf/config.yaml",
    "configuration.json",
    "model/NeuralGCM.py",
    "model/legacy",
    "model/reference_code",
    "scripts/common.py"
  ]
}