1{2 "steps": 4000,3 "best_val": -6.449357611608497,4 "wall_s": 853.1240651607513,5 "ex_per_s": 300.07358888857834,6 "peak_mem_mb": 91.210752,7 "n_params": 3455619,8 "device": "mps",9 "data_time_frac": 0.01718328800849000210}
main