1{2 "steps": 4000,3 "best_val": -6.4504765663426005,4 "wall_s": 617.6293749809265,5 "ex_per_s": 414.48805767683206,6 "peak_mem_mb": 86.535168,7 "n_params": 3200259,8 "device": "mps",9 "data_time_frac": 0.02267045980408426510}
main