{
  "fit_method": "gradient",
  "io_init": {
    "in_dim": 3,
    "out_dim": 2,
    "representation": "model init dimensions (not runtime output shape semantics)",
    "x_names": [
      "x0",
      "x1",
      "x2"
    ],
    "y_names": [
      "y0",
      "y1"
    ]
  },
  "io_pattern": "array->array",
  "model_family_id": "mlp_v1",
  "provenance": {
    "dependencies": {
      "flax": "0.10.4",
      "jax": "0.4.38",
      "numpy": "2.3.5",
      "optax": "0.2.5"
    },
    "git_sha": "853a30387805735584ec72dd7274e0fdb48591ba",
    "platform": "macOS-26.3.1-x86_64-i386-64bit",
    "python_version": "3.12.11",
    "saved_at_utc": "2026-04-10T02:09:02.834895+00:00",
    "toolkit": "astro_emulators_toolkit",
    "toolkit_version": "0.1.0"
  },
  "resolved": {
    "model": {
      "name": "mlp",
      "params": {
        "activation": "gelu",
        "dtype": "float32",
        "hidden_sizes": [
          32,
          32
        ],
        "use_bias": true
      }
    },
    "solver": {
      "name": "gradient",
      "params": {}
    },
    "task": {
      "name": "regression",
      "params": {
        "loss": "mse",
        "loss_weights": null,
        "metric_axes": null,
        "metrics": [
          "mse",
          "mae"
        ]
      }
    }
  },
  "runtime_contract": {
    "affine_leaf_shapes": {
      "inputs/x0": [],
      "inputs/x1": [],
      "inputs/x2": [],
      "outputs/y0": [],
      "outputs/y1": []
    },
    "input_name_tree": [
      "x0",
      "x1",
      "x2"
    ],
    "io_pattern": "array->array",
    "output_name_tree": [
      "y0",
      "y1"
    ]
  },
  "schema_version": 1,
  "spec": {
    "reference_scaling": {
      "kind": "affine_v1",
      "parametrization": "minmax",
      "space": "runtime_tree_v1",
      "storage": {
        "filename": "reference_scaling.safetensors",
        "format": "safetensors_v1",
        "layout": "tree_minmax_v1"
      }
    },
    "spec_version": 2,
    "x": {
      "meaning": [
        "",
        "",
        ""
      ],
      "names": [
        "x0",
        "x1",
        "x2"
      ],
      "units": [
        "",
        "",
        ""
      ]
    },
    "y": {
      "meaning": [
        "",
        ""
      ],
      "names": [
        "y0",
        "y1"
      ],
      "units": [
        "",
        ""
      ]
    }
  },
  "tool": "astro_emulators_toolkit",
  "weights_layout": "params_plus_model_state_v1"
}