| { |
| "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" |
| ] |
| }, |
| "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": "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", |
| "toolkit": "astro_emulators_toolkit", |
| "toolkit_version": "0.1.0" |
| }, |
| "resolved": { |
| "model": { |
| "name": "mlp", |
| "params": { |
| "activation": "gelu", |
| "dtype": "float32", |
| "hidden_sizes": [ |
| 32, |
| 32 |
| ], |
| "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/x0": [], |
| "inputs/x1": [], |
| "inputs/x2": [], |
| "outputs/y0": [], |
| "outputs/y1": [] |
| }, |
| "input_name_tree": [ |
| "x0", |
| "x1", |
| "x2" |
| ], |
| "io_pattern": "array->array", |
| "output_name_tree": [ |
| "y0", |
| "y1" |
| ] |
| }, |
| "schema_version": 1, |
| "spec": { |
| "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": [ |
| "", |
| "", |
| "" |
| ], |
| "names": [ |
| "x0", |
| "x1", |
| "x2" |
| ], |
| "units": [ |
| "", |
| "", |
| "" |
| ] |
| }, |
| "y": { |
| "meaning": [ |
| "", |
| "" |
| ], |
| "names": [ |
| "y0", |
| "y1" |
| ], |
| "units": [ |
| "", |
| "" |
| ] |
| } |
| }, |
| "tool": "astro_emulators_toolkit", |
| "weights_layout": "params_plus_model_state_v1" |
| } |
