Upload 10 files

reference_bundle_release/README.txt

@@ -0,0 +1,103 @@
+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
+  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'}}
+  extras: ['notes', 'wavelength_angstrom']
+
+Provenance:
+  toolkit_version: 0.1.0
+  created_at: 2026-04-12T14:23:17.605616+00:00
+  python_version: 3.12.13
+  git_commit: e922f60957ded21c91aca36a1d11b2bd269cc082
+
+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:
+  kind: affine_v1
+  max_tree:
+    inputs:
+      parameters:
+        - 7000.0
+        - 5.0
+        - 0.30000001192092896
+    outputs:
+      flux: 1.0
+  min_tree:
+    inputs:
+      parameters:
+        - 4500.0
+        - 2.5
+        - -0.30000001192092896
+    outputs:
+      flux: 0.0
+  parametrization: minmax
+  space: canonical_dict_trees_v1
+  storage:
+    filename: reference_scaling.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.

reference_bundle_release/bundle_integrity.json

@@ -0,0 +1,43 @@
+{
+  "algorithm": "sha256",
+  "bundle_id": "sha256:47a25983d0def8a111961baf403dcdc635de9470b2ff1f69ad99a82a4d94078d",
+  "integrity_format_version": 1,
+  "tree": [
+    {
+      "path": "README.txt",
+      "sha256": "ed787ef41d2abd132cf4de82dafa48e68d71eb9857b03131bc4429cbe25887f8"
+    },
+    {
+      "path": "config.json",
+      "sha256": "4fdbda81f1412fb3a272b995c7fcebad3c5f8a29ce89ad3786671960a73985aa"
+    },
+    {
+      "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": "46a57f9fbe5065462e516724e6082a2f3546db0c80e59c01a6d7d28edc7a7f4e"
+    },
+    {
+      "path": "reference_scaling.safetensors",
+      "sha256": "47c702a067f7080838deb437b3cc9c78a446bc8ff8a57fcd7f5cb2ef195af459"
+    },
+    {
+      "path": "weights/weights.safetensors",
+      "sha256": "3583bb65fd0ffd091cf3f5d0236ae3e149ffa41c8ebc5b3c6589eecb7900e592"
+    }
+  ]
+}

reference_bundle_release/config.json

@@ -0,0 +1,110 @@
+{
+  "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": null
+  },
+  "model": {
+    "init_hints": {
+      "input_last_axis": 3,
+      "output_last_axis": 500
+    },
+    "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,
+    "evaluation_interval_steps": 500,
+    "logging_interval_steps": 50,
+    "max_saved_checkpoints": 5,
+    "num_steps": 10000,
+    "shuffle": true,
+    "shuffle_seed": 0,
+    "steps_per_epoch": null,
+    "val_fraction": 0.1,
+    "workdir": "./runs/from_bundle"
+  }
+}

reference_bundle_release/extras/wavelength_angstrom.safetensors

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

reference_bundle_release/fingerprint_evaluation/inputs.safetensors

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

reference_bundle_release/fingerprint_evaluation/outputs.safetensors

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

reference_bundle_release/input_domain.safetensors

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

reference_bundle_release/metadata.json

@@ -0,0 +1,212 @@
+{
+  "bundle_format_version": 2,
+  "config_schema_version": 1,
+  "extras": {
+    "notes": "Inputs are min-max scaled before training using the fixed bounds stored in this bundle. Output reference scaling is identity.",
+    "wavelength_angstrom": {
+      "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_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-12T14:23:17.605616+00:00",
+    "dependencies": {
+      "flax": "0.12.6",
+      "jax": "0.9.2",
+      "numpy": "2.4.4",
+      "optax": "0.2.8"
+    },
+    "git_commit": "e922f60957ded21c91aca36a1d11b2bd269cc082",
+    "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": {
+      "kind": "affine_v1",
+      "max_tree": {
+        "inputs": {
+          "parameters": [
+            7000.0,
+            5.0,
+            0.30000001192092896
+          ]
+        },
+        "outputs": {
+          "flux": 1.0
+        }
+      },
+      "min_tree": {
+        "inputs": {
+          "parameters": [
+            4500.0,
+            2.5,
+            -0.30000001192092896
+          ]
+        },
+        "outputs": {
+          "flux": 0.0
+        }
+      },
+      "parametrization": "minmax",
+      "space": "canonical_dict_trees_v1",
+      "storage": {
+        "filename": "reference_scaling.safetensors",
+        "format": "safetensors_v1",
+        "layout": "split_minmax_tree_v1"
+      }
+    },
+    "spec_version": 1
+  },
+  "weights_layout": "params_plus_model_state_v1"
+}

reference_bundle_release/reference_scaling.safetensors

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

reference_bundle_release/weights/weights.safetensors

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