{
  "era5_variables": [
    "u10m",
    "v10m",
    "u100m",
    "v100m",
    "t2m",
    "sp",
    "msl",
    "u300",
    "u400",
    "u500",
    "u600",
    "u700",
    "u850",
    "u925",
    "u1000",
    "v300",
    "v400",
    "v500",
    "v600",
    "v700",
    "v850",
    "v925",
    "v1000",
    "z300",
    "z400",
    "z500",
    "z600",
    "z700",
    "z850",
    "z925",
    "z1000",
    "t300",
    "t400",
    "t500",
    "t600",
    "t700",
    "t850",
    "t925",
    "t1000",
    "q300",
    "q400",
    "q500",
    "q600",
    "q700",
    "q850",
    "q925",
    "q1000"
  ],
  "output_variables": [
    "10U",
    "10V",
    "2MT",
    "TD_2M",
    "TOT_PRECIP",
    "CLCT",
    "ASWDIFD_S",
    "ASWDIR_S",
    "PS",
    "H_PBL",
    "U3D_L45",
    "U3D_L46",
    "U3D_L47",
    "U3D_L48",
    "U3D_L49",
    "U3D_L50",
    "V3D_L45",
    "V3D_L46",
    "V3D_L47",
    "V3D_L48",
    "V3D_L49",
    "V3D_L50"
  ],
  "pre_invariant_variables": [
    "sin_lat",
    "cos_lat",
    "sin_lon",
    "cos_lon",
    "elevation_norm",
    "land_fraction"
  ],
  "post_invariant_variables": [
    "z0_lu_norm",
    "slope_east_norm",
    "slope_north_norm"
  ],
  "channel_transforms": {
    "TOT_PRECIP": {
      "transform": "log_eps",
      "eps": 1e-05,
      "forward": "log(x / eps + 1)",
      "inverse": "eps * (exp(y) - 1) with eps=1e-05"
    },
    "ASWDIR_S": {
      "transform": "log_eps",
      "eps": 1.0,
      "forward": "log(x / eps + 1)",
      "inverse": "eps * (exp(y) - 1) with eps=1.0"
    },
    "ASWDIFD_S": {
      "transform": "log_eps",
      "eps": 1.0,
      "forward": "log(x / eps + 1)",
      "inverse": "eps * (exp(y) - 1) with eps=1.0"
    },
    "CLCT": {
      "transform": "logit_eps_percent",
      "eps": 0.01,
      "scale": 100.0,
      "forward": "logit(eps + (1 - 2*eps) * (x/100))",
      "inverse": "((sigmoid(y) - eps) / (1 - 2*eps)) * 100, clip to [0, 100]"
    },
    "H_PBL": {
      "transform": "log_eps",
      "eps": 10.0,
      "forward": "log(x / eps + 1)",
      "inverse": "eps * (exp(y) - 1) with eps=10.0"
    }
  },
  "modes": {
    "mean": {
      "stats": "output_physical",
      "channel_transforms": false,
      "space": "physical"
    },
    "diffusion": {
      "stats": "output",
      "channel_transforms": true,
      "space": "transformed"
    }
  },
  "constraints": {
    "bounds": {
      "TOT_PRECIP": {
        "min": 0.0,
        "max": null,
        "units": "mm/h"
      },
      "ASWDIFD_S": {
        "min": 0.0,
        "max": null,
        "units": "W/m^2"
      },
      "ASWDIR_S": {
        "min": 0.0,
        "max": null,
        "units": "W/m^2"
      },
      "CLCT": {
        "min": 0.0,
        "max": 100.0,
        "units": "%",
        "mode": "clamp"
      },
      "PS": {
        "min": 0.0,
        "max": null,
        "units": "Pa"
      },
      "2MT": {
        "min": 0.0,
        "max": null,
        "units": "K"
      },
      "TD_2M": {
        "min": 0.0,
        "max": null,
        "units": "K"
      },
      "H_PBL": {
        "min": 0.0,
        "max": null,
        "units": "m"
      }
    },
    "sza_gate": {
      "half_width": 0.05,
      "channels": {
        "ASWDIFD_S": {
          "threshold": 0.01,
          "half_width": 0.02
        },
        "ASWDIR_S": {
          "threshold": 0.05
        }
      }
    }
  },
  "invariants": {
    "file": "invariants_rea2_ext.nc",
    "norm_stats_file": "invariants_norm_stats.json",
    "channels": [
      "elevation",
      "land_fraction",
      "z0_lu",
      "slope_east",
      "slope_north"
    ],
    "native_offset": [
      1000,
      1000
    ],
    "native_shape": [
      780,
      724
    ],
    "extended_shape": [
      2780,
      2724
    ],
    "grid": {
      "rotated_pole_lat": 40.0,
      "rotated_pole_lon": -170.0,
      "dlat": 0.018,
      "dlon": 0.018,
      "startlat_ext": -24.0,
      "startlon_ext": -25.5
    },
    "values": "physical; wrapper z-scores with norm_stats_file after sampling"
  },
  "wind_levels": {
    "elevation_invariant": "elevation_norm",
    "levels": [
      {
        "u": "U3D_L45",
        "v": "V3D_L45",
        "a": 252.2799,
        "b": -9.6701
      },
      {
        "u": "U3D_L46",
        "v": "V3D_L46",
        "a": 179.7021,
        "b": -6.8879
      },
      {
        "u": "U3D_L47",
        "v": "V3D_L47",
        "a": 119.5022,
        "b": -4.5769
      },
      {
        "u": "U3D_L48",
        "v": "V3D_L48",
        "a": 71.3555,
        "b": -2.7345
      },
      {
        "u": "U3D_L49",
        "v": "V3D_L49",
        "a": 34.8919,
        "b": -1.3361
      },
      {
        "u": "U3D_L50",
        "v": "V3D_L50",
        "a": 9.7612,
        "b": -0.369
      }
    ]
  },
  "number_of_samples": 1,
  "checkpoints": {
    "rea2_mean": "regression.mdlus",
    "rea2_diffusion": "diffusion.mdlus"
  },
  "regression": {
    "hidden_size": 576,
    "depth": 16,
    "num_heads": 9,
    "patch_size": 2,
    "attn_kernel_size": 23,
    "rope_theta": 10000.0,
    "img_resolution": [
      256,
      256
    ],
    "qk_norm": false,
    "layernorm_backend": "torch",
    "detok_conv_layers": 2
  },
  "diffusion": {
    "hidden_size": 768,
    "depth": 16,
    "num_heads": 12,
    "patch_size": 2,
    "attn_kernel_size": 23,
    "sigma_data": 1.0,
    "rope_theta": 10000.0,
    "img_resolution": [
      256,
      256
    ],
    "qk_norm": false,
    "layernorm_backend": "torch"
  },
  "sampler": {
    "num_steps": 18,
    "sigma_min": 0.002,
    "sigma_max": 800.0,
    "rho": 7.0,
    "solver": "heun"
  }
}