{
 "steps": 4000,
 "best_val": -6.4504765663426005,
 "wall_s": 617.6293749809265,
 "ex_per_s": 414.48805767683206,
 "peak_mem_mb": 86.535168,
 "n_params": 3200259,
 "device": "mps",
 "data_time_frac": 0.022670459804084265
}