1{2 "steps": 4000,3 "best_val": -6.450148327536085,4 "wall_s": 853.3193740844727,5 "ex_per_s": 300.004907628709,6 "peak_mem_mb": 90.356224,7 "n_params": 3421643,8 "device": "mps",9 "data_time_frac": 0.0168698744997117810}
main