1{2 "steps": 60,3 "best_val": -6.087464488115836,4 "wall_s": 23.60936999320984,5 "ex_per_s": 81.32364398339305,6 "peak_mem_mb": 11.138048,7 "n_params": 301827,8 "device": "mps",9 "data_time_frac": 0.0489240607473627210}
main