{
  "wall_seconds": 5031.164765119553,
  "records_seen": 19994,
  "requested_records": 15320,
  "truncated_records": 0,
  "rejected_records": 0,
  "optimizer_steps": 1915,
  "forward_tokens": 8738926,
  "peak_device_bytes": 47737451520,
  "device": "cuda",
  "dtype": "bf16",
  "batch": 2,
  "peak_rss_bytes": 31552528384
}