{
  "device": "cuda",
  "math_single_residual": 5.9650624280038755e-06,
  "math_32_product_residual": 1.3061589925200678e-05,
  "KLU": {
    "parameters": 564,
    "train_loss_0_100_300": [
      0.07692967355251312,
      0.0011512121418491006,
      0.00042202480835840106
    ],
    "test_mse": 0.0005223150947131217,
    "norm_drift": 2.123415470123291e-07,
    "unitarity_residual": 8.673438287587487e-07,
    "seconds": 19.60685217997525,
    "mean_grad_norm": 0.017432200691068662
  },
  "dense": {
    "parameters": 4352,
    "train_loss_0_100_300": [
      0.07300132513046265,
      0.0014969450421631336,
      0.0002997505653183907
    ],
    "test_mse": 0.0003739910898730159,
    "norm_drift": 0.009013999253511429,
    "unitarity_residual": 0.543326735496521,
    "seconds": 13.261652046989184,
    "mean_grad_norm": 0.004672153878406284
  }
}