Upload 8 files

README.txt

@@ -2,52 +2,102 @@ 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
-  x_names: ('x0', 'x1', 'x2')
-  y_names: ('y0', 'y1')
+  role_paths: {'input_leaf': 'inputs/parameters', 'output_leaf': 'outputs/flux'}
 
 Domain:
-  input_domain: not provided
+  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
-  saved_at_utc: 2026-04-10T02:09:02.834895+00:00
-  python_version: 3.12.11
-  git_sha: 853a30387805735584ec72dd7274e0fdb48591ba
+  created_at: 2026-04-12T14:23:17.605616+00:00
+  python_version: 3.12.13
+  git_commit: e922f60957ded21c91aca36a1d11b2bd269cc082
 
 spec:
-spec_version: 2
-x:
-  names:
-    - x0
-    - x1
-    - x2
-  meaning:
-    - 
-    - 
-    - 
-  units:
-    - 
-    - 
-    - 
-y:
-  names:
-    - y0
-    - y1
-  meaning:
-    - 
-    - 
-  units:
-    - 
-    - 
+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: runtime_tree_v1
+  space: canonical_dict_trees_v1
   storage:
-    format: safetensors_v1
     filename: reference_scaling.safetensors
-    layout: tree_minmax_v1
+    format: safetensors_v1
+    layout: split_minmax_tree_v1
+spec_version: 1
 
 Note: this bundle is the canonical emulator artifact. Physical-space composition is external.

config.json

@@ -7,19 +7,65 @@
     "revision": null
   },
   "io": {
-    "x_dim": 3,
-    "x_names": null,
-    "y_dim": 2,
-    "y_names": null
+    "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": [
-        32,
-        32
+        128,
+        128
       ]
     }
   },
@@ -28,13 +74,13 @@
     "b2": 0.999,
     "decay_steps": 0,
     "eps": 1e-08,
-    "lr": 0.0003,
-    "name": "adamw",
+    "lr": 0.001,
+    "name": "soap",
     "precondition_1d": false,
     "precondition_frequency": 10,
-    "schedule": "constant",
-    "warmup_steps": 0,
-    "weight_decay": 0.0001
+    "schedule": "cosine",
+    "warmup_steps": 1000,
+    "weight_decay": 1e-05
   },
   "schema_version": 1,
   "seed": 0,
@@ -45,24 +91,20 @@
   "task": {
     "name": "regression",
     "params": {
-      "loss": "mse",
-      "metrics": [
-        "mse",
-        "mae"
-      ]
+      "loss": "mse"
     }
   },
   "training": {
-    "batch_size": 256,
-    "checkpoint_interval_steps": 200,
-    "evaluation_interval_steps": 200,
-    "logging_interval_steps": 25,
-    "max_saved_checkpoints": 3,
-    "num_steps": 2000,
+    "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": "/Users/tr/repos/astro-emulators-toolkit/examples/development/runs/rff_mlp"
+    "workdir": "./runs/from_bundle"
   }
 }

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

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

metadata.json

