| { |
| "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" |
| } |
