{
  "mic": "0.3",
  "name": "WeatherNext2 forecast",
  "usage": "Feature extraction, a batched forecast, physical postprocessing, and ensemble CRPS training",
  "architecture": "WeatherNext2ForWeatherForecasting",
  "checkpoints": {
    "kashif/weathernext2-mini": {
      "fixture": "huggingface-mic/weathernext2-mic"
    }
  },
  "fixtures": {
    "huggingface-mic/weathernext2-mic": {
      "built_from": "kashif/weathernext2-mini",
      "covers": [
        "kashif/weathernext2-mini"
      ],
      "build": {
        "seed": 17,
        "model_class": "WeatherNext2ForWeatherForecasting",
        "config_overrides": {
          "fields": {
            "graph_config.edge_hidden_size": 64,
            "grid_latitudes": 9,
            "grid_longitudes": 16,
            "mesh_splits": 2,
            "attention_k_hop": 2
          },
          "reason": "Shrink the network, weather grid and mesh together; regenerate genuine spherical geometry and its mask."
        }
      }
    }
  },
  "integrations": {
    "transformers": {
      "entry_point": "integrations/transformers/usage.md",
      "format": "runnable-markdown",
      "requires": [
        "transformers>=5",
        "torch"
      ],
      "targets": [
        "fixtures"
      ],
      "variants": {
        "default": {},
        "eager": {
          "attn_implementation": "eager"
        },
        "bfloat16": {
          "dtype": "bfloat16"
        }
      },
      "training": true
    }
  },
  "outputs": {
    "runtime": {
      "compare": "exact"
    },
    "shape": {
      "compare": "exact"
    },
    "prediction": {
      "compare": "allclose",
      "atol": 0.0001,
      "rtol": 0.0001
    },
    "temperature": {
      "compare": "allclose",
      "atol": 0.0001,
      "rtol": 0.0001
    },
    "finite": {
      "compare": "true"
    },
    "training.loss": {
      "compare": "allclose",
      "atol": 0.0001,
      "rtol": 0.0001
    },
    "training.gradients": {
      "compare": "all_true"
    },
    "training.updated": {
      "compare": "true"
    }
  }
}