Replace repo contents

README.md

@@ -1,3 +0,0 @@
----
-license: mit
----

README.txt

@@ -0,0 +1,124 @@
+Astro Emulators Toolkit Bundle
+
+Summary:
+  model: transformer_payne
+  release: maja-harps-intensity-tpayne-2ch-192x6@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: transformer_payne_v1
+  fingerprint_evaluation: present
+  task: regression
+  fit_method: gradient
+  solver_params: not provided
+  solver_diagnostics: not provided
+  solver_design_matrix: not provided
+  role_paths: {'output_leaf': 'outputs/flux', 'parameter_leaf': 'inputs/parameters', 'wavelength_leaf': 'inputs/wavelengths'}
+
+Domain:
+  input_domain: {'kind': 'box_v1', 'max_tree': {'parameters': [7500.0, 5.0, 0.5, 5.0, 0.5, 0.30000001192092896, 0.4000000059604645, 0.4000000059604645, 0.5, 0.5, 1.0], 'wavelengths': 3.838849090737255}, 'min_tree': {'parameters': [3800.0, 0.5, -3.0, 1.0, -0.20000000298023224, -0.30000001192092896, -0.20000000298023224, -0.20000000298023224, -0.20000000298023224, -0.20000000298023224, 0.1246189996600151], 'wavelengths': 3.57978359661681}, 'storage': {'filename': 'input_domain.safetensors', 'format': 'safetensors_v1', 'layout': 'split_minmax_tree_v1'}, 'value_space': 'physical_input_dict_tree_v1'}
+  reference_scaling_inputs: {'applies_to': 'inputs', 'kind': 'affine_minmax_v1', 'max_tree': {'parameters': [7500.0, 5.0, 0.5, 5.0, 0.5, 0.30000001192092896, 0.4000000059604645, 0.4000000059604645, 0.5, 0.5, 1.0], 'wavelengths': 3.838849090737255}, 'min_tree': {'parameters': [3800.0, 0.5, -3.0, 1.0, -0.20000000298023224, -0.30000001192092896, -0.20000000298023224, -0.20000000298023224, -0.20000000298023224, -0.20000000298023224, 0.1246189996600151], 'wavelengths': 3.57978359661681}, 'source_space': 'physical_input_dict_tree_v1', 'storage': {'filename': 'reference_scaling_inputs.safetensors', 'format': 'safetensors_v1', 'layout': 'split_minmax_tree_v1'}, 'target_space': 'canonical_input_dict_tree_v1'}
+  reference_scaling_outputs: {'applies_to': 'outputs', 'kind': 'affine_minmax_v1', 'max_tree': {'flux': [7.435205936431885, 7.439098834991455]}, 'min_tree': {'flux': [2.4139199256896973, 4.380068778991699]}, 'source_space': 'canonical_output_dict_tree_v1', 'storage': {'filename': 'reference_scaling_outputs.safetensors', 'format': 'safetensors_v1', 'layout': 'split_minmax_tree_v1'}, 'target_space': 'physical_output_dict_tree_v1'}
+  extras: ['evaluation_scaled_log10_wavelength', 'notes', 'parameter_source_units', 'preprocessing_recipe', 'source_log10_wavelength', 'source_log10_wavelength_by_channel', 'source_scaled_log10_wavelength', 'source_scaled_log10_wavelength_by_channel', 'source_store', 'source_stores', 'source_wavelength', 'source_wavelength_by_channel', 'training']
+
+Provenance:
+  toolkit_version: 0.1.0
+  created_at: 2026-05-01T13:33:41.693459+00:00
+  python_version: 3.12.13
+  git_commit: None
+
+spec:
+input_domain:
+  kind: box_v1
+  max_tree:
+    parameters:
+      array(shape=(11,), dtype=float64)
+    wavelengths: 3.838849090737255
+  min_tree:
+    parameters:
+      array(shape=(11,), dtype=float64)
+    wavelengths: 3.57978359661681
+  storage:
+    filename: input_domain.safetensors
+    format: safetensors_v1
+    layout: split_minmax_tree_v1
+  value_space: physical_input_dict_tree_v1
+inputs:
+  channel_meanings_tree:
+    parameters:
+      array(shape=(11,), dtype=<U28)
+    wavelengths: None
+  channel_names_tree:
+    parameters:
+      array(shape=(11,), dtype=<U6)
+    wavelengths: None
+  channel_units_tree:
+    parameters:
+      array(shape=(11,), dtype=<U13)
+    wavelengths: None
+  leaf_meanings_tree:
+    parameters: min-max scaled source-grid parameters; see reference_scaling_inputs for raw parameter bounds
+    wavelengths: min-max scaled log10 wavelength; the user applies log10 before reference_scaling_inputs
+  leaf_units_tree:
+    parameters: None
+    wavelengths: dimensionless
+  structure_tree:
+    parameters: None
+    wavelengths: None
+outputs:
+  channel_meanings_tree:
+    flux:
+      - min-max scaled log10 line intensity from source array flux
+      - min-max scaled log10 continuum intensity from source array continuum
+  channel_names_tree:
+    flux:
+      - log_flux_lines
+      - log_flux_continuum
+  channel_units_tree:
+    flux:
+      - dimensionless
+      - dimensionless
+  leaf_meanings_tree:
+    flux: two min-max scaled log10 intensity channels from the Maja archive arrays flux and continuum
+  leaf_units_tree:
+    flux: dimensionless
+  structure_tree:
+    flux: None
+reference_scaling_inputs:
+  applies_to: inputs
+  kind: affine_minmax_v1
+  max_tree:
+    parameters:
+      array(shape=(11,), dtype=float64)
+    wavelengths: 3.838849090737255
+  min_tree:
+    parameters:
+      array(shape=(11,), dtype=float64)
+    wavelengths: 3.57978359661681
+  source_space: physical_input_dict_tree_v1
+  storage:
+    filename: reference_scaling_inputs.safetensors
+    format: safetensors_v1
+    layout: split_minmax_tree_v1
+  target_space: canonical_input_dict_tree_v1
+reference_scaling_outputs:
+  applies_to: outputs
+  kind: affine_minmax_v1
+  max_tree:
+    flux:
+      - 7.435205936431885
+      - 7.439098834991455
+  min_tree:
+    flux:
+      - 2.4139199256896973
+      - 4.380068778991699
+  source_space: canonical_output_dict_tree_v1
+  storage:
+    filename: reference_scaling_outputs.safetensors
+    format: safetensors_v1
+    layout: split_minmax_tree_v1
+  target_space: physical_output_dict_tree_v1
+spec_version: 1
+
+Note: this bundle is the canonical emulator artifact. Physical-space composition is external.

