{
 "node": "m3u96b",
 "chip": "Apple M3 Ultra",
 "torch": "2.15.0",
 "device": "mps",
 "n_origins": 16299,
 "loader": {
  "workers_0": {
   "ex_per_s": 3970.9504465989544
  },
  "workers_4": {
   "ex_per_s": 13015.525703964993
  },
  "workers_8": {
   "ex_per_s": 38486.33564282975
  }
 },
 "train": {
  "small_bs32": {
   "params": 3455619,
   "steps_per_s": 20.808967283900596,
   "ex_per_s": 665.8869530848191,
   "peak_mem_mb": 53.906944,
   "loss_finite": true
  },
  "small_bs64": {
   "params": 3455619,
   "steps_per_s": 12.212416590445695,
   "ex_per_s": 781.5946617885245,
   "peak_mem_mb": 68.529152,
   "loss_finite": true
  },
  "small_bs128": {
   "params": 3455619,
   "steps_per_s": 5.925401298878898,
   "ex_per_s": 758.451366256499,
   "peak_mem_mb": 96.797184,
   "loss_finite": true
  },
  "spec_bs32": {
   "params": 7677955,
   "steps_per_s": 11.544136009460699,
   "ex_per_s": 369.41235230274236,
   "peak_mem_mb": 114.085376,
   "loss_finite": true
  },
  "spec_bs64": {
   "params": 7677955,
   "steps_per_s": 6.285685112498215,
   "ex_per_s": 402.28384719988577,
   "peak_mem_mb": 119.197184,
   "loss_finite": true
  },
  "spec_bs128": {
   "params": 7677955,
   "steps_per_s": 2.9673766457169934,
   "ex_per_s": 379.82421065177516,
   "peak_mem_mb": 146.711552,
   "loss_finite": true
  },
  "base_bs32": {
   "params": 20718339,
   "steps_per_s": 6.10752187132323,
   "ex_per_s": 195.44069988234335,
   "peak_mem_mb": 275.288576,
   "loss_finite": true
  },
  "base_bs64": {
   "params": 20718339,
   "steps_per_s": 3.239651990508832,
   "ex_per_s": 207.33772739256526,
   "peak_mem_mb": 283.539968,
   "loss_finite": true
  },
  "base_bs128": {
   "params": 20718339,
   "steps_per_s": 1.5058899942717616,
   "ex_per_s": 192.7539192667855,
   "peak_mem_mb": 315.058176,
   "loss_finite": true
  }
 },
 "inference": {
  "small_bs32": {
   "latency_ms_batch1": 4.309260845184326,
   "latency_ms_per_origin_batched": 0.34431926906108856
  },
  "small_bs64": {
   "latency_ms_batch1": 4.311156272888184,
   "latency_ms_per_origin_batched": 0.2541523426771164
  },
  "small_bs128": {
   "latency_ms_batch1": 4.303455352783203,
   "latency_ms_per_origin_batched": 0.27103349566459656
  },
  "spec_bs32": {
   "latency_ms_batch1": 5.084848403930664,
   "latency_ms_per_origin_batched": 0.5867015570402145
  },
  "spec_bs64": {
   "latency_ms_batch1": 5.049645900726318,
   "latency_ms_per_origin_batched": 0.4757307469844818
  },
  "spec_bs128": {
   "latency_ms_batch1": 5.115711688995361,
   "latency_ms_per_origin_batched": 0.5341288633644581
  },
  "base_bs32": {
   "latency_ms_batch1": 5.908381938934326,
   "latency_ms_per_origin_batched": 5.037868395447731
  },
  "base_bs64": {
   "latency_ms_batch1": 5.8892011642456055,
   "latency_ms_per_origin_batched": 1.585175283253193
  },
  "base_bs128": {
   "latency_ms_batch1": 5.894708633422852,
   "latency_ms_per_origin_batched": 1.7552338540554047
  }
 }
}