{
 "base": "google/gemma-4-31B-it",
 "method": "fsdp2-custom-loop",
 "n_rows": 41951,
 "n_self": 0,
 "n_sequences": 20409,
 "n_tokens": 32716533,
 "world": 8,
 "micro_batch": 1,
 "accum": 8,
 "steps": 318,
 "lr": 1e-05,
 "optim": "AdamW fused, fp32 master, bf16 compute, fp32 grad reduce",
 "loss_curve": [
  1.4239871501922607,
  1.5003830194473267,
  1.345046043395996,
  1.3999685049057007
 ],
 "loss_first": [
  1.4239871501922607
 ],
 "loss_last": [
  1.3999685049057007
 ],
 "train_seconds": 547.8836672306061,
 "tokens_per_second": 59714.37908593377,
 "resumed_from": "/workspace/sdf/out/ctrl-SG/ckpt-step300"
}