bundle_integrity.json

@@ -0,0 +1,83 @@
+{
+  "algorithm": "sha256",
+  "bundle_id": "sha256:3ad1fd4fab9f5175095bc1fe27ae00af8362e24ff2f2996e398d85fcc753c7f7",
+  "integrity_format_version": 1,
+  "tree": [
+    {
+      "path": "README.txt",
+      "sha256": "6bbcdc4102985ed9bfa1e1d0a825c7d881ed6d934c163f5e789248c4fcf8b818"
+    },
+    {
+      "path": "config.json",
+      "sha256": "c4768f022a4b54457ca92904daefcc1d55928bb21fe5a744a323f81de751e0c7"
+    },
+    {
+      "path": "extras/evaluation_scaled_log10_wavelength.safetensors",
+      "sha256": "b0f2378a297d7cd1db2b50fbac3c79e611155523f4c45eb695bae6c89d2dda2f"
+    },
+    {
+      "path": "extras/source_log10_wavelength.safetensors",
+      "sha256": "ddc357db985e77e3ad4809c0c5aabd3f0c31d11690cd2e971428270b57cd93d2"
+    },
+    {
+      "path": "extras/source_log10_wavelength_by_channel/continuum.safetensors",
+      "sha256": "ad2ab34dd20887bd9afa699d6d71f5172f45a2dbfd2c1b12b0d61b47a659c921"
+    },
+    {
+      "path": "extras/source_log10_wavelength_by_channel/lines.safetensors",
+      "sha256": "ddc357db985e77e3ad4809c0c5aabd3f0c31d11690cd2e971428270b57cd93d2"
+    },
+    {
+      "path": "extras/source_scaled_log10_wavelength.safetensors",
+      "sha256": "6dfe5eef935c3ff0e42a5922196d256a7b3a7841858f7a062db9ff7d4261f501"
+    },
+    {
+      "path": "extras/source_scaled_log10_wavelength_by_channel/continuum.safetensors",
+      "sha256": "680a00c75b6cd891aa9957f2f65c6e288b7c5c4e4e4ccac27b97aa48f3dcd280"
+    },
+    {
+      "path": "extras/source_scaled_log10_wavelength_by_channel/lines.safetensors",
+      "sha256": "6dfe5eef935c3ff0e42a5922196d256a7b3a7841858f7a062db9ff7d4261f501"
+    },
+    {
+      "path": "extras/source_wavelength.safetensors",
+      "sha256": "36c6efd17e078ac09102e6c38b7c69bd073ec4996ed503378590fe5a4b7e5a1d"
+    },
+    {
+      "path": "extras/source_wavelength_by_channel/continuum.safetensors",
+      "sha256": "763e92f6f3f7bad0112ab67bbc05458fa442c6ef9743fa5a0d07f5c3dbbe0d2d"
+    },
+    {
+      "path": "extras/source_wavelength_by_channel/lines.safetensors",
+      "sha256": "36c6efd17e078ac09102e6c38b7c69bd073ec4996ed503378590fe5a4b7e5a1d"
+    },
+    {
+      "path": "fingerprint_evaluation/inputs.safetensors",
+      "sha256": "042d91eb9ed80efc7ab2ea06a18f07fa71760dfe6a23a716276b588b12ce1933"
+    },
+    {
+      "path": "fingerprint_evaluation/outputs.safetensors",
+      "sha256": "6c1f5288c076d988ee588f89051d5549e49139bc4ceda2e92f35078aabf65f7d"
+    },
+    {
+      "path": "input_domain.safetensors",
+      "sha256": "693c587d99028ac6bc7380f370faa2feae562be303343d4fb899cb7e277fcb01"
+    },
+    {
+      "path": "metadata.json",
+      "sha256": "a4354673984f7aa3cbe810053c5b7a86275ee5fcb8ea16d3e82bcf4ef4af8a3e"
+    },
+    {
+      "path": "reference_scaling_inputs.safetensors",
+      "sha256": "693c587d99028ac6bc7380f370faa2feae562be303343d4fb899cb7e277fcb01"
+    },
+    {
+      "path": "reference_scaling_outputs.safetensors",
+      "sha256": "ee95154c49e270d77e7cb13b188e06f2b850be8ef70216440769828caba56adc"
+    },
+    {
+      "path": "weights/weights.safetensors",
+      "sha256": "0df11633dd341cbabafd092d0d47628bd7e7435e51002bde49a6868f23687461"
+    }
+  ]
+}

