Upload 11 files

README.txt

@@ -0,0 +1,114 @@
+Astro Emulators Toolkit Bundle
+
+Summary:
+  model: mlp
+  release: payne-flux-reference-example@0.1.0 (released)
+  bundle_format_version: 1
+  config_schema_version: 1
+  spec_version: 1
+  weights_layout: params_plus_model_state_v1
+  model_family_id: mlp_v1
+  fingerprint_evaluation: present
+  task: regression
+  fit_method: gradient
+  solver_params: not provided
+  solver_diagnostics: not provided
+  solver_design_matrix: not provided
+  role_paths: {'input_leaf': 'inputs/parameters', 'output_leaf': 'outputs/flux'}
+
+Domain:
+  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'}}
+  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'}}
+  extras: ['notes', 'wavelength_angstrom']
+
+Provenance:
+  toolkit_version: 0.1.0
+  created_at: 2026-04-17T13:11:14.921005+00:00
+  python_version: 3.12.13
+  git_commit: c786aa2f0453ba260c000411f586f943a3109976
+
+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: None
+  structure_tree:
+    parameters: None
+outputs:
+  channel_meanings_tree: None
+  channel_names_tree: None
+  channel_units_tree: None
+  leaf_meanings_tree:
+    flux: continuum-normalized flux vector on the shared wavelength grid
+  leaf_units_tree:
+    flux: dimensionless
+  structure_tree:
+    flux: None
+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
+
+Note: this bundle is the canonical emulator artifact. Physical-space composition is external.

bundle_integrity.json

@@ -0,0 +1,47 @@
+{
+  "algorithm": "sha256",
+  "bundle_id": "sha256:681336e5d114856ea625386cfa3b5d27480a49df42dd0ff39ac8e1fc45e415ed",
+  "integrity_format_version": 1,
+  "tree": [
+    {
+      "path": "README.txt",
+      "sha256": "846a4ccbd1603534301a9e8a4158d0560f6d132f2846af4b2cb6c6950a283cdf"
+    },
+    {
+      "path": "config.json",
+      "sha256": "f75a262db2c158750fea3f124c5d8c8ae590797d3d63e0250b792a841d1f7b26"
+    },
+    {
+      "path": "extras/wavelength_angstrom.safetensors",
+      "sha256": "194b2e671c6995b9532cafcdeb72fbf06365dba2259611be4985836f6c617896"
+    },
+    {
+      "path": "fingerprint_evaluation/inputs.safetensors",
+      "sha256": "e411d571388dd13d2c5c9f020386c14378f2a63e63a8a28182b3ac2a45f60e60"
+    },
+    {
+      "path": "fingerprint_evaluation/outputs.safetensors",
+      "sha256": "a9555eceddf85014098c7aad0ef8052e385d008ee68582a26ba830f642d7dcc7"
+    },
+    {
+      "path": "input_domain.safetensors",
+      "sha256": "62290cabed8c655431439b63139df4d72f1f1fdc13772b1f1e306d4a5a5abade"
+    },
+    {
+      "path": "metadata.json",
+      "sha256": "b253d863e20a045a69bb8e4b9032d8e969226cfae1cea71b65dbc77ba14bd33c"
+    },
+    {
+      "path": "reference_scaling_inputs.safetensors",
+      "sha256": "2ea76cae2ecfd992e59e04361d2c15591c2f511aa8e484e7610b4536cce55444"
+    },
+    {
+      "path": "reference_scaling_outputs.safetensors",
+      "sha256": "d1441a6518c789aff00d5c53d76898cc98b7bbdbbf39e34704102e529a923313"
+    },
+    {
+      "path": "weights/weights.safetensors",
+      "sha256": "78c6b15002b7eeae2fee03c08d4dcf19f429d9742e8710c3aa013e980ae372d9"
+    }
+  ]
+}

config.json

@@ -0,0 +1,111 @@
+{
+  "bundle": {
+    "bundle_subdir": "bundle"
+  },
+  "hub": {
+    "repo_id": null,
+    "revision": null
+  },
+  "io": {
+    "input_domain": null,
+    "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": null,
+    "reference_scaling_outputs": null
+  },
+  "model": {
+    "init_hints": {},
+    "name": "mlp",
+    "params": {
+      "activation": "gelu",
+      "dtype": "float32",
+      "hidden_sizes": [
+        128,
+        128
+      ]
+    }
+  },
+  "optim": {
+    "b1": 0.9,
+    "b2": 0.999,
+    "decay_steps": 0,
+    "eps": 1e-08,
+    "lr": 0.001,
+    "name": "soap",
+    "precondition_1d": false,
+    "precondition_frequency": 10,
+    "schedule": "cosine",
+    "warmup_steps": 1000,
+    "weight_decay": 1e-05
+  },
+  "schema_version": 1,
+  "seed": 0,
+  "solver": {
+    "name": "auto",
+    "params": {}
+  },
+  "task": {
+    "name": "regression",
+    "params": {
+      "loss": "mse"
+    }
+  },
+  "training": {
+    "batch_size": 128,
+    "checkpoint_interval_steps": 500,
+    "checkpoint_steps": null,
+    "evaluation_interval_steps": 500,
+    "evaluation_steps": null,
+    "logging_interval_steps": 50,
+    "logging_steps": null,
+    "max_saved_checkpoints": 5,
+    "num_steps": 10000,
+    "shuffle": true,
+    "shuffle_seed": 0,
+    "steps_per_epoch": null,
+    "val_fraction": 0.1,
+    "workdir": "./runs/from_bundle"
+  }
+}

extras/wavelength_angstrom.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:194b2e671c6995b9532cafcdeb72fbf06365dba2259611be4985836f6c617896
+size 4072

fingerprint_evaluation/inputs.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:e411d571388dd13d2c5c9f020386c14378f2a63e63a8a28182b3ac2a45f60e60
+size 92

fingerprint_evaluation/outputs.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:a9555eceddf85014098c7aad0ef8052e385d008ee68582a26ba830f642d7dcc7
+size 2072

input_domain.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:62290cabed8c655431439b63139df4d72f1f1fdc13772b1f1e306d4a5a5abade
+size 208

metadata.json

@@ -0,0 +1,222 @@
+{
+  "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"
+}

reference_scaling_inputs.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:2ea76cae2ecfd992e59e04361d2c15591c2f511aa8e484e7610b4536cce55444
+size 192

reference_scaling_outputs.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:d1441a6518c789aff00d5c53d76898cc98b7bbdbbf39e34704102e529a923313
+size 144

weights/weights.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:78c6b15002b7eeae2fee03c08d4dcf19f429d9742e8710c3aa013e980ae372d9
+size 326784