1{2 "steps": 600,3 "best_val": -6.066347980807233,4 "wall_s": 206.8860673904419,5 "ex_per_s": 185.6094056229041,6 "peak_mem_mb": 62.975488,7 "n_params": 3455619,8 "device": "mps",9 "data_time_frac": 0.00970268779792949310}
main