[
  {
    "id": "arc:train:ARC-Challenge:696",
    "family": "arc",
    "branches": 4,
    "max_branch_tokens": 221,
    "loss": 1.148118019104004,
    "gradient_norm": 42.69694900512695,
    "peak_cuda_allocated_bytes": 8166141952
  },
  {
    "id": "banking:train:3917",
    "family": "banking",
    "branches": 4,
    "max_branch_tokens": 179,
    "loss": 0.0007213136414065957,
    "gradient_norm": 0.3080456256866455,
    "peak_cuda_allocated_bytes": 8166141952
  },
  {
    "id": "boolq:train:2141",
    "family": "boolq",
    "branches": 2,
    "max_branch_tokens": 700,
    "loss": 0.6513704657554626,
    "gradient_norm": 62.59121322631836,
    "peak_cuda_allocated_bytes": 8726854144
  },
  {
    "id": "snli:train:37567",
    "family": "snli",
    "branches": 3,
    "max_branch_tokens": 205,
    "loss": 0.3688506484031677,
    "gradient_norm": 21.545278549194336,
    "peak_cuda_allocated_bytes": 8726854144
  },
  {
    "id": "arc:train:ARC-Challenge:626",
    "family": "arc",
    "branches": 5,
    "max_branch_tokens": 148,
    "loss": 0.407499760389328,
    "gradient_norm": 16.394695281982422,
    "peak_cuda_allocated_bytes": 8726854144
  }
]