{
  "schema_version": 1,
  "model": "FUSE-EFIT",
  "protocol": "in era (random split of shots 125500-205774; DECISIONS D50, D55)",
  "form": "single",
  "form_meaning": "single: replicate 0 of the in-era finals, the frozen delivered network",
  "members": [
    {
      "run": "runs/FULLextfinal_era_r0",
      "replicate": 0,
      "seed": 100,
      "checkpoint_sha256": "28390d5950ec18477394a85dc73498afa0f847502514251b7c96c97569632d96",
      "config_sha256": "ddec260834c0f5fdaa5303382140e9c0bca1e3028ccd478c2c2a65f62d6f2398",
      "trained_code_sha256": "1095cd5859ce819840f3e35ead3243d28aae832dbd3f54b04b90d98330493290"
    }
  ],
  "cadence": {
    "period_ms": 50.0,
    "hz": 20,
    "rule": "one step-graph call per 50 ms of simulated time; each tick describes the trailing interval (t - 50 ms, t]; reset the state to zeros at the start of every shot"
  },
  "inputs": {
    "inputs": {
      "shape": [
        "batch",
        25
      ],
      "dtype": "float32",
      "names": [
        "pinj",
        "tinj",
        "ech_mw",
        "gasa_cal",
        "gasb_cal",
        "gasc_cal",
        "gasd_cal",
        "gase_cal",
        "bt",
        "ip",
        "aminor",
        "rsurf",
        "zsurf",
        "kappa",
        "tritop",
        "tribot",
        "drsep",
        "rvsout",
        "gapin",
        "rvsou",
        "rvsiu",
        "shafranov",
        "zeta_ou",
        "zeta_ol",
        "vloop"
      ],
      "units": {
        "pinj": "MW",
        "tinj": "N m",
        "ech_mw": "MW",
        "gasa_cal": "Torr L/s",
        "gasb_cal": "Torr L/s",
        "gasc_cal": "Torr L/s",
        "gasd_cal": "Torr L/s",
        "gase_cal": "Torr L/s",
        "bt": "T",
        "ip": "MA",
        "aminor": "m",
        "rsurf": "m",
        "zsurf": "m",
        "kappa": "dimensionless",
        "tritop": "dimensionless",
        "tribot": "dimensionless",
        "drsep": "m",
        "rvsout": "m",
        "gapin": "m",
        "rvsou": "m",
        "rvsiu": "m",
        "shafranov": "m",
        "zeta_ou": "dimensionless",
        "zeta_ol": "dimensionless",
        "vloop": "V"
      },
      "sources": {
        "pinj": "PtData bmspinj: neutral-beam injected power, all sources",
        "tinj": "PtData bmstinj: neutral-beam injected torque",
        "ech_mw": "MDSplus \\RF::ECHPWRC: calibrated total ECH power injected",
        "gasa_cal": "MDSplus NEUTRALS GASFLOW.GASA:FLOW: calibrated gas flow, valve A",
        "gasb_cal": "MDSplus NEUTRALS GASFLOW.GASB:FLOW: calibrated gas flow, valve B",
        "gasc_cal": "MDSplus NEUTRALS GASFLOW.GASC:FLOW: calibrated gas flow, valve C",
        "gasd_cal": "MDSplus NEUTRALS GASFLOW.GASD:FLOW: calibrated gas flow, valve D",
        "gase_cal": "MDSplus NEUTRALS GASFLOW.GASE:FLOW: calibrated gas flow, valve E",
        "bt": "PtData bt: vacuum toroidal field, signed",
        "ip": "PtData ip: measured plasma current (amperes / 1e6)",
        "aminor": "EFIT01 AMINOR: minor radius",
        "rsurf": "EFIT01 RSURF: major radius of the boundary's geometric centre",
        "zsurf": "EFIT01 ZSURF: height of the boundary's geometric centre",
        "kappa": "EFIT01 KAPPA: elongation",
        "tritop": "EFIT01 TRITOP: upper triangularity",
        "tribot": "EFIT01 TRIBOT: lower triangularity",
        "drsep": "EFIT01 DRSEP: separatrix separation at the outer midplane, clipped to +-0.1 m per slice",
        "rvsout": "EFIT01 RVSOUT: major radius of the outer (lower) strike point",
        "gapin": "EFIT01 GAPIN: inner gap",
        "rvsou": "EFIT01 RVSOU: major radius of the upper outer strike point",
        "rvsiu": "EFIT01 RVSIU: major radius of the upper inner strike point",
        "shafranov": "EFIT01 R0 - RSURF: Shafranov shift",
        "zeta_ou": "Luce squareness, outer-upper quadrant, from the EFIT01 boundary vertices",
        "zeta_ol": "Luce squareness, outer-lower quadrant, from the EFIT01 boundary vertices",
        "vloop": "PtData VLOOP: loop voltage"
      },
      "grid_rule": {
        "pinj": "plain mean of the three newest 20 ms samples at or before the tick, after a causal low-pass and hold onto 20 ms (INPUT_CONTRACT 3.1)",
        "tinj": "plain mean of the three newest 20 ms samples at or before the tick, after a causal low-pass and hold onto 20 ms (INPUT_CONTRACT 3.1)",
        "ech_mw": "plain mean of the three newest 20 ms samples at or before the tick, after a causal low-pass and hold onto 20 ms (INPUT_CONTRACT 3.1)",
        "gasa_cal": "plain mean of the three newest 20 ms samples at or before the tick, after a causal low-pass and hold onto 20 ms (INPUT_CONTRACT 3.1)",
        "gasb_cal": "plain mean of the three newest 20 ms samples at or before the tick, after a causal low-pass and hold onto 20 ms (INPUT_CONTRACT 3.1)",
        "gasc_cal": "plain mean of the three newest 20 ms samples at or before the tick, after a causal low-pass and hold onto 20 ms (INPUT_CONTRACT 3.1)",
        "gasd_cal": "plain mean of the three newest 20 ms samples at or before the tick, after a causal low-pass and hold onto 20 ms (INPUT_CONTRACT 3.1)",
        "gase_cal": "plain mean of the three newest 20 ms samples at or before the tick, after a causal low-pass and hold onto 20 ms (INPUT_CONTRACT 3.1)",
        "bt": "plain mean of the three newest 20 ms samples at or before the tick, after a causal low-pass and hold onto 20 ms (INPUT_CONTRACT 3.1)",
        "ip": "plain mean of the three newest 20 ms samples at or before the tick, after a causal low-pass and hold onto 20 ms (INPUT_CONTRACT 3.1)",
        "aminor": "mean of the per-slice values of the EFIT01 slices in (t - 50 ms, t] (INPUT_CONTRACT 3.2)",
        "rsurf": "mean of the per-slice values of the EFIT01 slices in (t - 50 ms, t] (INPUT_CONTRACT 3.2)",
        "zsurf": "mean of the per-slice values of the EFIT01 slices in (t - 50 ms, t] (INPUT_CONTRACT 3.2)",
        "kappa": "mean of the per-slice values of the EFIT01 slices in (t - 50 ms, t] (INPUT_CONTRACT 3.2)",
        "tritop": "mean of the per-slice values of the EFIT01 slices in (t - 50 ms, t] (INPUT_CONTRACT 3.2)",
        "tribot": "mean of the per-slice values of the EFIT01 slices in (t - 50 ms, t] (INPUT_CONTRACT 3.2)",
        "drsep": "mean of the per-slice values of the EFIT01 slices in (t - 50 ms, t] (INPUT_CONTRACT 3.2)",
        "rvsout": "mean of the per-slice values of the EFIT01 slices in (t - 50 ms, t] (INPUT_CONTRACT 3.2)",
        "gapin": "mean of the per-slice values of the EFIT01 slices in (t - 50 ms, t] (INPUT_CONTRACT 3.2)",
        "rvsou": "mean of the per-slice values of the EFIT01 slices in (t - 50 ms, t] (INPUT_CONTRACT 3.2)",
        "rvsiu": "mean of the per-slice values of the EFIT01 slices in (t - 50 ms, t] (INPUT_CONTRACT 3.2)",
        "shafranov": "mean of the per-slice values of the EFIT01 slices in (t - 50 ms, t] (INPUT_CONTRACT 3.2)",
        "zeta_ou": "mean of the per-slice values of the EFIT01 slices in (t - 50 ms, t] (INPUT_CONTRACT 3.2)",
        "zeta_ol": "mean of the per-slice values of the EFIT01 slices in (t - 50 ms, t] (INPUT_CONTRACT 3.2)",
        "vloop": "mean of the native samples in (t - 50 ms, t] (INPUT_CONTRACT 3.3)"
      },
      "zeroed_when_indicator_is_0": {
        "pinj": [],
        "tinj": [],
        "ech_mw": [],
        "gasa_cal": [],
        "gasb_cal": [],
        "gasc_cal": [],
        "gasd_cal": [],
        "gase_cal": [],
        "bt": [],
        "ip": [],
        "aminor": [
          "efit_valid"
        ],
        "rsurf": [
          "efit_valid"
        ],
        "zsurf": [
          "efit_valid"
        ],
        "kappa": [
          "efit_valid"
        ],
        "tritop": [
          "efit_valid"
        ],
        "tribot": [
          "efit_valid"
        ],
        "drsep": [
          "efit_valid"
        ],
        "rvsout": [
          "efit_valid",
          "rvsout_present"
        ],
        "gapin": [
          "efit_valid"
        ],
        "rvsou": [
          "efit_valid",
          "rvsou_present"
        ],
        "rvsiu": [
          "efit_valid",
          "rvsiu_present"
        ],
        "shafranov": [
          "efit_valid"
        ],
        "zeta_ou": [
          "efit_valid"
        ],
        "zeta_ol": [
          "efit_valid"
        ],
        "vloop": [
          "vloop_valid"
        ]
      },
      "note": "Physical units, contract order (docs/INPUT_CONTRACT.md \u00a72). Where a governing indicator is 0 the value is ignored (any number, or NaN). NaN anywhere is read as the training mean, so never pass NaN for a value you have."
    },
    "indicators": {
      "shape": [
        "batch",
        5
      ],
      "dtype": "float32",
      "values": "0 or 1, per tick",
      "names": [
        "efit_valid",
        "vloop_valid",
        "rvsout_present",
        "rvsou_present",
        "rvsiu_present"
      ],
      "meaning": {
        "efit_valid": "1 when at least one usable EFIT slice lies in (t - 50 ms, t]; 0 otherwise",
        "vloop_valid": "1 when at least one loop-voltage sample lies in (t - 50 ms, t]; 0 otherwise",
        "rvsout_present": "1 when at least one slice in the window has a real outer strike point; 0 otherwise",
        "rvsou_present": "1 when at least one slice in the window has a real upper outer strike point; 0 otherwise",
        "rvsiu_present": "1 when at least one slice in the window has a real upper inner strike point; 0 otherwise"
      }
    },
    "available": {
      "shape": [
        "batch",
        25
      ],
      "dtype": "float32",
      "values": "0 or 1, per shot",
      "names": [
        "pinj",
        "tinj",
        "ech_mw",
        "gasa_cal",
        "gasb_cal",
        "gasc_cal",
        "gasd_cal",
        "gase_cal",
        "bt",
        "ip",
        "aminor",
        "rsurf",
        "zsurf",
        "kappa",
        "tritop",
        "tribot",
        "drsep",
        "rvsout",
        "gapin",
        "rvsou",
        "rvsiu",
        "shafranov",
        "zeta_ou",
        "zeta_ol",
        "vloop"
      ],
      "default": 1,
      "note": "1 = a usable record of this channel exists for the shot (the normal case). 0 = no record for the whole shot: the value is ignored and the model receives the flag, as it did in training for missing records. Keep it constant within a shot."
    }
  },
  "outputs": {
    "predictions": {
      "shape": [
        "batch",
        9
      ],
      "dtype": "float32",
      "names": [
        "ne",
        "te_ped",
        "ti_ped",
        "t_rot_ped",
        "ne_knee",
        "te_knee",
        "rho_sym",
        "ne_top_loc",
        "hmode"
      ],
      "units": {
        "ne": "1e19 m^-3",
        "te_ped": "keV",
        "ti_ped": "keV",
        "t_rot_ped": "krad/s",
        "ne_knee": "1e19 m^-3",
        "te_knee": "keV",
        "rho_sym": "rho (normalised radius)",
        "ne_top_loc": "rho (normalised radius)",
        "hmode": "probability in [0, 1]"
      },
      "meaning": {
        "ne": "electron density at rho = 0.85",
        "te_ped": "electron temperature at rho = 0.85",
        "ti_ped": "ion temperature at rho = 0.85",
        "t_rot_ped": "toroidal rotation at rho = 0.85",
        "ne_knee": "electron density on the pedestal fit at its knee (H-mode only)",
        "te_knee": "electron temperature on its pedestal fit at its knee (H-mode only)",
        "rho_sym": "symmetry point of the density pedestal fit (H-mode only)",
        "ne_top_loc": "knee (pedestal top) of the density pedestal fit (H-mode only)",
        "hmode": "the plasma is in H-mode (sigmoid of the averaged or single logit)"
      }
    },
    "hmode_gated_heads": [
      "ne_knee",
      "te_knee",
      "rho_sym",
      "ne_top_loc"
    ],
    "hmode_gated_note": "Trained on H-mode ticks only and undefined in L-mode; the graph emits a number anyway. Report them only where predictions[hmode] >= threshold; otherwise treat them as undefined, never as 0.",
    "hmode_thresholds": {
      "frozen": {
        "balanced": 0.54,
        "conservative": 0.5
      },
      "definitions": {
        "balanced": "maximum Youden's J on validation (the best trade between missed H-mode and admitted L-mode ticks)",
        "conservative": "the lowest threshold keeping L-mode ticks at or below 5 % of admitted validation ticks"
      },
      "set_on": "replicate 0, in-era validation split",
      "source": "receipts/G5_freeze.md section 3; receipts/G5_freeze.json gate_thresholds.era",
      "freeze_record_sha256": "1fbd776750736a6995a5de48b74b16bc13e0a2a1539972e46be23e8dbb49316e",
      "note": "Gate the four H-mode-only heads on predictions[hmode] >= threshold; MODEL_CARD.md \u00a76."
    }
  },
  "graphs": {
    "step": {
      "file": "fuse_efit_single_step.onnx",
      "purpose": "one 50 ms tick per call; unbounded shot length",
      "inputs": {
        "inputs": [
          "batch",
          25
        ],
        "indicators": [
          "batch",
          5
        ],
        "available": [
          "batch",
          25
        ],
        "conv_state": [
          6,
          "batch",
          1792,
          3
        ],
        "ssm_state": [
          6,
          "batch",
          12,
          128,
          64
        ]
      },
      "outputs": {
        "predictions": [
          "batch",
          9
        ],
        "conv_state_out": [
          6,
          "batch",
          1792,
          3
        ],
        "ssm_state_out": [
          6,
          "batch",
          12,
          128,
          64
        ]
      }
    },
    "sequence": {
      "file": "fuse_efit_single_sequence.onnx",
      "seq_len": 256,
      "purpose": "a whole window of exactly seq_len ticks; right-pad a shorter shot (any finite values, indicators 1) and keep the first K outputs, which is exact because the model is causal; use the step graph for longer shots",
      "inputs": {
        "inputs": [
          "batch",
          256,
          25
        ],
        "indicators": [
          "batch",
          256,
          5
        ],
        "available": [
          "batch",
          25
        ]
      },
      "outputs": {
        "predictions": [
          "batch",
          256,
          9
        ]
      }
    }
  },
  "state": {
    "conv_state": [
      "n_members x n_layers",
      "batch",
      1792,
      3
    ],
    "ssm_state": [
      "n_members x n_layers",
      "batch",
      12,
      128,
      64
    ],
    "leading_axis": 6,
    "n_members": 1,
    "n_layers": 6,
    "initial": "all zeros at the start of every shot",
    "bytes_per_shot": 2488320,
    "note": "Opaque: store what the step graph returns and pass it back on the next call. The two forms differ only in the leading axis of these two tensors."
  },
  "architecture": {
    "family": "Mamba-2 state-space sequence model",
    "d_model": 768,
    "n_layers": 6,
    "d_state": 64,
    "d_conv": 4,
    "expand": 2,
    "headdim": 128,
    "nheads": 12,
    "ngroups": 2,
    "d_inner": 1536,
    "conv_dim": 1792,
    "chunk_size": 100,
    "n_inputs_trunk": 30,
    "n_params_per_member": 22882785,
    "n_members": 1
  },
  "input_transform": {
    "clip_sigma": 20.0,
    "std_floor": 1e-06,
    "file": "norms.json",
    "applied": "inside the graphs"
  },
  "dtype": "float32",
  "onnx_opset": 17,
  "documentation": "docs/EXPORT.md, docs/INPUT_CONTRACT.md, MODEL_CARD.md"
}