{
  "seed": 3145,
  "device": "cuda",
  "math_check": {
    "c4_composition_max_abs": 0.0,
    "true_mechanism_equivariance_max_abs": 0.0,
    "equivariance_penalty_exact_linear": 0.0,
    "equivariance_penalty_non_equivariant_linear": 1.9284594058990479
  },
  "params_each_model": 2690,
  "baseline": {
    "train_mse": 0.000663807790260762,
    "in_distribution_mse": 0.000933704199269414,
    "heldout_rotation_mse": 0.0010833757696673274,
    "heldout_equivariance_penalty": 0.0005097697721794248
  },
  "equivariant_shared": {
    "train_mse": 0.0007101972005330026,
    "in_distribution_mse": 0.0006173581350594759,
    "heldout_rotation_mse": 0.0007219823892228305,
    "heldout_equivariance_penalty": 9.187469549942762e-05
  }
}