{
  "device": "cuda",
  "math_checks": [
    {
      "gamma": 0.0,
      "pred_logdet": 0.0,
      "obs_logdet": 0.0,
      "pred_3x": 0.0,
      "obs_3x": 0.0,
      "recon_max": 2.384185791015625e-07
    },
    {
      "gamma": 0.25,
      "pred_logdet": 1.0,
      "obs_logdet": 1.0,
      "pred_3x": 3.0,
      "obs_3x": 3.0,
      "recon_max": 4.76837158203125e-07
    },
    {
      "gamma": 0.5,
      "pred_logdet": 2.0,
      "obs_logdet": 2.0,
      "pred_3x": 6.0,
      "obs_3x": 6.0,
      "recon_max": 2.384185791015625e-07
    },
    {
      "gamma": 1.0,
      "pred_logdet": 4.0,
      "obs_logdet": 4.0,
      "pred_3x": 12.0,
      "obs_3x": 12.0,
      "recon_max": 2.384185791015625e-07
    },
    {
      "gamma": 1.5,
      "pred_logdet": 6.0,
      "obs_logdet": 6.0,
      "pred_3x": 18.0,
      "obs_3x": 18.0,
      "recon_max": 2.384185791015625e-07
    },
    {
      "nonlinear_jacobian_logdet": 2.0182530879974365,
      "analytic_logdet": 2.0182530879974365,
      "jacobian_abs_error": 0.0,
      "jacobian_sign": 1.0,
      "nonlinear_inverse_max_error": 0.0
    }
  ],
  "models": {
    "2": {
      "flow": [
        1.5364100933074951,
        1.2448464632034302
      ],
      "diag_mlp": [
        4.465583324432373,
        2.818814277648926
      ]
    },
    "8": {
      "flow": [
        4.148821830749512,
        1.2992744445800781
      ],
      "diag_mlp": [
        17.779361724853516,
        2.9177722930908203
      ]
    }
  }
}