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