{
 "steps": 12000,
 "best_val": -6.452572455371413,
 "wall_s": 4579.964827299118,
 "ex_per_s": 167.6868772926587,
 "peak_mem_mb": 402.962432,
 "n_params": 20718339,
 "device": "mps",
 "data_time_frac": 0.004159492937716627
}