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