1{
 2  "math_checks": {
 3    "identity_max_error": 0.0,
 4    "remainder_loglog_slope": 2.002695747665,
 5    "remainder_quadratic_slope_predicted": 2.0,
 6    "remainder_coeff_est": 0.33344723957635847,
 7    "remainder_coeff_predicted": 0.33343481750184895,
 8    "gate_midpoint_observed": 0.08035261221856173,
 9    "gate_midpoint_predicted": 0.07999999999999999,
10    "gate_logit_slope_observed": 2.9999999999999996,
11    "gate_logit_slope_predicted": 3.0
12  },
13  "comparison": {
14    "full": {
15      "one_step_rmse": 0.023955633863806725,
16      "rollout_rmse": 0.3769861200835387,
17      "device": "cuda"
18    },
19    "residual": {
20      "one_step_rmse": 0.014161617495119572,
21      "rollout_rmse": 0.24085574749420596,
22      "device": "cuda"
23    }
24  }
25}