{
 "steps": 4000,
 "best_val": -6.451585810243266,
 "wall_s": 853.8766720294952,
 "ex_per_s": 299.8091040378687,
 "peak_mem_mb": 91.948032,
 "n_params": 3501699,
 "device": "mps",
 "data_time_frac": 0.016974973938125592
}