1{2 "steps": 12000,3 "best_val": -6.453461130264837,4 "wall_s": 1759.3876690864563,5 "ex_per_s": 436.5155067835482,6 "peak_mem_mb": 63.172096,7 "n_params": 3455619,8 "device": "mps",9 "data_time_frac": 0.00985826931389833410}
bab86036e93a82a6f322d24c81650aa6a2bb946e