{
  "step": 50000,
  "weights_source": "ema",
  "loss": 0.3709731566905975,
  "edge_loss": 0.19136443853378296,
  "diag_loss": 0.1796087145805359,
  "jvp_rms": 2.2817168676853177,
  "examples": 5120,
  "loss_nfe_1": 0.39239816665649413,
  "edge_loss_nfe_1": 0.21810964345932007,
  "diag_loss_nfe_1": 0.1742885172367096,
  "jvp_rms_nfe_1": 2.700194501876831,
  "loss_nfe_2": 0.3679115891456604,
  "edge_loss_nfe_2": 0.18770996630191802,
  "diag_loss_nfe_2": 0.18020162284374236,
  "jvp_rms_nfe_2": 2.536956226825714,
  "loss_nfe_3": 0.35463637113571167,
  "edge_loss_nfe_3": 0.1791027843952179,
  "diag_loss_nfe_3": 0.17553358674049377,
  "jvp_rms_nfe_3": 2.0643750607967375,
  "loss_nfe_4": 0.38753209710121156,
  "edge_loss_nfe_4": 0.19491184055805205,
  "diag_loss_nfe_4": 0.19262024462223054,
  "jvp_rms_nfe_4": 2.4871247947216033,
  "loss_nfe_5": 0.3523875594139099,
  "edge_loss_nfe_5": 0.17698795795440675,
  "diag_loss_nfe_5": 0.17539960145950317,
  "jvp_rms_nfe_5": 1.6199337542057037,
  "raw_loss": 0.37272677540779114,
  "raw_edge_loss": 0.19294691383838652,
  "raw_diag_loss": 0.17977986514568328,
  "raw_jvp_rms": 2.4338310968875883,
  "raw_loss_nfe_1": 0.39853897094726565,
  "raw_edge_loss_nfe_1": 0.2240687519311905,
  "raw_diag_loss_nfe_1": 0.1744702249765396,
  "raw_jvp_rms_nfe_1": 2.9251581192016602,
  "raw_loss_nfe_2": 0.3692192077636719,
  "raw_edge_loss_nfe_2": 0.18882804214954377,
  "raw_diag_loss_nfe_2": 0.18039117157459258,
  "raw_jvp_rms_nfe_2": 2.7281934142112734,
  "raw_loss_nfe_3": 0.35528815984725953,
  "raw_edge_loss_nfe_3": 0.1796249717473984,
  "raw_diag_loss_nfe_3": 0.17566319108009337,
  "raw_jvp_rms_nfe_3": 2.1835286557674407,
  "raw_loss_nfe_4": 0.3877676367759705,
  "raw_edge_loss_nfe_4": 0.19502896666526795,
  "raw_diag_loss_nfe_4": 0.1927386701107025,
  "raw_jvp_rms_nfe_4": 2.6403229236602783,
  "raw_loss_nfe_5": 0.3528199017047882,
  "raw_edge_loss_nfe_5": 0.1771838366985321,
  "raw_diag_loss_nfe_5": 0.17563606798648834,
  "raw_jvp_rms_nfe_5": 1.69195237159729
}