{
  "$schema": "https://json-schema.org/draft/2020-12/schema",
  "title": "Diffusion language baseline experiment registry",
  "type": "object",
  "additionalProperties": false,
  "required": ["step_grids", "models"],
  "properties": {
    "step_grids": {
      "type": "object",
      "additionalProperties": false,
      "required": ["many", "few", "fixed_1024"],
      "properties": {
        "many": {"const": [1, 2, 4, 8, 16, 32, 1024]},
        "few": {"const": [1, 2, 4, 8, 16, 32]},
        "fixed_1024": {"const": [1024]}
      }
    },
    "models": {
      "type": "object",
      "additionalProperties": {
        "type": "object",
        "additionalProperties": false,
        "required": ["category", "environment", "adapter", "source", "datasets"],
        "properties": {
          "category": {"enum": ["many", "few", "fixed_1024"]},
          "step_override": {
            "type": "array",
            "minItems": 1,
            "uniqueItems": true,
            "items": {"type": "integer", "minimum": 1}
          },
          "environment": {"pattern": "^[a-z][a-z0-9-]*$"},
          "adapter": {"pattern": "^[a-z][a-z0-9-]*$"},
          "source": {"pattern": "^[a-z][a-z0-9-]*$"},
          "datasets": {
            "type": "object",
            "additionalProperties": false,
            "required": ["lm1b", "owt"],
            "properties": {
              "lm1b": {"$ref": "#/$defs/dataset_support"},
              "owt": {"$ref": "#/$defs/dataset_support"}
            }
          }
        }
      }
    }
  },
  "$defs": {
    "dataset_support": {
      "type": "object",
      "additionalProperties": false,
      "required": ["status"],
      "properties": {
        "status": {"enum": ["supported", "unsupported"]},
        "provenance": {"enum": ["official", "reference_reproduction", "self_trained"]},
        "reason": {"type": "string", "minLength": 1},
        "train_recipe": {"type": "string", "pattern": "^[a-z][a-z0-9_]*$"}
      },
      "allOf": [
        {
          "if": {"properties": {"status": {"const": "supported"}}, "required": ["status"]},
          "then": {
            "required": ["provenance"],
            "not": {"required": ["reason"]}
          }
        },
        {
          "if": {"properties": {"status": {"const": "unsupported"}}, "required": ["status"]},
          "then": {
            "required": ["reason"],
            "not": {
              "anyOf": [
                {"required": ["provenance"]},
                {"required": ["train_recipe"]}
              ]
            }
          }
        }
      ]
    }
  }
}