[
  {
    "id": "arc:train:ARC-Challenge:696",
    "family": "arc",
    "branches": 4,
    "max_branch_tokens": 212,
    "loss": 5.821491718292236,
    "gradient_norm": 67.16741180419922,
    "peak_cuda_allocated_bytes": 16531817472
  },
  {
    "id": "banking:train:3917",
    "family": "banking",
    "branches": 4,
    "max_branch_tokens": 170,
    "loss": -0.0,
    "gradient_norm": 2.849806833182811e-06,
    "peak_cuda_allocated_bytes": 16531817472
  },
  {
    "id": "boolq:train:6648",
    "family": "boolq",
    "branches": 2,
    "max_branch_tokens": 509,
    "loss": 1.0364792346954346,
    "gradient_norm": 31.71306037902832,
    "peak_cuda_allocated_bytes": 16776178176
  },
  {
    "id": "snli:train:37567",
    "family": "snli",
    "branches": 3,
    "max_branch_tokens": 195,
    "loss": 0.0006712805479764938,
    "gradient_norm": 0.029433242976665497,
    "peak_cuda_allocated_bytes": 16776178176
  },
  {
    "id": "arc:train:ARC-Challenge:626",
    "family": "arc",
    "branches": 5,
    "max_branch_tokens": 139,
    "loss": 4.494089080253616e-05,
    "gradient_norm": 0.0032675806432962418,
    "peak_cuda_allocated_bytes": 16776178176
  }
]