1{2 "steps": 4000,3 "best_val": -6.451585810243266,4 "wall_s": 853.8766720294952,5 "ex_per_s": 299.8091040378687,6 "peak_mem_mb": 91.948032,7 "n_params": 3501699,8 "device": "mps",9 "data_time_frac": 0.01697497393812559210}
main