1{2 "steps": 4000,3 "best_val": -6.450862349156063,4 "wall_s": 854.9879620075226,5 "ex_per_s": 299.4194203610876,6 "peak_mem_mb": 90.587136,7 "n_params": 3416643,8 "device": "mps",9 "data_time_frac": 0.0170687373807190710}
main