Upload 5 files

README.txt

@@ -4,49 +4,43 @@ Summary:
   model: mlp
   task: regression
   fit_method: gradient
-  x_names: ('teff', 'logg', 'feh')
-  y_names: ('flux',)
+  x_names: ('x0', 'x1', 'x2')
+  y_names: ('y0', 'y1')
 
 Domain:
-  input_domain: {'kind': 'box_v1', 'space': 'runtime_tree_v1', 'storage': {'format': 'safetensors_v1', 'filename': 'input_domain.safetensors', 'layout': 'tree_minmax_v1'}}
+  input_domain: not provided
 
 Provenance:
   toolkit_version: 0.1.0
-  saved_at_utc: 2026-03-13T02:55:33.972802+00:00
+  saved_at_utc: 2026-04-10T02:09:02.834895+00:00
   python_version: 3.12.11
-  git_sha: 72e167d2c8d68abc1e2c6c4b2503e06f3a375537
+  git_sha: 853a30387805735584ec72dd7274e0fdb48591ba
 
 spec:
 spec_version: 2
 x:
   names:
-    - teff
-    - logg
-    - feh
+    - x0
+    - x1
+    - x2
   meaning:
-    - effective temperature
-    - surface gravity
-    - metallicity [Fe/H]
+    - 
+    - 
+    - 
   units:
-    - K
-    - dex
-    - dex
+    - 
+    - 
+    - 
 y:
   names:
-    - flux
+    - y0
+    - y1
   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
@@ -55,6 +49,5 @@ reference_scaling:
     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

@@ -8,15 +8,9 @@
   },
   "io": {
     "x_dim": 3,
-    "x_names": [
-      "teff",
-      "logg",
-      "feh"
-    ],
-    "y_dim": 500,
-    "y_names": [
-      "flux"
-    ]
+    "x_names": null,
+    "y_dim": 2,
+    "y_names": null
   },
   "model": {
     "name": "mlp",
@@ -24,8 +18,8 @@
       "activation": "gelu",
       "dtype": "float32",
       "hidden_sizes": [
-        128,
-        128
+        32,
+        32
       ]
     }
   },
@@ -34,13 +28,13 @@
     "b2": 0.999,
     "decay_steps": 0,
     "eps": 1e-08,
-    "lr": 0.001,
-    "name": "soap",
+    "lr": 0.0003,
+    "name": "adamw",
     "precondition_1d": false,
     "precondition_frequency": 10,
-    "schedule": "cosine",
-    "warmup_steps": 300,
-    "weight_decay": 1e-05
+    "schedule": "constant",
+    "warmup_steps": 0,
+    "weight_decay": 0.0001
   },
   "schema_version": 1,
   "seed": 0,
@@ -51,20 +45,24 @@
   "task": {
     "name": "regression",
     "params": {
-      "loss": "mse"
+      "loss": "mse",
+      "metrics": [
+        "mse",
+        "mae"
+      ]
     }
   },
   "training": {
-    "batch_size": 128,
-    "checkpoint_interval_steps": 500,
-    "evaluation_interval_steps": 500,
-    "logging_interval_steps": 50,
-    "max_saved_checkpoints": 5,
-    "num_steps": 3000,
+    "batch_size": 256,
+    "checkpoint_interval_steps": 200,
+    "evaluation_interval_steps": 200,
+    "logging_interval_steps": 25,
+    "max_saved_checkpoints": 3,
+    "num_steps": 2000,
     "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"
+    "workdir": "/Users/tr/repos/astro-emulators-toolkit/examples/development/runs/rff_mlp"
   }
 }

metadata.json

@@ -2,15 +2,16 @@
   "fit_method": "gradient",
   "io_init": {
     "in_dim": 3,
-    "out_dim": 500,
+    "out_dim": 2,
     "representation": "model init dimensions (not runtime output shape semantics)",
     "x_names": [
-      "teff",
-      "logg",
-      "feh"
+      "x0",
+      "x1",
+      "x2"
     ],
     "y_names": [
-      "flux"
+      "y0",
+      "y1"
     ]
   },
   "io_pattern": "array->array",
@@ -22,10 +23,10 @@
       "numpy": "2.3.5",
       "optax": "0.2.5"
     },
-    "git_sha": "72e167d2c8d68abc1e2c6c4b2503e06f3a375537",
+    "git_sha": "853a30387805735584ec72dd7274e0fdb48591ba",
     "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",
+    "saved_at_utc": "2026-04-10T02:09:02.834895+00:00",
     "toolkit": "astro_emulators_toolkit",
     "toolkit_version": "0.1.0"
   },
@@ -36,8 +37,8 @@
         "activation": "gelu",
         "dtype": "float32",
         "hidden_sizes": [
-          128,
-          128
+          32,
+          32
         ],
         "use_bias": true
       }
@@ -61,33 +62,25 @@
   },
   "runtime_contract": {
     "affine_leaf_shapes": {
-      "inputs/feh": [],
-      "inputs/logg": [],
-      "inputs/teff": [],
-      "outputs/flux": [
-        500
-      ]
+      "inputs/x0": [],
+      "inputs/x1": [],
+      "inputs/x2": [],
+      "outputs/y0": [],
+      "outputs/y1": []
     },
     "input_name_tree": [
-      "teff",
-      "logg",
-      "feh"
+      "x0",
+      "x1",
+      "x2"
     ],
     "io_pattern": "array->array",
-    "output_name_tree": "flux"
+    "output_name_tree": [
+      "y0",
+      "y1"
+    ]
   },
   "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",
@@ -101,532 +94,33 @@
     "spec_version": 2,
     "x": {
       "meaning": [
-        "effective temperature",
-        "surface gravity",
-        "metallicity [Fe/H]"
+        "",
+        "",
+        ""
       ],
       "names": [
-        "teff",
-        "logg",
-        "feh"
+        "x0",
+        "x1",
+        "x2"
       ],
       "units": [
-        "K",
-        "dex",
-        "dex"
+        "",
+        "",
+        ""
       ]
     },
     "y": {
       "meaning": [
-        "continuum-normalized flux vector on the shared wavelength grid"
+        "",
+        ""
       ],
       "names": [
-        "flux"
+        "y0",
+        "y1"
       ],
       "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
+        "",
+        ""
       ]
     }
   },

reference_scaling.safetensors

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

weights/weights.safetensors

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