{
 "steps": 12000,
 "best_val": -6.453461130264837,
 "wall_s": 1759.3876690864563,
 "ex_per_s": 436.5155067835482,
 "peak_mem_mb": 63.172096,
 "n_params": 3455619,
 "device": "mps",
 "data_time_frac": 0.009858269313898334
}