@@ -1,35 +1,558 @@
 {
-  "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"
+  "bundle_format_version": 1,
+  "bundle_id": "sha256:0c3351d9344510b3bd084e8ebb43a0039ea0c813c9bf2613fc34363d199a0a3b",
+  "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": [
+      4999.999999999999,
+      5000.020020106839,
+      5000.040040293839,
+      5000.060060561006,
+      5000.080080908329,
+      5000.1001013358145,
+      5000.120121843462,
+      5000.140142431276,
+      5000.160163099249,
+      5000.180183847385,
+      5000.200204675684,
+      5000.220225584153,
+      5000.24024657278,
+      5000.260267641572,
+      5000.280288790529,
+      5000.30031001965,
+      5000.320331328942,
+      5000.3403527183955,
+      5000.360374188014,
+      5000.3803957377995,
+      5000.400417367757,
+      5000.420439077876,
+      5000.4404608681625,
+      5000.4604827386165,
+      5000.480504689244,
+      5000.500526720035,
+      5000.520548830994,
+      5000.540571022122,
+      5000.560593293424,
+      5000.580615644892,
+      5000.600638076528,
+      5000.620660588335,
+      5000.640683180313,
+      5000.660705852467,
+      5000.680728604786,
+      5000.700751437278,
+      5000.720774349941,
+      5000.740797342782,
+      5000.76082041579,
+      5000.780843568969,
+      5000.800866802323,
+      5000.820890115856,
+      5000.840913509557,
+      5000.860936983432,
+      5000.880960537482,
+      5000.900984171706,
+      5000.92100788611,
+      5000.941031680685,
+      5000.961055555435,
+      5000.981079510361,
+      5001.001103545469,
+      5001.021127660749,
+      5001.041151856205,
+      5001.061176131839,
+      5001.081200487655,
+      5001.101224923645,
+      5001.121249439812,
+      5001.141274036158,
+      5001.161298712684,
+      5001.181323469393,
+      5001.2013483062765,
+      5001.22137322334,
+      5001.241398220584,
+      5001.261423298014,
+      5001.281448455619,
+      5001.301473693406,
+      5001.321499011374,
+      5001.341524409529,
+      5001.36154988786,
+      5001.381575446374,
+      5001.401601085072,
+      5001.421626803956,
+      5001.44165260302,
+      5001.461678482267,
+      5001.481704441699,
+      5001.501730481315,
+      5001.52175660112,
+      5001.541782801105,
+      5001.561809081276,
+      5001.581835441632,
+      5001.60186188218,
+      5001.621888402909,
+      5001.641915003824,
+      5001.661941684926,
+      5001.681968446221,
+      5001.701995287697,
+      5001.722022209363,
+      5001.742049211216,
+      5001.762076293257,
+      5001.782103455493,
+      5001.802130697913,
+      5001.822158020522,
+      5001.842185423321,
+      5001.862212906315,
+      5001.882240469495,
+      5001.902268112865,
+      5001.922295836427,
+      5001.942323640185,
+      5001.96235152413,
+      5001.982379488267,
+      5002.002407532596,
+      5002.022435657118,
+      5002.042463861839,
+      5002.062492146747,
+      5002.082520511849,
+      5002.102548957145,
+      5002.122577482641,
+      5002.142606088325,
+      5002.162634774205,
+      5002.18266354028,
+      5002.202692386557,
+      5002.222721313024,
+      5002.242750319687,
+      5002.2627794065465,
+      5002.282808573607,
+      5002.302837820862,
+      5002.322867148313,
+      5002.3428965559615,
+      5002.362926043809,
+      5002.38295561186,
+      5002.402985260104,
+      5002.423014988549,
+      5002.443044797192,
+      5002.463074686039,
+      5002.483104655083,
+      5002.503134704327,
+      5002.52316483377,
+      5002.543195043421,
+      5002.563225333268,
+      5002.583255703316,
+      5002.603286153567,
+      5002.623316684019,
+      5002.64334729468,
+      5002.663377985538,
+      5002.6834087566,
+      5002.703439607866,
+      5002.72347053934,
+      5002.743501551014,
+      5002.763532642893,
+      5002.783563814975,
+      5002.803595067269,
+      5002.823626399763,
+      5002.843657812463,
+      5002.863689305369,
+      5002.883720878482,
+      5002.903752531806,
+      5002.923784265333,
+      5002.943816079067,
+      5002.963847973008,
+      5002.983879947164,
+      5003.003912001523,
+      5003.02394413609,
+      5003.043976350866,
+      5003.064008645857,
+      5003.084041021052,
+      5003.104073476458,
+      5003.124106012074,
+      5003.1441386279,
+      5003.164171323943,
+      5003.184204100193,
+      5003.204236956653,
+      5003.224269893326,
+      5003.244302910216,
+      5003.264336007313,
+      5003.284369184624,
+      5003.304402442148,
+      5003.32443577989,
+      5003.344469197842,
+      5003.364502696008,
+      5003.384536274388,
+      5003.404569932987,
+      5003.424603671798,
+      5003.444637490824,
+      5003.464671390065,
+      5003.484705369523,
+      5003.504739429202,
+      5003.524773569093,
+      5003.5448077892015,
+      5003.564842089527,
+      5003.584876470075,
+      5003.604910930837,
+      5003.624945471816,
+      5003.644980093015,
+      5003.665014794437,
+      5003.685049576075,
+      5003.705084437931,
+      5003.725119380008,
+      5003.745154402305,
+      5003.7651895048275,
+      5003.785224687566,
+      5003.805259950526,
+      5003.825295293708,
+      5003.845330717117,
+      5003.8653662207425,
+      5003.88540180459,
+      5003.905437468662,
+      5003.925473212962,
+      5003.9455090374795,
+      5003.965544942222,
+      5003.985580927188,
+      5004.005616992378,
+      5004.025653137799,
+      5004.04568936344,
+      5004.065725669306,
+      5004.085762055398,
+      5004.10579852172,
+      5004.125835068266,
+      5004.1458716950365,
+      5004.165908402035,
+      5004.185945189265,
+      5004.205982056719,
+      5004.2260190044,
+      5004.24605603231,
+      5004.266093140453,
+      5004.286130328821,
+      5004.306167597418,
+      5004.326204946245,
+      5004.346242375301,
+      5004.366279884593,
+      5004.38631747411,
+      5004.4063551438585,
+      5004.426392893838,
+      5004.446430724054,
+      5004.466468634498,
+      5004.486506625173,
+      5004.506544696081,
+      5004.526582847227,
+      5004.546621078601,
+      5004.566659390208,
+      5004.586697782049,
+      5004.606736254124,
+      5004.626774806439,
+      5004.646813438984,
+      5004.666852151764,
+      5004.6868909447785,
+      5004.706929818035,
+      5004.726968771522,
+      5004.747007805246,
+      5004.767046919205,
+      5004.787086113408,
+      5004.8071253878425,
+      5004.8271647425145,
+      5004.847204177425,
+      5004.867243692573,
+      5004.887283287965,
+      5004.9073229635915,
+      5004.927362719457,
+      5004.947402555562,
+      5004.967442471912,
+      5004.987482468498,
+      5005.007522545323,
+      5005.02756270239,
+      5005.047602939703,
+      5005.067643257253,
+      5005.087683655043,
+      5005.107724133077,
+      5005.127764691359,
+      5005.147805329877,
+      5005.167846048639,
+      5005.187886847644,
+      5005.207927726894,
+      5005.227968686392,
+      5005.24800972613,
+      5005.268050846113,
+      5005.28809204634,
+      5005.308133326818,
+      5005.328174687536,
+      5005.348216128502,
+      5005.368257649712,
+      5005.388299251175,
+      5005.4083409328805,
+      5005.428382694832,
+      5005.448424537032,
+      5005.46846645948,
+      5005.488508462181,
+      5005.508550545125,
+      5005.528592708319,
+      5005.548634951761,
+      5005.568677275459,
+      5005.588719679401,
+      5005.608762163594,
+      5005.628804728037,
+      5005.648847372736,
+      5005.668890097681,
+      5005.688932902877,
+      5005.708975788326,
+      5005.729018754027,
+      5005.749061799984,
+      5005.76910492619,
+      5005.789148132649,
+      5005.809191419361,
+      5005.829234786333,
+      5005.8492782335525,
+      5005.869321761027,
+      5005.8893653687555,
+      5005.909409056745,
+      5005.929452824985,
+      5005.94949667348,
+      5005.969540602231,
+      5005.989584611244,
+      5006.009628700508,
+      5006.02967287003,
+      5006.049717119808,
+      5006.069761449844,
+      5006.089805860142,
+      5006.109850350695,
+      5006.129894921506,
+      5006.149939572576,
+      5006.16998430391,
+      5006.190029115498,
+      5006.2100740073465,
+      5006.230118979454,
+      5006.250164031829,
+      5006.270209164459,
+      5006.29025437735,
+      5006.310299670502,
+      5006.330345043916,
+      5006.350390497597,
+      5006.370436031536,
+      5006.390481645737,
+      5006.410527340202,
+      5006.430573114935,
+      5006.450618969927,
+      5006.470664905182,
+      5006.490710920702,
+      5006.510757016492,
+      5006.5308031925415,
+      5006.550849448857,
+      5006.570895785437,
+      5006.590942202283,
+      5006.610988699402,
+      5006.631035276781,
+      5006.651081934428,
+      5006.671128672341,
+      5006.691175490528,
+      5006.711222388976,
+      5006.731269367694,
+      5006.751316426679,
+      5006.771363565939,
+      5006.791410785463,
+      5006.811458085256,
+      5006.831505465319,
+      5006.851552925658,
+      5006.871600466262,
+      5006.891648087136,
+      5006.911695788282,
+      5006.9317435696985,
+      5006.951791431393,
+      5006.971839373354,
+      5006.991887395587,
+      5007.0119354980925,
+      5007.031983680877,
+      5007.052031943929,
+      5007.072080287256,
+      5007.092128710856,
+      5007.112177214734,
+      5007.132225798884,
+      5007.152274463308,
+      5007.172323208007,
+      5007.192372032981,
+      5007.212420938237,
+      5007.232469923764,
+      5007.252518989568,
+      5007.272568135649,
+      5007.292617362012,
+      5007.312666668647,
+      5007.332716055559,
+      5007.352765522751,
+      5007.372815070225,
+      5007.392864697974,
+      5007.412914406002,
+      5007.432964194309,
+      5007.453014062896,
+      5007.473064011768,
+      5007.493114040915,
+      5007.513164150343,
+      5007.533214340053,
+      5007.553264610048,
+      5007.573314960321,
+      5007.593365390875,
+      5007.613415901712,
+      5007.633466492836,
+      5007.653517164239,
+      5007.673567915925,
+      5007.693618747894,
+      5007.713669660147,
+      5007.73372065269,
+      5007.753771725512,
+      5007.773822878619,
+      5007.7938741120115,
+      5007.813925425694,
+      5007.833976819658,
+      5007.854028293908,
+      5007.874079848445,
+      5007.894131483273,
+      5007.914183198383,
+      5007.9342349937815,
+      5007.954286869468,
+      5007.9743388254465,
+      5007.99439086171,
+      5008.01444297826,
+      5008.034495175101,
+      5008.0545474522305,
+      5008.074599809655,
+      5008.094652247365,
+      5008.114704765365,
+      5008.134757363656,
+      5008.154810042242,
+      5008.1748628011155,
+      5008.194915640281,
+      5008.214968559737,
+      5008.2350215594915,
+      5008.255074639533,
+      5008.275127799868,
+      5008.295181040496,
+      5008.315234361417,
+      5008.335287762638,
+      5008.355341244148,
+      5008.375394805952,
+      5008.395448448051,
+      5008.415502170451,
+      5008.43555597314,
+      5008.455609856126,
+      5008.4756638194085,
+      5008.495717862992,
+      5008.515771986868,
+      5008.53582619104,
+      5008.555880475509,
+      5008.575934840277,
+      5008.595989285348,
+      5008.616043810713,
+      5008.636098416376,
+      5008.656153102337,
+      5008.676207868604,
+      5008.696262715165,
+      5008.716317642026,
+      5008.736372649188,
+      5008.756427736655,
+      5008.776482904418,
+      5008.796538152483,
+      5008.8165934808485,
+      5008.836648889522,
+      5008.856704378492,
+      5008.876759947765,
+      5008.896815597342,
+      5008.91687132722,
+      5008.936927137409,
+      5008.956983027895,
+      5008.977038998686,
+      5008.997095049782,
+      5009.017151181187,
+      5009.037207392892,
+      5009.057263684903,
+      5009.07732005722,
+      5009.097376509848,
+      5009.1174330427775,
+      5009.137489656014,
+      5009.1575463495565,
+      5009.177603123408,
+      5009.197659977571,
+      5009.217716912038,
+      5009.237773926812,
+      5009.257831021896,
+      5009.277888197294,
+      5009.297945452996,
+      5009.318002789008,
+      5009.338060205329,
+      5009.358117701966,
+      5009.378175278909,
+      5009.3982329361625,
+      5009.418290673728,
+      5009.438348491604,
+      5009.458406389797,
+      5009.478464368299,
+      5009.498522427111,
+      5009.518580566238,
+      5009.538638785682,
+      5009.558697085436,
+      5009.578755465503,
+      5009.598813925883,
+      5009.6188724665835,
+      5009.638931087594,
+      5009.65898978892,
+      5009.679048570561,
+      5009.699107432522,
+      5009.719166374795,
+      5009.739225397385,
+      5009.759284500291,
+      5009.779343683514,
+      5009.799402947059,
+      5009.819462290918,
+      5009.839521715094,
+      5009.859581219588,
+      5009.879640804407,
+      5009.899700469539,
+      5009.919760214991,
+      5009.939820040762,
+      5009.959879946859,
+      5009.97993993327,
+      5010.000000000002
     ]
   },
