{
 "steps": 12000,
 "best_val": -6.451788332870508,
 "wall_s": 927.3784019947052,
 "ex_per_s": 828.1409167477947,
 "peak_mem_mb": 199.197696,
 "n_params": 231459,
 "device": "mps",
 "data_time_frac": 0.045620226071422174
}