config.json

@@ -0,0 +1,176 @@
+{
+  "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]",
+          "microturbulence velocity",
+          "source-grid parameter [a/Fe]",
+          "source-grid parameter [C/Fe]",
+          "source-grid parameter [N/Fe]",
+          "source-grid parameter [O/Fe]",
+          "source-grid parameter [r/Fe]",
+          "source-grid parameter [s/Fe]",
+          "cosine of viewing angle"
+        ],
+        "wavelengths": null
+      },
+      "channel_names_tree": {
+        "parameters": [
+          "teff",
+          "logg",
+          "[Fe/H]",
+          "vmicro",
+          "[a/Fe]",
+          "[C/Fe]",
+          "[N/Fe]",
+          "[O/Fe]",
+          "[r/Fe]",
+          "[s/Fe]",
+          "mu"
+        ],
+        "wavelengths": null
+      },
+      "channel_units_tree": {
+        "parameters": [
+          "dimensionless",
+          "dimensionless",
+          "dimensionless",
+          "dimensionless",
+          "dimensionless",
+          "dimensionless",
+          "dimensionless",
+          "dimensionless",
+          "dimensionless",
+          "dimensionless",
+          "dimensionless"
+        ],
+        "wavelengths": null
+      },
+      "leaf_meanings_tree": {
+        "parameters": "min-max scaled source-grid parameters; see reference_scaling_inputs for raw parameter bounds",
+        "wavelengths": "min-max scaled log10 wavelength; the user applies log10 before reference_scaling_inputs"
+      },
+      "leaf_units_tree": {
+        "parameters": null,
+        "wavelengths": "dimensionless"
+      },
+      "structure_tree": {
+        "parameters": null,
+        "wavelengths": null
+      }
+    },
+    "outputs": {
+      "channel_meanings_tree": {
+        "flux": [
+          "min-max scaled log10 line intensity from source array flux",
+          "min-max scaled log10 continuum intensity from source array continuum"
+        ]
+      },
+      "channel_names_tree": {
+        "flux": [
+          "log_flux_lines",
+          "log_flux_continuum"
+        ]
+      },
+      "channel_units_tree": {
+        "flux": [
+          "dimensionless",
+          "dimensionless"
+        ]
+      },
+      "leaf_meanings_tree": {
+        "flux": "two min-max scaled log10 intensity channels from the Maja archive arrays flux and continuum"
+      },
+      "leaf_units_tree": {
+        "flux": "dimensionless"
+      },
+      "structure_tree": {
+        "flux": null
+      }
+    },
+    "reference_scaling_inputs": null,
+    "reference_scaling_outputs": null
+  },
+  "model": {
+    "init_hints": {
+      "parameter_dim": 11
+    },
+    "name": "transformer_payne",
+    "params": {
+      "channels": 2,
+      "dim": 192,
+      "dim_ff_multiplier": 4,
+      "dim_head": 32,
+      "dtype": "float32",
+      "max_period": 1.0,
+      "min_period": 1e-06,
+      "no_layers": 6,
+      "no_tokens": 8
+    }
+  },
+  "optim": {
+    "b1": 0.9,
+    "b2": 0.999,
+    "decay_steps": 0,
+    "eps": 1e-08,
+    "grad_clip": 0.0,
+    "lr": 0.001,
+    "lr_scaling": null,
+    "name": "soap",
+    "precondition_1d": false,
+    "precondition_frequency": 10,
+    "scale_embedding_lr": 1.0,
+    "schedule": "cosine",
+    "warmup_steps": 10000,
+    "weight_decay": 0.0001
+  },
+  "schema_version": 1,
+  "seed": 0,
+  "solver": {
+    "name": "auto",
+    "params": {}
+  },
+  "task": {
+    "name": "regression",
+    "params": {
+      "loss": "mse",
+      "metric_axes": {
+        "channel": [
+          0
+        ],
+        "global": "all"
+      },
+      "metrics": [
+        "mse",
+        "mae"
+      ]
+    }
+  },
+  "training": {
+    "batch_size": 64,
+    "checkpoint_interval_steps": 10000,
+    "checkpoint_steps": null,
+    "evaluation_interval_steps": 5000,
+    "evaluation_steps": null,
+    "logging_interval_steps": 1000,
+    "logging_steps": null,
+    "max_saved_checkpoints": 5,
+    "num_steps": 100000,
+    "shuffle": true,
+    "shuffle_seed": 0,
+    "steps_per_epoch": null,
+    "val_fraction": 0.0,
+    "workdir": "./runs/from_bundle"
+  }
+}