-  "io_pattern": "array->array",
+  "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.10.4",
-      "jax": "0.4.38",
-      "numpy": "2.3.5",
-      "optax": "0.2.5"
+      "flax": "0.12.6",
+      "jax": "0.9.2",
+      "numpy": "2.4.4",
+      "optax": "0.2.8"
     },
-    "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",
+    "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",
@@ -37,8 +560,8 @@
         "activation": "gelu",
         "dtype": "float32",
         "hidden_sizes": [
-          32,
-          32
+          128,
+          128
         ],
         "use_bias": true
       }
@@ -61,69 +584,129 @@
     }
   },
   "runtime_contract": {
-    "affine_leaf_shapes": {
-      "inputs/x0": [],
-      "inputs/x1": [],
-      "inputs/x2": [],
-      "outputs/y0": [],
-      "outputs/y1": []
+    "affine_leaf_specs": {
+      "inputs/parameters": {
+        "last_axis": 3,
+        "mode": "scalar_or_last_axis"
+      },
+      "outputs/flux": {
+        "last_axis": 500,
+        "mode": "scalar_or_last_axis"
+      }
     },
-    "input_name_tree": [
-      "x0",
-      "x1",
-      "x2"
-    ],
-    "io_pattern": "array->array",
-    "output_name_tree": [
-      "y0",
-      "y1"
-    ]
+    "role_paths": {
+      "input_leaf": "inputs/parameters",
+      "output_leaf": "outputs/flux"
+    },
+    "surface": "canonical_dict_trees_v1"
   },
-  "schema_version": 1,
   "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": "runtime_tree_v1",
+      "space": "canonical_dict_trees_v1",
       "storage": {
         "filename": "reference_scaling.safetensors",
         "format": "safetensors_v1",
-        "layout": "tree_minmax_v1"
+        "layout": "split_minmax_tree_v1"
       }
     },
-    "spec_version": 2,
-    "x": {
-      "meaning": [
-        "",
-        "",
-        ""
-      ],
-      "names": [
-        "x0",
-        "x1",
-        "x2"
-      ],
-      "units": [
-        "",
-        "",
-        ""
-      ]
-    },
-    "y": {
-      "meaning": [
-        "",
-        ""
-      ],
-      "names": [
-        "y0",
-        "y1"
-      ],
-      "units": [
-        "",
-        ""
-      ]
-    }
+    "spec_version": 1
   },
-  "tool": "astro_emulators_toolkit",
   "weights_layout": "params_plus_model_state_v1"
 }

reference_scaling.safetensors

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

weights/weights.safetensors

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