1{2 "steps": 600,3 "best_val": -6.068539780236767,4 "wall_s": 157.81467294692993,5 "ex_per_s": 243.32338231257614,6 "peak_mem_mb": 35.361792,7 "n_params": 231459,8 "device": "mps",9 "data_time_frac": 0.01176309837284115310}
main