1{2 "steps": 600,3 "best_val": -6.066584629400399,4 "wall_s": 151.7184603214264,5 "ex_per_s": 253.10038026122106,6 "peak_mem_mb": 14.961152,7 "n_params": 301827,8 "device": "mps",9 "data_time_frac": 0.01752184393719184410}
main