Upload 6 files

README.txt

@@ -0,0 +1,60 @@
+Astro Emulators Toolkit Bundle
+
+Summary:
+  model: mlp
+  task: regression
+  fit_method: gradient
+  x_names: ('teff', 'logg', 'feh')
+  y_names: ('flux',)
+
+Domain:
+  input_domain: {'kind': 'box_v1', 'space': 'runtime_tree_v1', 'storage': {'format': 'safetensors_v1', 'filename': 'input_domain.safetensors', 'layout': 'tree_minmax_v1'}}
+
+Provenance:
+  toolkit_version: 0.1.0
+  saved_at_utc: 2026-03-13T02:55:33.972802+00:00
+  python_version: 3.12.11
+  git_sha: 72e167d2c8d68abc1e2c6c4b2503e06f3a375537
+
+spec:
+spec_version: 2
+x:
+  names:
+    - teff
+    - logg
+    - feh
+  meaning:
+    - effective temperature
+    - surface gravity
+    - metallicity [Fe/H]
+  units:
+    - K
+    - dex
+    - dex
+y:
+  names:
+    - flux
+  meaning:
+    - continuum-normalized flux vector on the shared wavelength grid
+  units:
+    - dimensionless
+  wavelength_angstrom:
+    array(shape=(500,), dtype=float64)
+input_domain:
+  kind: box_v1
+  space: runtime_tree_v1
+  storage:
+    format: safetensors_v1
+    filename: input_domain.safetensors
+    layout: tree_minmax_v1
+reference_scaling:
+  kind: affine_v1
+  parametrization: minmax
+  space: runtime_tree_v1
+  storage:
+    format: safetensors_v1
+    filename: reference_scaling.safetensors
+    layout: tree_minmax_v1
+notes: Inputs are min-max scaled before training using the fixed bounds stored in this bundle. Output reference scaling is identity.
+
+Note: this bundle is the canonical emulator artifact. Physical-space composition is external.

config.json

@@ -0,0 +1,70 @@
+{
+  "bundle": {
+    "bundle_subdir": "bundle"
+  },
+  "hub": {
+    "repo_id": null,
+    "revision": null
+  },
+  "io": {
+    "x_dim": 3,
+    "x_names": [
+      "teff",
+      "logg",
+      "feh"
+    ],
+    "y_dim": 500,
+    "y_names": [
+      "flux"
+    ]
+  },
+  "model": {
+    "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": 300,
+    "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": 3000,
+    "shuffle": true,
+    "shuffle_seed": 0,
+    "steps_per_epoch": null,
+    "val_fraction": 0.1,
+    "workdir": "/Users/tr/repos/astro-emulators-toolkit/examples/runs/development_payne_flux_mlp_soap"
+  }
+}

input_domain.safetensors

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

metadata.json

@@ -0,0 +1,635 @@
+{
+  "fit_method": "gradient",
+  "io_init": {
+    "in_dim": 3,
+    "out_dim": 500,
+    "representation": "model init dimensions (not runtime output shape semantics)",
+    "x_names": [
+      "teff",
+      "logg",
+      "feh"
+    ],
+    "y_names": [
+      "flux"
+    ]
+  },
+  "io_pattern": "array->array",
+  "model_family_id": "mlp_v1",
+  "provenance": {
+    "dependencies": {
+      "flax": "0.10.4",
+      "jax": "0.4.38",
+      "numpy": "2.3.5",
+      "optax": "0.2.5"
+    },
+    "git_sha": "72e167d2c8d68abc1e2c6c4b2503e06f3a375537",
+    "platform": "macOS-26.3.1-x86_64-i386-64bit",
+    "python_version": "3.12.11",
+    "saved_at_utc": "2026-03-13T02:55:33.972802+00:00",
+    "toolkit": "astro_emulators_toolkit",
+    "toolkit_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_shapes": {
+      "inputs/feh": [],
+      "inputs/logg": [],
+      "inputs/teff": [],
+      "outputs/flux": [
+        500
+      ]
+    },
+    "input_name_tree": [
+      "teff",
+      "logg",
+      "feh"
+    ],
+    "io_pattern": "array->array",
+    "output_name_tree": "flux"
+  },
+  "schema_version": 1,
+  "spec": {
+    "input_domain": {
+      "kind": "box_v1",
+      "space": "runtime_tree_v1",
+      "storage": {
+        "filename": "input_domain.safetensors",
+        "format": "safetensors_v1",
+        "layout": "tree_minmax_v1"
+      }
+    },
+    "notes": "Inputs are min-max scaled before training using the fixed bounds stored in this bundle. Output reference scaling is identity.",
+    "reference_scaling": {
+      "kind": "affine_v1",
+      "parametrization": "minmax",
+      "space": "runtime_tree_v1",
+      "storage": {
+        "filename": "reference_scaling.safetensors",
+        "format": "safetensors_v1",
+        "layout": "tree_minmax_v1"
+      }
+    },
+    "spec_version": 2,
+    "x": {
+      "meaning": [
+        "effective temperature",
+        "surface gravity",
+        "metallicity [Fe/H]"
+      ],
+      "names": [
+        "teff",
+        "logg",
+        "feh"
+      ],
+      "units": [
+        "K",
+        "dex",
+        "dex"
+      ]
+    },
+    "y": {
+      "meaning": [
+        "continuum-normalized flux vector on the shared wavelength grid"
+      ],
+      "names": [
+        "flux"
+      ],
+      "units": [
+        "dimensionless"
+      ],
+      "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
+      ]
+    }
+  },
+  "tool": "astro_emulators_toolkit",
+  "weights_layout": "params_plus_model_state_v1"
+}

reference_scaling.safetensors

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

weights/weights.safetensors

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