extras/evaluation_scaled_log10_wavelength.safetensors

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

extras/source_log10_wavelength.safetensors

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

extras/source_log10_wavelength_by_channel/continuum.safetensors

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

extras/source_log10_wavelength_by_channel/lines.safetensors

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

extras/source_scaled_log10_wavelength.safetensors

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

extras/source_scaled_log10_wavelength_by_channel/continuum.safetensors

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

extras/source_scaled_log10_wavelength_by_channel/lines.safetensors

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

extras/source_wavelength.safetensors

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

extras/source_wavelength_by_channel/continuum.safetensors

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

extras/source_wavelength_by_channel/lines.safetensors

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

fingerprint_evaluation/inputs.safetensors

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

fingerprint_evaluation/outputs.safetensors

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

input_domain.safetensors

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

metadata.json

@@ -0,0 +1,457 @@
+{
+  "bundle_format_version": 1,
+  "config_schema_version": 1,
+  "extras": {
+    "evaluation_scaled_log10_wavelength": {
+      "__aet_sidecar__": {
+        "format": "safetensors_v1",
+        "layout": "single_array_v1",
+        "path": "extras/evaluation_scaled_log10_wavelength.safetensors"
+      }
+    },
+    "notes": "The source array named 'flux' is treated as line intensity. The source array 'continuum' is treated as continuum intensity. The bundle records min-max scaling metadata; log10 transforms for wavelengths and outputs are explicit user-side preprocessing.",
+    "parameter_source_units": {
+      "[C/Fe]": "",
+      "[Fe/H]": "dex",
+      "[N/Fe]": "",
+      "[O/Fe]": "",
+      "[a/Fe]": "",
+      "[r/Fe]": "",
+      "[s/Fe]": "",
+      "logg": "dex",
+      "mu": "dimensionless",
+      "teff": "K",
+      "vmicro": "km/s"
+    },
+    "preprocessing_recipe": {
+      "outputs": "The model predicts min-max scaled log10 flux and min-max scaled log10 continuum. Invert the min-max transform first; apply 10**y outside the bundle if physical intensities are needed.",
+      "parameters": "Scale raw parameter vector with (x - parameter_min) / (parameter_max - parameter_min).",
+      "wavelengths": "Apply log10 to physical wavelength first, then min-max scale with the log10_wavelength bounds."
+    },
+    "source_log10_wavelength": {
+      "__aet_sidecar__": {
+        "format": "safetensors_v1",
+        "layout": "single_array_v1",
+        "path": "extras/source_log10_wavelength.safetensors"
+      }
+    },
+    "source_log10_wavelength_by_channel": {
+      "continuum": {
+        "__aet_sidecar__": {
+          "format": "safetensors_v1",
+          "layout": "single_array_v1",
+          "path": "extras/source_log10_wavelength_by_channel/continuum.safetensors"
+        }
+      },
+      "lines": {
+        "__aet_sidecar__": {
+          "format": "safetensors_v1",
+          "layout": "single_array_v1",
+          "path": "extras/source_log10_wavelength_by_channel/lines.safetensors"
+        }
+      }
+    },
+    "source_scaled_log10_wavelength": {
+      "__aet_sidecar__": {
+        "format": "safetensors_v1",
+        "layout": "single_array_v1",
+        "path": "extras/source_scaled_log10_wavelength.safetensors"
+      }
+    },
+    "source_scaled_log10_wavelength_by_channel": {
+      "continuum": {
+        "__aet_sidecar__": {
+          "format": "safetensors_v1",
+          "layout": "single_array_v1",
+          "path": "extras/source_scaled_log10_wavelength_by_channel/continuum.safetensors"
+        }
+      },
+      "lines": {
+        "__aet_sidecar__": {
+          "format": "safetensors_v1",
+          "layout": "single_array_v1",
+          "path": "extras/source_scaled_log10_wavelength_by_channel/lines.safetensors"
+        }
+      }
+    },
+    "source_store": "/g/data/y89/mj8805/harps.zarr",
+    "source_stores": [
+      "/g/data/y89/mj8805/harps.zarr",
+      "/g/data/y89/mj8805/harps2.zarr"
+    ],
+    "source_wavelength": {
+      "__aet_sidecar__": {
+        "format": "safetensors_v1",
+        "layout": "single_array_v1",
+        "path": "extras/source_wavelength.safetensors"
+      }
+    },
+    "source_wavelength_by_channel": {
+      "continuum": {
+        "__aet_sidecar__": {
+          "format": "safetensors_v1",
+          "layout": "single_array_v1",
+          "path": "extras/source_wavelength_by_channel/continuum.safetensors"
+        }
+      },
+      "lines": {
+        "__aet_sidecar__": {
+          "format": "safetensors_v1",
+          "layout": "single_array_v1",
+          "path": "extras/source_wavelength_by_channel/lines.safetensors"
+        }
+      }
+    },
+    "training": {
+      "continuum_stride": 10,
+      "dataset_output_keys": [
+        "lines",
+        "continuum"
+      ],
+      "model_output_channel_names": [
+        "log_flux_lines",
+        "log_flux_continuum"
+      ],
+      "n_training_samples": 54000,
+      "n_validation_samples": 6000,
+      "n_wavelength_per_step": 2048,
+      "source_wavelength_samples_by_channel": {
+        "continuum": 31001,
+        "lines": 310001
+      },
+      "validation_policy": "first 10.0% of spectra by concatenated source index"
+    }
+  },
+  "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": "transformer_payne_v1",
+  "model_init": {
+    "hints": {
+      "parameter_dim": 11
+    },
+    "representation": "model-local init hints only"
+  },
+  "provenance": {
+    "created_at": "2026-05-01T13:33:41.693459+00:00",
+    "dependencies": {
+      "flax": "0.12.7",
+      "jax": "0.10.0",
+      "numpy": "2.4.4",
+      "optax": "0.2.8"
+    },
+    "git_commit": null,
+    "platform": "Linux-4.18.0-553.92.1.el8.nci.x86_64-x86_64-with-glibc2.28",
+    "python_version": "3.12.13",
+    "toolkit": "astro_emulators_toolkit",
+    "toolkit_version": "0.1.0"
+  },
+  "release": {
+    "name": "maja-harps-intensity-tpayne-2ch-192x6",
+    "status": "released",
+    "version": "0.1.0"
+  },
+  "resolved": {
+    "model": {
+      "name": "transformer_payne",
+      "params": {
+        "activation": "gelu",
+        "alpha_att": 1.0,
+        "alpha_emb": 1.0,
+        "bias_attention": false,
+        "bias_dense": false,
+        "bias_feed_forward": null,
+        "bias_output_head": null,
+        "bias_parameter_embedding": null,
+        "channels": 2,
+        "dim": 192,
+        "dim_ff_multiplier": 4,
+        "dim_head": 32,
+        "dtype": "float32",
+        "emb_init": "si",
+        "ff_init": "si",
+        "head_init": "si",
+        "init_att_o": "si",
+        "init_att_q": "si",
+        "max_period": 1.0,
+        "min_period": 1e-06,
+        "no_layers": 6,
+        "no_tokens": 8,
+        "output_activation": "linear",
+        "reference_depth": null,
+        "reference_width": null,
+        "sigma": 1.0
+      }
+    },
+    "solver": {
+      "name": "gradient",
+      "params": {}
+    },
+    "task": {
+      "name": "regression",
+      "params": {
+        "loss": "mse",
+        "loss_weights": null,
+        "metric_axes": {
+          "channel": [
+            0
+          ],
+          "global": "all"
+        },
+        "metrics": [
+          "mse",
+          "mae"
+        ]
+      }
+    }
+  },
+  "runtime_contract": {
+    "affine_leaf_specs": {
+      "inputs/parameters": {
+        "last_axis": 11,
+        "mode": "scalar_or_last_axis"
+      },
+      "inputs/wavelengths": {
+        "mode": "scalar_only"
+      },
+      "outputs/flux": {
+        "last_axis": 2,
+        "mode": "scalar_or_last_axis"
+      }
+    },
+    "role_paths": {
+      "output_leaf": "outputs/flux",
+      "parameter_leaf": "inputs/parameters",
+      "wavelength_leaf": "inputs/wavelengths"
+    },
+    "surface": "canonical_dict_trees_v1",
+    "transformer_payne_channels": [
+      {
+        "dataset_key": "lines",
+        "name": "log_flux_lines"
+      },
+      {
+        "dataset_key": "continuum",
+        "name": "log_flux_continuum"
+      }
+    ]
+  },
+  "spec": {
+    "input_domain": {
+      "kind": "box_v1",
+      "max_tree": {
+        "parameters": [
+          7500.0,
+          5.0,
+          0.5,
+          5.0,
+          0.5,
+          0.30000001192092896,
+          0.4000000059604645,
+          0.4000000059604645,
+          0.5,
+          0.5,
+          1.0
+        ],
+        "wavelengths": 3.838849090737255
+      },
+      "min_tree": {
+        "parameters": [
+          3800.0,
+          0.5,
+          -3.0,
+          1.0,
+          -0.20000000298023224,
+          -0.30000001192092896,
+          -0.20000000298023224,
+          -0.20000000298023224,
+          -0.20000000298023224,
+          -0.20000000298023224,
+          0.1246189996600151
+        ],
+        "wavelengths": 3.57978359661681
+      },
+      "storage": {
+        "filename": "input_domain.safetensors",
+        "format": "safetensors_v1",
+        "layout": "split_minmax_tree_v1"
+      },
+      "value_space": "physical_input_dict_tree_v1"
+    },
+    "inputs": {
+      "channel_meanings_tree": {
+        "parameters": [
+          "effective temperature",
+          "surface gravity",
+          "metallicity [Fe/H]",
+          "microturbulence velocity",
+          "source-grid parameter [a/Fe]",
+          "source-grid parameter [C/Fe]",
+          "source-grid parameter [N/Fe]",
+          "source-grid parameter [O/Fe]",
+          "source-grid parameter [r/Fe]",
+          "source-grid parameter [s/Fe]",
+          "cosine of viewing angle"
+        ],
+        "wavelengths": null
+      },
+      "channel_names_tree": {
+        "parameters": [
+          "teff",
+          "logg",
+          "[Fe/H]",
+          "vmicro",
+          "[a/Fe]",
+          "[C/Fe]",
+          "[N/Fe]",
+          "[O/Fe]",
+          "[r/Fe]",
+          "[s/Fe]",
+          "mu"
+        ],
+        "wavelengths": null
+      },
+      "channel_units_tree": {
+        "parameters": [
+          "dimensionless",
+          "dimensionless",
+          "dimensionless",
+          "dimensionless",
+          "dimensionless",
+          "dimensionless",
+          "dimensionless",
+          "dimensionless",
+          "dimensionless",
+          "dimensionless",
+          "dimensionless"
+        ],
+        "wavelengths": null
+      },
+      "leaf_meanings_tree": {
+        "parameters": "min-max scaled source-grid parameters; see reference_scaling_inputs for raw parameter bounds",
+        "wavelengths": "min-max scaled log10 wavelength; the user applies log10 before reference_scaling_inputs"
+      },
+      "leaf_units_tree": {
+        "parameters": null,
+        "wavelengths": "dimensionless"
+      },
+      "structure_tree": {
+        "parameters": null,
+        "wavelengths": null
+      }
+    },
+    "outputs": {
+      "channel_meanings_tree": {
+        "flux": [
+          "min-max scaled log10 line intensity from source array flux",
+          "min-max scaled log10 continuum intensity from source array continuum"
+        ]
+      },
+      "channel_names_tree": {
+        "flux": [
+          "log_flux_lines",
+          "log_flux_continuum"
+        ]
+      },
+      "channel_units_tree": {
+        "flux": [
+          "dimensionless",
+          "dimensionless"
+        ]
+      },
+      "leaf_meanings_tree": {
+        "flux": "two min-max scaled log10 intensity channels from the Maja archive arrays flux and continuum"
+      },
+      "leaf_units_tree": {
+        "flux": "dimensionless"
+      },
+      "structure_tree": {
+        "flux": null
+      }
+    },
+    "reference_scaling_inputs": {
+      "applies_to": "inputs",
+      "kind": "affine_minmax_v1",
+      "max_tree": {
+        "parameters": [
+          7500.0,
+          5.0,
+          0.5,
+          5.0,
+          0.5,
+          0.30000001192092896,
+          0.4000000059604645,
+          0.4000000059604645,
+          0.5,
+          0.5,
+          1.0
+        ],
+        "wavelengths": 3.838849090737255
+      },
+      "min_tree": {
+        "parameters": [
+          3800.0,
+          0.5,
+          -3.0,
+          1.0,
+          -0.20000000298023224,
+          -0.30000001192092896,
+          -0.20000000298023224,
+          -0.20000000298023224,
+          -0.20000000298023224,
+          -0.20000000298023224,
+          0.1246189996600151
+        ],
+        "wavelengths": 3.57978359661681
+      },
+      "source_space": "physical_input_dict_tree_v1",
+      "storage": {
+        "filename": "reference_scaling_inputs.safetensors",
+        "format": "safetensors_v1",
+        "layout": "split_minmax_tree_v1"
+      },
+      "target_space": "canonical_input_dict_tree_v1"
+    },
+    "reference_scaling_outputs": {
+      "applies_to": "outputs",
+      "kind": "affine_minmax_v1",
+      "max_tree": {
+        "flux": [
+          7.435205936431885,
+          7.439098834991455
+        ]
+      },
+      "min_tree": {
+        "flux": [
+          2.4139199256896973,
+          4.380068778991699
+        ]
+      },
+      "source_space": "canonical_output_dict_tree_v1",
+      "storage": {
+        "filename": "reference_scaling_outputs.safetensors",
+        "format": "safetensors_v1",
+        "layout": "split_minmax_tree_v1"
+      },
+      "target_space": "physical_output_dict_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:693c587d99028ac6bc7380f370faa2feae562be303343d4fb899cb7e277fcb01
+size 480

reference_scaling_outputs.safetensors

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

weights/weights.safetensors

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