checkpoints/step_5000/state.json

{"step": 5000, "loss": 2.5539231300354004, "time": 1790664764.634873}