{
  "bundle_format_version": 1,
  "config_schema_version": 1,
  "extras": {
    "notes": "Inputs are min-max scaled before training using the fixed bounds stored in this bundle. Output reference scaling metadata is recorded separately and is identity.",
    "wavelength_angstrom": {
      "__aet_sidecar__": {
        "format": "safetensors_v1",
        "layout": "single_array_v1",
        "path": "extras/wavelength_angstrom.safetensors"
      }
    }
  },
  "fingerprint_evaluation": {
    "atol": 1e-07,
    "inputs": {
      "filename": "fingerprint_evaluation/inputs.safetensors",
      "format": "safetensors_v1",
      "layout": "numeric_dict_tree_v1",
      "space": "canonical_input_dict_trees_v1"
    },
    "kind": "canonical_inputs_outputs_v1",
    "outputs": {
      "filename": "fingerprint_evaluation/outputs.safetensors",
      "format": "safetensors_v1",
      "layout": "numeric_dict_tree_v1",
      "space": "canonical_output_dict_trees_v1"
    },
    "rtol": 1e-05,
    "selection_strategy": "midpoint_from_input_domain_then_reference_scaling_inputs_v1"
  },
  "fit_method": "gradient",
  "model_family_id": "mlp_v1",
  "model_init": {
    "hints": {
      "input_last_axis": 3,
      "output_last_axis": 500
    },
    "representation": "model-local init hints only"
  },
  "provenance": {
    "created_at": "2026-04-17T13:11:14.921005+00:00",
    "dependencies": {
      "flax": "0.12.6",
      "jax": "0.9.2",
      "numpy": "2.4.4",
      "optax": "0.2.8"
    },
    "git_commit": "c786aa2f0453ba260c000411f586f943a3109976",
    "platform": "macOS-26.4.1-arm64-arm-64bit",
    "python_version": "3.12.13",
    "toolkit": "astro_emulators_toolkit",
    "toolkit_version": "0.1.0"
  },
  "release": {
    "name": "payne-flux-reference-example",
    "status": "released",
    "version": "0.1.0"
  },
  "resolved": {
    "model": {
      "name": "mlp",
      "params": {
        "activation": "gelu",
        "dtype": "float32",
        "hidden_sizes": [
          128,
          128
        ],
        "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_specs": {
      "inputs/parameters": {
        "last_axis": 3,
        "mode": "scalar_or_last_axis"
      },
      "outputs/flux": {
        "last_axis": 500,
        "mode": "scalar_or_last_axis"
      }
    },
    "role_paths": {
      "input_leaf": "inputs/parameters",
      "output_leaf": "outputs/flux"
    },
    "surface": "canonical_dict_trees_v1"
  },
  "spec": {
    "input_domain": {
      "kind": "box_v1",
      "max_tree": {
        "inputs": {
          "parameters": [
            7000.0,
            5.0,
            0.30000001192092896
          ]
        }
      },
      "min_tree": {
        "inputs": {
          "parameters": [
            4500.0,
            2.5,
            -0.30000001192092896
          ]
        }
      },
      "space": "canonical_input_dict_trees_v1",
      "storage": {
        "filename": "input_domain.safetensors",
        "format": "safetensors_v1",
        "layout": "split_minmax_tree_v1"
      }
    },
    "inputs": {
      "channel_meanings_tree": {
        "parameters": [
          "effective temperature",
          "surface gravity",
          "metallicity [Fe/H]"
        ]
      },
      "channel_names_tree": {
        "parameters": [
          "teff",
          "logg",
          "feh"
        ]
      },
      "channel_units_tree": {
        "parameters": [
          "K",
          "dex",
          "dex"
        ]
      },
      "leaf_meanings_tree": {
        "parameters": "stellar labels"
      },
      "leaf_units_tree": null,
      "structure_tree": {
        "parameters": null
      }
    },
    "outputs": {
      "channel_meanings_tree": null,
      "channel_names_tree": null,
      "channel_units_tree": null,
      "leaf_meanings_tree": {
        "flux": "continuum-normalized flux vector on the shared wavelength grid"
      },
      "leaf_units_tree": {
        "flux": "dimensionless"
      },
      "structure_tree": {
        "flux": null
      }
    },
    "reference_scaling_inputs": {
      "kind": "affine_v1",
      "max_tree": {
        "parameters": [
          7000.0,
          5.0,
          0.30000001192092896
        ]
      },
      "min_tree": {
        "parameters": [
          4500.0,
          2.5,
          -0.30000001192092896
        ]
      },
      "parametrization": "minmax",
      "space": "canonical_input_dict_trees_v1",
      "storage": {
        "filename": "reference_scaling_inputs.safetensors",
        "format": "safetensors_v1",
        "layout": "split_minmax_tree_v1"
      }
    },
    "reference_scaling_outputs": {
      "kind": "affine_v1",
      "max_tree": {
        "flux": 1.0
      },
      "min_tree": {
        "flux": 0.0
      },
      "parametrization": "minmax",
      "space": "canonical_output_dict_trees_v1",
      "storage": {
        "filename": "reference_scaling_outputs.safetensors",
        "format": "safetensors_v1",
        "layout": "split_minmax_tree_v1"
      }
    },
    "spec_version": 1
  },
  "weights_layout": "params_plus_model_state_v1"
}