1{2 "steps": 4000,3 "best_val": -6.45034669881108,4 "wall_s": 793.9436521530151,5 "ex_per_s": 322.4410187118187,6 "peak_mem_mb": 122.998784,7 "n_params": 5533443,8 "device": "mps",9 "data_time_frac": 0.01849493274532728510}
main