{
 "steps": 4000,
 "best_val": -6.3999923274301365,
 "wall_s": 852.7440068721771,
 "ex_per_s": 300.20732826841595,
 "peak_mem_mb": 107.833344,
 "n_params": 3455619,
 "device": "mps",
 "data_time_frac": 0.016831904932074735
}