{
 "steps": 12000,
 "best_val": -6.449690639457265,
 "wall_s": 838.5925340652466,
 "ex_per_s": 915.8202211471702,
 "peak_mem_mb": 14.961152,
 "n_params": 301827,
 "device": "mps",
 "data_time_frac": 0.05081952609936494
}