{
 "steps": 12000,
 "best_val": -6.452020036474576,
 "wall_s": 1087.6674129962921,
 "ex_per_s": 706.0981976874011,
 "peak_mem_mb": 87.387136,
 "n_params": 3200259,
 "device": "mps",
 "data_time_frac": 0.01726834735532425
}