1{2 "steps": 4000,3 "best_val": -6.451227885057247,4 "wall_s": 719.5306529998779,5 "ex_per_s": 355.7874830386739,6 "peak_mem_mb": 87.04,7 "n_params": 3330819,8 "device": "mps",9 "data_time_frac": 0.01925191117800422610}
main