1{2 "steps": 12000,3 "best_val": -6.449690639457265,4 "wall_s": 838.5925340652466,5 "ex_per_s": 915.8202211471702,6 "peak_mem_mb": 14.961152,7 "n_params": 301827,8 "device": "mps",9 "data_time_frac": 0.0508195260993649410}
e976cfd0ab0ee7e59d46d9e75fd8b8fb0a861c50