{
  "best_global_step": null,
  "best_metric": null,
  "best_model_checkpoint": null,
  "epoch": 0.9333333333333333,
  "eval_steps": 500,
  "global_step": 30000,
  "is_hyper_param_search": false,
  "is_local_process_zero": true,
  "is_world_process_zero": true,
  "log_history": [
    {
      "step": 0,
      "train_flow_matching_loss": 0.326171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 0,
      "train_flow_matching_loss": 0.1591796875,
      "train_physics_loss": 0.0
    },
    {
      "step": 0,
      "train_flow_matching_loss": 0.490234375,
      "train_physics_loss": 1.2741376161575317
    },
    {
      "step": 0,
      "train_flow_matching_loss": 0.21875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 3.98442006111145,
      "learning_rate": 3.0000000000000004e-07,
      "loss": 0.294,
      "step": 10
    },
    {
      "step": 10,
      "train_flow_matching_loss": 0.28125,
      "train_physics_loss": 0.0
    },
    {
      "step": 10,
      "train_flow_matching_loss": 0.25,
      "train_physics_loss": 0.0
    },
    {
      "step": 10,
      "train_flow_matching_loss": 0.21875,
      "train_physics_loss": 1.2196308374404907
    },
    {
      "step": 10,
      "train_flow_matching_loss": 0.203125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 3.9263219833374023,
      "learning_rate": 6.333333333333333e-07,
      "loss": 0.2746,
      "step": 20
    },
    {
      "step": 20,
      "train_flow_matching_loss": 0.1474609375,
      "train_physics_loss": 0.0
    },
    {
      "step": 20,
      "train_flow_matching_loss": 0.330078125,
      "train_physics_loss": 0.0
    },
    {
      "step": 20,
      "train_flow_matching_loss": 0.26953125,
      "train_physics_loss": 1.2393964529037476
    },
    {
      "step": 20,
      "train_flow_matching_loss": 0.1923828125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 1.1313203573226929,
      "learning_rate": 9.666666666666668e-07,
      "loss": 0.2785,
      "step": 30
    },
    {
      "step": 30,
      "train_flow_matching_loss": 0.30078125,
      "train_physics_loss": 0.0
    },
    {
      "step": 30,
      "train_flow_matching_loss": 0.2216796875,
      "train_physics_loss": 0.0
    },
    {
      "step": 30,
      "train_flow_matching_loss": 0.228515625,
      "train_physics_loss": 1.1477241516113281
    },
    {
      "step": 30,
      "train_flow_matching_loss": 0.2333984375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.4835752844810486,
      "learning_rate": 1.3e-06,
      "loss": 0.2698,
      "step": 40
    },
    {
      "step": 40,
      "train_flow_matching_loss": 0.1845703125,
      "train_physics_loss": 0.0
    },
    {
      "step": 40,
      "train_flow_matching_loss": 0.28515625,
      "train_physics_loss": 0.0
    },
    {
      "step": 40,
      "train_flow_matching_loss": 0.326171875,
      "train_physics_loss": 1.1630665063858032
    },
    {
      "step": 40,
      "train_flow_matching_loss": 0.255859375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 1.1689863204956055,
      "learning_rate": 1.6333333333333333e-06,
      "loss": 0.2732,
      "step": 50
    },
    {
      "step": 50,
      "train_flow_matching_loss": 0.2236328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 50,
      "train_flow_matching_loss": 0.2099609375,
      "train_physics_loss": 0.0
    },
    {
      "step": 50,
      "train_flow_matching_loss": 0.2041015625,
      "train_physics_loss": 1.3365275859832764
    },
    {
      "step": 50,
      "train_flow_matching_loss": 0.2431640625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.9448999166488647,
      "learning_rate": 1.9666666666666668e-06,
      "loss": 0.268,
      "step": 60
    },
    {
      "step": 60,
      "train_flow_matching_loss": 0.15625,
      "train_physics_loss": 0.0
    },
    {
      "step": 60,
      "train_flow_matching_loss": 0.25390625,
      "train_physics_loss": 0.0
    },
    {
      "step": 60,
      "train_flow_matching_loss": 0.306640625,
      "train_physics_loss": 1.2744060754776
    },
    {
      "step": 60,
      "train_flow_matching_loss": 0.2392578125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.32942628860473633,
      "learning_rate": 2.3e-06,
      "loss": 0.2597,
      "step": 70
    },
    {
      "step": 70,
      "train_flow_matching_loss": 0.259765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 70,
      "train_flow_matching_loss": 0.255859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 70,
      "train_flow_matching_loss": 0.376953125,
      "train_physics_loss": 1.251474380493164
    },
    {
      "step": 70,
      "train_flow_matching_loss": 0.2294921875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.7263258099555969,
      "learning_rate": 2.6333333333333337e-06,
      "loss": 0.2581,
      "step": 80
    },
    {
      "step": 80,
      "train_flow_matching_loss": 0.1904296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 80,
      "train_flow_matching_loss": 0.33984375,
      "train_physics_loss": 0.0
    },
    {
      "step": 80,
      "train_flow_matching_loss": 0.384765625,
      "train_physics_loss": 1.1229314804077148
    },
    {
      "step": 80,
      "train_flow_matching_loss": 0.1923828125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.23911438882350922,
      "learning_rate": 2.966666666666667e-06,
      "loss": 0.2504,
      "step": 90
    },
    {
      "step": 90,
      "train_flow_matching_loss": 0.333984375,
      "train_physics_loss": 0.0
    },
    {
      "step": 90,
      "train_flow_matching_loss": 0.1865234375,
      "train_physics_loss": 0.0
    },
    {
      "step": 90,
      "train_flow_matching_loss": 0.310546875,
      "train_physics_loss": 1.182525873184204
    },
    {
      "step": 90,
      "train_flow_matching_loss": 0.271484375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.16914765536785126,
      "learning_rate": 3.3e-06,
      "loss": 0.2475,
      "step": 100
    },
    {
      "step": 100,
      "train_flow_matching_loss": 0.201171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 100,
      "train_flow_matching_loss": 0.2236328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 100,
      "train_flow_matching_loss": 0.296875,
      "train_physics_loss": 1.1718263626098633
    },
    {
      "step": 100,
      "train_flow_matching_loss": 0.205078125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.2843989431858063,
      "learning_rate": 3.633333333333334e-06,
      "loss": 0.2462,
      "step": 110
    },
    {
      "step": 110,
      "train_flow_matching_loss": 0.26171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 110,
      "train_flow_matching_loss": 0.2197265625,
      "train_physics_loss": 0.0
    },
    {
      "step": 110,
      "train_flow_matching_loss": 0.244140625,
      "train_physics_loss": 1.0591553449630737
    },
    {
      "step": 110,
      "train_flow_matching_loss": 0.2275390625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.18271377682685852,
      "learning_rate": 3.966666666666667e-06,
      "loss": 0.2454,
      "step": 120
    },
    {
      "step": 120,
      "train_flow_matching_loss": 0.234375,
      "train_physics_loss": 0.0
    },
    {
      "step": 120,
      "train_flow_matching_loss": 0.267578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 120,
      "train_flow_matching_loss": 0.330078125,
      "train_physics_loss": 1.124878168106079
    },
    {
      "step": 120,
      "train_flow_matching_loss": 0.208984375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.1412096917629242,
      "learning_rate": 4.2999999999999995e-06,
      "loss": 0.2437,
      "step": 130
    },
    {
      "step": 130,
      "train_flow_matching_loss": 0.2109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 130,
      "train_flow_matching_loss": 0.26953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 130,
      "train_flow_matching_loss": 0.203125,
      "train_physics_loss": 1.2839598655700684
    },
    {
      "step": 130,
      "train_flow_matching_loss": 0.1513671875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.210030660033226,
      "learning_rate": 4.633333333333334e-06,
      "loss": 0.2469,
      "step": 140
    },
    {
      "step": 140,
      "train_flow_matching_loss": 0.1474609375,
      "train_physics_loss": 0.0
    },
    {
      "step": 140,
      "train_flow_matching_loss": 0.2265625,
      "train_physics_loss": 0.0
    },
    {
      "step": 140,
      "train_flow_matching_loss": 0.1640625,
      "train_physics_loss": 1.1520222425460815
    },
    {
      "step": 140,
      "train_flow_matching_loss": 0.228515625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.1495743989944458,
      "learning_rate": 4.966666666666667e-06,
      "loss": 0.2349,
      "step": 150
    },
    {
      "step": 150,
      "train_flow_matching_loss": 0.1826171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 150,
      "train_flow_matching_loss": 0.1533203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 150,
      "train_flow_matching_loss": 0.30859375,
      "train_physics_loss": 1.1289371252059937
    },
    {
      "step": 150,
      "train_flow_matching_loss": 0.318359375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.22715237736701965,
      "learning_rate": 5.3e-06,
      "loss": 0.2325,
      "step": 160
    },
    {
      "step": 160,
      "train_flow_matching_loss": 0.251953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 160,
      "train_flow_matching_loss": 0.1533203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 160,
      "train_flow_matching_loss": 0.1455078125,
      "train_physics_loss": 1.0798701047897339
    },
    {
      "step": 160,
      "train_flow_matching_loss": 0.1845703125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.2102261781692505,
      "learning_rate": 5.633333333333333e-06,
      "loss": 0.235,
      "step": 170
    },
    {
      "step": 170,
      "train_flow_matching_loss": 0.1552734375,
      "train_physics_loss": 0.0
    },
    {
      "step": 170,
      "train_flow_matching_loss": 0.1455078125,
      "train_physics_loss": 0.0
    },
    {
      "step": 170,
      "train_flow_matching_loss": 0.25390625,
      "train_physics_loss": 1.0951461791992188
    },
    {
      "step": 170,
      "train_flow_matching_loss": 0.1611328125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11585128307342529,
      "learning_rate": 5.9666666666666666e-06,
      "loss": 0.2366,
      "step": 180
    },
    {
      "step": 180,
      "train_flow_matching_loss": 0.1708984375,
      "train_physics_loss": 0.0
    },
    {
      "step": 180,
      "train_flow_matching_loss": 0.150390625,
      "train_physics_loss": 0.0
    },
    {
      "step": 180,
      "train_flow_matching_loss": 0.283203125,
      "train_physics_loss": 1.1284942626953125
    },
    {
      "step": 180,
      "train_flow_matching_loss": 0.216796875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.17999766767024994,
      "learning_rate": 6.300000000000001e-06,
      "loss": 0.2389,
      "step": 190
    },
    {
      "step": 190,
      "train_flow_matching_loss": 0.1806640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 190,
      "train_flow_matching_loss": 0.1767578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 190,
      "train_flow_matching_loss": 0.302734375,
      "train_physics_loss": 1.2676578760147095
    },
    {
      "step": 190,
      "train_flow_matching_loss": 0.1826171875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.1244242861866951,
      "learning_rate": 6.633333333333333e-06,
      "loss": 0.2343,
      "step": 200
    },
    {
      "step": 200,
      "train_flow_matching_loss": 0.212890625,
      "train_physics_loss": 0.0
    },
    {
      "step": 200,
      "train_flow_matching_loss": 0.158203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 200,
      "train_flow_matching_loss": 0.142578125,
      "train_physics_loss": 1.1330649852752686
    },
    {
      "step": 200,
      "train_flow_matching_loss": 0.1728515625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.20014578104019165,
      "learning_rate": 6.966666666666667e-06,
      "loss": 0.2375,
      "step": 210
    },
    {
      "step": 210,
      "train_flow_matching_loss": 0.2373046875,
      "train_physics_loss": 0.0
    },
    {
      "step": 210,
      "train_flow_matching_loss": 0.169921875,
      "train_physics_loss": 0.0
    },
    {
      "step": 210,
      "train_flow_matching_loss": 0.27734375,
      "train_physics_loss": 1.196747899055481
    },
    {
      "step": 210,
      "train_flow_matching_loss": 0.177734375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.14675188064575195,
      "learning_rate": 7.2999999999999996e-06,
      "loss": 0.2385,
      "step": 220
    },
    {
      "step": 220,
      "train_flow_matching_loss": 0.2177734375,
      "train_physics_loss": 0.0
    },
    {
      "step": 220,
      "train_flow_matching_loss": 0.1552734375,
      "train_physics_loss": 0.0
    },
    {
      "step": 220,
      "train_flow_matching_loss": 0.1953125,
      "train_physics_loss": 1.1038117408752441
    },
    {
      "step": 220,
      "train_flow_matching_loss": 0.177734375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.126113623380661,
      "learning_rate": 7.633333333333334e-06,
      "loss": 0.2282,
      "step": 230
    },
    {
      "step": 230,
      "train_flow_matching_loss": 0.291015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 230,
      "train_flow_matching_loss": 0.1640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 230,
      "train_flow_matching_loss": 0.263671875,
      "train_physics_loss": 1.051543951034546
    },
    {
      "step": 230,
      "train_flow_matching_loss": 0.2041015625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11941422522068024,
      "learning_rate": 7.966666666666666e-06,
      "loss": 0.2335,
      "step": 240
    },
    {
      "step": 240,
      "train_flow_matching_loss": 0.150390625,
      "train_physics_loss": 0.0
    },
    {
      "step": 240,
      "train_flow_matching_loss": 0.255859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 240,
      "train_flow_matching_loss": 0.236328125,
      "train_physics_loss": 1.2647684812545776
    },
    {
      "step": 240,
      "train_flow_matching_loss": 0.1591796875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.13348649442195892,
      "learning_rate": 8.3e-06,
      "loss": 0.231,
      "step": 250
    },
    {
      "step": 250,
      "train_flow_matching_loss": 0.15625,
      "train_physics_loss": 0.0
    },
    {
      "step": 250,
      "train_flow_matching_loss": 0.177734375,
      "train_physics_loss": 0.0
    },
    {
      "step": 250,
      "train_flow_matching_loss": 0.1357421875,
      "train_physics_loss": 1.12180757522583
    },
    {
      "step": 250,
      "train_flow_matching_loss": 0.263671875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09305528551340103,
      "learning_rate": 8.633333333333334e-06,
      "loss": 0.2393,
      "step": 260
    },
    {
      "step": 260,
      "train_flow_matching_loss": 0.2734375,
      "train_physics_loss": 0.0
    },
    {
      "step": 260,
      "train_flow_matching_loss": 0.1865234375,
      "train_physics_loss": 0.0
    },
    {
      "step": 260,
      "train_flow_matching_loss": 0.234375,
      "train_physics_loss": 0.0
    },
    {
      "step": 260,
      "train_flow_matching_loss": 0.1474609375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.1336001753807068,
      "learning_rate": 8.966666666666668e-06,
      "loss": 0.2354,
      "step": 270
    },
    {
      "step": 270,
      "train_flow_matching_loss": 0.189453125,
      "train_physics_loss": 0.0
    },
    {
      "step": 270,
      "train_flow_matching_loss": 0.181640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 270,
      "train_flow_matching_loss": 0.1787109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 270,
      "train_flow_matching_loss": 0.1982421875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09680461138486862,
      "learning_rate": 9.3e-06,
      "loss": 0.2323,
      "step": 280
    },
    {
      "step": 280,
      "train_flow_matching_loss": 0.1796875,
      "train_physics_loss": 0.0
    },
    {
      "step": 280,
      "train_flow_matching_loss": 0.26953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 280,
      "train_flow_matching_loss": 0.296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 280,
      "train_flow_matching_loss": 0.2294921875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11074908822774887,
      "learning_rate": 9.633333333333335e-06,
      "loss": 0.2377,
      "step": 290
    },
    {
      "step": 290,
      "train_flow_matching_loss": 0.169921875,
      "train_physics_loss": 0.0
    },
    {
      "step": 290,
      "train_flow_matching_loss": 0.1748046875,
      "train_physics_loss": 0.0
    },
    {
      "step": 290,
      "train_flow_matching_loss": 0.1962890625,
      "train_physics_loss": 0.0
    },
    {
      "step": 290,
      "train_flow_matching_loss": 0.25390625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.19325394928455353,
      "learning_rate": 9.966666666666667e-06,
      "loss": 0.2345,
      "step": 300
    },
    {
      "step": 300,
      "train_flow_matching_loss": 0.1396484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 300,
      "train_flow_matching_loss": 0.275390625,
      "train_physics_loss": 0.0
    },
    {
      "step": 300,
      "train_flow_matching_loss": 0.1201171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 300,
      "train_flow_matching_loss": 0.21875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.1510547399520874,
      "learning_rate": 1.03e-05,
      "loss": 0.2361,
      "step": 310
    },
    {
      "step": 310,
      "train_flow_matching_loss": 0.1611328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 310,
      "train_flow_matching_loss": 0.205078125,
      "train_physics_loss": 0.0
    },
    {
      "step": 310,
      "train_flow_matching_loss": 0.25390625,
      "train_physics_loss": 0.0
    },
    {
      "step": 310,
      "train_flow_matching_loss": 0.1923828125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.14867986738681793,
      "learning_rate": 1.0633333333333334e-05,
      "loss": 0.2333,
      "step": 320
    },
    {
      "step": 320,
      "train_flow_matching_loss": 0.2392578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 320,
      "train_flow_matching_loss": 0.189453125,
      "train_physics_loss": 0.0
    },
    {
      "step": 320,
      "train_flow_matching_loss": 0.232421875,
      "train_physics_loss": 0.0
    },
    {
      "step": 320,
      "train_flow_matching_loss": 0.1513671875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11669163405895233,
      "learning_rate": 1.0966666666666666e-05,
      "loss": 0.234,
      "step": 330
    },
    {
      "step": 330,
      "train_flow_matching_loss": 0.171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 330,
      "train_flow_matching_loss": 0.1767578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 330,
      "train_flow_matching_loss": 0.28125,
      "train_physics_loss": 0.0
    },
    {
      "step": 330,
      "train_flow_matching_loss": 0.2431640625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.15880121290683746,
      "learning_rate": 1.13e-05,
      "loss": 0.2382,
      "step": 340
    },
    {
      "step": 340,
      "train_flow_matching_loss": 0.21875,
      "train_physics_loss": 0.0
    },
    {
      "step": 340,
      "train_flow_matching_loss": 0.2158203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 340,
      "train_flow_matching_loss": 0.3125,
      "train_physics_loss": 0.0
    },
    {
      "step": 340,
      "train_flow_matching_loss": 0.2412109375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.15435479581356049,
      "learning_rate": 1.1633333333333334e-05,
      "loss": 0.2303,
      "step": 350
    },
    {
      "step": 350,
      "train_flow_matching_loss": 0.1845703125,
      "train_physics_loss": 0.0
    },
    {
      "step": 350,
      "train_flow_matching_loss": 0.25,
      "train_physics_loss": 0.0
    },
    {
      "step": 350,
      "train_flow_matching_loss": 0.28125,
      "train_physics_loss": 0.0
    },
    {
      "step": 350,
      "train_flow_matching_loss": 0.177734375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.18996568024158478,
      "learning_rate": 1.1966666666666668e-05,
      "loss": 0.2355,
      "step": 360
    },
    {
      "step": 360,
      "train_flow_matching_loss": 0.1943359375,
      "train_physics_loss": 0.0
    },
    {
      "step": 360,
      "train_flow_matching_loss": 0.2158203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 360,
      "train_flow_matching_loss": 0.189453125,
      "train_physics_loss": 0.0
    },
    {
      "step": 360,
      "train_flow_matching_loss": 0.2421875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.07093728333711624,
      "learning_rate": 1.23e-05,
      "loss": 0.2305,
      "step": 370
    },
    {
      "step": 370,
      "train_flow_matching_loss": 0.1376953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 370,
      "train_flow_matching_loss": 0.177734375,
      "train_physics_loss": 0.0
    },
    {
      "step": 370,
      "train_flow_matching_loss": 0.263671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 370,
      "train_flow_matching_loss": 0.1591796875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.12340039759874344,
      "learning_rate": 1.2633333333333333e-05,
      "loss": 0.2263,
      "step": 380
    },
    {
      "step": 380,
      "train_flow_matching_loss": 0.142578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 380,
      "train_flow_matching_loss": 0.2158203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 380,
      "train_flow_matching_loss": 0.2578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 380,
      "train_flow_matching_loss": 0.2099609375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.1154441088438034,
      "learning_rate": 1.2966666666666669e-05,
      "loss": 0.2339,
      "step": 390
    },
    {
      "step": 390,
      "train_flow_matching_loss": 0.12451171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 390,
      "train_flow_matching_loss": 0.1904296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 390,
      "train_flow_matching_loss": 0.267578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 390,
      "train_flow_matching_loss": 0.2431640625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.14932315051555634,
      "learning_rate": 1.3300000000000001e-05,
      "loss": 0.2334,
      "step": 400
    },
    {
      "step": 400,
      "train_flow_matching_loss": 0.1474609375,
      "train_physics_loss": 0.0
    },
    {
      "step": 400,
      "train_flow_matching_loss": 0.220703125,
      "train_physics_loss": 0.0
    },
    {
      "step": 400,
      "train_flow_matching_loss": 0.1533203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 400,
      "train_flow_matching_loss": 0.298828125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11459579318761826,
      "learning_rate": 1.3633333333333334e-05,
      "loss": 0.2316,
      "step": 410
    },
    {
      "step": 410,
      "train_flow_matching_loss": 0.1806640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 410,
      "train_flow_matching_loss": 0.1796875,
      "train_physics_loss": 0.0
    },
    {
      "step": 410,
      "train_flow_matching_loss": 0.1806640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 410,
      "train_flow_matching_loss": 0.3125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11602616310119629,
      "learning_rate": 1.3966666666666666e-05,
      "loss": 0.2256,
      "step": 420
    },
    {
      "step": 420,
      "train_flow_matching_loss": 0.201171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 420,
      "train_flow_matching_loss": 0.203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 420,
      "train_flow_matching_loss": 0.1669921875,
      "train_physics_loss": 0.0
    },
    {
      "step": 420,
      "train_flow_matching_loss": 0.1865234375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.16709451377391815,
      "learning_rate": 1.43e-05,
      "loss": 0.2338,
      "step": 430
    },
    {
      "step": 430,
      "train_flow_matching_loss": 0.2001953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 430,
      "train_flow_matching_loss": 0.158203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 430,
      "train_flow_matching_loss": 0.2099609375,
      "train_physics_loss": 0.0
    },
    {
      "step": 430,
      "train_flow_matching_loss": 0.2080078125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10652045905590057,
      "learning_rate": 1.4633333333333334e-05,
      "loss": 0.2239,
      "step": 440
    },
    {
      "step": 440,
      "train_flow_matching_loss": 0.1845703125,
      "train_physics_loss": 0.0
    },
    {
      "step": 440,
      "train_flow_matching_loss": 0.134765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 440,
      "train_flow_matching_loss": 0.1708984375,
      "train_physics_loss": 0.0
    },
    {
      "step": 440,
      "train_flow_matching_loss": 0.328125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09710021317005157,
      "learning_rate": 1.4966666666666668e-05,
      "loss": 0.2322,
      "step": 450
    },
    {
      "step": 450,
      "train_flow_matching_loss": 0.208984375,
      "train_physics_loss": 0.0
    },
    {
      "step": 450,
      "train_flow_matching_loss": 0.1416015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 450,
      "train_flow_matching_loss": 0.2333984375,
      "train_physics_loss": 0.0
    },
    {
      "step": 450,
      "train_flow_matching_loss": 0.1845703125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10346579551696777,
      "learning_rate": 1.53e-05,
      "loss": 0.2299,
      "step": 460
    },
    {
      "step": 460,
      "train_flow_matching_loss": 0.2041015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 460,
      "train_flow_matching_loss": 0.2197265625,
      "train_physics_loss": 0.0
    },
    {
      "step": 460,
      "train_flow_matching_loss": 0.294921875,
      "train_physics_loss": 0.0
    },
    {
      "step": 460,
      "train_flow_matching_loss": 0.220703125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.12442238628864288,
      "learning_rate": 1.563333333333333e-05,
      "loss": 0.2325,
      "step": 470
    },
    {
      "step": 470,
      "train_flow_matching_loss": 0.267578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 470,
      "train_flow_matching_loss": 0.173828125,
      "train_physics_loss": 0.0
    },
    {
      "step": 470,
      "train_flow_matching_loss": 0.1455078125,
      "train_physics_loss": 0.0
    },
    {
      "step": 470,
      "train_flow_matching_loss": 0.2333984375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08412368595600128,
      "learning_rate": 1.5966666666666667e-05,
      "loss": 0.2305,
      "step": 480
    },
    {
      "step": 480,
      "train_flow_matching_loss": 0.267578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 480,
      "train_flow_matching_loss": 0.2060546875,
      "train_physics_loss": 0.0
    },
    {
      "step": 480,
      "train_flow_matching_loss": 0.208984375,
      "train_physics_loss": 0.0
    },
    {
      "step": 480,
      "train_flow_matching_loss": 0.1337890625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.15383101999759674,
      "learning_rate": 1.63e-05,
      "loss": 0.2306,
      "step": 490
    },
    {
      "step": 490,
      "train_flow_matching_loss": 0.1875,
      "train_physics_loss": 0.0
    },
    {
      "step": 490,
      "train_flow_matching_loss": 0.1669921875,
      "train_physics_loss": 0.0
    },
    {
      "step": 490,
      "train_flow_matching_loss": 0.2236328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 490,
      "train_flow_matching_loss": 0.2275390625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.13231289386749268,
      "learning_rate": 1.6633333333333336e-05,
      "loss": 0.2327,
      "step": 500
    },
    {
      "step": 500,
      "train_flow_matching_loss": 0.216796875,
      "train_physics_loss": 0.0
    },
    {
      "step": 500,
      "train_flow_matching_loss": 0.1865234375,
      "train_physics_loss": 0.0
    },
    {
      "step": 500,
      "train_flow_matching_loss": 0.1796875,
      "train_physics_loss": 0.0
    },
    {
      "step": 500,
      "train_flow_matching_loss": 0.1845703125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.07903650403022766,
      "learning_rate": 1.6966666666666668e-05,
      "loss": 0.2399,
      "step": 510
    },
    {
      "step": 510,
      "train_flow_matching_loss": 0.255859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 510,
      "train_flow_matching_loss": 0.1357421875,
      "train_physics_loss": 0.0
    },
    {
      "step": 510,
      "train_flow_matching_loss": 0.2412109375,
      "train_physics_loss": 1.1193593740463257
    },
    {
      "step": 510,
      "train_flow_matching_loss": 0.2392578125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.1139911413192749,
      "learning_rate": 1.73e-05,
      "loss": 0.2257,
      "step": 520
    },
    {
      "step": 520,
      "train_flow_matching_loss": 0.1953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 520,
      "train_flow_matching_loss": 0.2119140625,
      "train_physics_loss": 0.0
    },
    {
      "step": 520,
      "train_flow_matching_loss": 0.193359375,
      "train_physics_loss": 1.126528024673462
    },
    {
      "step": 520,
      "train_flow_matching_loss": 0.267578125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.07190673053264618,
      "learning_rate": 1.7633333333333336e-05,
      "loss": 0.2333,
      "step": 530
    },
    {
      "step": 530,
      "train_flow_matching_loss": 0.2578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 530,
      "train_flow_matching_loss": 0.154296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 530,
      "train_flow_matching_loss": 0.30078125,
      "train_physics_loss": 1.3095051050186157
    },
    {
      "step": 530,
      "train_flow_matching_loss": 0.287109375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11326662451028824,
      "learning_rate": 1.796666666666667e-05,
      "loss": 0.229,
      "step": 540
    },
    {
      "step": 540,
      "train_flow_matching_loss": 0.2021484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 540,
      "train_flow_matching_loss": 0.2177734375,
      "train_physics_loss": 0.0
    },
    {
      "step": 540,
      "train_flow_matching_loss": 0.2021484375,
      "train_physics_loss": 1.1746515035629272
    },
    {
      "step": 540,
      "train_flow_matching_loss": 0.216796875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11712835729122162,
      "learning_rate": 1.83e-05,
      "loss": 0.2259,
      "step": 550
    },
    {
      "step": 550,
      "train_flow_matching_loss": 0.1201171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 550,
      "train_flow_matching_loss": 0.1689453125,
      "train_physics_loss": 0.0
    },
    {
      "step": 550,
      "train_flow_matching_loss": 0.109375,
      "train_physics_loss": 1.1187113523483276
    },
    {
      "step": 550,
      "train_flow_matching_loss": 0.30078125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.07239869982004166,
      "learning_rate": 1.8633333333333333e-05,
      "loss": 0.2271,
      "step": 560
    },
    {
      "step": 560,
      "train_flow_matching_loss": 0.1845703125,
      "train_physics_loss": 0.0
    },
    {
      "step": 560,
      "train_flow_matching_loss": 0.318359375,
      "train_physics_loss": 0.0
    },
    {
      "step": 560,
      "train_flow_matching_loss": 0.1318359375,
      "train_physics_loss": 1.022248387336731
    },
    {
      "step": 560,
      "train_flow_matching_loss": 0.166015625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10492479801177979,
      "learning_rate": 1.896666666666667e-05,
      "loss": 0.2252,
      "step": 570
    },
    {
      "step": 570,
      "train_flow_matching_loss": 0.26171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 570,
      "train_flow_matching_loss": 0.1708984375,
      "train_physics_loss": 0.0
    },
    {
      "step": 570,
      "train_flow_matching_loss": 0.240234375,
      "train_physics_loss": 1.1364072561264038
    },
    {
      "step": 570,
      "train_flow_matching_loss": 0.2197265625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11665148288011551,
      "learning_rate": 1.93e-05,
      "loss": 0.2276,
      "step": 580
    },
    {
      "step": 580,
      "train_flow_matching_loss": 0.1943359375,
      "train_physics_loss": 0.0
    },
    {
      "step": 580,
      "train_flow_matching_loss": 0.10009765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 580,
      "train_flow_matching_loss": 0.28515625,
      "train_physics_loss": 1.0542774200439453
    },
    {
      "step": 580,
      "train_flow_matching_loss": 0.255859375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09312278032302856,
      "learning_rate": 1.9633333333333334e-05,
      "loss": 0.2271,
      "step": 590
    },
    {
      "step": 590,
      "train_flow_matching_loss": 0.177734375,
      "train_physics_loss": 0.0
    },
    {
      "step": 590,
      "train_flow_matching_loss": 0.2392578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 590,
      "train_flow_matching_loss": 0.37109375,
      "train_physics_loss": 1.0916481018066406
    },
    {
      "step": 590,
      "train_flow_matching_loss": 0.166015625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.06942244619131088,
      "learning_rate": 1.9966666666666666e-05,
      "loss": 0.2265,
      "step": 600
    },
    {
      "step": 600,
      "train_flow_matching_loss": 0.26953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 600,
      "train_flow_matching_loss": 0.25,
      "train_physics_loss": 0.0
    },
    {
      "step": 600,
      "train_flow_matching_loss": 0.205078125,
      "train_physics_loss": 1.0734905004501343
    },
    {
      "step": 600,
      "train_flow_matching_loss": 0.205078125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10529624670743942,
      "learning_rate": 2.0300000000000002e-05,
      "loss": 0.2342,
      "step": 610
    },
    {
      "step": 610,
      "train_flow_matching_loss": 0.2236328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 610,
      "train_flow_matching_loss": 0.2099609375,
      "train_physics_loss": 0.0
    },
    {
      "step": 610,
      "train_flow_matching_loss": 0.306640625,
      "train_physics_loss": 1.0934414863586426
    },
    {
      "step": 610,
      "train_flow_matching_loss": 0.1611328125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.06641392409801483,
      "learning_rate": 2.0633333333333335e-05,
      "loss": 0.2278,
      "step": 620
    },
    {
      "step": 620,
      "train_flow_matching_loss": 0.2060546875,
      "train_physics_loss": 0.0
    },
    {
      "step": 620,
      "train_flow_matching_loss": 0.1259765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 620,
      "train_flow_matching_loss": 0.1494140625,
      "train_physics_loss": 1.0211650133132935
    },
    {
      "step": 620,
      "train_flow_matching_loss": 0.212890625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.12207586318254471,
      "learning_rate": 2.0966666666666667e-05,
      "loss": 0.2228,
      "step": 630
    },
    {
      "step": 630,
      "train_flow_matching_loss": 0.2041015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 630,
      "train_flow_matching_loss": 0.1689453125,
      "train_physics_loss": 0.0
    },
    {
      "step": 630,
      "train_flow_matching_loss": 0.2109375,
      "train_physics_loss": 1.1233752965927124
    },
    {
      "step": 630,
      "train_flow_matching_loss": 0.1845703125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.07163283228874207,
      "learning_rate": 2.13e-05,
      "loss": 0.2266,
      "step": 640
    },
    {
      "step": 640,
      "train_flow_matching_loss": 0.2099609375,
      "train_physics_loss": 0.0
    },
    {
      "step": 640,
      "train_flow_matching_loss": 0.181640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 640,
      "train_flow_matching_loss": 0.1943359375,
      "train_physics_loss": 1.070631742477417
    },
    {
      "step": 640,
      "train_flow_matching_loss": 0.2236328125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.06662950664758682,
      "learning_rate": 2.1633333333333332e-05,
      "loss": 0.2272,
      "step": 650
    },
    {
      "step": 650,
      "train_flow_matching_loss": 0.255859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 650,
      "train_flow_matching_loss": 0.1865234375,
      "train_physics_loss": 0.0
    },
    {
      "step": 650,
      "train_flow_matching_loss": 0.166015625,
      "train_physics_loss": 1.1911576986312866
    },
    {
      "step": 650,
      "train_flow_matching_loss": 0.15625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.07315566390752792,
      "learning_rate": 2.1966666666666668e-05,
      "loss": 0.2279,
      "step": 660
    },
    {
      "step": 660,
      "train_flow_matching_loss": 0.1943359375,
      "train_physics_loss": 0.0
    },
    {
      "step": 660,
      "train_flow_matching_loss": 0.1728515625,
      "train_physics_loss": 0.0
    },
    {
      "step": 660,
      "train_flow_matching_loss": 0.1787109375,
      "train_physics_loss": 1.0218944549560547
    },
    {
      "step": 660,
      "train_flow_matching_loss": 0.205078125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.07452845573425293,
      "learning_rate": 2.23e-05,
      "loss": 0.234,
      "step": 670
    },
    {
      "step": 670,
      "train_flow_matching_loss": 0.1591796875,
      "train_physics_loss": 0.0
    },
    {
      "step": 670,
      "train_flow_matching_loss": 0.1982421875,
      "train_physics_loss": 0.0
    },
    {
      "step": 670,
      "train_flow_matching_loss": 0.1962890625,
      "train_physics_loss": 1.0345114469528198
    },
    {
      "step": 670,
      "train_flow_matching_loss": 0.2138671875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.13680411875247955,
      "learning_rate": 2.2633333333333336e-05,
      "loss": 0.2268,
      "step": 680
    },
    {
      "step": 680,
      "train_flow_matching_loss": 0.1298828125,
      "train_physics_loss": 0.0
    },
    {
      "step": 680,
      "train_flow_matching_loss": 0.259765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 680,
      "train_flow_matching_loss": 0.1728515625,
      "train_physics_loss": 1.039867877960205
    },
    {
      "step": 680,
      "train_flow_matching_loss": 0.265625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.06050241366028786,
      "learning_rate": 2.2966666666666668e-05,
      "loss": 0.2264,
      "step": 690
    },
    {
      "step": 690,
      "train_flow_matching_loss": 0.1875,
      "train_physics_loss": 0.0
    },
    {
      "step": 690,
      "train_flow_matching_loss": 0.1708984375,
      "train_physics_loss": 0.0
    },
    {
      "step": 690,
      "train_flow_matching_loss": 0.1650390625,
      "train_physics_loss": 1.0541917085647583
    },
    {
      "step": 690,
      "train_flow_matching_loss": 0.10986328125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.23056082427501678,
      "learning_rate": 2.3300000000000004e-05,
      "loss": 0.2255,
      "step": 700
    },
    {
      "step": 700,
      "train_flow_matching_loss": 0.302734375,
      "train_physics_loss": 0.0
    },
    {
      "step": 700,
      "train_flow_matching_loss": 0.287109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 700,
      "train_flow_matching_loss": 0.1689453125,
      "train_physics_loss": 0.9928261041641235
    },
    {
      "step": 700,
      "train_flow_matching_loss": 0.177734375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.24343863129615784,
      "learning_rate": 2.3633333333333336e-05,
      "loss": 0.2276,
      "step": 710
    },
    {
      "step": 710,
      "train_flow_matching_loss": 0.2197265625,
      "train_physics_loss": 0.0
    },
    {
      "step": 710,
      "train_flow_matching_loss": 0.216796875,
      "train_physics_loss": 0.0
    },
    {
      "step": 710,
      "train_flow_matching_loss": 0.09912109375,
      "train_physics_loss": 1.0637205839157104
    },
    {
      "step": 710,
      "train_flow_matching_loss": 0.298828125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.0798896923661232,
      "learning_rate": 2.396666666666667e-05,
      "loss": 0.2235,
      "step": 720
    },
    {
      "step": 720,
      "train_flow_matching_loss": 0.2392578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 720,
      "train_flow_matching_loss": 0.2392578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 720,
      "train_flow_matching_loss": 0.173828125,
      "train_physics_loss": 1.0625635385513306
    },
    {
      "step": 720,
      "train_flow_matching_loss": 0.162109375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.1560012847185135,
      "learning_rate": 2.43e-05,
      "loss": 0.2255,
      "step": 730
    },
    {
      "step": 730,
      "train_flow_matching_loss": 0.216796875,
      "train_physics_loss": 0.0
    },
    {
      "step": 730,
      "train_flow_matching_loss": 0.1123046875,
      "train_physics_loss": 0.0
    },
    {
      "step": 730,
      "train_flow_matching_loss": 0.2353515625,
      "train_physics_loss": 0.9938982725143433
    },
    {
      "step": 730,
      "train_flow_matching_loss": 0.30859375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.0801643505692482,
      "learning_rate": 2.4633333333333334e-05,
      "loss": 0.2211,
      "step": 740
    },
    {
      "step": 740,
      "train_flow_matching_loss": 0.1474609375,
      "train_physics_loss": 0.0
    },
    {
      "step": 740,
      "train_flow_matching_loss": 0.2197265625,
      "train_physics_loss": 0.0
    },
    {
      "step": 740,
      "train_flow_matching_loss": 0.287109375,
      "train_physics_loss": 0.972010612487793
    },
    {
      "step": 740,
      "train_flow_matching_loss": 0.2041015625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.07676834613084793,
      "learning_rate": 2.496666666666667e-05,
      "loss": 0.221,
      "step": 750
    },
    {
      "step": 750,
      "train_flow_matching_loss": 0.1533203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 750,
      "train_flow_matching_loss": 0.154296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 750,
      "train_flow_matching_loss": 0.142578125,
      "train_physics_loss": 1.073029637336731
    },
    {
      "step": 750,
      "train_flow_matching_loss": 0.1845703125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.07197835296392441,
      "learning_rate": 2.5300000000000002e-05,
      "loss": 0.2243,
      "step": 760
    },
    {
      "step": 760,
      "train_flow_matching_loss": 0.1396484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 760,
      "train_flow_matching_loss": 0.244140625,
      "train_physics_loss": 0.0
    },
    {
      "step": 760,
      "train_flow_matching_loss": 0.2021484375,
      "train_physics_loss": 1.1910450458526611
    },
    {
      "step": 760,
      "train_flow_matching_loss": 0.2451171875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08525007963180542,
      "learning_rate": 2.5633333333333338e-05,
      "loss": 0.229,
      "step": 770
    },
    {
      "step": 770,
      "train_flow_matching_loss": 0.212890625,
      "train_physics_loss": 0.0
    },
    {
      "step": 770,
      "train_flow_matching_loss": 0.1904296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 770,
      "train_flow_matching_loss": 0.2353515625,
      "train_physics_loss": 1.0544323921203613
    },
    {
      "step": 770,
      "train_flow_matching_loss": 0.173828125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.13165563344955444,
      "learning_rate": 2.5966666666666667e-05,
      "loss": 0.2286,
      "step": 780
    },
    {
      "step": 780,
      "train_flow_matching_loss": 0.1767578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 780,
      "train_flow_matching_loss": 0.138671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 780,
      "train_flow_matching_loss": 0.1572265625,
      "train_physics_loss": 1.0932003259658813
    },
    {
      "step": 780,
      "train_flow_matching_loss": 0.2021484375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.16382718086242676,
      "learning_rate": 2.6300000000000002e-05,
      "loss": 0.2274,
      "step": 790
    },
    {
      "step": 790,
      "train_flow_matching_loss": 0.1826171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 790,
      "train_flow_matching_loss": 0.140625,
      "train_physics_loss": 0.0
    },
    {
      "step": 790,
      "train_flow_matching_loss": 0.279296875,
      "train_physics_loss": 1.1805977821350098
    },
    {
      "step": 790,
      "train_flow_matching_loss": 0.1611328125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.0747007206082344,
      "learning_rate": 2.663333333333333e-05,
      "loss": 0.2257,
      "step": 800
    },
    {
      "step": 800,
      "train_flow_matching_loss": 0.162109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 800,
      "train_flow_matching_loss": 0.314453125,
      "train_physics_loss": 0.0
    },
    {
      "step": 800,
      "train_flow_matching_loss": 0.181640625,
      "train_physics_loss": 0.9864815473556519
    },
    {
      "step": 800,
      "train_flow_matching_loss": 0.1943359375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08578569442033768,
      "learning_rate": 2.6966666666666667e-05,
      "loss": 0.2288,
      "step": 810
    },
    {
      "step": 810,
      "train_flow_matching_loss": 0.2099609375,
      "train_physics_loss": 0.0
    },
    {
      "step": 810,
      "train_flow_matching_loss": 0.27734375,
      "train_physics_loss": 0.0
    },
    {
      "step": 810,
      "train_flow_matching_loss": 0.203125,
      "train_physics_loss": 1.0710862874984741
    },
    {
      "step": 810,
      "train_flow_matching_loss": 0.2275390625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.04827818274497986,
      "learning_rate": 2.7300000000000003e-05,
      "loss": 0.2233,
      "step": 820
    },
    {
      "step": 820,
      "train_flow_matching_loss": 0.267578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 820,
      "train_flow_matching_loss": 0.1357421875,
      "train_physics_loss": 0.0
    },
    {
      "step": 820,
      "train_flow_matching_loss": 0.2490234375,
      "train_physics_loss": 1.0107275247573853
    },
    {
      "step": 820,
      "train_flow_matching_loss": 0.2353515625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.05919807404279709,
      "learning_rate": 2.7633333333333332e-05,
      "loss": 0.2264,
      "step": 830
    },
    {
      "step": 830,
      "train_flow_matching_loss": 0.134765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 830,
      "train_flow_matching_loss": 0.134765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 830,
      "train_flow_matching_loss": 0.2177734375,
      "train_physics_loss": 1.0691488981246948
    },
    {
      "step": 830,
      "train_flow_matching_loss": 0.1484375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.1365600824356079,
      "learning_rate": 2.7966666666666668e-05,
      "loss": 0.2212,
      "step": 840
    },
    {
      "step": 840,
      "train_flow_matching_loss": 0.1650390625,
      "train_physics_loss": 0.0
    },
    {
      "step": 840,
      "train_flow_matching_loss": 0.158203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 840,
      "train_flow_matching_loss": 0.1884765625,
      "train_physics_loss": 0.9634192585945129
    },
    {
      "step": 840,
      "train_flow_matching_loss": 0.263671875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10558287054300308,
      "learning_rate": 2.83e-05,
      "loss": 0.2308,
      "step": 850
    },
    {
      "step": 850,
      "train_flow_matching_loss": 0.1494140625,
      "train_physics_loss": 0.0
    },
    {
      "step": 850,
      "train_flow_matching_loss": 0.201171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 850,
      "train_flow_matching_loss": 0.1953125,
      "train_physics_loss": 1.0432966947555542
    },
    {
      "step": 850,
      "train_flow_matching_loss": 0.25,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.06894788891077042,
      "learning_rate": 2.8633333333333336e-05,
      "loss": 0.2279,
      "step": 860
    },
    {
      "step": 860,
      "train_flow_matching_loss": 0.189453125,
      "train_physics_loss": 0.0
    },
    {
      "step": 860,
      "train_flow_matching_loss": 0.2578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 860,
      "train_flow_matching_loss": 0.267578125,
      "train_physics_loss": 1.0717509984970093
    },
    {
      "step": 860,
      "train_flow_matching_loss": 0.1416015625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11610148102045059,
      "learning_rate": 2.8966666666666668e-05,
      "loss": 0.2202,
      "step": 870
    },
    {
      "step": 870,
      "train_flow_matching_loss": 0.154296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 870,
      "train_flow_matching_loss": 0.138671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 870,
      "train_flow_matching_loss": 0.126953125,
      "train_physics_loss": 1.050184726715088
    },
    {
      "step": 870,
      "train_flow_matching_loss": 0.1572265625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11412333697080612,
      "learning_rate": 2.93e-05,
      "loss": 0.2247,
      "step": 880
    },
    {
      "step": 880,
      "train_flow_matching_loss": 0.220703125,
      "train_physics_loss": 0.0
    },
    {
      "step": 880,
      "train_flow_matching_loss": 0.09814453125,
      "train_physics_loss": 0.0
    },
    {
      "step": 880,
      "train_flow_matching_loss": 0.337890625,
      "train_physics_loss": 1.1567102670669556
    },
    {
      "step": 880,
      "train_flow_matching_loss": 0.1650390625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10204999893903732,
      "learning_rate": 2.9633333333333336e-05,
      "loss": 0.223,
      "step": 890
    },
    {
      "step": 890,
      "train_flow_matching_loss": 0.244140625,
      "train_physics_loss": 0.0
    },
    {
      "step": 890,
      "train_flow_matching_loss": 0.1533203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 890,
      "train_flow_matching_loss": 0.279296875,
      "train_physics_loss": 1.1194419860839844
    },
    {
      "step": 890,
      "train_flow_matching_loss": 0.150390625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.05254870653152466,
      "learning_rate": 2.9966666666666672e-05,
      "loss": 0.2259,
      "step": 900
    },
    {
      "step": 900,
      "train_flow_matching_loss": 0.142578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 900,
      "train_flow_matching_loss": 0.166015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 900,
      "train_flow_matching_loss": 0.255859375,
      "train_physics_loss": 1.1503714323043823
    },
    {
      "step": 900,
      "train_flow_matching_loss": 0.263671875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.12015681713819504,
      "learning_rate": 3.03e-05,
      "loss": 0.2223,
      "step": 910
    },
    {
      "step": 910,
      "train_flow_matching_loss": 0.1416015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 910,
      "train_flow_matching_loss": 0.1875,
      "train_physics_loss": 0.0
    },
    {
      "step": 910,
      "train_flow_matching_loss": 0.447265625,
      "train_physics_loss": 0.9776902198791504
    },
    {
      "step": 910,
      "train_flow_matching_loss": 0.1279296875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.04058997333049774,
      "learning_rate": 3.063333333333334e-05,
      "loss": 0.2219,
      "step": 920
    },
    {
      "step": 920,
      "train_flow_matching_loss": 0.140625,
      "train_physics_loss": 0.0
    },
    {
      "step": 920,
      "train_flow_matching_loss": 0.2109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 920,
      "train_flow_matching_loss": 0.1826171875,
      "train_physics_loss": 0.9938367009162903
    },
    {
      "step": 920,
      "train_flow_matching_loss": 0.138671875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.07445070892572403,
      "learning_rate": 3.096666666666666e-05,
      "loss": 0.224,
      "step": 930
    },
    {
      "step": 930,
      "train_flow_matching_loss": 0.2041015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 930,
      "train_flow_matching_loss": 0.2177734375,
      "train_physics_loss": 0.0
    },
    {
      "step": 930,
      "train_flow_matching_loss": 0.2041015625,
      "train_physics_loss": 1.004056453704834
    },
    {
      "step": 930,
      "train_flow_matching_loss": 0.11865234375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11243508011102676,
      "learning_rate": 3.13e-05,
      "loss": 0.2254,
      "step": 940
    },
    {
      "step": 940,
      "train_flow_matching_loss": 0.166015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 940,
      "train_flow_matching_loss": 0.1044921875,
      "train_physics_loss": 0.0
    },
    {
      "step": 940,
      "train_flow_matching_loss": 0.384765625,
      "train_physics_loss": 1.1279563903808594
    },
    {
      "step": 940,
      "train_flow_matching_loss": 0.185546875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.13521738350391388,
      "learning_rate": 3.1633333333333334e-05,
      "loss": 0.2295,
      "step": 950
    },
    {
      "step": 950,
      "train_flow_matching_loss": 0.12060546875,
      "train_physics_loss": 0.0
    },
    {
      "step": 950,
      "train_flow_matching_loss": 0.259765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 950,
      "train_flow_matching_loss": 0.1767578125,
      "train_physics_loss": 1.2156647443771362
    },
    {
      "step": 950,
      "train_flow_matching_loss": 0.181640625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.05210377648472786,
      "learning_rate": 3.196666666666667e-05,
      "loss": 0.2178,
      "step": 960
    },
    {
      "step": 960,
      "train_flow_matching_loss": 0.1416015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 960,
      "train_flow_matching_loss": 0.2197265625,
      "train_physics_loss": 0.0
    },
    {
      "step": 960,
      "train_flow_matching_loss": 0.30078125,
      "train_physics_loss": 1.0666940212249756
    },
    {
      "step": 960,
      "train_flow_matching_loss": 0.1640625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.05845818668603897,
      "learning_rate": 3.2300000000000006e-05,
      "loss": 0.2176,
      "step": 970
    },
    {
      "step": 970,
      "train_flow_matching_loss": 0.2158203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 970,
      "train_flow_matching_loss": 0.12353515625,
      "train_physics_loss": 0.0
    },
    {
      "step": 970,
      "train_flow_matching_loss": 0.287109375,
      "train_physics_loss": 1.0803557634353638
    },
    {
      "step": 970,
      "train_flow_matching_loss": 0.234375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.12706217169761658,
      "learning_rate": 3.263333333333333e-05,
      "loss": 0.2216,
      "step": 980
    },
    {
      "step": 980,
      "train_flow_matching_loss": 0.2490234375,
      "train_physics_loss": 0.0
    },
    {
      "step": 980,
      "train_flow_matching_loss": 0.0751953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 980,
      "train_flow_matching_loss": 0.1044921875,
      "train_physics_loss": 1.0513802766799927
    },
    {
      "step": 980,
      "train_flow_matching_loss": 0.2001953125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.045442089438438416,
      "learning_rate": 3.296666666666667e-05,
      "loss": 0.2236,
      "step": 990
    },
    {
      "step": 990,
      "train_flow_matching_loss": 0.1298828125,
      "train_physics_loss": 0.0
    },
    {
      "step": 990,
      "train_flow_matching_loss": 0.205078125,
      "train_physics_loss": 0.0
    },
    {
      "step": 990,
      "train_flow_matching_loss": 0.294921875,
      "train_physics_loss": 1.0516427755355835
    },
    {
      "step": 990,
      "train_flow_matching_loss": 0.267578125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.12765483558177948,
      "learning_rate": 3.33e-05,
      "loss": 0.2217,
      "step": 1000
    },
    {
      "step": 1000,
      "train_flow_matching_loss": 0.1533203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 1000,
      "train_flow_matching_loss": 0.1689453125,
      "train_physics_loss": 0.0
    },
    {
      "step": 1000,
      "train_flow_matching_loss": 0.234375,
      "train_physics_loss": 1.0302354097366333
    },
    {
      "step": 1000,
      "train_flow_matching_loss": 0.1484375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.06382420659065247,
      "learning_rate": 3.3633333333333335e-05,
      "loss": 0.217,
      "step": 1010
    },
    {
      "step": 1010,
      "train_flow_matching_loss": 0.177734375,
      "train_physics_loss": 0.0
    },
    {
      "step": 1010,
      "train_flow_matching_loss": 0.173828125,
      "train_physics_loss": 0.0
    },
    {
      "step": 1010,
      "train_flow_matching_loss": 0.169921875,
      "train_physics_loss": 1.0554667711257935
    },
    {
      "step": 1010,
      "train_flow_matching_loss": 0.2216796875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.12127324193716049,
      "learning_rate": 3.396666666666667e-05,
      "loss": 0.2141,
      "step": 1020
    },
    {
      "step": 1020,
      "train_flow_matching_loss": 0.1826171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 1020,
      "train_flow_matching_loss": 0.2080078125,
      "train_physics_loss": 0.0
    },
    {
      "step": 1020,
      "train_flow_matching_loss": 0.1669921875,
      "train_physics_loss": 1.0674573183059692
    },
    {
      "step": 1020,
      "train_flow_matching_loss": 0.1494140625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.045330170542001724,
      "learning_rate": 3.430000000000001e-05,
      "loss": 0.2168,
      "step": 1030
    },
    {
      "step": 1030,
      "train_flow_matching_loss": 0.251953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 1030,
      "train_flow_matching_loss": 0.21875,
      "train_physics_loss": 0.0
    },
    {
      "step": 1030,
      "train_flow_matching_loss": 0.26953125,
      "train_physics_loss": 1.0853160619735718
    },
    {
      "step": 1030,
      "train_flow_matching_loss": 0.240234375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08531888574361801,
      "learning_rate": 3.463333333333333e-05,
      "loss": 0.2142,
      "step": 1040
    },
    {
      "step": 1040,
      "train_flow_matching_loss": 0.2021484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 1040,
      "train_flow_matching_loss": 0.1630859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 1040,
      "train_flow_matching_loss": 0.185546875,
      "train_physics_loss": 1.0815349817276
    },
    {
      "step": 1040,
      "train_flow_matching_loss": 0.146484375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10388952493667603,
      "learning_rate": 3.496666666666667e-05,
      "loss": 0.2168,
      "step": 1050
    },
    {
      "step": 1050,
      "train_flow_matching_loss": 0.1435546875,
      "train_physics_loss": 0.0
    },
    {
      "step": 1050,
      "train_flow_matching_loss": 0.1259765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 1050,
      "train_flow_matching_loss": 0.2119140625,
      "train_physics_loss": 1.0548597574234009
    },
    {
      "step": 1050,
      "train_flow_matching_loss": 0.1953125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.07275362312793732,
      "learning_rate": 3.53e-05,
      "loss": 0.2181,
      "step": 1060
    },
    {
      "step": 1060,
      "train_flow_matching_loss": 0.10693359375,
      "train_physics_loss": 0.0
    },
    {
      "step": 1060,
      "train_flow_matching_loss": 0.1884765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 1060,
      "train_flow_matching_loss": 0.29296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 1060,
      "train_flow_matching_loss": 0.216796875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.05583971366286278,
      "learning_rate": 3.563333333333334e-05,
      "loss": 0.221,
      "step": 1070
    },
    {
      "step": 1070,
      "train_flow_matching_loss": 0.203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 1070,
      "train_flow_matching_loss": 0.236328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 1070,
      "train_flow_matching_loss": 0.1435546875,
      "train_physics_loss": 1.1803914308547974
    },
    {
      "step": 1070,
      "train_flow_matching_loss": 0.201171875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.45944881439208984,
      "learning_rate": 3.596666666666667e-05,
      "loss": 0.2229,
      "step": 1080
    },
    {
      "step": 1080,
      "train_flow_matching_loss": 0.1591796875,
      "train_physics_loss": 0.0
    },
    {
      "step": 1080,
      "train_flow_matching_loss": 0.1484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 1080,
      "train_flow_matching_loss": 0.2255859375,
      "train_physics_loss": 1.018959879875183
    },
    {
      "step": 1080,
      "train_flow_matching_loss": 0.2265625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.07284148782491684,
      "learning_rate": 3.63e-05,
      "loss": 0.2112,
      "step": 1090
    },
    {
      "step": 1090,
      "train_flow_matching_loss": 0.2099609375,
      "train_physics_loss": 0.0
    },
    {
      "step": 1090,
      "train_flow_matching_loss": 0.13671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 1090,
      "train_flow_matching_loss": 0.1669921875,
      "train_physics_loss": 1.0806039571762085
    },
    {
      "step": 1090,
      "train_flow_matching_loss": 0.171875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09040670841932297,
      "learning_rate": 3.6633333333333334e-05,
      "loss": 0.2158,
      "step": 1100
    },
    {
      "step": 1100,
      "train_flow_matching_loss": 0.2119140625,
      "train_physics_loss": 0.0
    },
    {
      "step": 1100,
      "train_flow_matching_loss": 0.173828125,
      "train_physics_loss": 0.0
    },
    {
      "step": 1100,
      "train_flow_matching_loss": 0.185546875,
      "train_physics_loss": 0.0
    },
    {
      "step": 1100,
      "train_flow_matching_loss": 0.1884765625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.16789332032203674,
      "learning_rate": 3.6966666666666666e-05,
      "loss": 0.2142,
      "step": 1110
    },
    {
      "step": 1110,
      "train_flow_matching_loss": 0.1435546875,
      "train_physics_loss": 0.0
    },
    {
      "step": 1110,
      "train_flow_matching_loss": 0.12451171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 1110,
      "train_flow_matching_loss": 0.189453125,
      "train_physics_loss": 0.9871474504470825
    },
    {
      "step": 1110,
      "train_flow_matching_loss": 0.1953125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.0856097862124443,
      "learning_rate": 3.73e-05,
      "loss": 0.2163,
      "step": 1120
    },
    {
      "step": 1120,
      "train_flow_matching_loss": 0.15625,
      "train_physics_loss": 0.0
    },
    {
      "step": 1120,
      "train_flow_matching_loss": 0.25390625,
      "train_physics_loss": 0.0
    },
    {
      "step": 1120,
      "train_flow_matching_loss": 0.2001953125,
      "train_physics_loss": 1.0331249237060547
    },
    {
      "step": 1120,
      "train_flow_matching_loss": 0.1630859375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.07229320704936981,
      "learning_rate": 3.763333333333334e-05,
      "loss": 0.2203,
      "step": 1130
    },
    {
      "step": 1130,
      "train_flow_matching_loss": 0.1611328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 1130,
      "train_flow_matching_loss": 0.1923828125,
      "train_physics_loss": 0.0
    },
    {
      "step": 1130,
      "train_flow_matching_loss": 0.2451171875,
      "train_physics_loss": 1.0513014793395996
    },
    {
      "step": 1130,
      "train_flow_matching_loss": 0.1962890625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.0732717216014862,
      "learning_rate": 3.796666666666667e-05,
      "loss": 0.214,
      "step": 1140
    },
    {
      "step": 1140,
      "train_flow_matching_loss": 0.1611328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 1140,
      "train_flow_matching_loss": 0.166015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 1140,
      "train_flow_matching_loss": 0.216796875,
      "train_physics_loss": 1.2523974180221558
    },
    {
      "step": 1140,
      "train_flow_matching_loss": 0.2294921875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.14727656543254852,
      "learning_rate": 3.83e-05,
      "loss": 0.2141,
      "step": 1150
    },
    {
      "step": 1150,
      "train_flow_matching_loss": 0.154296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 1150,
      "train_flow_matching_loss": 0.1044921875,
      "train_physics_loss": 0.0
    },
    {
      "step": 1150,
      "train_flow_matching_loss": 0.1953125,
      "train_physics_loss": 1.1823188066482544
    },
    {
      "step": 1150,
      "train_flow_matching_loss": 0.1279296875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.05762432515621185,
      "learning_rate": 3.8633333333333335e-05,
      "loss": 0.2126,
      "step": 1160
    },
    {
      "step": 1160,
      "train_flow_matching_loss": 0.12255859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 1160,
      "train_flow_matching_loss": 0.171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 1160,
      "train_flow_matching_loss": 0.1875,
      "train_physics_loss": 1.3618494272232056
    },
    {
      "step": 1160,
      "train_flow_matching_loss": 0.142578125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.07968246936798096,
      "learning_rate": 3.896666666666667e-05,
      "loss": 0.2061,
      "step": 1170
    },
    {
      "step": 1170,
      "train_flow_matching_loss": 0.1455078125,
      "train_physics_loss": 0.0
    },
    {
      "step": 1170,
      "train_flow_matching_loss": 0.162109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 1170,
      "train_flow_matching_loss": 0.1015625,
      "train_physics_loss": 1.064009666442871
    },
    {
      "step": 1170,
      "train_flow_matching_loss": 0.2236328125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.4024865925312042,
      "learning_rate": 3.9300000000000007e-05,
      "loss": 0.2114,
      "step": 1180
    },
    {
      "step": 1180,
      "train_flow_matching_loss": 0.154296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 1180,
      "train_flow_matching_loss": 0.1650390625,
      "train_physics_loss": 0.0
    },
    {
      "step": 1180,
      "train_flow_matching_loss": 0.1728515625,
      "train_physics_loss": 1.0951364040374756
    },
    {
      "step": 1180,
      "train_flow_matching_loss": 0.1796875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.28760161995887756,
      "learning_rate": 3.963333333333333e-05,
      "loss": 0.2156,
      "step": 1190
    },
    {
      "step": 1190,
      "train_flow_matching_loss": 0.197265625,
      "train_physics_loss": 0.0
    },
    {
      "step": 1190,
      "train_flow_matching_loss": 0.205078125,
      "train_physics_loss": 0.0
    },
    {
      "step": 1190,
      "train_flow_matching_loss": 0.212890625,
      "train_physics_loss": 1.1147454977035522
    },
    {
      "step": 1190,
      "train_flow_matching_loss": 0.1845703125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.12427179515361786,
      "learning_rate": 3.996666666666667e-05,
      "loss": 0.2228,
      "step": 1200
    },
    {
      "step": 1200,
      "train_flow_matching_loss": 0.1474609375,
      "train_physics_loss": 0.0
    },
    {
      "step": 1200,
      "train_flow_matching_loss": 0.27734375,
      "train_physics_loss": 0.0
    },
    {
      "step": 1200,
      "train_flow_matching_loss": 0.328125,
      "train_physics_loss": 1.0422614812850952
    },
    {
      "step": 1200,
      "train_flow_matching_loss": 0.228515625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.0794922262430191,
      "learning_rate": 4.0300000000000004e-05,
      "loss": 0.2146,
      "step": 1210
    },
    {
      "step": 1210,
      "train_flow_matching_loss": 0.1953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 1210,
      "train_flow_matching_loss": 0.1650390625,
      "train_physics_loss": 0.0
    },
    {
      "step": 1210,
      "train_flow_matching_loss": 0.21875,
      "train_physics_loss": 1.156057596206665
    },
    {
      "step": 1210,
      "train_flow_matching_loss": 0.1669921875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.05675072595477104,
      "learning_rate": 4.0633333333333336e-05,
      "loss": 0.2073,
      "step": 1220
    },
    {
      "step": 1220,
      "train_flow_matching_loss": 0.1640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 1220,
      "train_flow_matching_loss": 0.1611328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 1220,
      "train_flow_matching_loss": 0.341796875,
      "train_physics_loss": 1.0398201942443848
    },
    {
      "step": 1220,
      "train_flow_matching_loss": 0.255859375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.12678121030330658,
      "learning_rate": 4.096666666666667e-05,
      "loss": 0.2129,
      "step": 1230
    },
    {
      "step": 1230,
      "train_flow_matching_loss": 0.1611328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 1230,
      "train_flow_matching_loss": 0.1943359375,
      "train_physics_loss": 0.0
    },
    {
      "step": 1230,
      "train_flow_matching_loss": 0.1923828125,
      "train_physics_loss": 1.0318430662155151
    },
    {
      "step": 1230,
      "train_flow_matching_loss": 0.1474609375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.0983707532286644,
      "learning_rate": 4.13e-05,
      "loss": 0.2109,
      "step": 1240
    },
    {
      "step": 1240,
      "train_flow_matching_loss": 0.251953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 1240,
      "train_flow_matching_loss": 0.255859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 1240,
      "train_flow_matching_loss": 0.2431640625,
      "train_physics_loss": 1.037642240524292
    },
    {
      "step": 1240,
      "train_flow_matching_loss": 0.1357421875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.222572922706604,
      "learning_rate": 4.1633333333333333e-05,
      "loss": 0.2139,
      "step": 1250
    },
    {
      "step": 1250,
      "train_flow_matching_loss": 0.1298828125,
      "train_physics_loss": 0.0
    },
    {
      "step": 1250,
      "train_flow_matching_loss": 0.208984375,
      "train_physics_loss": 0.0
    },
    {
      "step": 1250,
      "train_flow_matching_loss": 0.287109375,
      "train_physics_loss": 0.9962090253829956
    },
    {
      "step": 1250,
      "train_flow_matching_loss": 0.267578125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.4652150571346283,
      "learning_rate": 4.196666666666667e-05,
      "loss": 0.2129,
      "step": 1260
    },
    {
      "step": 1260,
      "train_flow_matching_loss": 0.2109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 1260,
      "train_flow_matching_loss": 0.1826171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 1260,
      "train_flow_matching_loss": 0.2158203125,
      "train_physics_loss": 1.0768146514892578
    },
    {
      "step": 1260,
      "train_flow_matching_loss": 0.185546875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.07618173211812973,
      "learning_rate": 4.23e-05,
      "loss": 0.2138,
      "step": 1270
    },
    {
      "step": 1270,
      "train_flow_matching_loss": 0.1552734375,
      "train_physics_loss": 0.0
    },
    {
      "step": 1270,
      "train_flow_matching_loss": 0.2041015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 1270,
      "train_flow_matching_loss": 0.1748046875,
      "train_physics_loss": 1.0029113292694092
    },
    {
      "step": 1270,
      "train_flow_matching_loss": 0.1748046875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.13008639216423035,
      "learning_rate": 4.263333333333334e-05,
      "loss": 0.219,
      "step": 1280
    },
    {
      "step": 1280,
      "train_flow_matching_loss": 0.2177734375,
      "train_physics_loss": 0.0
    },
    {
      "step": 1280,
      "train_flow_matching_loss": 0.1953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 1280,
      "train_flow_matching_loss": 0.185546875,
      "train_physics_loss": 0.0
    },
    {
      "step": 1280,
      "train_flow_matching_loss": 0.2431640625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.12124402076005936,
      "learning_rate": 4.296666666666666e-05,
      "loss": 0.2137,
      "step": 1290
    },
    {
      "step": 1290,
      "train_flow_matching_loss": 0.259765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 1290,
      "train_flow_matching_loss": 0.1455078125,
      "train_physics_loss": 0.0
    },
    {
      "step": 1290,
      "train_flow_matching_loss": 0.1884765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 1290,
      "train_flow_matching_loss": 0.177734375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.1783202886581421,
      "learning_rate": 4.33e-05,
      "loss": 0.21,
      "step": 1300
    },
    {
      "step": 1300,
      "train_flow_matching_loss": 0.173828125,
      "train_physics_loss": 0.0
    },
    {
      "step": 1300,
      "train_flow_matching_loss": 0.1865234375,
      "train_physics_loss": 0.0
    },
    {
      "step": 1300,
      "train_flow_matching_loss": 0.2001953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 1300,
      "train_flow_matching_loss": 0.220703125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.37712377309799194,
      "learning_rate": 4.3633333333333335e-05,
      "loss": 0.2128,
      "step": 1310
    },
    {
      "step": 1310,
      "train_flow_matching_loss": 0.212890625,
      "train_physics_loss": 0.0
    },
    {
      "step": 1310,
      "train_flow_matching_loss": 0.1962890625,
      "train_physics_loss": 0.0
    },
    {
      "step": 1310,
      "train_flow_matching_loss": 0.1982421875,
      "train_physics_loss": 0.0
    },
    {
      "step": 1310,
      "train_flow_matching_loss": 0.2236328125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.0910569578409195,
      "learning_rate": 4.396666666666667e-05,
      "loss": 0.2137,
      "step": 1320
    },
    {
      "step": 1320,
      "train_flow_matching_loss": 0.166015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 1320,
      "train_flow_matching_loss": 0.224609375,
      "train_physics_loss": 0.0
    },
    {
      "step": 1320,
      "train_flow_matching_loss": 0.1689453125,
      "train_physics_loss": 0.0
    },
    {
      "step": 1320,
      "train_flow_matching_loss": 0.251953125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09554701298475266,
      "learning_rate": 4.43e-05,
      "loss": 0.2064,
      "step": 1330
    },
    {
      "step": 1330,
      "train_flow_matching_loss": 0.1796875,
      "train_physics_loss": 0.0
    },
    {
      "step": 1330,
      "train_flow_matching_loss": 0.1748046875,
      "train_physics_loss": 0.0
    },
    {
      "step": 1330,
      "train_flow_matching_loss": 0.1865234375,
      "train_physics_loss": 0.0
    },
    {
      "step": 1330,
      "train_flow_matching_loss": 0.2255859375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.4159400463104248,
      "learning_rate": 4.463333333333334e-05,
      "loss": 0.2097,
      "step": 1340
    },
    {
      "step": 1340,
      "train_flow_matching_loss": 0.220703125,
      "train_physics_loss": 0.0
    },
    {
      "step": 1340,
      "train_flow_matching_loss": 0.181640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 1340,
      "train_flow_matching_loss": 0.2109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 1340,
      "train_flow_matching_loss": 0.205078125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.17200617492198944,
      "learning_rate": 4.496666666666667e-05,
      "loss": 0.2112,
      "step": 1350
    },
    {
      "step": 1350,
      "train_flow_matching_loss": 0.1728515625,
      "train_physics_loss": 0.0
    },
    {
      "step": 1350,
      "train_flow_matching_loss": 0.1611328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 1350,
      "train_flow_matching_loss": 0.1787109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 1350,
      "train_flow_matching_loss": 0.1875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11631922423839569,
      "learning_rate": 4.53e-05,
      "loss": 0.2124,
      "step": 1360
    },
    {
      "step": 1360,
      "train_flow_matching_loss": 0.2119140625,
      "train_physics_loss": 0.0
    },
    {
      "step": 1360,
      "train_flow_matching_loss": 0.181640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 1360,
      "train_flow_matching_loss": 0.2255859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 1360,
      "train_flow_matching_loss": 0.189453125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.6013944149017334,
      "learning_rate": 4.5633333333333336e-05,
      "loss": 0.2069,
      "step": 1370
    },
    {
      "step": 1370,
      "train_flow_matching_loss": 0.1708984375,
      "train_physics_loss": 0.0
    },
    {
      "step": 1370,
      "train_flow_matching_loss": 0.2490234375,
      "train_physics_loss": 0.0
    },
    {
      "step": 1370,
      "train_flow_matching_loss": 0.212890625,
      "train_physics_loss": 0.0
    },
    {
      "step": 1370,
      "train_flow_matching_loss": 0.193359375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.6487261056900024,
      "learning_rate": 4.596666666666667e-05,
      "loss": 0.2113,
      "step": 1380
    },
    {
      "step": 1380,
      "train_flow_matching_loss": 0.1708984375,
      "train_physics_loss": 0.0
    },
    {
      "step": 1380,
      "train_flow_matching_loss": 0.1728515625,
      "train_physics_loss": 0.0
    },
    {
      "step": 1380,
      "train_flow_matching_loss": 0.232421875,
      "train_physics_loss": 0.0
    },
    {
      "step": 1380,
      "train_flow_matching_loss": 0.279296875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.2988802194595337,
      "learning_rate": 4.630000000000001e-05,
      "loss": 0.2054,
      "step": 1390
    },
    {
      "step": 1390,
      "train_flow_matching_loss": 0.15625,
      "train_physics_loss": 0.0
    },
    {
      "step": 1390,
      "train_flow_matching_loss": 0.2392578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 1390,
      "train_flow_matching_loss": 0.2041015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 1390,
      "train_flow_matching_loss": 0.1748046875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.20543885231018066,
      "learning_rate": 4.663333333333333e-05,
      "loss": 0.2155,
      "step": 1400
    },
    {
      "step": 1400,
      "train_flow_matching_loss": 0.2255859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 1400,
      "train_flow_matching_loss": 0.2333984375,
      "train_physics_loss": 0.0
    },
    {
      "step": 1400,
      "train_flow_matching_loss": 0.201171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 1400,
      "train_flow_matching_loss": 0.2197265625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.24119925498962402,
      "learning_rate": 4.696666666666667e-05,
      "loss": 0.208,
      "step": 1410
    },
    {
      "step": 1410,
      "train_flow_matching_loss": 0.130859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 1410,
      "train_flow_matching_loss": 0.1962890625,
      "train_physics_loss": 0.0
    },
    {
      "step": 1410,
      "train_flow_matching_loss": 0.326171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 1410,
      "train_flow_matching_loss": 0.12353515625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.2632496654987335,
      "learning_rate": 4.73e-05,
      "loss": 0.2158,
      "step": 1420
    },
    {
      "step": 1420,
      "train_flow_matching_loss": 0.11376953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 1420,
      "train_flow_matching_loss": 0.2138671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 1420,
      "train_flow_matching_loss": 0.1630859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 1420,
      "train_flow_matching_loss": 0.11767578125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.1541750133037567,
      "learning_rate": 4.763333333333334e-05,
      "loss": 0.2096,
      "step": 1430
    },
    {
      "step": 1430,
      "train_flow_matching_loss": 0.1953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 1430,
      "train_flow_matching_loss": 0.1904296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 1430,
      "train_flow_matching_loss": 0.1923828125,
      "train_physics_loss": 0.0
    },
    {
      "step": 1430,
      "train_flow_matching_loss": 0.171875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.6172294616699219,
      "learning_rate": 4.796666666666667e-05,
      "loss": 0.209,
      "step": 1440
    },
    {
      "step": 1440,
      "train_flow_matching_loss": 0.181640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 1440,
      "train_flow_matching_loss": 0.1162109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 1440,
      "train_flow_matching_loss": 0.1181640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 1440,
      "train_flow_matching_loss": 0.1923828125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.2245466262102127,
      "learning_rate": 4.83e-05,
      "loss": 0.2092,
      "step": 1450
    },
    {
      "step": 1450,
      "train_flow_matching_loss": 0.134765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 1450,
      "train_flow_matching_loss": 0.146484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 1450,
      "train_flow_matching_loss": 0.2431640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 1450,
      "train_flow_matching_loss": 0.1220703125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.12819266319274902,
      "learning_rate": 4.8633333333333334e-05,
      "loss": 0.2111,
      "step": 1460
    },
    {
      "step": 1460,
      "train_flow_matching_loss": 0.1640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 1460,
      "train_flow_matching_loss": 0.1796875,
      "train_physics_loss": 0.0
    },
    {
      "step": 1460,
      "train_flow_matching_loss": 0.197265625,
      "train_physics_loss": 0.0
    },
    {
      "step": 1460,
      "train_flow_matching_loss": 0.1474609375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.07723420113325119,
      "learning_rate": 4.8966666666666667e-05,
      "loss": 0.211,
      "step": 1470
    },
    {
      "step": 1470,
      "train_flow_matching_loss": 0.271484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 1470,
      "train_flow_matching_loss": 0.263671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 1470,
      "train_flow_matching_loss": 0.142578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 1470,
      "train_flow_matching_loss": 0.1640625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11305595934391022,
      "learning_rate": 4.93e-05,
      "loss": 0.212,
      "step": 1480
    },
    {
      "step": 1480,
      "train_flow_matching_loss": 0.177734375,
      "train_physics_loss": 0.0
    },
    {
      "step": 1480,
      "train_flow_matching_loss": 0.2109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 1480,
      "train_flow_matching_loss": 0.06494140625,
      "train_physics_loss": 0.0
    },
    {
      "step": 1480,
      "train_flow_matching_loss": 0.228515625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09946904331445694,
      "learning_rate": 4.963333333333334e-05,
      "loss": 0.212,
      "step": 1490
    },
    {
      "step": 1490,
      "train_flow_matching_loss": 0.09619140625,
      "train_physics_loss": 0.0
    },
    {
      "step": 1490,
      "train_flow_matching_loss": 0.142578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 1490,
      "train_flow_matching_loss": 0.26953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 1490,
      "train_flow_matching_loss": 0.1748046875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.15560853481292725,
      "learning_rate": 4.996666666666667e-05,
      "loss": 0.2157,
      "step": 1500
    },
    {
      "step": 1500,
      "train_flow_matching_loss": 0.1591796875,
      "train_physics_loss": 0.0
    },
    {
      "step": 1500,
      "train_flow_matching_loss": 0.1630859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 1500,
      "train_flow_matching_loss": 0.1494140625,
      "train_physics_loss": 0.0
    },
    {
      "step": 1500,
      "train_flow_matching_loss": 0.1552734375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.16900543868541718,
      "learning_rate": 5e-05,
      "loss": 0.2088,
      "step": 1510
    },
    {
      "step": 1510,
      "train_flow_matching_loss": 0.1337890625,
      "train_physics_loss": 0.0
    },
    {
      "step": 1510,
      "train_flow_matching_loss": 0.228515625,
      "train_physics_loss": 0.0
    },
    {
      "step": 1510,
      "train_flow_matching_loss": 0.30859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 1510,
      "train_flow_matching_loss": 0.189453125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.0826060101389885,
      "learning_rate": 5e-05,
      "loss": 0.2156,
      "step": 1520
    },
    {
      "step": 1520,
      "train_flow_matching_loss": 0.337890625,
      "train_physics_loss": 1.0086265802383423
    },
    {
      "step": 1520,
      "train_flow_matching_loss": 0.166015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 1520,
      "train_flow_matching_loss": 0.1611328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 1520,
      "train_flow_matching_loss": 0.28515625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08655646443367004,
      "learning_rate": 5e-05,
      "loss": 0.2128,
      "step": 1530
    },
    {
      "step": 1530,
      "train_flow_matching_loss": 0.212890625,
      "train_physics_loss": 1.0115457773208618
    },
    {
      "step": 1530,
      "train_flow_matching_loss": 0.2099609375,
      "train_physics_loss": 0.0
    },
    {
      "step": 1530,
      "train_flow_matching_loss": 0.1533203125,
      "train_physics_loss": 1.1479488611221313
    },
    {
      "step": 1530,
      "train_flow_matching_loss": 0.1982421875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11506307870149612,
      "learning_rate": 5e-05,
      "loss": 0.2197,
      "step": 1540
    },
    {
      "step": 1540,
      "train_flow_matching_loss": 0.1875,
      "train_physics_loss": 1.0279303789138794
    },
    {
      "step": 1540,
      "train_flow_matching_loss": 0.1650390625,
      "train_physics_loss": 0.0
    },
    {
      "step": 1540,
      "train_flow_matching_loss": 0.2021484375,
      "train_physics_loss": 0.9439539313316345
    },
    {
      "step": 1540,
      "train_flow_matching_loss": 0.1572265625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.141235813498497,
      "learning_rate": 5e-05,
      "loss": 0.2102,
      "step": 1550
    },
    {
      "step": 1550,
      "train_flow_matching_loss": 0.1728515625,
      "train_physics_loss": 0.9935173392295837
    },
    {
      "step": 1550,
      "train_flow_matching_loss": 0.2001953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 1550,
      "train_flow_matching_loss": 0.234375,
      "train_physics_loss": 1.107974886894226
    },
    {
      "step": 1550,
      "train_flow_matching_loss": 0.1650390625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.26108312606811523,
      "learning_rate": 5e-05,
      "loss": 0.2152,
      "step": 1560
    },
    {
      "step": 1560,
      "train_flow_matching_loss": 0.189453125,
      "train_physics_loss": 1.0381748676300049
    },
    {
      "step": 1560,
      "train_flow_matching_loss": 0.1787109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 1560,
      "train_flow_matching_loss": 0.2265625,
      "train_physics_loss": 0.9372909069061279
    },
    {
      "step": 1560,
      "train_flow_matching_loss": 0.185546875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.06624602526426315,
      "learning_rate": 5e-05,
      "loss": 0.2125,
      "step": 1570
    },
    {
      "step": 1570,
      "train_flow_matching_loss": 0.1279296875,
      "train_physics_loss": 1.0075048208236694
    },
    {
      "step": 1570,
      "train_flow_matching_loss": 0.1826171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 1570,
      "train_flow_matching_loss": 0.201171875,
      "train_physics_loss": 0.9784711003303528
    },
    {
      "step": 1570,
      "train_flow_matching_loss": 0.2001953125,
      "train_physics_loss": 1.1423087120056152
    },
    {
      "grad_norm": 0.23788206279277802,
      "learning_rate": 5e-05,
      "loss": 0.2097,
      "step": 1580
    },
    {
      "step": 1580,
      "train_flow_matching_loss": 0.1982421875,
      "train_physics_loss": 0.9929946660995483
    },
    {
      "step": 1580,
      "train_flow_matching_loss": 0.1572265625,
      "train_physics_loss": 0.0
    },
    {
      "step": 1580,
      "train_flow_matching_loss": 0.2275390625,
      "train_physics_loss": 1.0579419136047363
    },
    {
      "step": 1580,
      "train_flow_matching_loss": 0.1484375,
      "train_physics_loss": 1.1398833990097046
    },
    {
      "grad_norm": 0.11952438950538635,
      "learning_rate": 5e-05,
      "loss": 0.209,
      "step": 1590
    },
    {
      "step": 1590,
      "train_flow_matching_loss": 0.287109375,
      "train_physics_loss": 0.9756933450698853
    },
    {
      "step": 1590,
      "train_flow_matching_loss": 0.1884765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 1590,
      "train_flow_matching_loss": 0.1298828125,
      "train_physics_loss": 1.1785887479782104
    },
    {
      "step": 1590,
      "train_flow_matching_loss": 0.201171875,
      "train_physics_loss": 1.0318148136138916
    },
    {
      "grad_norm": 0.08659040182828903,
      "learning_rate": 5e-05,
      "loss": 0.2152,
      "step": 1600
    },
    {
      "step": 1600,
      "train_flow_matching_loss": 0.25,
      "train_physics_loss": 0.0
    },
    {
      "step": 1600,
      "train_flow_matching_loss": 0.2216796875,
      "train_physics_loss": 0.0
    },
    {
      "step": 1600,
      "train_flow_matching_loss": 0.2412109375,
      "train_physics_loss": 1.1471138000488281
    },
    {
      "step": 1600,
      "train_flow_matching_loss": 0.10009765625,
      "train_physics_loss": 0.9602320790290833
    },
    {
      "grad_norm": 0.1354345977306366,
      "learning_rate": 5e-05,
      "loss": 0.2038,
      "step": 1610
    },
    {
      "step": 1610,
      "train_flow_matching_loss": 0.390625,
      "train_physics_loss": 0.978724479675293
    },
    {
      "step": 1610,
      "train_flow_matching_loss": 0.1904296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 1610,
      "train_flow_matching_loss": 0.10498046875,
      "train_physics_loss": 1.1302189826965332
    },
    {
      "step": 1610,
      "train_flow_matching_loss": 0.236328125,
      "train_physics_loss": 1.2785484790802002
    },
    {
      "grad_norm": 0.0811605378985405,
      "learning_rate": 5e-05,
      "loss": 0.2106,
      "step": 1620
    },
    {
      "step": 1620,
      "train_flow_matching_loss": 0.1513671875,
      "train_physics_loss": 1.0200682878494263
    },
    {
      "step": 1620,
      "train_flow_matching_loss": 0.1455078125,
      "train_physics_loss": 0.0
    },
    {
      "step": 1620,
      "train_flow_matching_loss": 0.203125,
      "train_physics_loss": 0.9813075661659241
    },
    {
      "step": 1620,
      "train_flow_matching_loss": 0.2041015625,
      "train_physics_loss": 0.9973756670951843
    },
    {
      "grad_norm": 0.054288480430841446,
      "learning_rate": 5e-05,
      "loss": 0.2118,
      "step": 1630
    },
    {
      "step": 1630,
      "train_flow_matching_loss": 0.1357421875,
      "train_physics_loss": 0.953217089176178
    },
    {
      "step": 1630,
      "train_flow_matching_loss": 0.2421875,
      "train_physics_loss": 0.0
    },
    {
      "step": 1630,
      "train_flow_matching_loss": 0.189453125,
      "train_physics_loss": 1.1055872440338135
    },
    {
      "step": 1630,
      "train_flow_matching_loss": 0.10888671875,
      "train_physics_loss": 1.0111334323883057
    },
    {
      "grad_norm": 0.08393033593893051,
      "learning_rate": 5e-05,
      "loss": 0.2113,
      "step": 1640
    },
    {
      "step": 1640,
      "train_flow_matching_loss": 0.1494140625,
      "train_physics_loss": 0.961430549621582
    },
    {
      "step": 1640,
      "train_flow_matching_loss": 0.27734375,
      "train_physics_loss": 0.0
    },
    {
      "step": 1640,
      "train_flow_matching_loss": 0.1435546875,
      "train_physics_loss": 1.148047924041748
    },
    {
      "step": 1640,
      "train_flow_matching_loss": 0.1865234375,
      "train_physics_loss": 0.9708103537559509
    },
    {
      "grad_norm": 0.1807229369878769,
      "learning_rate": 5e-05,
      "loss": 0.21,
      "step": 1650
    },
    {
      "step": 1650,
      "train_flow_matching_loss": 0.205078125,
      "train_physics_loss": 0.9905651211738586
    },
    {
      "step": 1650,
      "train_flow_matching_loss": 0.1513671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 1650,
      "train_flow_matching_loss": 0.154296875,
      "train_physics_loss": 1.165093183517456
    },
    {
      "step": 1650,
      "train_flow_matching_loss": 0.201171875,
      "train_physics_loss": 1.0655044317245483
    },
    {
      "grad_norm": 0.1599602997303009,
      "learning_rate": 5e-05,
      "loss": 0.203,
      "step": 1660
    },
    {
      "step": 1660,
      "train_flow_matching_loss": 0.1962890625,
      "train_physics_loss": 1.069008231163025
    },
    {
      "step": 1660,
      "train_flow_matching_loss": 0.1279296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 1660,
      "train_flow_matching_loss": 0.205078125,
      "train_physics_loss": 1.0266937017440796
    },
    {
      "step": 1660,
      "train_flow_matching_loss": 0.265625,
      "train_physics_loss": 1.3555809259414673
    },
    {
      "grad_norm": 0.10218081623315811,
      "learning_rate": 5e-05,
      "loss": 0.2098,
      "step": 1670
    },
    {
      "step": 1670,
      "train_flow_matching_loss": 0.154296875,
      "train_physics_loss": 1.0315470695495605
    },
    {
      "step": 1670,
      "train_flow_matching_loss": 0.232421875,
      "train_physics_loss": 0.0
    },
    {
      "step": 1670,
      "train_flow_matching_loss": 0.2138671875,
      "train_physics_loss": 0.9440473914146423
    },
    {
      "step": 1670,
      "train_flow_matching_loss": 0.1826171875,
      "train_physics_loss": 1.0545374155044556
    },
    {
      "grad_norm": 0.11910474300384521,
      "learning_rate": 5e-05,
      "loss": 0.2092,
      "step": 1680
    },
    {
      "step": 1680,
      "train_flow_matching_loss": 0.1728515625,
      "train_physics_loss": 0.9836764931678772
    },
    {
      "step": 1680,
      "train_flow_matching_loss": 0.173828125,
      "train_physics_loss": 0.0
    },
    {
      "step": 1680,
      "train_flow_matching_loss": 0.177734375,
      "train_physics_loss": 1.1742628812789917
    },
    {
      "step": 1680,
      "train_flow_matching_loss": 0.1484375,
      "train_physics_loss": 1.1151715517044067
    },
    {
      "grad_norm": 0.10141646862030029,
      "learning_rate": 5e-05,
      "loss": 0.2131,
      "step": 1690
    },
    {
      "step": 1690,
      "train_flow_matching_loss": 0.11181640625,
      "train_physics_loss": 0.9916908740997314
    },
    {
      "step": 1690,
      "train_flow_matching_loss": 0.177734375,
      "train_physics_loss": 0.0
    },
    {
      "step": 1690,
      "train_flow_matching_loss": 0.10546875,
      "train_physics_loss": 1.055820107460022
    },
    {
      "step": 1690,
      "train_flow_matching_loss": 0.130859375,
      "train_physics_loss": 1.0928077697753906
    },
    {
      "grad_norm": 0.1885375827550888,
      "learning_rate": 5e-05,
      "loss": 0.2101,
      "step": 1700
    },
    {
      "step": 1700,
      "train_flow_matching_loss": 0.1357421875,
      "train_physics_loss": 1.0799791812896729
    },
    {
      "step": 1700,
      "train_flow_matching_loss": 0.1708984375,
      "train_physics_loss": 0.0
    },
    {
      "step": 1700,
      "train_flow_matching_loss": 0.236328125,
      "train_physics_loss": 1.084865927696228
    },
    {
      "step": 1700,
      "train_flow_matching_loss": 0.0986328125,
      "train_physics_loss": 1.1126737594604492
    },
    {
      "grad_norm": 0.17969469726085663,
      "learning_rate": 5e-05,
      "loss": 0.2095,
      "step": 1710
    },
    {
      "step": 1710,
      "train_flow_matching_loss": 0.1494140625,
      "train_physics_loss": 0.9615604281425476
    },
    {
      "step": 1710,
      "train_flow_matching_loss": 0.1591796875,
      "train_physics_loss": 0.0
    },
    {
      "step": 1710,
      "train_flow_matching_loss": 0.21875,
      "train_physics_loss": 0.9370481967926025
    },
    {
      "step": 1710,
      "train_flow_matching_loss": 0.10107421875,
      "train_physics_loss": 0.993007242679596
    },
    {
      "grad_norm": 0.49959269165992737,
      "learning_rate": 5e-05,
      "loss": 0.2106,
      "step": 1720
    },
    {
      "step": 1720,
      "train_flow_matching_loss": 0.294921875,
      "train_physics_loss": 1.0844091176986694
    },
    {
      "step": 1720,
      "train_flow_matching_loss": 0.1552734375,
      "train_physics_loss": 0.0
    },
    {
      "step": 1720,
      "train_flow_matching_loss": 0.10693359375,
      "train_physics_loss": 1.0684846639633179
    },
    {
      "step": 1720,
      "train_flow_matching_loss": 0.216796875,
      "train_physics_loss": 1.0363413095474243
    },
    {
      "grad_norm": 0.44017910957336426,
      "learning_rate": 5e-05,
      "loss": 0.2112,
      "step": 1730
    },
    {
      "step": 1730,
      "train_flow_matching_loss": 0.279296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 1730,
      "train_flow_matching_loss": 0.197265625,
      "train_physics_loss": 0.0
    },
    {
      "step": 1730,
      "train_flow_matching_loss": 0.1318359375,
      "train_physics_loss": 1.0786086320877075
    },
    {
      "step": 1730,
      "train_flow_matching_loss": 0.1318359375,
      "train_physics_loss": 1.068070888519287
    },
    {
      "grad_norm": 0.07049677520990372,
      "learning_rate": 5e-05,
      "loss": 0.2074,
      "step": 1740
    },
    {
      "step": 1740,
      "train_flow_matching_loss": 0.1484375,
      "train_physics_loss": 1.0312950611114502
    },
    {
      "step": 1740,
      "train_flow_matching_loss": 0.2158203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 1740,
      "train_flow_matching_loss": 0.1630859375,
      "train_physics_loss": 1.0593585968017578
    },
    {
      "step": 1740,
      "train_flow_matching_loss": 0.1767578125,
      "train_physics_loss": 0.9937055706977844
    },
    {
      "grad_norm": 0.13912424445152283,
      "learning_rate": 5e-05,
      "loss": 0.2117,
      "step": 1750
    },
    {
      "step": 1750,
      "train_flow_matching_loss": 0.357421875,
      "train_physics_loss": 0.9887628555297852
    },
    {
      "step": 1750,
      "train_flow_matching_loss": 0.205078125,
      "train_physics_loss": 0.0
    },
    {
      "step": 1750,
      "train_flow_matching_loss": 0.1982421875,
      "train_physics_loss": 1.0935370922088623
    },
    {
      "step": 1750,
      "train_flow_matching_loss": 0.2001953125,
      "train_physics_loss": 1.0662139654159546
    },
    {
      "grad_norm": 0.2563222646713257,
      "learning_rate": 5e-05,
      "loss": 0.215,
      "step": 1760
    },
    {
      "step": 1760,
      "train_flow_matching_loss": 0.146484375,
      "train_physics_loss": 1.028956651687622
    },
    {
      "step": 1760,
      "train_flow_matching_loss": 0.267578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 1760,
      "train_flow_matching_loss": 0.228515625,
      "train_physics_loss": 1.1305193901062012
    },
    {
      "step": 1760,
      "train_flow_matching_loss": 0.15625,
      "train_physics_loss": 1.0461634397506714
    },
    {
      "grad_norm": 0.1380806714296341,
      "learning_rate": 5e-05,
      "loss": 0.2101,
      "step": 1770
    },
    {
      "step": 1770,
      "train_flow_matching_loss": 0.1728515625,
      "train_physics_loss": 1.2589088678359985
    },
    {
      "step": 1770,
      "train_flow_matching_loss": 0.12353515625,
      "train_physics_loss": 0.0
    },
    {
      "step": 1770,
      "train_flow_matching_loss": 0.1923828125,
      "train_physics_loss": 1.042083740234375
    },
    {
      "step": 1770,
      "train_flow_matching_loss": 0.248046875,
      "train_physics_loss": 1.1425487995147705
    },
    {
      "grad_norm": 0.10561925172805786,
      "learning_rate": 5e-05,
      "loss": 0.2101,
      "step": 1780
    },
    {
      "step": 1780,
      "train_flow_matching_loss": 0.150390625,
      "train_physics_loss": 1.0284115076065063
    },
    {
      "step": 1780,
      "train_flow_matching_loss": 0.1376953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 1780,
      "train_flow_matching_loss": 0.1416015625,
      "train_physics_loss": 1.2127968072891235
    },
    {
      "step": 1780,
      "train_flow_matching_loss": 0.12353515625,
      "train_physics_loss": 0.987876832485199
    },
    {
      "grad_norm": 0.15677882730960846,
      "learning_rate": 5e-05,
      "loss": 0.2147,
      "step": 1790
    },
    {
      "step": 1790,
      "train_flow_matching_loss": 0.185546875,
      "train_physics_loss": 1.0625725984573364
    },
    {
      "step": 1790,
      "train_flow_matching_loss": 0.1591796875,
      "train_physics_loss": 0.0
    },
    {
      "step": 1790,
      "train_flow_matching_loss": 0.1708984375,
      "train_physics_loss": 0.0
    },
    {
      "step": 1790,
      "train_flow_matching_loss": 0.142578125,
      "train_physics_loss": 1.0016729831695557
    },
    {
      "grad_norm": 0.12358389049768448,
      "learning_rate": 5e-05,
      "loss": 0.2059,
      "step": 1800
    },
    {
      "step": 1800,
      "train_flow_matching_loss": 0.1279296875,
      "train_physics_loss": 1.0174556970596313
    },
    {
      "step": 1800,
      "train_flow_matching_loss": 0.1357421875,
      "train_physics_loss": 1.0166481733322144
    },
    {
      "step": 1800,
      "train_flow_matching_loss": 0.1796875,
      "train_physics_loss": 0.0
    },
    {
      "step": 1800,
      "train_flow_matching_loss": 0.0947265625,
      "train_physics_loss": 1.0864923000335693
    },
    {
      "grad_norm": 0.057144880294799805,
      "learning_rate": 5e-05,
      "loss": 0.2062,
      "step": 1810
    },
    {
      "step": 1810,
      "train_flow_matching_loss": 0.1875,
      "train_physics_loss": 1.1154799461364746
    },
    {
      "step": 1810,
      "train_flow_matching_loss": 0.279296875,
      "train_physics_loss": 0.9987672567367554
    },
    {
      "step": 1810,
      "train_flow_matching_loss": 0.1796875,
      "train_physics_loss": 0.0
    },
    {
      "step": 1810,
      "train_flow_matching_loss": 0.2236328125,
      "train_physics_loss": 1.0262534618377686
    },
    {
      "grad_norm": 0.15459805727005005,
      "learning_rate": 5e-05,
      "loss": 0.2048,
      "step": 1820
    },
    {
      "step": 1820,
      "train_flow_matching_loss": 0.291015625,
      "train_physics_loss": 0.9829200506210327
    },
    {
      "step": 1820,
      "train_flow_matching_loss": 0.13671875,
      "train_physics_loss": 0.9650934338569641
    },
    {
      "step": 1820,
      "train_flow_matching_loss": 0.267578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 1820,
      "train_flow_matching_loss": 0.173828125,
      "train_physics_loss": 1.0454076528549194
    },
    {
      "grad_norm": 0.05521339550614357,
      "learning_rate": 5e-05,
      "loss": 0.2021,
      "step": 1830
    },
    {
      "step": 1830,
      "train_flow_matching_loss": 0.1435546875,
      "train_physics_loss": 0.9635727405548096
    },
    {
      "step": 1830,
      "train_flow_matching_loss": 0.1552734375,
      "train_physics_loss": 1.0410009622573853
    },
    {
      "step": 1830,
      "train_flow_matching_loss": 0.201171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 1830,
      "train_flow_matching_loss": 0.138671875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.4287504553794861,
      "learning_rate": 5e-05,
      "loss": 0.2046,
      "step": 1840
    },
    {
      "step": 1840,
      "train_flow_matching_loss": 0.1611328125,
      "train_physics_loss": 1.1161468029022217
    },
    {
      "step": 1840,
      "train_flow_matching_loss": 0.1806640625,
      "train_physics_loss": 1.0850998163223267
    },
    {
      "step": 1840,
      "train_flow_matching_loss": 0.1259765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 1840,
      "train_flow_matching_loss": 0.1708984375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.07172863930463791,
      "learning_rate": 5e-05,
      "loss": 0.2018,
      "step": 1850
    },
    {
      "step": 1850,
      "train_flow_matching_loss": 0.263671875,
      "train_physics_loss": 1.0622053146362305
    },
    {
      "step": 1850,
      "train_flow_matching_loss": 0.0966796875,
      "train_physics_loss": 1.081496238708496
    },
    {
      "step": 1850,
      "train_flow_matching_loss": 0.287109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 1850,
      "train_flow_matching_loss": 0.1982421875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.1438072770833969,
      "learning_rate": 5e-05,
      "loss": 0.2061,
      "step": 1860
    },
    {
      "step": 1860,
      "train_flow_matching_loss": 0.1953125,
      "train_physics_loss": 1.1372066736221313
    },
    {
      "step": 1860,
      "train_flow_matching_loss": 0.212890625,
      "train_physics_loss": 1.4837629795074463
    },
    {
      "step": 1860,
      "train_flow_matching_loss": 0.2470703125,
      "train_physics_loss": 0.0
    },
    {
      "step": 1860,
      "train_flow_matching_loss": 0.1689453125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.20877303183078766,
      "learning_rate": 5e-05,
      "loss": 0.2058,
      "step": 1870
    },
    {
      "step": 1870,
      "train_flow_matching_loss": 0.2177734375,
      "train_physics_loss": 0.9791767001152039
    },
    {
      "step": 1870,
      "train_flow_matching_loss": 0.185546875,
      "train_physics_loss": 0.9549384117126465
    },
    {
      "step": 1870,
      "train_flow_matching_loss": 0.1650390625,
      "train_physics_loss": 0.0
    },
    {
      "step": 1870,
      "train_flow_matching_loss": 0.10693359375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.13601672649383545,
      "learning_rate": 5e-05,
      "loss": 0.2031,
      "step": 1880
    },
    {
      "step": 1880,
      "train_flow_matching_loss": 0.2265625,
      "train_physics_loss": 0.9573955535888672
    },
    {
      "step": 1880,
      "train_flow_matching_loss": 0.142578125,
      "train_physics_loss": 1.139295220375061
    },
    {
      "step": 1880,
      "train_flow_matching_loss": 0.134765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 1880,
      "train_flow_matching_loss": 0.177734375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.1257515698671341,
      "learning_rate": 5e-05,
      "loss": 0.1994,
      "step": 1890
    },
    {
      "step": 1890,
      "train_flow_matching_loss": 0.146484375,
      "train_physics_loss": 1.0195871591567993
    },
    {
      "step": 1890,
      "train_flow_matching_loss": 0.29296875,
      "train_physics_loss": 1.0396056175231934
    },
    {
      "step": 1890,
      "train_flow_matching_loss": 0.1767578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 1890,
      "train_flow_matching_loss": 0.1162109375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11643112450838089,
      "learning_rate": 5e-05,
      "loss": 0.2058,
      "step": 1900
    },
    {
      "step": 1900,
      "train_flow_matching_loss": 0.189453125,
      "train_physics_loss": 1.080922245979309
    },
    {
      "step": 1900,
      "train_flow_matching_loss": 0.1474609375,
      "train_physics_loss": 1.0358521938323975
    },
    {
      "step": 1900,
      "train_flow_matching_loss": 0.177734375,
      "train_physics_loss": 0.0
    },
    {
      "step": 1900,
      "train_flow_matching_loss": 0.2255859375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.2539294958114624,
      "learning_rate": 5e-05,
      "loss": 0.2057,
      "step": 1910
    },
    {
      "step": 1910,
      "train_flow_matching_loss": 0.177734375,
      "train_physics_loss": 1.0584871768951416
    },
    {
      "step": 1910,
      "train_flow_matching_loss": 0.1845703125,
      "train_physics_loss": 1.0399830341339111
    },
    {
      "step": 1910,
      "train_flow_matching_loss": 0.1884765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 1910,
      "train_flow_matching_loss": 0.2177734375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11186906695365906,
      "learning_rate": 5e-05,
      "loss": 0.2059,
      "step": 1920
    },
    {
      "step": 1920,
      "train_flow_matching_loss": 0.1455078125,
      "train_physics_loss": 0.9879170060157776
    },
    {
      "step": 1920,
      "train_flow_matching_loss": 0.240234375,
      "train_physics_loss": 1.0161101818084717
    },
    {
      "step": 1920,
      "train_flow_matching_loss": 0.26171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 1920,
      "train_flow_matching_loss": 0.1572265625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.06667591631412506,
      "learning_rate": 5e-05,
      "loss": 0.2051,
      "step": 1930
    },
    {
      "step": 1930,
      "train_flow_matching_loss": 0.11474609375,
      "train_physics_loss": 1.0449310541152954
    },
    {
      "step": 1930,
      "train_flow_matching_loss": 0.240234375,
      "train_physics_loss": 0.9764738082885742
    },
    {
      "step": 1930,
      "train_flow_matching_loss": 0.173828125,
      "train_physics_loss": 0.0
    },
    {
      "step": 1930,
      "train_flow_matching_loss": 0.2734375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11948324739933014,
      "learning_rate": 5e-05,
      "loss": 0.2057,
      "step": 1940
    },
    {
      "step": 1940,
      "train_flow_matching_loss": 0.1484375,
      "train_physics_loss": 1.1471558809280396
    },
    {
      "step": 1940,
      "train_flow_matching_loss": 0.158203125,
      "train_physics_loss": 1.1136637926101685
    },
    {
      "step": 1940,
      "train_flow_matching_loss": 0.1611328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 1940,
      "train_flow_matching_loss": 0.310546875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.06825944036245346,
      "learning_rate": 5e-05,
      "loss": 0.2043,
      "step": 1950
    },
    {
      "step": 1950,
      "train_flow_matching_loss": 0.11328125,
      "train_physics_loss": 1.0639344453811646
    },
    {
      "step": 1950,
      "train_flow_matching_loss": 0.1513671875,
      "train_physics_loss": 0.9856606721878052
    },
    {
      "step": 1950,
      "train_flow_matching_loss": 0.26171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 1950,
      "train_flow_matching_loss": 0.171875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10421253740787506,
      "learning_rate": 5e-05,
      "loss": 0.1946,
      "step": 1960
    },
    {
      "step": 1960,
      "train_flow_matching_loss": 0.2119140625,
      "train_physics_loss": 0.9930260181427002
    },
    {
      "step": 1960,
      "train_flow_matching_loss": 0.232421875,
      "train_physics_loss": 1.0138272047042847
    },
    {
      "step": 1960,
      "train_flow_matching_loss": 0.1416015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 1960,
      "train_flow_matching_loss": 0.1923828125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.13114073872566223,
      "learning_rate": 5e-05,
      "loss": 0.2059,
      "step": 1970
    },
    {
      "step": 1970,
      "train_flow_matching_loss": 0.1552734375,
      "train_physics_loss": 1.0084782838821411
    },
    {
      "step": 1970,
      "train_flow_matching_loss": 0.25,
      "train_physics_loss": 1.1827160120010376
    },
    {
      "step": 1970,
      "train_flow_matching_loss": 0.1455078125,
      "train_physics_loss": 0.0
    },
    {
      "step": 1970,
      "train_flow_matching_loss": 0.1484375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10762210935354233,
      "learning_rate": 5e-05,
      "loss": 0.2075,
      "step": 1980
    },
    {
      "step": 1980,
      "train_flow_matching_loss": 0.08837890625,
      "train_physics_loss": 1.1506922245025635
    },
    {
      "step": 1980,
      "train_flow_matching_loss": 0.12353515625,
      "train_physics_loss": 1.078010082244873
    },
    {
      "step": 1980,
      "train_flow_matching_loss": 0.1767578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 1980,
      "train_flow_matching_loss": 0.138671875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08325135707855225,
      "learning_rate": 5e-05,
      "loss": 0.1972,
      "step": 1990
    },
    {
      "step": 1990,
      "train_flow_matching_loss": 0.181640625,
      "train_physics_loss": 1.1137199401855469
    },
    {
      "step": 1990,
      "train_flow_matching_loss": 0.1767578125,
      "train_physics_loss": 1.0067189931869507
    },
    {
      "step": 1990,
      "train_flow_matching_loss": 0.2041015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 1990,
      "train_flow_matching_loss": 0.1484375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.06720130890607834,
      "learning_rate": 5e-05,
      "loss": 0.2043,
      "step": 2000
    },
    {
      "step": 2000,
      "train_flow_matching_loss": 0.1826171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 2000,
      "train_flow_matching_loss": 0.07373046875,
      "train_physics_loss": 0.0
    },
    {
      "step": 2000,
      "train_flow_matching_loss": 0.2578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 2000,
      "train_flow_matching_loss": 0.2119140625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.30847451090812683,
      "learning_rate": 5e-05,
      "loss": 0.2844,
      "step": 2010
    },
    {
      "step": 2010,
      "train_flow_matching_loss": 0.271484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 2010,
      "train_flow_matching_loss": 0.341796875,
      "train_physics_loss": 0.0
    },
    {
      "step": 2010,
      "train_flow_matching_loss": 0.248046875,
      "train_physics_loss": 0.0
    },
    {
      "step": 2010,
      "train_flow_matching_loss": 0.1884765625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.31857556104660034,
      "learning_rate": 5e-05,
      "loss": 0.2543,
      "step": 2020
    },
    {
      "step": 2020,
      "train_flow_matching_loss": 0.296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 2020,
      "train_flow_matching_loss": 0.25,
      "train_physics_loss": 0.0
    },
    {
      "step": 2020,
      "train_flow_matching_loss": 0.33984375,
      "train_physics_loss": 0.0
    },
    {
      "step": 2020,
      "train_flow_matching_loss": 0.259765625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.18520279228687286,
      "learning_rate": 5e-05,
      "loss": 0.2572,
      "step": 2030
    },
    {
      "step": 2030,
      "train_flow_matching_loss": 0.201171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 2030,
      "train_flow_matching_loss": 0.314453125,
      "train_physics_loss": 0.0
    },
    {
      "step": 2030,
      "train_flow_matching_loss": 0.171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 2030,
      "train_flow_matching_loss": 0.2314453125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.1601283699274063,
      "learning_rate": 5e-05,
      "loss": 0.2579,
      "step": 2040
    },
    {
      "step": 2040,
      "train_flow_matching_loss": 0.2197265625,
      "train_physics_loss": 0.0
    },
    {
      "step": 2040,
      "train_flow_matching_loss": 0.234375,
      "train_physics_loss": 0.0
    },
    {
      "step": 2040,
      "train_flow_matching_loss": 0.208984375,
      "train_physics_loss": 0.0
    },
    {
      "step": 2040,
      "train_flow_matching_loss": 0.279296875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.12042475491762161,
      "learning_rate": 5e-05,
      "loss": 0.249,
      "step": 2050
    },
    {
      "step": 2050,
      "train_flow_matching_loss": 0.25,
      "train_physics_loss": 0.0
    },
    {
      "step": 2050,
      "train_flow_matching_loss": 0.1572265625,
      "train_physics_loss": 0.0
    },
    {
      "step": 2050,
      "train_flow_matching_loss": 0.2333984375,
      "train_physics_loss": 0.0
    },
    {
      "step": 2050,
      "train_flow_matching_loss": 0.2099609375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.32061532139778137,
      "learning_rate": 5e-05,
      "loss": 0.2418,
      "step": 2060
    },
    {
      "step": 2060,
      "train_flow_matching_loss": 0.255859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 2060,
      "train_flow_matching_loss": 0.26953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 2060,
      "train_flow_matching_loss": 0.330078125,
      "train_physics_loss": 0.0
    },
    {
      "step": 2060,
      "train_flow_matching_loss": 0.1474609375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.17935042083263397,
      "learning_rate": 5e-05,
      "loss": 0.2379,
      "step": 2070
    },
    {
      "step": 2070,
      "train_flow_matching_loss": 0.244140625,
      "train_physics_loss": 0.0
    },
    {
      "step": 2070,
      "train_flow_matching_loss": 0.185546875,
      "train_physics_loss": 0.0
    },
    {
      "step": 2070,
      "train_flow_matching_loss": 0.25390625,
      "train_physics_loss": 0.0
    },
    {
      "step": 2070,
      "train_flow_matching_loss": 0.1669921875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.16420985758304596,
      "learning_rate": 5e-05,
      "loss": 0.2279,
      "step": 2080
    },
    {
      "step": 2080,
      "train_flow_matching_loss": 0.216796875,
      "train_physics_loss": 0.0
    },
    {
      "step": 2080,
      "train_flow_matching_loss": 0.203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 2080,
      "train_flow_matching_loss": 0.1298828125,
      "train_physics_loss": 0.0
    },
    {
      "step": 2080,
      "train_flow_matching_loss": 0.220703125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.13994190096855164,
      "learning_rate": 5e-05,
      "loss": 0.2285,
      "step": 2090
    },
    {
      "step": 2090,
      "train_flow_matching_loss": 0.2138671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 2090,
      "train_flow_matching_loss": 0.1953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 2090,
      "train_flow_matching_loss": 0.2314453125,
      "train_physics_loss": 0.0
    },
    {
      "step": 2090,
      "train_flow_matching_loss": 0.2197265625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.1267644613981247,
      "learning_rate": 5e-05,
      "loss": 0.2299,
      "step": 2100
    },
    {
      "step": 2100,
      "train_flow_matching_loss": 0.2431640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 2100,
      "train_flow_matching_loss": 0.1953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 2100,
      "train_flow_matching_loss": 0.1943359375,
      "train_physics_loss": 0.0
    },
    {
      "step": 2100,
      "train_flow_matching_loss": 0.2236328125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.1392682045698166,
      "learning_rate": 5e-05,
      "loss": 0.2211,
      "step": 2110
    },
    {
      "step": 2110,
      "train_flow_matching_loss": 0.2373046875,
      "train_physics_loss": 0.0
    },
    {
      "step": 2110,
      "train_flow_matching_loss": 0.2412109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 2110,
      "train_flow_matching_loss": 0.2060546875,
      "train_physics_loss": 0.0
    },
    {
      "step": 2110,
      "train_flow_matching_loss": 0.1865234375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.13750292360782623,
      "learning_rate": 5e-05,
      "loss": 0.2275,
      "step": 2120
    },
    {
      "step": 2120,
      "train_flow_matching_loss": 0.2265625,
      "train_physics_loss": 0.0
    },
    {
      "step": 2120,
      "train_flow_matching_loss": 0.11083984375,
      "train_physics_loss": 0.0
    },
    {
      "step": 2120,
      "train_flow_matching_loss": 0.1728515625,
      "train_physics_loss": 0.0
    },
    {
      "step": 2120,
      "train_flow_matching_loss": 0.2197265625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.12855499982833862,
      "learning_rate": 5e-05,
      "loss": 0.2243,
      "step": 2130
    },
    {
      "step": 2130,
      "train_flow_matching_loss": 0.2333984375,
      "train_physics_loss": 0.0
    },
    {
      "step": 2130,
      "train_flow_matching_loss": 0.2138671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 2130,
      "train_flow_matching_loss": 0.2080078125,
      "train_physics_loss": 0.0
    },
    {
      "step": 2130,
      "train_flow_matching_loss": 0.359375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.14857999980449677,
      "learning_rate": 5e-05,
      "loss": 0.2234,
      "step": 2140
    },
    {
      "step": 2140,
      "train_flow_matching_loss": 0.38671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 2140,
      "train_flow_matching_loss": 0.2197265625,
      "train_physics_loss": 0.0
    },
    {
      "step": 2140,
      "train_flow_matching_loss": 0.384765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 2140,
      "train_flow_matching_loss": 0.189453125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.2177819311618805,
      "learning_rate": 5e-05,
      "loss": 0.2352,
      "step": 2150
    },
    {
      "step": 2150,
      "train_flow_matching_loss": 0.310546875,
      "train_physics_loss": 0.0
    },
    {
      "step": 2150,
      "train_flow_matching_loss": 0.1982421875,
      "train_physics_loss": 0.0
    },
    {
      "step": 2150,
      "train_flow_matching_loss": 0.328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 2150,
      "train_flow_matching_loss": 0.11376953125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.12572354078292847,
      "learning_rate": 5e-05,
      "loss": 0.2202,
      "step": 2160
    },
    {
      "step": 2160,
      "train_flow_matching_loss": 0.2177734375,
      "train_physics_loss": 0.0
    },
    {
      "step": 2160,
      "train_flow_matching_loss": 0.1962890625,
      "train_physics_loss": 0.0
    },
    {
      "step": 2160,
      "train_flow_matching_loss": 0.1640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 2160,
      "train_flow_matching_loss": 0.166015625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.12432493269443512,
      "learning_rate": 5e-05,
      "loss": 0.217,
      "step": 2170
    },
    {
      "step": 2170,
      "train_flow_matching_loss": 0.234375,
      "train_physics_loss": 0.0
    },
    {
      "step": 2170,
      "train_flow_matching_loss": 0.1884765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 2170,
      "train_flow_matching_loss": 0.166015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 2170,
      "train_flow_matching_loss": 0.1591796875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.27478423714637756,
      "learning_rate": 5e-05,
      "loss": 0.2221,
      "step": 2180
    },
    {
      "step": 2180,
      "train_flow_matching_loss": 0.1806640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 2180,
      "train_flow_matching_loss": 0.12353515625,
      "train_physics_loss": 0.0
    },
    {
      "step": 2180,
      "train_flow_matching_loss": 0.150390625,
      "train_physics_loss": 0.0
    },
    {
      "step": 2180,
      "train_flow_matching_loss": 0.1748046875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.2896011769771576,
      "learning_rate": 5e-05,
      "loss": 0.2182,
      "step": 2190
    },
    {
      "step": 2190,
      "train_flow_matching_loss": 0.2451171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 2190,
      "train_flow_matching_loss": 0.1943359375,
      "train_physics_loss": 0.0
    },
    {
      "step": 2190,
      "train_flow_matching_loss": 0.2109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 2190,
      "train_flow_matching_loss": 0.1748046875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.15162049233913422,
      "learning_rate": 5e-05,
      "loss": 0.2183,
      "step": 2200
    },
    {
      "step": 2200,
      "train_flow_matching_loss": 0.31640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 2200,
      "train_flow_matching_loss": 0.1474609375,
      "train_physics_loss": 0.0
    },
    {
      "step": 2200,
      "train_flow_matching_loss": 0.208984375,
      "train_physics_loss": 0.0
    },
    {
      "step": 2200,
      "train_flow_matching_loss": 0.1884765625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.12154624611139297,
      "learning_rate": 5e-05,
      "loss": 0.2208,
      "step": 2210
    },
    {
      "step": 2210,
      "train_flow_matching_loss": 0.1904296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 2210,
      "train_flow_matching_loss": 0.2109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 2210,
      "train_flow_matching_loss": 0.236328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 2210,
      "train_flow_matching_loss": 0.2041015625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.151609405875206,
      "learning_rate": 5e-05,
      "loss": 0.2157,
      "step": 2220
    },
    {
      "step": 2220,
      "train_flow_matching_loss": 0.1904296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 2220,
      "train_flow_matching_loss": 0.177734375,
      "train_physics_loss": 0.0
    },
    {
      "step": 2220,
      "train_flow_matching_loss": 0.2109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 2220,
      "train_flow_matching_loss": 0.1455078125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.1620166152715683,
      "learning_rate": 5e-05,
      "loss": 0.2129,
      "step": 2230
    },
    {
      "step": 2230,
      "train_flow_matching_loss": 0.234375,
      "train_physics_loss": 0.0
    },
    {
      "step": 2230,
      "train_flow_matching_loss": 0.2109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 2230,
      "train_flow_matching_loss": 0.1552734375,
      "train_physics_loss": 0.0
    },
    {
      "step": 2230,
      "train_flow_matching_loss": 0.177734375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.3134675621986389,
      "learning_rate": 5e-05,
      "loss": 0.212,
      "step": 2240
    },
    {
      "step": 2240,
      "train_flow_matching_loss": 0.26171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 2240,
      "train_flow_matching_loss": 0.298828125,
      "train_physics_loss": 0.0
    },
    {
      "step": 2240,
      "train_flow_matching_loss": 0.224609375,
      "train_physics_loss": 0.0
    },
    {
      "step": 2240,
      "train_flow_matching_loss": 0.1630859375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.22719989717006683,
      "learning_rate": 5e-05,
      "loss": 0.2179,
      "step": 2250
    },
    {
      "step": 2250,
      "train_flow_matching_loss": 0.1640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 2250,
      "train_flow_matching_loss": 0.1669921875,
      "train_physics_loss": 0.0
    },
    {
      "step": 2250,
      "train_flow_matching_loss": 0.2294921875,
      "train_physics_loss": 0.0
    },
    {
      "step": 2250,
      "train_flow_matching_loss": 0.154296875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.17337819933891296,
      "learning_rate": 5e-05,
      "loss": 0.2232,
      "step": 2260
    },
    {
      "step": 2260,
      "train_flow_matching_loss": 0.1630859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 2260,
      "train_flow_matching_loss": 0.171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 2260,
      "train_flow_matching_loss": 0.1572265625,
      "train_physics_loss": 0.0
    },
    {
      "step": 2260,
      "train_flow_matching_loss": 0.146484375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11921187490224838,
      "learning_rate": 5e-05,
      "loss": 0.2162,
      "step": 2270
    },
    {
      "step": 2270,
      "train_flow_matching_loss": 0.306640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 2270,
      "train_flow_matching_loss": 0.203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 2270,
      "train_flow_matching_loss": 0.2216796875,
      "train_physics_loss": 0.0
    },
    {
      "step": 2270,
      "train_flow_matching_loss": 0.1630859375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.15506839752197266,
      "learning_rate": 5e-05,
      "loss": 0.2121,
      "step": 2280
    },
    {
      "step": 2280,
      "train_flow_matching_loss": 0.1787109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 2280,
      "train_flow_matching_loss": 0.3125,
      "train_physics_loss": 0.0
    },
    {
      "step": 2280,
      "train_flow_matching_loss": 0.126953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 2280,
      "train_flow_matching_loss": 0.15625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.17118962109088898,
      "learning_rate": 5e-05,
      "loss": 0.2149,
      "step": 2290
    },
    {
      "step": 2290,
      "train_flow_matching_loss": 0.2412109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 2290,
      "train_flow_matching_loss": 0.158203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 2290,
      "train_flow_matching_loss": 0.1748046875,
      "train_physics_loss": 0.0
    },
    {
      "step": 2290,
      "train_flow_matching_loss": 0.1640625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.13489685952663422,
      "learning_rate": 5e-05,
      "loss": 0.2069,
      "step": 2300
    },
    {
      "step": 2300,
      "train_flow_matching_loss": 0.1533203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 2300,
      "train_flow_matching_loss": 0.1416015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 2300,
      "train_flow_matching_loss": 0.169921875,
      "train_physics_loss": 0.0
    },
    {
      "step": 2300,
      "train_flow_matching_loss": 0.208984375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.17016996443271637,
      "learning_rate": 5e-05,
      "loss": 0.2155,
      "step": 2310
    },
    {
      "step": 2310,
      "train_flow_matching_loss": 0.1318359375,
      "train_physics_loss": 0.0
    },
    {
      "step": 2310,
      "train_flow_matching_loss": 0.2119140625,
      "train_physics_loss": 0.0
    },
    {
      "step": 2310,
      "train_flow_matching_loss": 0.259765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 2310,
      "train_flow_matching_loss": 0.212890625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09526512771844864,
      "learning_rate": 5e-05,
      "loss": 0.207,
      "step": 2320
    },
    {
      "step": 2320,
      "train_flow_matching_loss": 0.2236328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 2320,
      "train_flow_matching_loss": 0.2060546875,
      "train_physics_loss": 0.0
    },
    {
      "step": 2320,
      "train_flow_matching_loss": 0.265625,
      "train_physics_loss": 0.0
    },
    {
      "step": 2320,
      "train_flow_matching_loss": 0.203125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.14650633931159973,
      "learning_rate": 5e-05,
      "loss": 0.2195,
      "step": 2330
    },
    {
      "step": 2330,
      "train_flow_matching_loss": 0.181640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 2330,
      "train_flow_matching_loss": 0.2275390625,
      "train_physics_loss": 0.0
    },
    {
      "step": 2330,
      "train_flow_matching_loss": 0.1845703125,
      "train_physics_loss": 0.0
    },
    {
      "step": 2330,
      "train_flow_matching_loss": 0.1787109375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.16266398131847382,
      "learning_rate": 5e-05,
      "loss": 0.2089,
      "step": 2340
    },
    {
      "step": 2340,
      "train_flow_matching_loss": 0.1396484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 2340,
      "train_flow_matching_loss": 0.248046875,
      "train_physics_loss": 0.0
    },
    {
      "step": 2340,
      "train_flow_matching_loss": 0.2060546875,
      "train_physics_loss": 0.0
    },
    {
      "step": 2340,
      "train_flow_matching_loss": 0.193359375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.13569049537181854,
      "learning_rate": 5e-05,
      "loss": 0.2078,
      "step": 2350
    },
    {
      "step": 2350,
      "train_flow_matching_loss": 0.21875,
      "train_physics_loss": 0.0
    },
    {
      "step": 2350,
      "train_flow_matching_loss": 0.220703125,
      "train_physics_loss": 0.0
    },
    {
      "step": 2350,
      "train_flow_matching_loss": 0.216796875,
      "train_physics_loss": 0.0
    },
    {
      "step": 2350,
      "train_flow_matching_loss": 0.1787109375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09778100997209549,
      "learning_rate": 5e-05,
      "loss": 0.2021,
      "step": 2360
    },
    {
      "step": 2360,
      "train_flow_matching_loss": 0.224609375,
      "train_physics_loss": 0.0
    },
    {
      "step": 2360,
      "train_flow_matching_loss": 0.1708984375,
      "train_physics_loss": 0.0
    },
    {
      "step": 2360,
      "train_flow_matching_loss": 0.2353515625,
      "train_physics_loss": 0.0
    },
    {
      "step": 2360,
      "train_flow_matching_loss": 0.12158203125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.2687607407569885,
      "learning_rate": 5e-05,
      "loss": 0.206,
      "step": 2370
    },
    {
      "step": 2370,
      "train_flow_matching_loss": 0.0751953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 2370,
      "train_flow_matching_loss": 0.2294921875,
      "train_physics_loss": 0.0
    },
    {
      "step": 2370,
      "train_flow_matching_loss": 0.193359375,
      "train_physics_loss": 0.0
    },
    {
      "step": 2370,
      "train_flow_matching_loss": 0.1796875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.21805670857429504,
      "learning_rate": 5e-05,
      "loss": 0.2061,
      "step": 2380
    },
    {
      "step": 2380,
      "train_flow_matching_loss": 0.1728515625,
      "train_physics_loss": 0.0
    },
    {
      "step": 2380,
      "train_flow_matching_loss": 0.1865234375,
      "train_physics_loss": 0.0
    },
    {
      "step": 2380,
      "train_flow_matching_loss": 0.2431640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 2380,
      "train_flow_matching_loss": 0.1923828125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.12510916590690613,
      "learning_rate": 5e-05,
      "loss": 0.2125,
      "step": 2390
    },
    {
      "step": 2390,
      "train_flow_matching_loss": 0.302734375,
      "train_physics_loss": 0.0
    },
    {
      "step": 2390,
      "train_flow_matching_loss": 0.126953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 2390,
      "train_flow_matching_loss": 0.32421875,
      "train_physics_loss": 0.0
    },
    {
      "step": 2390,
      "train_flow_matching_loss": 0.2041015625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.2001071721315384,
      "learning_rate": 5e-05,
      "loss": 0.217,
      "step": 2400
    },
    {
      "step": 2400,
      "train_flow_matching_loss": 0.1826171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 2400,
      "train_flow_matching_loss": 0.1787109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 2400,
      "train_flow_matching_loss": 0.181640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 2400,
      "train_flow_matching_loss": 0.12158203125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.13157357275485992,
      "learning_rate": 5e-05,
      "loss": 0.2007,
      "step": 2410
    },
    {
      "step": 2410,
      "train_flow_matching_loss": 0.1982421875,
      "train_physics_loss": 0.0
    },
    {
      "step": 2410,
      "train_flow_matching_loss": 0.234375,
      "train_physics_loss": 0.0
    },
    {
      "step": 2410,
      "train_flow_matching_loss": 0.251953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 2410,
      "train_flow_matching_loss": 0.326171875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.12587040662765503,
      "learning_rate": 5e-05,
      "loss": 0.214,
      "step": 2420
    },
    {
      "step": 2420,
      "train_flow_matching_loss": 0.224609375,
      "train_physics_loss": 0.0
    },
    {
      "step": 2420,
      "train_flow_matching_loss": 0.1943359375,
      "train_physics_loss": 0.0
    },
    {
      "step": 2420,
      "train_flow_matching_loss": 0.1611328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 2420,
      "train_flow_matching_loss": 0.228515625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.18620063364505768,
      "learning_rate": 5e-05,
      "loss": 0.2038,
      "step": 2430
    },
    {
      "step": 2430,
      "train_flow_matching_loss": 0.224609375,
      "train_physics_loss": 0.0
    },
    {
      "step": 2430,
      "train_flow_matching_loss": 0.1630859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 2430,
      "train_flow_matching_loss": 0.205078125,
      "train_physics_loss": 0.0
    },
    {
      "step": 2430,
      "train_flow_matching_loss": 0.2138671875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.13060826063156128,
      "learning_rate": 5e-05,
      "loss": 0.209,
      "step": 2440
    },
    {
      "step": 2440,
      "train_flow_matching_loss": 0.265625,
      "train_physics_loss": 0.0
    },
    {
      "step": 2440,
      "train_flow_matching_loss": 0.283203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 2440,
      "train_flow_matching_loss": 0.162109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 2440,
      "train_flow_matching_loss": 0.1630859375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.16581198573112488,
      "learning_rate": 5e-05,
      "loss": 0.2049,
      "step": 2450
    },
    {
      "step": 2450,
      "train_flow_matching_loss": 0.1328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 2450,
      "train_flow_matching_loss": 0.27734375,
      "train_physics_loss": 0.0
    },
    {
      "step": 2450,
      "train_flow_matching_loss": 0.1474609375,
      "train_physics_loss": 0.0
    },
    {
      "step": 2450,
      "train_flow_matching_loss": 0.1298828125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08156519383192062,
      "learning_rate": 5e-05,
      "loss": 0.2091,
      "step": 2460
    },
    {
      "step": 2460,
      "train_flow_matching_loss": 0.173828125,
      "train_physics_loss": 0.0
    },
    {
      "step": 2460,
      "train_flow_matching_loss": 0.162109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 2460,
      "train_flow_matching_loss": 0.150390625,
      "train_physics_loss": 0.0
    },
    {
      "step": 2460,
      "train_flow_matching_loss": 0.255859375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.1261570006608963,
      "learning_rate": 5e-05,
      "loss": 0.2029,
      "step": 2470
    },
    {
      "step": 2470,
      "train_flow_matching_loss": 0.2177734375,
      "train_physics_loss": 0.0
    },
    {
      "step": 2470,
      "train_flow_matching_loss": 0.2451171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 2470,
      "train_flow_matching_loss": 0.220703125,
      "train_physics_loss": 0.0
    },
    {
      "step": 2470,
      "train_flow_matching_loss": 0.1826171875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.13003569841384888,
      "learning_rate": 5e-05,
      "loss": 0.2075,
      "step": 2480
    },
    {
      "step": 2480,
      "train_flow_matching_loss": 0.1962890625,
      "train_physics_loss": 0.0
    },
    {
      "step": 2480,
      "train_flow_matching_loss": 0.1806640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 2480,
      "train_flow_matching_loss": 0.1787109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 2480,
      "train_flow_matching_loss": 0.2333984375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.13776874542236328,
      "learning_rate": 5e-05,
      "loss": 0.2038,
      "step": 2490
    },
    {
      "step": 2490,
      "train_flow_matching_loss": 0.1748046875,
      "train_physics_loss": 0.0
    },
    {
      "step": 2490,
      "train_flow_matching_loss": 0.1787109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 2490,
      "train_flow_matching_loss": 0.189453125,
      "train_physics_loss": 0.0
    },
    {
      "step": 2490,
      "train_flow_matching_loss": 0.150390625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.12646950781345367,
      "learning_rate": 5e-05,
      "loss": 0.2037,
      "step": 2500
    },
    {
      "step": 2500,
      "train_flow_matching_loss": 0.2412109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 2500,
      "train_flow_matching_loss": 0.2333984375,
      "train_physics_loss": 0.0
    },
    {
      "step": 2500,
      "train_flow_matching_loss": 0.1953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 2500,
      "train_flow_matching_loss": 0.224609375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.13291828334331512,
      "learning_rate": 5e-05,
      "loss": 0.2062,
      "step": 2510
    },
    {
      "step": 2510,
      "train_flow_matching_loss": 0.1533203125,
      "train_physics_loss": 0.9401140213012695
    },
    {
      "step": 2510,
      "train_flow_matching_loss": 0.251953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 2510,
      "train_flow_matching_loss": 0.1845703125,
      "train_physics_loss": 0.0
    },
    {
      "step": 2510,
      "train_flow_matching_loss": 0.21875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09229830652475357,
      "learning_rate": 5e-05,
      "loss": 0.2054,
      "step": 2520
    },
    {
      "step": 2520,
      "train_flow_matching_loss": 0.125,
      "train_physics_loss": 0.9552685022354126
    },
    {
      "step": 2520,
      "train_flow_matching_loss": 0.1357421875,
      "train_physics_loss": 0.0
    },
    {
      "step": 2520,
      "train_flow_matching_loss": 0.189453125,
      "train_physics_loss": 0.0
    },
    {
      "step": 2520,
      "train_flow_matching_loss": 0.302734375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.1923285722732544,
      "learning_rate": 5e-05,
      "loss": 0.2016,
      "step": 2530
    },
    {
      "step": 2530,
      "train_flow_matching_loss": 0.1552734375,
      "train_physics_loss": 0.9884737133979797
    },
    {
      "step": 2530,
      "train_flow_matching_loss": 0.12890625,
      "train_physics_loss": 0.0
    },
    {
      "step": 2530,
      "train_flow_matching_loss": 0.181640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 2530,
      "train_flow_matching_loss": 0.1220703125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.18301346898078918,
      "learning_rate": 5e-05,
      "loss": 0.2064,
      "step": 2540
    },
    {
      "step": 2540,
      "train_flow_matching_loss": 0.1494140625,
      "train_physics_loss": 0.9310185313224792
    },
    {
      "step": 2540,
      "train_flow_matching_loss": 0.142578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 2540,
      "train_flow_matching_loss": 0.1923828125,
      "train_physics_loss": 0.0
    },
    {
      "step": 2540,
      "train_flow_matching_loss": 0.1728515625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10958381742238998,
      "learning_rate": 5e-05,
      "loss": 0.2078,
      "step": 2550
    },
    {
      "step": 2550,
      "train_flow_matching_loss": 0.08935546875,
      "train_physics_loss": 0.9891444444656372
    },
    {
      "step": 2550,
      "train_flow_matching_loss": 0.1787109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 2550,
      "train_flow_matching_loss": 0.171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 2550,
      "train_flow_matching_loss": 0.1572265625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11116684228181839,
      "learning_rate": 5e-05,
      "loss": 0.2055,
      "step": 2560
    },
    {
      "step": 2560,
      "train_flow_matching_loss": 0.1611328125,
      "train_physics_loss": 1.057307243347168
    },
    {
      "step": 2560,
      "train_flow_matching_loss": 0.26953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 2560,
      "train_flow_matching_loss": 0.1796875,
      "train_physics_loss": 0.0
    },
    {
      "step": 2560,
      "train_flow_matching_loss": 0.0986328125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.15688340365886688,
      "learning_rate": 5e-05,
      "loss": 0.2051,
      "step": 2570
    },
    {
      "step": 2570,
      "train_flow_matching_loss": 0.1708984375,
      "train_physics_loss": 1.020572304725647
    },
    {
      "step": 2570,
      "train_flow_matching_loss": 0.2080078125,
      "train_physics_loss": 0.0
    },
    {
      "step": 2570,
      "train_flow_matching_loss": 0.2236328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 2570,
      "train_flow_matching_loss": 0.15625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.18636852502822876,
      "learning_rate": 5e-05,
      "loss": 0.2058,
      "step": 2580
    },
    {
      "step": 2580,
      "train_flow_matching_loss": 0.09423828125,
      "train_physics_loss": 1.0056198835372925
    },
    {
      "step": 2580,
      "train_flow_matching_loss": 0.220703125,
      "train_physics_loss": 0.0
    },
    {
      "step": 2580,
      "train_flow_matching_loss": 0.283203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 2580,
      "train_flow_matching_loss": 0.1435546875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11532717198133469,
      "learning_rate": 5e-05,
      "loss": 0.1995,
      "step": 2590
    },
    {
      "step": 2590,
      "train_flow_matching_loss": 0.12890625,
      "train_physics_loss": 1.0748783349990845
    },
    {
      "step": 2590,
      "train_flow_matching_loss": 0.1494140625,
      "train_physics_loss": 0.0
    },
    {
      "step": 2590,
      "train_flow_matching_loss": 0.197265625,
      "train_physics_loss": 0.0
    },
    {
      "step": 2590,
      "train_flow_matching_loss": 0.2021484375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.12218693643808365,
      "learning_rate": 5e-05,
      "loss": 0.1965,
      "step": 2600
    },
    {
      "step": 2600,
      "train_flow_matching_loss": 0.1376953125,
      "train_physics_loss": 0.9581121206283569
    },
    {
      "step": 2600,
      "train_flow_matching_loss": 0.2353515625,
      "train_physics_loss": 0.0
    },
    {
      "step": 2600,
      "train_flow_matching_loss": 0.236328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 2600,
      "train_flow_matching_loss": 0.1591796875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11486195772886276,
      "learning_rate": 5e-05,
      "loss": 0.1927,
      "step": 2610
    },
    {
      "step": 2610,
      "train_flow_matching_loss": 0.11962890625,
      "train_physics_loss": 0.9983457326889038
    },
    {
      "step": 2610,
      "train_flow_matching_loss": 0.197265625,
      "train_physics_loss": 0.0
    },
    {
      "step": 2610,
      "train_flow_matching_loss": 0.1962890625,
      "train_physics_loss": 0.0
    },
    {
      "step": 2610,
      "train_flow_matching_loss": 0.09619140625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.12302552908658981,
      "learning_rate": 5e-05,
      "loss": 0.1954,
      "step": 2620
    },
    {
      "step": 2620,
      "train_flow_matching_loss": 0.1064453125,
      "train_physics_loss": 0.9827998280525208
    },
    {
      "step": 2620,
      "train_flow_matching_loss": 0.1953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 2620,
      "train_flow_matching_loss": 0.1455078125,
      "train_physics_loss": 0.0
    },
    {
      "step": 2620,
      "train_flow_matching_loss": 0.1513671875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.108951635658741,
      "learning_rate": 5e-05,
      "loss": 0.1891,
      "step": 2630
    },
    {
      "step": 2630,
      "train_flow_matching_loss": 0.1005859375,
      "train_physics_loss": 0.9241379499435425
    },
    {
      "step": 2630,
      "train_flow_matching_loss": 0.1904296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 2630,
      "train_flow_matching_loss": 0.1962890625,
      "train_physics_loss": 0.0
    },
    {
      "step": 2630,
      "train_flow_matching_loss": 0.259765625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.18709032237529755,
      "learning_rate": 5e-05,
      "loss": 0.1901,
      "step": 2640
    },
    {
      "step": 2640,
      "train_flow_matching_loss": 0.1728515625,
      "train_physics_loss": 1.201164960861206
    },
    {
      "step": 2640,
      "train_flow_matching_loss": 0.1357421875,
      "train_physics_loss": 0.0
    },
    {
      "step": 2640,
      "train_flow_matching_loss": 0.171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 2640,
      "train_flow_matching_loss": 0.2001953125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.4626635015010834,
      "learning_rate": 5e-05,
      "loss": 0.1967,
      "step": 2650
    },
    {
      "step": 2650,
      "train_flow_matching_loss": 0.1865234375,
      "train_physics_loss": 1.0163475275039673
    },
    {
      "step": 2650,
      "train_flow_matching_loss": 0.287109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 2650,
      "train_flow_matching_loss": 0.1923828125,
      "train_physics_loss": 0.0
    },
    {
      "step": 2650,
      "train_flow_matching_loss": 0.173828125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.06189565360546112,
      "learning_rate": 5e-05,
      "loss": 0.2034,
      "step": 2660
    },
    {
      "step": 2660,
      "train_flow_matching_loss": 0.13671875,
      "train_physics_loss": 1.0290902853012085
    },
    {
      "step": 2660,
      "train_flow_matching_loss": 0.25390625,
      "train_physics_loss": 0.0
    },
    {
      "step": 2660,
      "train_flow_matching_loss": 0.197265625,
      "train_physics_loss": 0.0
    },
    {
      "step": 2660,
      "train_flow_matching_loss": 0.166015625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.1419411450624466,
      "learning_rate": 5e-05,
      "loss": 0.1945,
      "step": 2670
    },
    {
      "step": 2670,
      "train_flow_matching_loss": 0.07861328125,
      "train_physics_loss": 0.9300938248634338
    },
    {
      "step": 2670,
      "train_flow_matching_loss": 0.1787109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 2670,
      "train_flow_matching_loss": 0.197265625,
      "train_physics_loss": 0.0
    },
    {
      "step": 2670,
      "train_flow_matching_loss": 0.11279296875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11482983082532883,
      "learning_rate": 5e-05,
      "loss": 0.1969,
      "step": 2680
    },
    {
      "step": 2680,
      "train_flow_matching_loss": 0.1728515625,
      "train_physics_loss": 0.9940235018730164
    },
    {
      "step": 2680,
      "train_flow_matching_loss": 0.1748046875,
      "train_physics_loss": 0.0
    },
    {
      "step": 2680,
      "train_flow_matching_loss": 0.1962890625,
      "train_physics_loss": 0.0
    },
    {
      "step": 2680,
      "train_flow_matching_loss": 0.205078125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.14552277326583862,
      "learning_rate": 5e-05,
      "loss": 0.195,
      "step": 2690
    },
    {
      "step": 2690,
      "train_flow_matching_loss": 0.06982421875,
      "train_physics_loss": 0.994682788848877
    },
    {
      "step": 2690,
      "train_flow_matching_loss": 0.185546875,
      "train_physics_loss": 0.0
    },
    {
      "step": 2690,
      "train_flow_matching_loss": 0.203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 2690,
      "train_flow_matching_loss": 0.1376953125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.2408749759197235,
      "learning_rate": 5e-05,
      "loss": 0.1929,
      "step": 2700
    },
    {
      "step": 2700,
      "train_flow_matching_loss": 0.1025390625,
      "train_physics_loss": 0.8897818326950073
    },
    {
      "step": 2700,
      "train_flow_matching_loss": 0.1328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 2700,
      "train_flow_matching_loss": 0.1796875,
      "train_physics_loss": 0.0
    },
    {
      "step": 2700,
      "train_flow_matching_loss": 0.15625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.12755294144153595,
      "learning_rate": 5e-05,
      "loss": 0.1929,
      "step": 2710
    },
    {
      "step": 2710,
      "train_flow_matching_loss": 0.1064453125,
      "train_physics_loss": 1.008151888847351
    },
    {
      "step": 2710,
      "train_flow_matching_loss": 0.1689453125,
      "train_physics_loss": 0.0
    },
    {
      "step": 2710,
      "train_flow_matching_loss": 0.1298828125,
      "train_physics_loss": 0.0
    },
    {
      "step": 2710,
      "train_flow_matching_loss": 0.2158203125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.15350964665412903,
      "learning_rate": 5e-05,
      "loss": 0.1931,
      "step": 2720
    },
    {
      "step": 2720,
      "train_flow_matching_loss": 0.09326171875,
      "train_physics_loss": 1.0621260404586792
    },
    {
      "step": 2720,
      "train_flow_matching_loss": 0.1376953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 2720,
      "train_flow_matching_loss": 0.2001953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 2720,
      "train_flow_matching_loss": 0.1884765625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.12073871493339539,
      "learning_rate": 5e-05,
      "loss": 0.1922,
      "step": 2730
    },
    {
      "step": 2730,
      "train_flow_matching_loss": 0.154296875,
      "train_physics_loss": 0.9867119789123535
    },
    {
      "step": 2730,
      "train_flow_matching_loss": 0.12060546875,
      "train_physics_loss": 0.0
    },
    {
      "step": 2730,
      "train_flow_matching_loss": 0.2216796875,
      "train_physics_loss": 0.0
    },
    {
      "step": 2730,
      "train_flow_matching_loss": 0.1611328125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.15104609727859497,
      "learning_rate": 5e-05,
      "loss": 0.1914,
      "step": 2740
    },
    {
      "step": 2740,
      "train_flow_matching_loss": 0.1572265625,
      "train_physics_loss": 0.919122040271759
    },
    {
      "step": 2740,
      "train_flow_matching_loss": 0.2060546875,
      "train_physics_loss": 0.0
    },
    {
      "step": 2740,
      "train_flow_matching_loss": 0.173828125,
      "train_physics_loss": 0.0
    },
    {
      "step": 2740,
      "train_flow_matching_loss": 0.2060546875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10302640497684479,
      "learning_rate": 5e-05,
      "loss": 0.1889,
      "step": 2750
    },
    {
      "step": 2750,
      "train_flow_matching_loss": 0.1181640625,
      "train_physics_loss": 0.8963473439216614
    },
    {
      "step": 2750,
      "train_flow_matching_loss": 0.11474609375,
      "train_physics_loss": 0.0
    },
    {
      "step": 2750,
      "train_flow_matching_loss": 0.234375,
      "train_physics_loss": 0.0
    },
    {
      "step": 2750,
      "train_flow_matching_loss": 0.1845703125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09043946862220764,
      "learning_rate": 5e-05,
      "loss": 0.1944,
      "step": 2760
    },
    {
      "step": 2760,
      "train_flow_matching_loss": 0.1572265625,
      "train_physics_loss": 0.9852238297462463
    },
    {
      "step": 2760,
      "train_flow_matching_loss": 0.1884765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 2760,
      "train_flow_matching_loss": 0.1796875,
      "train_physics_loss": 0.0
    },
    {
      "step": 2760,
      "train_flow_matching_loss": 0.2041015625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.14381259679794312,
      "learning_rate": 5e-05,
      "loss": 0.1927,
      "step": 2770
    },
    {
      "step": 2770,
      "train_flow_matching_loss": 0.12451171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 2770,
      "train_flow_matching_loss": 0.1943359375,
      "train_physics_loss": 0.0
    },
    {
      "step": 2770,
      "train_flow_matching_loss": 0.203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 2770,
      "train_flow_matching_loss": 0.193359375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.13287146389484406,
      "learning_rate": 5e-05,
      "loss": 0.1946,
      "step": 2780
    },
    {
      "step": 2780,
      "train_flow_matching_loss": 0.150390625,
      "train_physics_loss": 0.0
    },
    {
      "step": 2780,
      "train_flow_matching_loss": 0.232421875,
      "train_physics_loss": 0.0
    },
    {
      "step": 2780,
      "train_flow_matching_loss": 0.234375,
      "train_physics_loss": 0.0
    },
    {
      "step": 2780,
      "train_flow_matching_loss": 0.2275390625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.12652908265590668,
      "learning_rate": 5e-05,
      "loss": 0.1876,
      "step": 2790
    },
    {
      "step": 2790,
      "train_flow_matching_loss": 0.11083984375,
      "train_physics_loss": 0.0
    },
    {
      "step": 2790,
      "train_flow_matching_loss": 0.1533203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 2790,
      "train_flow_matching_loss": 0.11181640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 2790,
      "train_flow_matching_loss": 0.1142578125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.1504422128200531,
      "learning_rate": 5e-05,
      "loss": 0.1928,
      "step": 2800
    },
    {
      "step": 2800,
      "train_flow_matching_loss": 0.1796875,
      "train_physics_loss": 0.0
    },
    {
      "step": 2800,
      "train_flow_matching_loss": 0.08837890625,
      "train_physics_loss": 0.0
    },
    {
      "step": 2800,
      "train_flow_matching_loss": 0.1474609375,
      "train_physics_loss": 0.0
    },
    {
      "step": 2800,
      "train_flow_matching_loss": 0.166015625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.1849260777235031,
      "learning_rate": 5e-05,
      "loss": 0.1917,
      "step": 2810
    },
    {
      "step": 2810,
      "train_flow_matching_loss": 0.11572265625,
      "train_physics_loss": 0.0
    },
    {
      "step": 2810,
      "train_flow_matching_loss": 0.1455078125,
      "train_physics_loss": 0.0
    },
    {
      "step": 2810,
      "train_flow_matching_loss": 0.1943359375,
      "train_physics_loss": 0.0
    },
    {
      "step": 2810,
      "train_flow_matching_loss": 0.15625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.16501669585704803,
      "learning_rate": 5e-05,
      "loss": 0.1859,
      "step": 2820
    },
    {
      "step": 2820,
      "train_flow_matching_loss": 0.166015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 2820,
      "train_flow_matching_loss": 0.09326171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 2820,
      "train_flow_matching_loss": 0.1474609375,
      "train_physics_loss": 0.0
    },
    {
      "step": 2820,
      "train_flow_matching_loss": 0.130859375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.21754291653633118,
      "learning_rate": 5e-05,
      "loss": 0.1877,
      "step": 2830
    },
    {
      "step": 2830,
      "train_flow_matching_loss": 0.1044921875,
      "train_physics_loss": 0.0
    },
    {
      "step": 2830,
      "train_flow_matching_loss": 0.232421875,
      "train_physics_loss": 0.0
    },
    {
      "step": 2830,
      "train_flow_matching_loss": 0.09130859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 2830,
      "train_flow_matching_loss": 0.2392578125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.15903332829475403,
      "learning_rate": 5e-05,
      "loss": 0.1895,
      "step": 2840
    },
    {
      "step": 2840,
      "train_flow_matching_loss": 0.1953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 2840,
      "train_flow_matching_loss": 0.1494140625,
      "train_physics_loss": 0.0
    },
    {
      "step": 2840,
      "train_flow_matching_loss": 0.1708984375,
      "train_physics_loss": 0.0
    },
    {
      "step": 2840,
      "train_flow_matching_loss": 0.224609375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.0906161218881607,
      "learning_rate": 5e-05,
      "loss": 0.1866,
      "step": 2850
    },
    {
      "step": 2850,
      "train_flow_matching_loss": 0.162109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 2850,
      "train_flow_matching_loss": 0.2470703125,
      "train_physics_loss": 0.0
    },
    {
      "step": 2850,
      "train_flow_matching_loss": 0.11328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 2850,
      "train_flow_matching_loss": 0.142578125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10960433632135391,
      "learning_rate": 5e-05,
      "loss": 0.1875,
      "step": 2860
    },
    {
      "step": 2860,
      "train_flow_matching_loss": 0.1708984375,
      "train_physics_loss": 0.0
    },
    {
      "step": 2860,
      "train_flow_matching_loss": 0.1630859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 2860,
      "train_flow_matching_loss": 0.1640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 2860,
      "train_flow_matching_loss": 0.1650390625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10510174930095673,
      "learning_rate": 5e-05,
      "loss": 0.1896,
      "step": 2870
    },
    {
      "step": 2870,
      "train_flow_matching_loss": 0.1591796875,
      "train_physics_loss": 0.0
    },
    {
      "step": 2870,
      "train_flow_matching_loss": 0.1748046875,
      "train_physics_loss": 0.0
    },
    {
      "step": 2870,
      "train_flow_matching_loss": 0.1474609375,
      "train_physics_loss": 0.0
    },
    {
      "step": 2870,
      "train_flow_matching_loss": 0.1796875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.15505878627300262,
      "learning_rate": 5e-05,
      "loss": 0.1876,
      "step": 2880
    },
    {
      "step": 2880,
      "train_flow_matching_loss": 0.11083984375,
      "train_physics_loss": 0.0
    },
    {
      "step": 2880,
      "train_flow_matching_loss": 0.1826171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 2880,
      "train_flow_matching_loss": 0.205078125,
      "train_physics_loss": 0.0
    },
    {
      "step": 2880,
      "train_flow_matching_loss": 0.1640625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.1738130748271942,
      "learning_rate": 5e-05,
      "loss": 0.1867,
      "step": 2890
    },
    {
      "step": 2890,
      "train_flow_matching_loss": 0.1806640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 2890,
      "train_flow_matching_loss": 0.1220703125,
      "train_physics_loss": 0.0
    },
    {
      "step": 2890,
      "train_flow_matching_loss": 0.171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 2890,
      "train_flow_matching_loss": 0.1396484375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.14162543416023254,
      "learning_rate": 5e-05,
      "loss": 0.1949,
      "step": 2900
    },
    {
      "step": 2900,
      "train_flow_matching_loss": 0.1650390625,
      "train_physics_loss": 0.0
    },
    {
      "step": 2900,
      "train_flow_matching_loss": 0.283203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 2900,
      "train_flow_matching_loss": 0.2294921875,
      "train_physics_loss": 0.0
    },
    {
      "step": 2900,
      "train_flow_matching_loss": 0.2138671875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.13592840731143951,
      "learning_rate": 5e-05,
      "loss": 0.1923,
      "step": 2910
    },
    {
      "step": 2910,
      "train_flow_matching_loss": 0.1435546875,
      "train_physics_loss": 0.0
    },
    {
      "step": 2910,
      "train_flow_matching_loss": 0.171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 2910,
      "train_flow_matching_loss": 0.1376953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 2910,
      "train_flow_matching_loss": 0.333984375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09782455861568451,
      "learning_rate": 5e-05,
      "loss": 0.1907,
      "step": 2920
    },
    {
      "step": 2920,
      "train_flow_matching_loss": 0.1708984375,
      "train_physics_loss": 0.0
    },
    {
      "step": 2920,
      "train_flow_matching_loss": 0.1455078125,
      "train_physics_loss": 0.0
    },
    {
      "step": 2920,
      "train_flow_matching_loss": 0.173828125,
      "train_physics_loss": 0.0
    },
    {
      "step": 2920,
      "train_flow_matching_loss": 0.1513671875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11259377747774124,
      "learning_rate": 5e-05,
      "loss": 0.1839,
      "step": 2930
    },
    {
      "step": 2930,
      "train_flow_matching_loss": 0.1533203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 2930,
      "train_flow_matching_loss": 0.154296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 2930,
      "train_flow_matching_loss": 0.1279296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 2930,
      "train_flow_matching_loss": 0.15625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10918223857879639,
      "learning_rate": 5e-05,
      "loss": 0.1885,
      "step": 2940
    },
    {
      "step": 2940,
      "train_flow_matching_loss": 0.267578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 2940,
      "train_flow_matching_loss": 0.11865234375,
      "train_physics_loss": 0.0
    },
    {
      "step": 2940,
      "train_flow_matching_loss": 0.212890625,
      "train_physics_loss": 0.0
    },
    {
      "step": 2940,
      "train_flow_matching_loss": 0.142578125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08569598197937012,
      "learning_rate": 5e-05,
      "loss": 0.1849,
      "step": 2950
    },
    {
      "step": 2950,
      "train_flow_matching_loss": 0.1650390625,
      "train_physics_loss": 0.0
    },
    {
      "step": 2950,
      "train_flow_matching_loss": 0.1435546875,
      "train_physics_loss": 0.0
    },
    {
      "step": 2950,
      "train_flow_matching_loss": 0.2099609375,
      "train_physics_loss": 0.0
    },
    {
      "step": 2950,
      "train_flow_matching_loss": 0.2353515625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.2844478189945221,
      "learning_rate": 5e-05,
      "loss": 0.1932,
      "step": 2960
    },
    {
      "step": 2960,
      "train_flow_matching_loss": 0.1845703125,
      "train_physics_loss": 0.0
    },
    {
      "step": 2960,
      "train_flow_matching_loss": 0.1796875,
      "train_physics_loss": 0.0
    },
    {
      "step": 2960,
      "train_flow_matching_loss": 0.08447265625,
      "train_physics_loss": 0.0
    },
    {
      "step": 2960,
      "train_flow_matching_loss": 0.267578125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08413069695234299,
      "learning_rate": 5e-05,
      "loss": 0.1928,
      "step": 2970
    },
    {
      "step": 2970,
      "train_flow_matching_loss": 0.09326171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 2970,
      "train_flow_matching_loss": 0.208984375,
      "train_physics_loss": 0.0
    },
    {
      "step": 2970,
      "train_flow_matching_loss": 0.1396484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 2970,
      "train_flow_matching_loss": 0.1962890625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.15576767921447754,
      "learning_rate": 5e-05,
      "loss": 0.1869,
      "step": 2980
    },
    {
      "step": 2980,
      "train_flow_matching_loss": 0.1650390625,
      "train_physics_loss": 0.0
    },
    {
      "step": 2980,
      "train_flow_matching_loss": 0.251953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 2980,
      "train_flow_matching_loss": 0.1103515625,
      "train_physics_loss": 0.0
    },
    {
      "step": 2980,
      "train_flow_matching_loss": 0.1484375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.14395292103290558,
      "learning_rate": 5e-05,
      "loss": 0.1839,
      "step": 2990
    },
    {
      "step": 2990,
      "train_flow_matching_loss": 0.1005859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 2990,
      "train_flow_matching_loss": 0.2109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 2990,
      "train_flow_matching_loss": 0.1884765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 2990,
      "train_flow_matching_loss": 0.1708984375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11649023741483688,
      "learning_rate": 5e-05,
      "loss": 0.1913,
      "step": 3000
    },
    {
      "step": 3000,
      "train_flow_matching_loss": 0.0947265625,
      "train_physics_loss": 0.0
    },
    {
      "step": 3000,
      "train_flow_matching_loss": 0.25,
      "train_physics_loss": 0.0
    },
    {
      "step": 3000,
      "train_flow_matching_loss": 0.1513671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 3000,
      "train_flow_matching_loss": 0.15625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11192353069782257,
      "learning_rate": 5e-05,
      "loss": 0.1832,
      "step": 3010
    },
    {
      "step": 3010,
      "train_flow_matching_loss": 0.1572265625,
      "train_physics_loss": 0.0
    },
    {
      "step": 3010,
      "train_flow_matching_loss": 0.1533203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 3010,
      "train_flow_matching_loss": 0.1689453125,
      "train_physics_loss": 0.0
    },
    {
      "step": 3010,
      "train_flow_matching_loss": 0.279296875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11475014686584473,
      "learning_rate": 5e-05,
      "loss": 0.1934,
      "step": 3020
    },
    {
      "step": 3020,
      "train_flow_matching_loss": 0.11669921875,
      "train_physics_loss": 0.0
    },
    {
      "step": 3020,
      "train_flow_matching_loss": 0.0869140625,
      "train_physics_loss": 0.0
    },
    {
      "step": 3020,
      "train_flow_matching_loss": 0.1298828125,
      "train_physics_loss": 0.0
    },
    {
      "step": 3020,
      "train_flow_matching_loss": 0.134765625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.16289831697940826,
      "learning_rate": 5e-05,
      "loss": 0.1836,
      "step": 3030
    },
    {
      "step": 3030,
      "train_flow_matching_loss": 0.212890625,
      "train_physics_loss": 0.0
    },
    {
      "step": 3030,
      "train_flow_matching_loss": 0.2255859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 3030,
      "train_flow_matching_loss": 0.09326171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 3030,
      "train_flow_matching_loss": 0.189453125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.13720302283763885,
      "learning_rate": 5e-05,
      "loss": 0.1851,
      "step": 3040
    },
    {
      "step": 3040,
      "train_flow_matching_loss": 0.1689453125,
      "train_physics_loss": 0.0
    },
    {
      "step": 3040,
      "train_flow_matching_loss": 0.1923828125,
      "train_physics_loss": 0.0
    },
    {
      "step": 3040,
      "train_flow_matching_loss": 0.11865234375,
      "train_physics_loss": 0.0
    },
    {
      "step": 3040,
      "train_flow_matching_loss": 0.244140625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.27521172165870667,
      "learning_rate": 5e-05,
      "loss": 0.1862,
      "step": 3050
    },
    {
      "step": 3050,
      "train_flow_matching_loss": 0.1572265625,
      "train_physics_loss": 0.0
    },
    {
      "step": 3050,
      "train_flow_matching_loss": 0.1689453125,
      "train_physics_loss": 0.0
    },
    {
      "step": 3050,
      "train_flow_matching_loss": 0.1865234375,
      "train_physics_loss": 0.0
    },
    {
      "step": 3050,
      "train_flow_matching_loss": 0.2109375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.19177265465259552,
      "learning_rate": 5e-05,
      "loss": 0.1844,
      "step": 3060
    },
    {
      "step": 3060,
      "train_flow_matching_loss": 0.11474609375,
      "train_physics_loss": 0.0
    },
    {
      "step": 3060,
      "train_flow_matching_loss": 0.25390625,
      "train_physics_loss": 0.0
    },
    {
      "step": 3060,
      "train_flow_matching_loss": 0.1611328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 3060,
      "train_flow_matching_loss": 0.2099609375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10352201014757156,
      "learning_rate": 5e-05,
      "loss": 0.1905,
      "step": 3070
    },
    {
      "step": 3070,
      "train_flow_matching_loss": 0.26171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 3070,
      "train_flow_matching_loss": 0.1572265625,
      "train_physics_loss": 0.0
    },
    {
      "step": 3070,
      "train_flow_matching_loss": 0.1845703125,
      "train_physics_loss": 0.0
    },
    {
      "step": 3070,
      "train_flow_matching_loss": 0.2119140625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.23573462665081024,
      "learning_rate": 5e-05,
      "loss": 0.1832,
      "step": 3080
    },
    {
      "step": 3080,
      "train_flow_matching_loss": 0.1474609375,
      "train_physics_loss": 0.0
    },
    {
      "step": 3080,
      "train_flow_matching_loss": 0.1953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 3080,
      "train_flow_matching_loss": 0.1142578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 3080,
      "train_flow_matching_loss": 0.1767578125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08758249878883362,
      "learning_rate": 5e-05,
      "loss": 0.1881,
      "step": 3090
    },
    {
      "step": 3090,
      "train_flow_matching_loss": 0.1552734375,
      "train_physics_loss": 0.0
    },
    {
      "step": 3090,
      "train_flow_matching_loss": 0.1669921875,
      "train_physics_loss": 0.0
    },
    {
      "step": 3090,
      "train_flow_matching_loss": 0.08935546875,
      "train_physics_loss": 0.0
    },
    {
      "step": 3090,
      "train_flow_matching_loss": 0.224609375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.13579989969730377,
      "learning_rate": 5e-05,
      "loss": 0.1905,
      "step": 3100
    },
    {
      "step": 3100,
      "train_flow_matching_loss": 0.259765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 3100,
      "train_flow_matching_loss": 0.142578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 3100,
      "train_flow_matching_loss": 0.2236328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 3100,
      "train_flow_matching_loss": 0.1455078125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.5149375200271606,
      "learning_rate": 5e-05,
      "loss": 0.1899,
      "step": 3110
    },
    {
      "step": 3110,
      "train_flow_matching_loss": 0.11767578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 3110,
      "train_flow_matching_loss": 0.1376953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 3110,
      "train_flow_matching_loss": 0.193359375,
      "train_physics_loss": 0.0
    },
    {
      "step": 3110,
      "train_flow_matching_loss": 0.12060546875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.15543027222156525,
      "learning_rate": 5e-05,
      "loss": 0.1912,
      "step": 3120
    },
    {
      "step": 3120,
      "train_flow_matching_loss": 0.1259765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 3120,
      "train_flow_matching_loss": 0.1689453125,
      "train_physics_loss": 0.0
    },
    {
      "step": 3120,
      "train_flow_matching_loss": 0.1875,
      "train_physics_loss": 0.0
    },
    {
      "step": 3120,
      "train_flow_matching_loss": 0.201171875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.15977627038955688,
      "learning_rate": 5e-05,
      "loss": 0.1877,
      "step": 3130
    },
    {
      "step": 3130,
      "train_flow_matching_loss": 0.09716796875,
      "train_physics_loss": 0.0
    },
    {
      "step": 3130,
      "train_flow_matching_loss": 0.1572265625,
      "train_physics_loss": 0.0
    },
    {
      "step": 3130,
      "train_flow_matching_loss": 0.142578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 3130,
      "train_flow_matching_loss": 0.1640625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.12678298354148865,
      "learning_rate": 5e-05,
      "loss": 0.1873,
      "step": 3140
    },
    {
      "step": 3140,
      "train_flow_matching_loss": 0.173828125,
      "train_physics_loss": 0.0
    },
    {
      "step": 3140,
      "train_flow_matching_loss": 0.1484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 3140,
      "train_flow_matching_loss": 0.1669921875,
      "train_physics_loss": 0.0
    },
    {
      "step": 3140,
      "train_flow_matching_loss": 0.1962890625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10928988456726074,
      "learning_rate": 5e-05,
      "loss": 0.1865,
      "step": 3150
    },
    {
      "step": 3150,
      "train_flow_matching_loss": 0.1767578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 3150,
      "train_flow_matching_loss": 0.2353515625,
      "train_physics_loss": 0.0
    },
    {
      "step": 3150,
      "train_flow_matching_loss": 0.142578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 3150,
      "train_flow_matching_loss": 0.166015625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11322753876447678,
      "learning_rate": 5e-05,
      "loss": 0.189,
      "step": 3160
    },
    {
      "step": 3160,
      "train_flow_matching_loss": 0.228515625,
      "train_physics_loss": 0.0
    },
    {
      "step": 3160,
      "train_flow_matching_loss": 0.15625,
      "train_physics_loss": 0.0
    },
    {
      "step": 3160,
      "train_flow_matching_loss": 0.07861328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 3160,
      "train_flow_matching_loss": 0.1396484375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.23447705805301666,
      "learning_rate": 5e-05,
      "loss": 0.1821,
      "step": 3170
    },
    {
      "step": 3170,
      "train_flow_matching_loss": 0.2158203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 3170,
      "train_flow_matching_loss": 0.201171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 3170,
      "train_flow_matching_loss": 0.134765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 3170,
      "train_flow_matching_loss": 0.1455078125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09865536540746689,
      "learning_rate": 5e-05,
      "loss": 0.1829,
      "step": 3180
    },
    {
      "step": 3180,
      "train_flow_matching_loss": 0.255859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 3180,
      "train_flow_matching_loss": 0.1484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 3180,
      "train_flow_matching_loss": 0.162109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 3180,
      "train_flow_matching_loss": 0.224609375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.21032637357711792,
      "learning_rate": 5e-05,
      "loss": 0.1841,
      "step": 3190
    },
    {
      "step": 3190,
      "train_flow_matching_loss": 0.255859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 3190,
      "train_flow_matching_loss": 0.1630859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 3190,
      "train_flow_matching_loss": 0.2138671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 3190,
      "train_flow_matching_loss": 0.150390625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.14306220412254333,
      "learning_rate": 5e-05,
      "loss": 0.1894,
      "step": 3200
    },
    {
      "step": 3200,
      "train_flow_matching_loss": 0.1865234375,
      "train_physics_loss": 0.0
    },
    {
      "step": 3200,
      "train_flow_matching_loss": 0.1484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 3200,
      "train_flow_matching_loss": 0.12158203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 3200,
      "train_flow_matching_loss": 0.169921875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.13046635687351227,
      "learning_rate": 5e-05,
      "loss": 0.184,
      "step": 3210
    },
    {
      "step": 3210,
      "train_flow_matching_loss": 0.1923828125,
      "train_physics_loss": 0.0
    },
    {
      "step": 3210,
      "train_flow_matching_loss": 0.189453125,
      "train_physics_loss": 0.0
    },
    {
      "step": 3210,
      "train_flow_matching_loss": 0.1103515625,
      "train_physics_loss": 0.0
    },
    {
      "step": 3210,
      "train_flow_matching_loss": 0.2109375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.1036713644862175,
      "learning_rate": 5e-05,
      "loss": 0.1761,
      "step": 3220
    },
    {
      "step": 3220,
      "train_flow_matching_loss": 0.0654296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 3220,
      "train_flow_matching_loss": 0.203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 3220,
      "train_flow_matching_loss": 0.1650390625,
      "train_physics_loss": 0.0
    },
    {
      "step": 3220,
      "train_flow_matching_loss": 0.185546875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.2297234684228897,
      "learning_rate": 5e-05,
      "loss": 0.1834,
      "step": 3230
    },
    {
      "step": 3230,
      "train_flow_matching_loss": 0.1552734375,
      "train_physics_loss": 0.0
    },
    {
      "step": 3230,
      "train_flow_matching_loss": 0.1767578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 3230,
      "train_flow_matching_loss": 0.09765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 3230,
      "train_flow_matching_loss": 0.2109375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.12770432233810425,
      "learning_rate": 5e-05,
      "loss": 0.1819,
      "step": 3240
    },
    {
      "step": 3240,
      "train_flow_matching_loss": 0.228515625,
      "train_physics_loss": 0.0
    },
    {
      "step": 3240,
      "train_flow_matching_loss": 0.1298828125,
      "train_physics_loss": 0.0
    },
    {
      "step": 3240,
      "train_flow_matching_loss": 0.138671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 3240,
      "train_flow_matching_loss": 0.10595703125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.06653868407011032,
      "learning_rate": 5e-05,
      "loss": 0.1792,
      "step": 3250
    },
    {
      "step": 3250,
      "train_flow_matching_loss": 0.146484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 3250,
      "train_flow_matching_loss": 0.1171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 3250,
      "train_flow_matching_loss": 0.1533203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 3250,
      "train_flow_matching_loss": 0.171875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11012265086174011,
      "learning_rate": 5e-05,
      "loss": 0.1868,
      "step": 3260
    },
    {
      "step": 3260,
      "train_flow_matching_loss": 0.189453125,
      "train_physics_loss": 0.0
    },
    {
      "step": 3260,
      "train_flow_matching_loss": 0.146484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 3260,
      "train_flow_matching_loss": 0.140625,
      "train_physics_loss": 0.0
    },
    {
      "step": 3260,
      "train_flow_matching_loss": 0.1396484375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11391524970531464,
      "learning_rate": 5e-05,
      "loss": 0.1809,
      "step": 3270
    },
    {
      "step": 3270,
      "train_flow_matching_loss": 0.29296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 3270,
      "train_flow_matching_loss": 0.138671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 3270,
      "train_flow_matching_loss": 0.169921875,
      "train_physics_loss": 0.0
    },
    {
      "step": 3270,
      "train_flow_matching_loss": 0.1591796875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.15761251747608185,
      "learning_rate": 5e-05,
      "loss": 0.1836,
      "step": 3280
    },
    {
      "step": 3280,
      "train_flow_matching_loss": 0.1591796875,
      "train_physics_loss": 0.0
    },
    {
      "step": 3280,
      "train_flow_matching_loss": 0.07666015625,
      "train_physics_loss": 0.8966520428657532
    },
    {
      "step": 3280,
      "train_flow_matching_loss": 0.1416015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 3280,
      "train_flow_matching_loss": 0.1494140625,
      "train_physics_loss": 0.9670743942260742
    },
    {
      "grad_norm": 0.16461706161499023,
      "learning_rate": 5e-05,
      "loss": 0.182,
      "step": 3290
    },
    {
      "step": 3290,
      "train_flow_matching_loss": 0.197265625,
      "train_physics_loss": 0.0
    },
    {
      "step": 3290,
      "train_flow_matching_loss": 0.05615234375,
      "train_physics_loss": 0.9249544739723206
    },
    {
      "step": 3290,
      "train_flow_matching_loss": 0.1640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 3290,
      "train_flow_matching_loss": 0.09375,
      "train_physics_loss": 0.9386805295944214
    },
    {
      "grad_norm": 0.1427697241306305,
      "learning_rate": 5e-05,
      "loss": 0.1798,
      "step": 3300
    },
    {
      "step": 3300,
      "train_flow_matching_loss": 0.2060546875,
      "train_physics_loss": 0.0
    },
    {
      "step": 3300,
      "train_flow_matching_loss": 0.09228515625,
      "train_physics_loss": 0.9660555124282837
    },
    {
      "step": 3300,
      "train_flow_matching_loss": 0.1669921875,
      "train_physics_loss": 0.0
    },
    {
      "step": 3300,
      "train_flow_matching_loss": 0.08447265625,
      "train_physics_loss": 0.9521654844284058
    },
    {
      "grad_norm": 0.25299328565597534,
      "learning_rate": 5e-05,
      "loss": 0.1813,
      "step": 3310
    },
    {
      "step": 3310,
      "train_flow_matching_loss": 0.2236328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 3310,
      "train_flow_matching_loss": 0.10107421875,
      "train_physics_loss": 1.0018904209136963
    },
    {
      "step": 3310,
      "train_flow_matching_loss": 0.154296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 3310,
      "train_flow_matching_loss": 0.0771484375,
      "train_physics_loss": 0.9028459191322327
    },
    {
      "grad_norm": 0.10584941506385803,
      "learning_rate": 5e-05,
      "loss": 0.1816,
      "step": 3320
    },
    {
      "step": 3320,
      "train_flow_matching_loss": 0.158203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 3320,
      "train_flow_matching_loss": 0.09619140625,
      "train_physics_loss": 0.9473683834075928
    },
    {
      "step": 3320,
      "train_flow_matching_loss": 0.1552734375,
      "train_physics_loss": 0.0
    },
    {
      "step": 3320,
      "train_flow_matching_loss": 0.06884765625,
      "train_physics_loss": 0.8724405169487
    },
    {
      "grad_norm": 0.44164302945137024,
      "learning_rate": 5e-05,
      "loss": 0.1837,
      "step": 3330
    },
    {
      "step": 3330,
      "train_flow_matching_loss": 0.2041015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 3330,
      "train_flow_matching_loss": 0.0966796875,
      "train_physics_loss": 0.9694034457206726
    },
    {
      "step": 3330,
      "train_flow_matching_loss": 0.158203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 3330,
      "train_flow_matching_loss": 0.10107421875,
      "train_physics_loss": 1.1266826391220093
    },
    {
      "grad_norm": 0.11820387095212936,
      "learning_rate": 5e-05,
      "loss": 0.1802,
      "step": 3340
    },
    {
      "step": 3340,
      "train_flow_matching_loss": 0.212890625,
      "train_physics_loss": 0.0
    },
    {
      "step": 3340,
      "train_flow_matching_loss": 0.09716796875,
      "train_physics_loss": 0.9679458141326904
    },
    {
      "step": 3340,
      "train_flow_matching_loss": 0.2470703125,
      "train_physics_loss": 0.0
    },
    {
      "step": 3340,
      "train_flow_matching_loss": 0.0712890625,
      "train_physics_loss": 1.003904938697815
    },
    {
      "grad_norm": 0.1247769147157669,
      "learning_rate": 5e-05,
      "loss": 0.1875,
      "step": 3350
    },
    {
      "step": 3350,
      "train_flow_matching_loss": 0.10205078125,
      "train_physics_loss": 0.8769217729568481
    },
    {
      "step": 3350,
      "train_flow_matching_loss": 0.058349609375,
      "train_physics_loss": 0.9055207371711731
    },
    {
      "step": 3350,
      "train_flow_matching_loss": 0.1962890625,
      "train_physics_loss": 0.0
    },
    {
      "step": 3350,
      "train_flow_matching_loss": 0.1103515625,
      "train_physics_loss": 0.9417082667350769
    },
    {
      "grad_norm": 0.15405488014221191,
      "learning_rate": 5e-05,
      "loss": 0.1821,
      "step": 3360
    },
    {
      "step": 3360,
      "train_flow_matching_loss": 0.09033203125,
      "train_physics_loss": 0.8951643109321594
    },
    {
      "step": 3360,
      "train_flow_matching_loss": 0.11669921875,
      "train_physics_loss": 0.9672820568084717
    },
    {
      "step": 3360,
      "train_flow_matching_loss": 0.1953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 3360,
      "train_flow_matching_loss": 0.1376953125,
      "train_physics_loss": 1.0212256908416748
    },
    {
      "grad_norm": 0.4048524498939514,
      "learning_rate": 5e-05,
      "loss": 0.188,
      "step": 3370
    },
    {
      "step": 3370,
      "train_flow_matching_loss": 0.09375,
      "train_physics_loss": 0.9767728447914124
    },
    {
      "step": 3370,
      "train_flow_matching_loss": 0.2734375,
      "train_physics_loss": 1.0479967594146729
    },
    {
      "step": 3370,
      "train_flow_matching_loss": 0.130859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 3370,
      "train_flow_matching_loss": 0.08984375,
      "train_physics_loss": 1.007917046546936
    },
    {
      "grad_norm": 0.12323581427335739,
      "learning_rate": 5e-05,
      "loss": 0.1894,
      "step": 3380
    },
    {
      "step": 3380,
      "train_flow_matching_loss": 0.2177734375,
      "train_physics_loss": 0.9289358854293823
    },
    {
      "step": 3380,
      "train_flow_matching_loss": 0.09228515625,
      "train_physics_loss": 1.015120506286621
    },
    {
      "step": 3380,
      "train_flow_matching_loss": 0.1669921875,
      "train_physics_loss": 0.0
    },
    {
      "step": 3380,
      "train_flow_matching_loss": 0.1533203125,
      "train_physics_loss": 0.9578828811645508
    },
    {
      "grad_norm": 0.1372767686843872,
      "learning_rate": 5e-05,
      "loss": 0.1838,
      "step": 3390
    },
    {
      "step": 3390,
      "train_flow_matching_loss": 0.1318359375,
      "train_physics_loss": 0.9206938743591309
    },
    {
      "step": 3390,
      "train_flow_matching_loss": 0.1142578125,
      "train_physics_loss": 0.9875980615615845
    },
    {
      "step": 3390,
      "train_flow_matching_loss": 0.171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 3390,
      "train_flow_matching_loss": 0.047607421875,
      "train_physics_loss": 1.0130962133407593
    },
    {
      "grad_norm": 0.12011703848838806,
      "learning_rate": 5e-05,
      "loss": 0.1891,
      "step": 3400
    },
    {
      "step": 3400,
      "train_flow_matching_loss": 0.11279296875,
      "train_physics_loss": 0.9582490921020508
    },
    {
      "step": 3400,
      "train_flow_matching_loss": 0.126953125,
      "train_physics_loss": 0.9319745898246765
    },
    {
      "step": 3400,
      "train_flow_matching_loss": 0.08740234375,
      "train_physics_loss": 0.0
    },
    {
      "step": 3400,
      "train_flow_matching_loss": 0.10107421875,
      "train_physics_loss": 0.9819138050079346
    },
    {
      "grad_norm": 0.1547250747680664,
      "learning_rate": 5e-05,
      "loss": 0.1829,
      "step": 3410
    },
    {
      "step": 3410,
      "train_flow_matching_loss": 0.12060546875,
      "train_physics_loss": 0.9307671785354614
    },
    {
      "step": 3410,
      "train_flow_matching_loss": 0.08251953125,
      "train_physics_loss": 1.069166898727417
    },
    {
      "step": 3410,
      "train_flow_matching_loss": 0.201171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 3410,
      "train_flow_matching_loss": 0.1162109375,
      "train_physics_loss": 0.9524369835853577
    },
    {
      "grad_norm": 0.13315348327159882,
      "learning_rate": 5e-05,
      "loss": 0.185,
      "step": 3420
    },
    {
      "step": 3420,
      "train_flow_matching_loss": 0.0888671875,
      "train_physics_loss": 1.0432740449905396
    },
    {
      "step": 3420,
      "train_flow_matching_loss": 0.10302734375,
      "train_physics_loss": 0.8951514959335327
    },
    {
      "step": 3420,
      "train_flow_matching_loss": 0.171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 3420,
      "train_flow_matching_loss": 0.06494140625,
      "train_physics_loss": 1.0045738220214844
    },
    {
      "grad_norm": 0.08044639229774475,
      "learning_rate": 5e-05,
      "loss": 0.1847,
      "step": 3430
    },
    {
      "step": 3430,
      "train_flow_matching_loss": 0.0712890625,
      "train_physics_loss": 0.9586800336837769
    },
    {
      "step": 3430,
      "train_flow_matching_loss": 0.10302734375,
      "train_physics_loss": 0.963385820388794
    },
    {
      "step": 3430,
      "train_flow_matching_loss": 0.1474609375,
      "train_physics_loss": 0.0
    },
    {
      "step": 3430,
      "train_flow_matching_loss": 0.10888671875,
      "train_physics_loss": 0.8822524547576904
    },
    {
      "grad_norm": 0.11478058993816376,
      "learning_rate": 5e-05,
      "loss": 0.1852,
      "step": 3440
    },
    {
      "step": 3440,
      "train_flow_matching_loss": 0.1162109375,
      "train_physics_loss": 1.0182164907455444
    },
    {
      "step": 3440,
      "train_flow_matching_loss": 0.09521484375,
      "train_physics_loss": 0.9546613693237305
    },
    {
      "step": 3440,
      "train_flow_matching_loss": 0.203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 3440,
      "train_flow_matching_loss": 0.08740234375,
      "train_physics_loss": 0.9382050037384033
    },
    {
      "grad_norm": 0.23136599361896515,
      "learning_rate": 5e-05,
      "loss": 0.1829,
      "step": 3450
    },
    {
      "step": 3450,
      "train_flow_matching_loss": 0.06787109375,
      "train_physics_loss": 0.9353094696998596
    },
    {
      "step": 3450,
      "train_flow_matching_loss": 0.07080078125,
      "train_physics_loss": 0.9613434672355652
    },
    {
      "step": 3450,
      "train_flow_matching_loss": 0.2294921875,
      "train_physics_loss": 0.0
    },
    {
      "step": 3450,
      "train_flow_matching_loss": 0.2001953125,
      "train_physics_loss": 0.9247453808784485
    },
    {
      "grad_norm": 0.14401018619537354,
      "learning_rate": 5e-05,
      "loss": 0.1865,
      "step": 3460
    },
    {
      "step": 3460,
      "train_flow_matching_loss": 0.05517578125,
      "train_physics_loss": 0.9244439005851746
    },
    {
      "step": 3460,
      "train_flow_matching_loss": 0.064453125,
      "train_physics_loss": 0.9316753149032593
    },
    {
      "step": 3460,
      "train_flow_matching_loss": 0.248046875,
      "train_physics_loss": 0.0
    },
    {
      "step": 3460,
      "train_flow_matching_loss": 0.06591796875,
      "train_physics_loss": 0.9548230767250061
    },
    {
      "grad_norm": 0.15282097458839417,
      "learning_rate": 5e-05,
      "loss": 0.177,
      "step": 3470
    },
    {
      "step": 3470,
      "train_flow_matching_loss": 0.138671875,
      "train_physics_loss": 0.9596405029296875
    },
    {
      "step": 3470,
      "train_flow_matching_loss": 0.08544921875,
      "train_physics_loss": 1.0851150751113892
    },
    {
      "step": 3470,
      "train_flow_matching_loss": 0.162109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 3470,
      "train_flow_matching_loss": 0.06982421875,
      "train_physics_loss": 0.8927248120307922
    },
    {
      "grad_norm": 0.14173978567123413,
      "learning_rate": 5e-05,
      "loss": 0.1857,
      "step": 3480
    },
    {
      "step": 3480,
      "train_flow_matching_loss": 0.1318359375,
      "train_physics_loss": 0.9348333477973938
    },
    {
      "step": 3480,
      "train_flow_matching_loss": 0.06298828125,
      "train_physics_loss": 0.8068230152130127
    },
    {
      "step": 3480,
      "train_flow_matching_loss": 0.11669921875,
      "train_physics_loss": 0.0
    },
    {
      "step": 3480,
      "train_flow_matching_loss": 0.0654296875,
      "train_physics_loss": 0.8593722581863403
    },
    {
      "grad_norm": 0.10678332298994064,
      "learning_rate": 5e-05,
      "loss": 0.1838,
      "step": 3490
    },
    {
      "step": 3490,
      "train_flow_matching_loss": 0.1376953125,
      "train_physics_loss": 1.0169451236724854
    },
    {
      "step": 3490,
      "train_flow_matching_loss": 0.12255859375,
      "train_physics_loss": 1.048445463180542
    },
    {
      "step": 3490,
      "train_flow_matching_loss": 0.150390625,
      "train_physics_loss": 0.0
    },
    {
      "step": 3490,
      "train_flow_matching_loss": 0.1357421875,
      "train_physics_loss": 0.9141648411750793
    },
    {
      "grad_norm": 0.11102515459060669,
      "learning_rate": 5e-05,
      "loss": 0.1924,
      "step": 3500
    },
    {
      "step": 3500,
      "train_flow_matching_loss": 0.09033203125,
      "train_physics_loss": 0.9214987754821777
    },
    {
      "step": 3500,
      "train_flow_matching_loss": 0.07275390625,
      "train_physics_loss": 0.9205106496810913
    },
    {
      "step": 3500,
      "train_flow_matching_loss": 0.1435546875,
      "train_physics_loss": 0.0
    },
    {
      "step": 3500,
      "train_flow_matching_loss": 0.1640625,
      "train_physics_loss": 0.8773791193962097
    },
    {
      "grad_norm": 0.09460658580064774,
      "learning_rate": 5e-05,
      "loss": 0.1845,
      "step": 3510
    },
    {
      "step": 3510,
      "train_flow_matching_loss": 0.1005859375,
      "train_physics_loss": 0.910069465637207
    },
    {
      "step": 3510,
      "train_flow_matching_loss": 0.036865234375,
      "train_physics_loss": 0.9135082364082336
    },
    {
      "step": 3510,
      "train_flow_matching_loss": 0.09716796875,
      "train_physics_loss": 0.0
    },
    {
      "step": 3510,
      "train_flow_matching_loss": 0.04248046875,
      "train_physics_loss": 0.8721149563789368
    },
    {
      "grad_norm": 0.11733721941709518,
      "learning_rate": 5e-05,
      "loss": 0.1774,
      "step": 3520
    },
    {
      "step": 3520,
      "train_flow_matching_loss": 0.078125,
      "train_physics_loss": 0.9687926769256592
    },
    {
      "step": 3520,
      "train_flow_matching_loss": 0.1103515625,
      "train_physics_loss": 0.9532767534255981
    },
    {
      "step": 3520,
      "train_flow_matching_loss": 0.201171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 3520,
      "train_flow_matching_loss": 0.0947265625,
      "train_physics_loss": 0.9744883179664612
    },
    {
      "grad_norm": 0.11856431514024734,
      "learning_rate": 5e-05,
      "loss": 0.1817,
      "step": 3530
    },
    {
      "step": 3530,
      "train_flow_matching_loss": 0.09423828125,
      "train_physics_loss": 0.8917995095252991
    },
    {
      "step": 3530,
      "train_flow_matching_loss": 0.07373046875,
      "train_physics_loss": 0.9399599432945251
    },
    {
      "step": 3530,
      "train_flow_matching_loss": 0.1435546875,
      "train_physics_loss": 0.0
    },
    {
      "step": 3530,
      "train_flow_matching_loss": 0.2265625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.097434863448143,
      "learning_rate": 5e-05,
      "loss": 0.1875,
      "step": 3540
    },
    {
      "step": 3540,
      "train_flow_matching_loss": 0.1484375,
      "train_physics_loss": 0.9371412396430969
    },
    {
      "step": 3540,
      "train_flow_matching_loss": 0.140625,
      "train_physics_loss": 0.0
    },
    {
      "step": 3540,
      "train_flow_matching_loss": 0.1943359375,
      "train_physics_loss": 0.0
    },
    {
      "step": 3540,
      "train_flow_matching_loss": 0.2265625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.12749400734901428,
      "learning_rate": 5e-05,
      "loss": 0.18,
      "step": 3550
    },
    {
      "step": 3550,
      "train_flow_matching_loss": 0.15625,
      "train_physics_loss": 0.9271191954612732
    },
    {
      "step": 3550,
      "train_flow_matching_loss": 0.1533203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 3550,
      "train_flow_matching_loss": 0.150390625,
      "train_physics_loss": 0.0
    },
    {
      "step": 3550,
      "train_flow_matching_loss": 0.1591796875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.13078083097934723,
      "learning_rate": 5e-05,
      "loss": 0.1828,
      "step": 3560
    },
    {
      "step": 3560,
      "train_flow_matching_loss": 0.09326171875,
      "train_physics_loss": 0.9115901589393616
    },
    {
      "step": 3560,
      "train_flow_matching_loss": 0.11376953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 3560,
      "train_flow_matching_loss": 0.205078125,
      "train_physics_loss": 0.0
    },
    {
      "step": 3560,
      "train_flow_matching_loss": 0.1962890625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11255446076393127,
      "learning_rate": 5e-05,
      "loss": 0.1849,
      "step": 3570
    },
    {
      "step": 3570,
      "train_flow_matching_loss": 0.11083984375,
      "train_physics_loss": 0.8932960033416748
    },
    {
      "step": 3570,
      "train_flow_matching_loss": 0.2236328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 3570,
      "train_flow_matching_loss": 0.197265625,
      "train_physics_loss": 0.0
    },
    {
      "step": 3570,
      "train_flow_matching_loss": 0.171875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.12100052833557129,
      "learning_rate": 5e-05,
      "loss": 0.1845,
      "step": 3580
    },
    {
      "step": 3580,
      "train_flow_matching_loss": 0.1552734375,
      "train_physics_loss": 0.9550878405570984
    },
    {
      "step": 3580,
      "train_flow_matching_loss": 0.138671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 3580,
      "train_flow_matching_loss": 0.1455078125,
      "train_physics_loss": 0.0
    },
    {
      "step": 3580,
      "train_flow_matching_loss": 0.1826171875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.25061145424842834,
      "learning_rate": 5e-05,
      "loss": 0.1805,
      "step": 3590
    },
    {
      "step": 3590,
      "train_flow_matching_loss": 0.11328125,
      "train_physics_loss": 0.9252359867095947
    },
    {
      "step": 3590,
      "train_flow_matching_loss": 0.240234375,
      "train_physics_loss": 0.0
    },
    {
      "step": 3590,
      "train_flow_matching_loss": 0.201171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 3590,
      "train_flow_matching_loss": 0.1357421875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10971014946699142,
      "learning_rate": 5e-05,
      "loss": 0.1835,
      "step": 3600
    },
    {
      "step": 3600,
      "train_flow_matching_loss": 0.12890625,
      "train_physics_loss": 0.0
    },
    {
      "step": 3600,
      "train_flow_matching_loss": 0.234375,
      "train_physics_loss": 0.0
    },
    {
      "step": 3600,
      "train_flow_matching_loss": 0.1552734375,
      "train_physics_loss": 0.0
    },
    {
      "step": 3600,
      "train_flow_matching_loss": 0.1748046875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.14067979156970978,
      "learning_rate": 5e-05,
      "loss": 0.1732,
      "step": 3610
    },
    {
      "step": 3610,
      "train_flow_matching_loss": 0.1787109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 3610,
      "train_flow_matching_loss": 0.1865234375,
      "train_physics_loss": 0.0
    },
    {
      "step": 3610,
      "train_flow_matching_loss": 0.1982421875,
      "train_physics_loss": 0.0
    },
    {
      "step": 3610,
      "train_flow_matching_loss": 0.162109375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.13658078014850616,
      "learning_rate": 5e-05,
      "loss": 0.1824,
      "step": 3620
    },
    {
      "step": 3620,
      "train_flow_matching_loss": 0.10400390625,
      "train_physics_loss": 0.0
    },
    {
      "step": 3620,
      "train_flow_matching_loss": 0.1474609375,
      "train_physics_loss": 0.0
    },
    {
      "step": 3620,
      "train_flow_matching_loss": 0.1748046875,
      "train_physics_loss": 0.0
    },
    {
      "step": 3620,
      "train_flow_matching_loss": 0.150390625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11328816413879395,
      "learning_rate": 5e-05,
      "loss": 0.1802,
      "step": 3630
    },
    {
      "step": 3630,
      "train_flow_matching_loss": 0.1689453125,
      "train_physics_loss": 0.0
    },
    {
      "step": 3630,
      "train_flow_matching_loss": 0.1796875,
      "train_physics_loss": 0.0
    },
    {
      "step": 3630,
      "train_flow_matching_loss": 0.1845703125,
      "train_physics_loss": 0.0
    },
    {
      "step": 3630,
      "train_flow_matching_loss": 0.310546875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11547716706991196,
      "learning_rate": 5e-05,
      "loss": 0.1847,
      "step": 3640
    },
    {
      "step": 3640,
      "train_flow_matching_loss": 0.2236328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 3640,
      "train_flow_matching_loss": 0.1494140625,
      "train_physics_loss": 0.0
    },
    {
      "step": 3640,
      "train_flow_matching_loss": 0.1689453125,
      "train_physics_loss": 0.0
    },
    {
      "step": 3640,
      "train_flow_matching_loss": 0.1611328125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.1788681596517563,
      "learning_rate": 5e-05,
      "loss": 0.1802,
      "step": 3650
    },
    {
      "step": 3650,
      "train_flow_matching_loss": 0.11669921875,
      "train_physics_loss": 0.0
    },
    {
      "step": 3650,
      "train_flow_matching_loss": 0.11767578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 3650,
      "train_flow_matching_loss": 0.2333984375,
      "train_physics_loss": 0.0
    },
    {
      "step": 3650,
      "train_flow_matching_loss": 0.32421875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.1475323587656021,
      "learning_rate": 5e-05,
      "loss": 0.1757,
      "step": 3660
    },
    {
      "step": 3660,
      "train_flow_matching_loss": 0.220703125,
      "train_physics_loss": 0.0
    },
    {
      "step": 3660,
      "train_flow_matching_loss": 0.11279296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 3660,
      "train_flow_matching_loss": 0.208984375,
      "train_physics_loss": 0.0
    },
    {
      "step": 3660,
      "train_flow_matching_loss": 0.11767578125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08071353286504745,
      "learning_rate": 5e-05,
      "loss": 0.178,
      "step": 3670
    },
    {
      "step": 3670,
      "train_flow_matching_loss": 0.2109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 3670,
      "train_flow_matching_loss": 0.150390625,
      "train_physics_loss": 0.0
    },
    {
      "step": 3670,
      "train_flow_matching_loss": 0.1591796875,
      "train_physics_loss": 0.0
    },
    {
      "step": 3670,
      "train_flow_matching_loss": 0.10205078125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11042577028274536,
      "learning_rate": 5e-05,
      "loss": 0.182,
      "step": 3680
    },
    {
      "step": 3680,
      "train_flow_matching_loss": 0.1572265625,
      "train_physics_loss": 0.0
    },
    {
      "step": 3680,
      "train_flow_matching_loss": 0.1669921875,
      "train_physics_loss": 0.0
    },
    {
      "step": 3680,
      "train_flow_matching_loss": 0.1826171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 3680,
      "train_flow_matching_loss": 0.177734375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11853291094303131,
      "learning_rate": 5e-05,
      "loss": 0.1819,
      "step": 3690
    },
    {
      "step": 3690,
      "train_flow_matching_loss": 0.1279296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 3690,
      "train_flow_matching_loss": 0.169921875,
      "train_physics_loss": 0.0
    },
    {
      "step": 3690,
      "train_flow_matching_loss": 0.2734375,
      "train_physics_loss": 0.0
    },
    {
      "step": 3690,
      "train_flow_matching_loss": 0.1298828125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.18063868582248688,
      "learning_rate": 5e-05,
      "loss": 0.1787,
      "step": 3700
    },
    {
      "step": 3700,
      "train_flow_matching_loss": 0.2421875,
      "train_physics_loss": 0.0
    },
    {
      "step": 3700,
      "train_flow_matching_loss": 0.11083984375,
      "train_physics_loss": 0.0
    },
    {
      "step": 3700,
      "train_flow_matching_loss": 0.1904296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 3700,
      "train_flow_matching_loss": 0.197265625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.18284043669700623,
      "learning_rate": 5e-05,
      "loss": 0.1828,
      "step": 3710
    },
    {
      "step": 3710,
      "train_flow_matching_loss": 0.162109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 3710,
      "train_flow_matching_loss": 0.177734375,
      "train_physics_loss": 0.0
    },
    {
      "step": 3710,
      "train_flow_matching_loss": 0.1474609375,
      "train_physics_loss": 0.0
    },
    {
      "step": 3710,
      "train_flow_matching_loss": 0.109375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.1356421858072281,
      "learning_rate": 5e-05,
      "loss": 0.1831,
      "step": 3720
    },
    {
      "step": 3720,
      "train_flow_matching_loss": 0.2109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 3720,
      "train_flow_matching_loss": 0.134765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 3720,
      "train_flow_matching_loss": 0.166015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 3720,
      "train_flow_matching_loss": 0.12890625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11630476266145706,
      "learning_rate": 5e-05,
      "loss": 0.1734,
      "step": 3730
    },
    {
      "step": 3730,
      "train_flow_matching_loss": 0.1669921875,
      "train_physics_loss": 0.0
    },
    {
      "step": 3730,
      "train_flow_matching_loss": 0.1396484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 3730,
      "train_flow_matching_loss": 0.1376953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 3730,
      "train_flow_matching_loss": 0.158203125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.25317779183387756,
      "learning_rate": 5e-05,
      "loss": 0.1819,
      "step": 3740
    },
    {
      "step": 3740,
      "train_flow_matching_loss": 0.2373046875,
      "train_physics_loss": 0.0
    },
    {
      "step": 3740,
      "train_flow_matching_loss": 0.2119140625,
      "train_physics_loss": 0.0
    },
    {
      "step": 3740,
      "train_flow_matching_loss": 0.328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 3740,
      "train_flow_matching_loss": 0.140625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.07579941302537918,
      "learning_rate": 5e-05,
      "loss": 0.183,
      "step": 3750
    },
    {
      "step": 3750,
      "train_flow_matching_loss": 0.1396484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 3750,
      "train_flow_matching_loss": 0.1396484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 3750,
      "train_flow_matching_loss": 0.2158203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 3750,
      "train_flow_matching_loss": 0.08447265625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.2237834632396698,
      "learning_rate": 5e-05,
      "loss": 0.1769,
      "step": 3760
    },
    {
      "step": 3760,
      "train_flow_matching_loss": 0.1748046875,
      "train_physics_loss": 0.0
    },
    {
      "step": 3760,
      "train_flow_matching_loss": 0.09765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 3760,
      "train_flow_matching_loss": 0.10498046875,
      "train_physics_loss": 0.0
    },
    {
      "step": 3760,
      "train_flow_matching_loss": 0.1337890625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09263826906681061,
      "learning_rate": 5e-05,
      "loss": 0.1801,
      "step": 3770
    },
    {
      "step": 3770,
      "train_flow_matching_loss": 0.138671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 3770,
      "train_flow_matching_loss": 0.150390625,
      "train_physics_loss": 0.0
    },
    {
      "step": 3770,
      "train_flow_matching_loss": 0.162109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 3770,
      "train_flow_matching_loss": 0.146484375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.1786985844373703,
      "learning_rate": 5e-05,
      "loss": 0.1816,
      "step": 3780
    },
    {
      "step": 3780,
      "train_flow_matching_loss": 0.1630859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 3780,
      "train_flow_matching_loss": 0.1708984375,
      "train_physics_loss": 0.0
    },
    {
      "step": 3780,
      "train_flow_matching_loss": 0.142578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 3780,
      "train_flow_matching_loss": 0.244140625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.14360485970973969,
      "learning_rate": 5e-05,
      "loss": 0.1769,
      "step": 3790
    },
    {
      "step": 3790,
      "train_flow_matching_loss": 0.1240234375,
      "train_physics_loss": 0.0
    },
    {
      "step": 3790,
      "train_flow_matching_loss": 0.10986328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 3790,
      "train_flow_matching_loss": 0.138671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 3790,
      "train_flow_matching_loss": 0.07763671875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.17948947846889496,
      "learning_rate": 5e-05,
      "loss": 0.1806,
      "step": 3800
    },
    {
      "step": 3800,
      "train_flow_matching_loss": 0.1806640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 3800,
      "train_flow_matching_loss": 0.181640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 3800,
      "train_flow_matching_loss": 0.177734375,
      "train_physics_loss": 0.0
    },
    {
      "step": 3800,
      "train_flow_matching_loss": 0.154296875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.30103299021720886,
      "learning_rate": 5e-05,
      "loss": 0.1791,
      "step": 3810
    },
    {
      "step": 3810,
      "train_flow_matching_loss": 0.1884765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 3810,
      "train_flow_matching_loss": 0.150390625,
      "train_physics_loss": 0.0
    },
    {
      "step": 3810,
      "train_flow_matching_loss": 0.203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 3810,
      "train_flow_matching_loss": 0.1484375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.17341583967208862,
      "learning_rate": 5e-05,
      "loss": 0.183,
      "step": 3820
    },
    {
      "step": 3820,
      "train_flow_matching_loss": 0.1396484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 3820,
      "train_flow_matching_loss": 0.25390625,
      "train_physics_loss": 0.0
    },
    {
      "step": 3820,
      "train_flow_matching_loss": 0.134765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 3820,
      "train_flow_matching_loss": 0.2021484375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08199319243431091,
      "learning_rate": 5e-05,
      "loss": 0.1825,
      "step": 3830
    },
    {
      "step": 3830,
      "train_flow_matching_loss": 0.1669921875,
      "train_physics_loss": 0.0
    },
    {
      "step": 3830,
      "train_flow_matching_loss": 0.1337890625,
      "train_physics_loss": 0.0
    },
    {
      "step": 3830,
      "train_flow_matching_loss": 0.146484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 3830,
      "train_flow_matching_loss": 0.1533203125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09134262800216675,
      "learning_rate": 5e-05,
      "loss": 0.1807,
      "step": 3840
    },
    {
      "step": 3840,
      "train_flow_matching_loss": 0.12109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 3840,
      "train_flow_matching_loss": 0.15625,
      "train_physics_loss": 0.0
    },
    {
      "step": 3840,
      "train_flow_matching_loss": 0.142578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 3840,
      "train_flow_matching_loss": 0.275390625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09349419176578522,
      "learning_rate": 5e-05,
      "loss": 0.1804,
      "step": 3850
    },
    {
      "step": 3850,
      "train_flow_matching_loss": 0.166015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 3850,
      "train_flow_matching_loss": 0.1611328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 3850,
      "train_flow_matching_loss": 0.1884765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 3850,
      "train_flow_matching_loss": 0.1796875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.17276503145694733,
      "learning_rate": 5e-05,
      "loss": 0.1853,
      "step": 3860
    },
    {
      "step": 3860,
      "train_flow_matching_loss": 0.173828125,
      "train_physics_loss": 1.0060995817184448
    },
    {
      "step": 3860,
      "train_flow_matching_loss": 0.1884765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 3860,
      "train_flow_matching_loss": 0.1943359375,
      "train_physics_loss": 0.0
    },
    {
      "step": 3860,
      "train_flow_matching_loss": 0.1455078125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.17768064141273499,
      "learning_rate": 5e-05,
      "loss": 0.1833,
      "step": 3870
    },
    {
      "step": 3870,
      "train_flow_matching_loss": 0.12451171875,
      "train_physics_loss": 0.9463674426078796
    },
    {
      "step": 3870,
      "train_flow_matching_loss": 0.1416015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 3870,
      "train_flow_matching_loss": 0.169921875,
      "train_physics_loss": 0.0
    },
    {
      "step": 3870,
      "train_flow_matching_loss": 0.2109375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08461593836545944,
      "learning_rate": 5e-05,
      "loss": 0.1786,
      "step": 3880
    },
    {
      "step": 3880,
      "train_flow_matching_loss": 0.056884765625,
      "train_physics_loss": 0.8878611922264099
    },
    {
      "step": 3880,
      "train_flow_matching_loss": 0.1845703125,
      "train_physics_loss": 0.0
    },
    {
      "step": 3880,
      "train_flow_matching_loss": 0.1044921875,
      "train_physics_loss": 0.0
    },
    {
      "step": 3880,
      "train_flow_matching_loss": 0.1875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.12082281708717346,
      "learning_rate": 5e-05,
      "loss": 0.1784,
      "step": 3890
    },
    {
      "step": 3890,
      "train_flow_matching_loss": 0.1494140625,
      "train_physics_loss": 0.9381943345069885
    },
    {
      "step": 3890,
      "train_flow_matching_loss": 0.185546875,
      "train_physics_loss": 0.0
    },
    {
      "step": 3890,
      "train_flow_matching_loss": 0.1875,
      "train_physics_loss": 0.0
    },
    {
      "step": 3890,
      "train_flow_matching_loss": 0.1484375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.7535666227340698,
      "learning_rate": 5e-05,
      "loss": 0.1808,
      "step": 3900
    },
    {
      "step": 3900,
      "train_flow_matching_loss": 0.0732421875,
      "train_physics_loss": 0.9274042248725891
    },
    {
      "step": 3900,
      "train_flow_matching_loss": 0.1533203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 3900,
      "train_flow_matching_loss": 0.2216796875,
      "train_physics_loss": 0.0
    },
    {
      "step": 3900,
      "train_flow_matching_loss": 0.177734375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.1646895408630371,
      "learning_rate": 5e-05,
      "loss": 0.1802,
      "step": 3910
    },
    {
      "step": 3910,
      "train_flow_matching_loss": 0.1455078125,
      "train_physics_loss": 1.0361418724060059
    },
    {
      "step": 3910,
      "train_flow_matching_loss": 0.1923828125,
      "train_physics_loss": 0.0
    },
    {
      "step": 3910,
      "train_flow_matching_loss": 0.130859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 3910,
      "train_flow_matching_loss": 0.177734375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.12679371237754822,
      "learning_rate": 5e-05,
      "loss": 0.1787,
      "step": 3920
    },
    {
      "step": 3920,
      "train_flow_matching_loss": 0.140625,
      "train_physics_loss": 0.9505107402801514
    },
    {
      "step": 3920,
      "train_flow_matching_loss": 0.1865234375,
      "train_physics_loss": 0.0
    },
    {
      "step": 3920,
      "train_flow_matching_loss": 0.146484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 3920,
      "train_flow_matching_loss": 0.142578125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.13790521025657654,
      "learning_rate": 5e-05,
      "loss": 0.1795,
      "step": 3930
    },
    {
      "step": 3930,
      "train_flow_matching_loss": 0.09228515625,
      "train_physics_loss": 0.8959740400314331
    },
    {
      "step": 3930,
      "train_flow_matching_loss": 0.1474609375,
      "train_physics_loss": 0.0
    },
    {
      "step": 3930,
      "train_flow_matching_loss": 0.2099609375,
      "train_physics_loss": 0.0
    },
    {
      "step": 3930,
      "train_flow_matching_loss": 0.1357421875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11422476917505264,
      "learning_rate": 5e-05,
      "loss": 0.1797,
      "step": 3940
    },
    {
      "step": 3940,
      "train_flow_matching_loss": 0.09375,
      "train_physics_loss": 0.9150725603103638
    },
    {
      "step": 3940,
      "train_flow_matching_loss": 0.2021484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 3940,
      "train_flow_matching_loss": 0.12060546875,
      "train_physics_loss": 0.0
    },
    {
      "step": 3940,
      "train_flow_matching_loss": 0.1806640625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11356256902217865,
      "learning_rate": 5e-05,
      "loss": 0.1836,
      "step": 3950
    },
    {
      "step": 3950,
      "train_flow_matching_loss": 0.11181640625,
      "train_physics_loss": 0.9463791251182556
    },
    {
      "step": 3950,
      "train_flow_matching_loss": 0.1435546875,
      "train_physics_loss": 0.0
    },
    {
      "step": 3950,
      "train_flow_matching_loss": 0.236328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 3950,
      "train_flow_matching_loss": 0.1357421875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.1441309154033661,
      "learning_rate": 5e-05,
      "loss": 0.1815,
      "step": 3960
    },
    {
      "step": 3960,
      "train_flow_matching_loss": 0.09619140625,
      "train_physics_loss": 0.903617799282074
    },
    {
      "step": 3960,
      "train_flow_matching_loss": 0.173828125,
      "train_physics_loss": 0.0
    },
    {
      "step": 3960,
      "train_flow_matching_loss": 0.185546875,
      "train_physics_loss": 0.0
    },
    {
      "step": 3960,
      "train_flow_matching_loss": 0.1796875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.1439417153596878,
      "learning_rate": 5e-05,
      "loss": 0.1771,
      "step": 3970
    },
    {
      "step": 3970,
      "train_flow_matching_loss": 0.1416015625,
      "train_physics_loss": 0.9713739156723022
    },
    {
      "step": 3970,
      "train_flow_matching_loss": 0.1982421875,
      "train_physics_loss": 0.0
    },
    {
      "step": 3970,
      "train_flow_matching_loss": 0.1787109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 3970,
      "train_flow_matching_loss": 0.189453125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.1573357880115509,
      "learning_rate": 5e-05,
      "loss": 0.1783,
      "step": 3980
    },
    {
      "step": 3980,
      "train_flow_matching_loss": 0.04541015625,
      "train_physics_loss": 0.8999875783920288
    },
    {
      "step": 3980,
      "train_flow_matching_loss": 0.1904296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 3980,
      "train_flow_matching_loss": 0.1865234375,
      "train_physics_loss": 0.0
    },
    {
      "step": 3980,
      "train_flow_matching_loss": 0.1904296875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.14731168746948242,
      "learning_rate": 5e-05,
      "loss": 0.1772,
      "step": 3990
    },
    {
      "step": 3990,
      "train_flow_matching_loss": 0.1826171875,
      "train_physics_loss": 0.9496592879295349
    },
    {
      "step": 3990,
      "train_flow_matching_loss": 0.126953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 3990,
      "train_flow_matching_loss": 0.1708984375,
      "train_physics_loss": 0.0
    },
    {
      "step": 3990,
      "train_flow_matching_loss": 0.171875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.1716415137052536,
      "learning_rate": 5e-05,
      "loss": 0.1812,
      "step": 4000
    },
    {
      "step": 4000,
      "train_flow_matching_loss": 0.09814453125,
      "train_physics_loss": 0.9800876379013062
    },
    {
      "step": 4000,
      "train_flow_matching_loss": 0.1728515625,
      "train_physics_loss": 0.0
    },
    {
      "step": 4000,
      "train_flow_matching_loss": 0.1630859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 4000,
      "train_flow_matching_loss": 0.2333984375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09472537785768509,
      "learning_rate": 5e-05,
      "loss": 0.1802,
      "step": 4010
    },
    {
      "step": 4010,
      "train_flow_matching_loss": 0.130859375,
      "train_physics_loss": 1.0106498003005981
    },
    {
      "step": 4010,
      "train_flow_matching_loss": 0.1533203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 4010,
      "train_flow_matching_loss": 0.1708984375,
      "train_physics_loss": 0.0
    },
    {
      "step": 4010,
      "train_flow_matching_loss": 0.1494140625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.0730404406785965,
      "learning_rate": 5e-05,
      "loss": 0.1783,
      "step": 4020
    },
    {
      "step": 4020,
      "train_flow_matching_loss": 0.138671875,
      "train_physics_loss": 0.8561689257621765
    },
    {
      "step": 4020,
      "train_flow_matching_loss": 0.11669921875,
      "train_physics_loss": 0.0
    },
    {
      "step": 4020,
      "train_flow_matching_loss": 0.10107421875,
      "train_physics_loss": 0.0
    },
    {
      "step": 4020,
      "train_flow_matching_loss": 0.2041015625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.1298755407333374,
      "learning_rate": 5e-05,
      "loss": 0.1832,
      "step": 4030
    },
    {
      "step": 4030,
      "train_flow_matching_loss": 0.08740234375,
      "train_physics_loss": 1.1208477020263672
    },
    {
      "step": 4030,
      "train_flow_matching_loss": 0.1708984375,
      "train_physics_loss": 0.0
    },
    {
      "step": 4030,
      "train_flow_matching_loss": 0.1572265625,
      "train_physics_loss": 0.0
    },
    {
      "step": 4030,
      "train_flow_matching_loss": 0.11328125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09599177539348602,
      "learning_rate": 5e-05,
      "loss": 0.1802,
      "step": 4040
    },
    {
      "step": 4040,
      "train_flow_matching_loss": 0.05224609375,
      "train_physics_loss": 1.1105875968933105
    },
    {
      "step": 4040,
      "train_flow_matching_loss": 0.2158203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 4040,
      "train_flow_matching_loss": 0.1728515625,
      "train_physics_loss": 0.0
    },
    {
      "step": 4040,
      "train_flow_matching_loss": 0.1552734375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.15470537543296814,
      "learning_rate": 5e-05,
      "loss": 0.1785,
      "step": 4050
    },
    {
      "step": 4050,
      "train_flow_matching_loss": 0.1806640625,
      "train_physics_loss": 1.001538634300232
    },
    {
      "step": 4050,
      "train_flow_matching_loss": 0.1572265625,
      "train_physics_loss": 0.0
    },
    {
      "step": 4050,
      "train_flow_matching_loss": 0.173828125,
      "train_physics_loss": 0.0
    },
    {
      "step": 4050,
      "train_flow_matching_loss": 0.1650390625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.24192854762077332,
      "learning_rate": 5e-05,
      "loss": 0.1805,
      "step": 4060
    },
    {
      "step": 4060,
      "train_flow_matching_loss": 0.08642578125,
      "train_physics_loss": 0.8746528625488281
    },
    {
      "step": 4060,
      "train_flow_matching_loss": 0.306640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 4060,
      "train_flow_matching_loss": 0.1650390625,
      "train_physics_loss": 0.0
    },
    {
      "step": 4060,
      "train_flow_matching_loss": 0.1494140625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.41680583357810974,
      "learning_rate": 5e-05,
      "loss": 0.1805,
      "step": 4070
    },
    {
      "step": 4070,
      "train_flow_matching_loss": 0.060791015625,
      "train_physics_loss": 1.0302751064300537
    },
    {
      "step": 4070,
      "train_flow_matching_loss": 0.1513671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 4070,
      "train_flow_matching_loss": 0.173828125,
      "train_physics_loss": 0.0
    },
    {
      "step": 4070,
      "train_flow_matching_loss": 0.203125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.12752090394496918,
      "learning_rate": 5e-05,
      "loss": 0.1754,
      "step": 4080
    },
    {
      "step": 4080,
      "train_flow_matching_loss": 0.171875,
      "train_physics_loss": 1.019238829612732
    },
    {
      "step": 4080,
      "train_flow_matching_loss": 0.2470703125,
      "train_physics_loss": 0.0
    },
    {
      "step": 4080,
      "train_flow_matching_loss": 0.181640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 4080,
      "train_flow_matching_loss": 0.2421875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.29685306549072266,
      "learning_rate": 5e-05,
      "loss": 0.1803,
      "step": 4090
    },
    {
      "step": 4090,
      "train_flow_matching_loss": 0.09033203125,
      "train_physics_loss": 0.8914633393287659
    },
    {
      "step": 4090,
      "train_flow_matching_loss": 0.1982421875,
      "train_physics_loss": 0.0
    },
    {
      "step": 4090,
      "train_flow_matching_loss": 0.1142578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 4090,
      "train_flow_matching_loss": 0.287109375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.14128266274929047,
      "learning_rate": 5e-05,
      "loss": 0.181,
      "step": 4100
    },
    {
      "step": 4100,
      "train_flow_matching_loss": 0.0712890625,
      "train_physics_loss": 0.9053237438201904
    },
    {
      "step": 4100,
      "train_flow_matching_loss": 0.130859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 4100,
      "train_flow_matching_loss": 0.138671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 4100,
      "train_flow_matching_loss": 0.177734375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09959839284420013,
      "learning_rate": 5e-05,
      "loss": 0.1774,
      "step": 4110
    },
    {
      "step": 4110,
      "train_flow_matching_loss": 0.201171875,
      "train_physics_loss": 1.0016236305236816
    },
    {
      "step": 4110,
      "train_flow_matching_loss": 0.2099609375,
      "train_physics_loss": 0.0
    },
    {
      "step": 4110,
      "train_flow_matching_loss": 0.1484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 4110,
      "train_flow_matching_loss": 0.2197265625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08540687710046768,
      "learning_rate": 5e-05,
      "loss": 0.1822,
      "step": 4120
    },
    {
      "step": 4120,
      "train_flow_matching_loss": 0.2421875,
      "train_physics_loss": 0.0
    },
    {
      "step": 4120,
      "train_flow_matching_loss": 0.09765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 4120,
      "train_flow_matching_loss": 0.2138671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 4120,
      "train_flow_matching_loss": 0.10888671875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.27542200684547424,
      "learning_rate": 5e-05,
      "loss": 0.1827,
      "step": 4130
    },
    {
      "step": 4130,
      "train_flow_matching_loss": 0.1640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 4130,
      "train_flow_matching_loss": 0.2265625,
      "train_physics_loss": 0.0
    },
    {
      "step": 4130,
      "train_flow_matching_loss": 0.1806640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 4130,
      "train_flow_matching_loss": 0.1826171875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11680976301431656,
      "learning_rate": 5e-05,
      "loss": 0.1791,
      "step": 4140
    },
    {
      "step": 4140,
      "train_flow_matching_loss": 0.212890625,
      "train_physics_loss": 0.0
    },
    {
      "step": 4140,
      "train_flow_matching_loss": 0.2119140625,
      "train_physics_loss": 0.0
    },
    {
      "step": 4140,
      "train_flow_matching_loss": 0.1533203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 4140,
      "train_flow_matching_loss": 0.1826171875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.19247537851333618,
      "learning_rate": 5e-05,
      "loss": 0.1745,
      "step": 4150
    },
    {
      "step": 4150,
      "train_flow_matching_loss": 0.203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 4150,
      "train_flow_matching_loss": 0.1357421875,
      "train_physics_loss": 0.0
    },
    {
      "step": 4150,
      "train_flow_matching_loss": 0.2431640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 4150,
      "train_flow_matching_loss": 0.130859375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.16049708425998688,
      "learning_rate": 5e-05,
      "loss": 0.1735,
      "step": 4160
    },
    {
      "step": 4160,
      "train_flow_matching_loss": 0.1591796875,
      "train_physics_loss": 0.0
    },
    {
      "step": 4160,
      "train_flow_matching_loss": 0.1630859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 4160,
      "train_flow_matching_loss": 0.2373046875,
      "train_physics_loss": 0.0
    },
    {
      "step": 4160,
      "train_flow_matching_loss": 0.138671875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11431142687797546,
      "learning_rate": 5e-05,
      "loss": 0.1795,
      "step": 4170
    },
    {
      "step": 4170,
      "train_flow_matching_loss": 0.1982421875,
      "train_physics_loss": 0.0
    },
    {
      "step": 4170,
      "train_flow_matching_loss": 0.2216796875,
      "train_physics_loss": 0.0
    },
    {
      "step": 4170,
      "train_flow_matching_loss": 0.177734375,
      "train_physics_loss": 0.0
    },
    {
      "step": 4170,
      "train_flow_matching_loss": 0.1513671875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.1240689605474472,
      "learning_rate": 5e-05,
      "loss": 0.179,
      "step": 4180
    },
    {
      "step": 4180,
      "train_flow_matching_loss": 0.11279296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 4180,
      "train_flow_matching_loss": 0.2001953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 4180,
      "train_flow_matching_loss": 0.1357421875,
      "train_physics_loss": 0.0
    },
    {
      "step": 4180,
      "train_flow_matching_loss": 0.162109375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.15638680756092072,
      "learning_rate": 5e-05,
      "loss": 0.1782,
      "step": 4190
    },
    {
      "step": 4190,
      "train_flow_matching_loss": 0.435546875,
      "train_physics_loss": 0.0
    },
    {
      "step": 4190,
      "train_flow_matching_loss": 0.10791015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 4190,
      "train_flow_matching_loss": 0.1630859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 4190,
      "train_flow_matching_loss": 0.134765625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10618533194065094,
      "learning_rate": 5e-05,
      "loss": 0.1766,
      "step": 4200
    },
    {
      "step": 4200,
      "train_flow_matching_loss": 0.1630859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 4200,
      "train_flow_matching_loss": 0.142578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 4200,
      "train_flow_matching_loss": 0.10595703125,
      "train_physics_loss": 0.0
    },
    {
      "step": 4200,
      "train_flow_matching_loss": 0.12451171875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.6027449369430542,
      "learning_rate": 5e-05,
      "loss": 0.1726,
      "step": 4210
    },
    {
      "step": 4210,
      "train_flow_matching_loss": 0.1572265625,
      "train_physics_loss": 0.0
    },
    {
      "step": 4210,
      "train_flow_matching_loss": 0.1484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 4210,
      "train_flow_matching_loss": 0.10498046875,
      "train_physics_loss": 0.0
    },
    {
      "step": 4210,
      "train_flow_matching_loss": 0.2421875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.14286677539348602,
      "learning_rate": 5e-05,
      "loss": 0.1758,
      "step": 4220
    },
    {
      "step": 4220,
      "train_flow_matching_loss": 0.1767578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 4220,
      "train_flow_matching_loss": 0.205078125,
      "train_physics_loss": 0.0
    },
    {
      "step": 4220,
      "train_flow_matching_loss": 0.1748046875,
      "train_physics_loss": 0.0
    },
    {
      "step": 4220,
      "train_flow_matching_loss": 0.1982421875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08186361193656921,
      "learning_rate": 5e-05,
      "loss": 0.1735,
      "step": 4230
    },
    {
      "step": 4230,
      "train_flow_matching_loss": 0.11376953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 4230,
      "train_flow_matching_loss": 0.1552734375,
      "train_physics_loss": 0.0
    },
    {
      "step": 4230,
      "train_flow_matching_loss": 0.150390625,
      "train_physics_loss": 0.0
    },
    {
      "step": 4230,
      "train_flow_matching_loss": 0.232421875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.13311541080474854,
      "learning_rate": 5e-05,
      "loss": 0.1754,
      "step": 4240
    },
    {
      "step": 4240,
      "train_flow_matching_loss": 0.1669921875,
      "train_physics_loss": 0.0
    },
    {
      "step": 4240,
      "train_flow_matching_loss": 0.2421875,
      "train_physics_loss": 0.0
    },
    {
      "step": 4240,
      "train_flow_matching_loss": 0.1015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 4240,
      "train_flow_matching_loss": 0.1474609375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.13006454706192017,
      "learning_rate": 5e-05,
      "loss": 0.1792,
      "step": 4250
    },
    {
      "step": 4250,
      "train_flow_matching_loss": 0.1630859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 4250,
      "train_flow_matching_loss": 0.2001953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 4250,
      "train_flow_matching_loss": 0.1025390625,
      "train_physics_loss": 0.0
    },
    {
      "step": 4250,
      "train_flow_matching_loss": 0.1416015625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.17941251397132874,
      "learning_rate": 5e-05,
      "loss": 0.1782,
      "step": 4260
    },
    {
      "step": 4260,
      "train_flow_matching_loss": 0.1865234375,
      "train_physics_loss": 0.0
    },
    {
      "step": 4260,
      "train_flow_matching_loss": 0.1630859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 4260,
      "train_flow_matching_loss": 0.09765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 4260,
      "train_flow_matching_loss": 0.1630859375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10183332115411758,
      "learning_rate": 5e-05,
      "loss": 0.1744,
      "step": 4270
    },
    {
      "step": 4270,
      "train_flow_matching_loss": 0.11572265625,
      "train_physics_loss": 0.0
    },
    {
      "step": 4270,
      "train_flow_matching_loss": 0.201171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 4270,
      "train_flow_matching_loss": 0.158203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 4270,
      "train_flow_matching_loss": 0.171875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10019482672214508,
      "learning_rate": 5e-05,
      "loss": 0.1756,
      "step": 4280
    },
    {
      "step": 4280,
      "train_flow_matching_loss": 0.1689453125,
      "train_physics_loss": 0.0
    },
    {
      "step": 4280,
      "train_flow_matching_loss": 0.2109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 4280,
      "train_flow_matching_loss": 0.2353515625,
      "train_physics_loss": 0.0
    },
    {
      "step": 4280,
      "train_flow_matching_loss": 0.08935546875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.1097140833735466,
      "learning_rate": 5e-05,
      "loss": 0.1769,
      "step": 4290
    },
    {
      "step": 4290,
      "train_flow_matching_loss": 0.2197265625,
      "train_physics_loss": 0.0
    },
    {
      "step": 4290,
      "train_flow_matching_loss": 0.1279296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 4290,
      "train_flow_matching_loss": 0.1728515625,
      "train_physics_loss": 0.0
    },
    {
      "step": 4290,
      "train_flow_matching_loss": 0.220703125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.1273731142282486,
      "learning_rate": 5e-05,
      "loss": 0.1744,
      "step": 4300
    },
    {
      "step": 4300,
      "train_flow_matching_loss": 0.2119140625,
      "train_physics_loss": 0.0
    },
    {
      "step": 4300,
      "train_flow_matching_loss": 0.138671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 4300,
      "train_flow_matching_loss": 0.2275390625,
      "train_physics_loss": 0.0
    },
    {
      "step": 4300,
      "train_flow_matching_loss": 0.1669921875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10870736092329025,
      "learning_rate": 5e-05,
      "loss": 0.1755,
      "step": 4310
    },
    {
      "step": 4310,
      "train_flow_matching_loss": 0.1474609375,
      "train_physics_loss": 0.0
    },
    {
      "step": 4310,
      "train_flow_matching_loss": 0.1025390625,
      "train_physics_loss": 0.0
    },
    {
      "step": 4310,
      "train_flow_matching_loss": 0.193359375,
      "train_physics_loss": 0.0
    },
    {
      "step": 4310,
      "train_flow_matching_loss": 0.265625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10348469018936157,
      "learning_rate": 5e-05,
      "loss": 0.1743,
      "step": 4320
    },
    {
      "step": 4320,
      "train_flow_matching_loss": 0.197265625,
      "train_physics_loss": 0.0
    },
    {
      "step": 4320,
      "train_flow_matching_loss": 0.279296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 4320,
      "train_flow_matching_loss": 0.279296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 4320,
      "train_flow_matching_loss": 0.208984375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11655665189027786,
      "learning_rate": 5e-05,
      "loss": 0.1767,
      "step": 4330
    },
    {
      "step": 4330,
      "train_flow_matching_loss": 0.134765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 4330,
      "train_flow_matching_loss": 0.240234375,
      "train_physics_loss": 0.0
    },
    {
      "step": 4330,
      "train_flow_matching_loss": 0.1484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 4330,
      "train_flow_matching_loss": 0.166015625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.3567488193511963,
      "learning_rate": 5e-05,
      "loss": 0.176,
      "step": 4340
    },
    {
      "step": 4340,
      "train_flow_matching_loss": 0.224609375,
      "train_physics_loss": 0.0
    },
    {
      "step": 4340,
      "train_flow_matching_loss": 0.1533203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 4340,
      "train_flow_matching_loss": 0.1484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 4340,
      "train_flow_matching_loss": 0.1728515625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.1547512710094452,
      "learning_rate": 5e-05,
      "loss": 0.1775,
      "step": 4350
    },
    {
      "step": 4350,
      "train_flow_matching_loss": 0.234375,
      "train_physics_loss": 0.0
    },
    {
      "step": 4350,
      "train_flow_matching_loss": 0.162109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 4350,
      "train_flow_matching_loss": 0.1005859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 4350,
      "train_flow_matching_loss": 0.203125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10026616603136063,
      "learning_rate": 5e-05,
      "loss": 0.1759,
      "step": 4360
    },
    {
      "step": 4360,
      "train_flow_matching_loss": 0.310546875,
      "train_physics_loss": 0.0
    },
    {
      "step": 4360,
      "train_flow_matching_loss": 0.1884765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 4360,
      "train_flow_matching_loss": 0.1328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 4360,
      "train_flow_matching_loss": 0.1923828125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.3045263886451721,
      "learning_rate": 5e-05,
      "loss": 0.1755,
      "step": 4370
    },
    {
      "step": 4370,
      "train_flow_matching_loss": 0.11083984375,
      "train_physics_loss": 0.0
    },
    {
      "step": 4370,
      "train_flow_matching_loss": 0.142578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 4370,
      "train_flow_matching_loss": 0.1875,
      "train_physics_loss": 0.0
    },
    {
      "step": 4370,
      "train_flow_matching_loss": 0.2431640625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.13467566668987274,
      "learning_rate": 5e-05,
      "loss": 0.1789,
      "step": 4380
    },
    {
      "step": 4380,
      "train_flow_matching_loss": 0.1826171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 4380,
      "train_flow_matching_loss": 0.11328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 4380,
      "train_flow_matching_loss": 0.328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 4380,
      "train_flow_matching_loss": 0.2265625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.1008230447769165,
      "learning_rate": 5e-05,
      "loss": 0.1759,
      "step": 4390
    },
    {
      "step": 4390,
      "train_flow_matching_loss": 0.21875,
      "train_physics_loss": 0.0
    },
    {
      "step": 4390,
      "train_flow_matching_loss": 0.2119140625,
      "train_physics_loss": 0.0
    },
    {
      "step": 4390,
      "train_flow_matching_loss": 0.1611328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 4390,
      "train_flow_matching_loss": 0.1259765625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.13932126760482788,
      "learning_rate": 5e-05,
      "loss": 0.1826,
      "step": 4400
    },
    {
      "step": 4400,
      "train_flow_matching_loss": 0.212890625,
      "train_physics_loss": 0.0
    },
    {
      "step": 4400,
      "train_flow_matching_loss": 0.2080078125,
      "train_physics_loss": 0.0
    },
    {
      "step": 4400,
      "train_flow_matching_loss": 0.1728515625,
      "train_physics_loss": 0.0
    },
    {
      "step": 4400,
      "train_flow_matching_loss": 0.2041015625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09741009026765823,
      "learning_rate": 5e-05,
      "loss": 0.1764,
      "step": 4410
    },
    {
      "step": 4410,
      "train_flow_matching_loss": 0.1474609375,
      "train_physics_loss": 0.0
    },
    {
      "step": 4410,
      "train_flow_matching_loss": 0.1845703125,
      "train_physics_loss": 0.0
    },
    {
      "step": 4410,
      "train_flow_matching_loss": 0.169921875,
      "train_physics_loss": 0.0
    },
    {
      "step": 4410,
      "train_flow_matching_loss": 0.2275390625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.13578851521015167,
      "learning_rate": 5e-05,
      "loss": 0.1804,
      "step": 4420
    },
    {
      "step": 4420,
      "train_flow_matching_loss": 0.173828125,
      "train_physics_loss": 0.0
    },
    {
      "step": 4420,
      "train_flow_matching_loss": 0.2099609375,
      "train_physics_loss": 0.0
    },
    {
      "step": 4420,
      "train_flow_matching_loss": 0.12451171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 4420,
      "train_flow_matching_loss": 0.1875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10226403176784515,
      "learning_rate": 5e-05,
      "loss": 0.1756,
      "step": 4430
    },
    {
      "step": 4430,
      "train_flow_matching_loss": 0.1435546875,
      "train_physics_loss": 0.0
    },
    {
      "step": 4430,
      "train_flow_matching_loss": 0.10205078125,
      "train_physics_loss": 0.0
    },
    {
      "step": 4430,
      "train_flow_matching_loss": 0.1484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 4430,
      "train_flow_matching_loss": 0.1845703125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.17536622285842896,
      "learning_rate": 5e-05,
      "loss": 0.1812,
      "step": 4440
    },
    {
      "step": 4440,
      "train_flow_matching_loss": 0.275390625,
      "train_physics_loss": 0.0
    },
    {
      "step": 4440,
      "train_flow_matching_loss": 0.1689453125,
      "train_physics_loss": 0.0
    },
    {
      "step": 4440,
      "train_flow_matching_loss": 0.1669921875,
      "train_physics_loss": 0.0
    },
    {
      "step": 4440,
      "train_flow_matching_loss": 0.1962890625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.1514994353055954,
      "learning_rate": 5e-05,
      "loss": 0.1742,
      "step": 4450
    },
    {
      "step": 4450,
      "train_flow_matching_loss": 0.10205078125,
      "train_physics_loss": 0.0
    },
    {
      "step": 4450,
      "train_flow_matching_loss": 0.138671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 4450,
      "train_flow_matching_loss": 0.1318359375,
      "train_physics_loss": 0.0
    },
    {
      "step": 4450,
      "train_flow_matching_loss": 0.2109375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.13234438002109528,
      "learning_rate": 5e-05,
      "loss": 0.1759,
      "step": 4460
    },
    {
      "step": 4460,
      "train_flow_matching_loss": 0.11767578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 4460,
      "train_flow_matching_loss": 0.1640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 4460,
      "train_flow_matching_loss": 0.1591796875,
      "train_physics_loss": 0.0
    },
    {
      "step": 4460,
      "train_flow_matching_loss": 0.1494140625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.14948183298110962,
      "learning_rate": 5e-05,
      "loss": 0.1767,
      "step": 4470
    },
    {
      "step": 4470,
      "train_flow_matching_loss": 0.251953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 4470,
      "train_flow_matching_loss": 0.1357421875,
      "train_physics_loss": 0.0
    },
    {
      "step": 4470,
      "train_flow_matching_loss": 0.208984375,
      "train_physics_loss": 0.0
    },
    {
      "step": 4470,
      "train_flow_matching_loss": 0.2021484375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.12246604263782501,
      "learning_rate": 5e-05,
      "loss": 0.1731,
      "step": 4480
    },
    {
      "step": 4480,
      "train_flow_matching_loss": 0.2177734375,
      "train_physics_loss": 0.0
    },
    {
      "step": 4480,
      "train_flow_matching_loss": 0.126953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 4480,
      "train_flow_matching_loss": 0.2431640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 4480,
      "train_flow_matching_loss": 0.1396484375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09427829831838608,
      "learning_rate": 5e-05,
      "loss": 0.1737,
      "step": 4490
    },
    {
      "step": 4490,
      "train_flow_matching_loss": 0.1474609375,
      "train_physics_loss": 0.0
    },
    {
      "step": 4490,
      "train_flow_matching_loss": 0.1708984375,
      "train_physics_loss": 0.0
    },
    {
      "step": 4490,
      "train_flow_matching_loss": 0.1572265625,
      "train_physics_loss": 0.0
    },
    {
      "step": 4490,
      "train_flow_matching_loss": 0.216796875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10739727318286896,
      "learning_rate": 5e-05,
      "loss": 0.1731,
      "step": 4500
    },
    {
      "step": 4500,
      "train_flow_matching_loss": 0.2412109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 4500,
      "train_flow_matching_loss": 0.1142578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 4500,
      "train_flow_matching_loss": 0.12890625,
      "train_physics_loss": 0.0
    },
    {
      "step": 4500,
      "train_flow_matching_loss": 0.212890625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11342237889766693,
      "learning_rate": 5e-05,
      "loss": 0.177,
      "step": 4510
    },
    {
      "step": 4510,
      "train_flow_matching_loss": 0.169921875,
      "train_physics_loss": 0.0
    },
    {
      "step": 4510,
      "train_flow_matching_loss": 0.1591796875,
      "train_physics_loss": 0.0
    },
    {
      "step": 4510,
      "train_flow_matching_loss": 0.1494140625,
      "train_physics_loss": 0.0
    },
    {
      "step": 4510,
      "train_flow_matching_loss": 0.1796875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.1415872722864151,
      "learning_rate": 5e-05,
      "loss": 0.1782,
      "step": 4520
    },
    {
      "step": 4520,
      "train_flow_matching_loss": 0.1982421875,
      "train_physics_loss": 0.0
    },
    {
      "step": 4520,
      "train_flow_matching_loss": 0.1494140625,
      "train_physics_loss": 0.0
    },
    {
      "step": 4520,
      "train_flow_matching_loss": 0.1787109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 4520,
      "train_flow_matching_loss": 0.205078125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.1435115486383438,
      "learning_rate": 5e-05,
      "loss": 0.1797,
      "step": 4530
    },
    {
      "step": 4530,
      "train_flow_matching_loss": 0.169921875,
      "train_physics_loss": 0.0
    },
    {
      "step": 4530,
      "train_flow_matching_loss": 0.11328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 4530,
      "train_flow_matching_loss": 0.1513671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 4530,
      "train_flow_matching_loss": 0.1689453125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.12431605160236359,
      "learning_rate": 5e-05,
      "loss": 0.1782,
      "step": 4540
    },
    {
      "step": 4540,
      "train_flow_matching_loss": 0.12158203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 4540,
      "train_flow_matching_loss": 0.2119140625,
      "train_physics_loss": 0.0
    },
    {
      "step": 4540,
      "train_flow_matching_loss": 0.0712890625,
      "train_physics_loss": 0.9974063634872437
    },
    {
      "step": 4540,
      "train_flow_matching_loss": 0.2158203125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11855581402778625,
      "learning_rate": 5e-05,
      "loss": 0.1728,
      "step": 4550
    },
    {
      "step": 4550,
      "train_flow_matching_loss": 0.154296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 4550,
      "train_flow_matching_loss": 0.232421875,
      "train_physics_loss": 0.0
    },
    {
      "step": 4550,
      "train_flow_matching_loss": 0.05615234375,
      "train_physics_loss": 0.9624087810516357
    },
    {
      "step": 4550,
      "train_flow_matching_loss": 0.07666015625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.26133957505226135,
      "learning_rate": 5e-05,
      "loss": 0.1744,
      "step": 4560
    },
    {
      "step": 4560,
      "train_flow_matching_loss": 0.189453125,
      "train_physics_loss": 0.0
    },
    {
      "step": 4560,
      "train_flow_matching_loss": 0.162109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 4560,
      "train_flow_matching_loss": 0.1162109375,
      "train_physics_loss": 0.9050702452659607
    },
    {
      "step": 4560,
      "train_flow_matching_loss": 0.1435546875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.12637054920196533,
      "learning_rate": 5e-05,
      "loss": 0.1769,
      "step": 4570
    },
    {
      "step": 4570,
      "train_flow_matching_loss": 0.154296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 4570,
      "train_flow_matching_loss": 0.1259765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 4570,
      "train_flow_matching_loss": 0.08447265625,
      "train_physics_loss": 0.9267836809158325
    },
    {
      "step": 4570,
      "train_flow_matching_loss": 0.208984375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.16317026317119598,
      "learning_rate": 5e-05,
      "loss": 0.1779,
      "step": 4580
    },
    {
      "step": 4580,
      "train_flow_matching_loss": 0.2158203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 4580,
      "train_flow_matching_loss": 0.11865234375,
      "train_physics_loss": 0.0
    },
    {
      "step": 4580,
      "train_flow_matching_loss": 0.06689453125,
      "train_physics_loss": 0.9319851994514465
    },
    {
      "step": 4580,
      "train_flow_matching_loss": 0.150390625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.1696740686893463,
      "learning_rate": 5e-05,
      "loss": 0.1723,
      "step": 4590
    },
    {
      "step": 4590,
      "train_flow_matching_loss": 0.1640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 4590,
      "train_flow_matching_loss": 0.142578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 4590,
      "train_flow_matching_loss": 0.1201171875,
      "train_physics_loss": 0.9935142993927002
    },
    {
      "step": 4590,
      "train_flow_matching_loss": 0.1728515625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.15905436873435974,
      "learning_rate": 5e-05,
      "loss": 0.1789,
      "step": 4600
    },
    {
      "step": 4600,
      "train_flow_matching_loss": 0.1591796875,
      "train_physics_loss": 0.0
    },
    {
      "step": 4600,
      "train_flow_matching_loss": 0.154296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 4600,
      "train_flow_matching_loss": 0.08544921875,
      "train_physics_loss": 0.8724650740623474
    },
    {
      "step": 4600,
      "train_flow_matching_loss": 0.21875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.13974303007125854,
      "learning_rate": 5e-05,
      "loss": 0.1745,
      "step": 4610
    },
    {
      "step": 4610,
      "train_flow_matching_loss": 0.1435546875,
      "train_physics_loss": 0.0
    },
    {
      "step": 4610,
      "train_flow_matching_loss": 0.1455078125,
      "train_physics_loss": 0.0
    },
    {
      "step": 4610,
      "train_flow_matching_loss": 0.08251953125,
      "train_physics_loss": 1.0085567235946655
    },
    {
      "step": 4610,
      "train_flow_matching_loss": 0.1787109375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.15402817726135254,
      "learning_rate": 5e-05,
      "loss": 0.1774,
      "step": 4620
    },
    {
      "step": 4620,
      "train_flow_matching_loss": 0.181640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 4620,
      "train_flow_matching_loss": 0.1708984375,
      "train_physics_loss": 0.0
    },
    {
      "step": 4620,
      "train_flow_matching_loss": 0.0966796875,
      "train_physics_loss": 1.0649657249450684
    },
    {
      "step": 4620,
      "train_flow_matching_loss": 0.2578125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.16031663119792938,
      "learning_rate": 5e-05,
      "loss": 0.1769,
      "step": 4630
    },
    {
      "step": 4630,
      "train_flow_matching_loss": 0.126953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 4630,
      "train_flow_matching_loss": 0.1455078125,
      "train_physics_loss": 0.0
    },
    {
      "step": 4630,
      "train_flow_matching_loss": 0.08935546875,
      "train_physics_loss": 0.8540801405906677
    },
    {
      "step": 4630,
      "train_flow_matching_loss": 0.169921875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.13534587621688843,
      "learning_rate": 5e-05,
      "loss": 0.1738,
      "step": 4640
    },
    {
      "step": 4640,
      "train_flow_matching_loss": 0.240234375,
      "train_physics_loss": 0.0
    },
    {
      "step": 4640,
      "train_flow_matching_loss": 0.11767578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 4640,
      "train_flow_matching_loss": 0.041015625,
      "train_physics_loss": 0.8988040685653687
    },
    {
      "step": 4640,
      "train_flow_matching_loss": 0.2158203125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.12189143896102905,
      "learning_rate": 5e-05,
      "loss": 0.1757,
      "step": 4650
    },
    {
      "step": 4650,
      "train_flow_matching_loss": 0.1728515625,
      "train_physics_loss": 0.0
    },
    {
      "step": 4650,
      "train_flow_matching_loss": 0.1611328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 4650,
      "train_flow_matching_loss": 0.060546875,
      "train_physics_loss": 0.8855880498886108
    },
    {
      "step": 4650,
      "train_flow_matching_loss": 0.177734375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10096998512744904,
      "learning_rate": 5e-05,
      "loss": 0.1837,
      "step": 4660
    },
    {
      "step": 4660,
      "train_flow_matching_loss": 0.220703125,
      "train_physics_loss": 0.0
    },
    {
      "step": 4660,
      "train_flow_matching_loss": 0.162109375,
      "train_physics_loss": 1.0028913021087646
    },
    {
      "step": 4660,
      "train_flow_matching_loss": 0.05908203125,
      "train_physics_loss": 0.888073742389679
    },
    {
      "step": 4660,
      "train_flow_matching_loss": 0.130859375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.15601292252540588,
      "learning_rate": 5e-05,
      "loss": 0.1765,
      "step": 4670
    },
    {
      "step": 4670,
      "train_flow_matching_loss": 0.1953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 4670,
      "train_flow_matching_loss": 0.1083984375,
      "train_physics_loss": 0.8391485810279846
    },
    {
      "step": 4670,
      "train_flow_matching_loss": 0.08203125,
      "train_physics_loss": 1.0692362785339355
    },
    {
      "step": 4670,
      "train_flow_matching_loss": 0.1806640625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11337900161743164,
      "learning_rate": 5e-05,
      "loss": 0.1707,
      "step": 4680
    },
    {
      "step": 4680,
      "train_flow_matching_loss": 0.2138671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 4680,
      "train_flow_matching_loss": 0.09716796875,
      "train_physics_loss": 0.9377226829528809
    },
    {
      "step": 4680,
      "train_flow_matching_loss": 0.08447265625,
      "train_physics_loss": 0.8907145857810974
    },
    {
      "step": 4680,
      "train_flow_matching_loss": 0.181640625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.1265861988067627,
      "learning_rate": 5e-05,
      "loss": 0.1775,
      "step": 4690
    },
    {
      "step": 4690,
      "train_flow_matching_loss": 0.171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 4690,
      "train_flow_matching_loss": 0.146484375,
      "train_physics_loss": 0.8886890411376953
    },
    {
      "step": 4690,
      "train_flow_matching_loss": 0.10009765625,
      "train_physics_loss": 0.9320847988128662
    },
    {
      "step": 4690,
      "train_flow_matching_loss": 0.1875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09483996033668518,
      "learning_rate": 5e-05,
      "loss": 0.1818,
      "step": 4700
    },
    {
      "step": 4700,
      "train_flow_matching_loss": 0.2275390625,
      "train_physics_loss": 0.0
    },
    {
      "step": 4700,
      "train_flow_matching_loss": 0.0673828125,
      "train_physics_loss": 0.8910539746284485
    },
    {
      "step": 4700,
      "train_flow_matching_loss": 0.0849609375,
      "train_physics_loss": 0.9514535665512085
    },
    {
      "step": 4700,
      "train_flow_matching_loss": 0.1875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.20675057172775269,
      "learning_rate": 5e-05,
      "loss": 0.1774,
      "step": 4710
    },
    {
      "step": 4710,
      "train_flow_matching_loss": 0.12451171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 4710,
      "train_flow_matching_loss": 0.06103515625,
      "train_physics_loss": 0.8934500813484192
    },
    {
      "step": 4710,
      "train_flow_matching_loss": 0.046630859375,
      "train_physics_loss": 0.9878296852111816
    },
    {
      "step": 4710,
      "train_flow_matching_loss": 0.216796875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11447131633758545,
      "learning_rate": 5e-05,
      "loss": 0.1804,
      "step": 4720
    },
    {
      "step": 4720,
      "train_flow_matching_loss": 0.1904296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 4720,
      "train_flow_matching_loss": 0.07763671875,
      "train_physics_loss": 0.8672616481781006
    },
    {
      "step": 4720,
      "train_flow_matching_loss": 0.07177734375,
      "train_physics_loss": 0.981846272945404
    },
    {
      "step": 4720,
      "train_flow_matching_loss": 0.201171875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.14005020260810852,
      "learning_rate": 5e-05,
      "loss": 0.1728,
      "step": 4730
    },
    {
      "step": 4730,
      "train_flow_matching_loss": 0.1513671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 4730,
      "train_flow_matching_loss": 0.0732421875,
      "train_physics_loss": 0.8480814099311829
    },
    {
      "step": 4730,
      "train_flow_matching_loss": 0.041259765625,
      "train_physics_loss": 0.8787682056427002
    },
    {
      "step": 4730,
      "train_flow_matching_loss": 0.1640625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.15997736155986786,
      "learning_rate": 5e-05,
      "loss": 0.1731,
      "step": 4740
    },
    {
      "step": 4740,
      "train_flow_matching_loss": 0.205078125,
      "train_physics_loss": 0.0
    },
    {
      "step": 4740,
      "train_flow_matching_loss": 0.1015625,
      "train_physics_loss": 0.8661871552467346
    },
    {
      "step": 4740,
      "train_flow_matching_loss": 0.09814453125,
      "train_physics_loss": 0.8925744295120239
    },
    {
      "step": 4740,
      "train_flow_matching_loss": 0.2265625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.14518240094184875,
      "learning_rate": 5e-05,
      "loss": 0.175,
      "step": 4750
    },
    {
      "step": 4750,
      "train_flow_matching_loss": 0.1552734375,
      "train_physics_loss": 0.0
    },
    {
      "step": 4750,
      "train_flow_matching_loss": 0.1083984375,
      "train_physics_loss": 0.9054712057113647
    },
    {
      "step": 4750,
      "train_flow_matching_loss": 0.05078125,
      "train_physics_loss": 0.7920320630073547
    },
    {
      "step": 4750,
      "train_flow_matching_loss": 0.1513671875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10349708795547485,
      "learning_rate": 5e-05,
      "loss": 0.1731,
      "step": 4760
    },
    {
      "step": 4760,
      "train_flow_matching_loss": 0.28515625,
      "train_physics_loss": 0.0
    },
    {
      "step": 4760,
      "train_flow_matching_loss": 0.216796875,
      "train_physics_loss": 0.8921160101890564
    },
    {
      "step": 4760,
      "train_flow_matching_loss": 0.04638671875,
      "train_physics_loss": 0.8429822325706482
    },
    {
      "step": 4760,
      "train_flow_matching_loss": 0.142578125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.2147286832332611,
      "learning_rate": 5e-05,
      "loss": 0.1777,
      "step": 4770
    },
    {
      "step": 4770,
      "train_flow_matching_loss": 0.1259765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 4770,
      "train_flow_matching_loss": 0.10888671875,
      "train_physics_loss": 0.9488556981086731
    },
    {
      "step": 4770,
      "train_flow_matching_loss": 0.07470703125,
      "train_physics_loss": 0.9857398867607117
    },
    {
      "step": 4770,
      "train_flow_matching_loss": 0.1962890625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.1567244529724121,
      "learning_rate": 5e-05,
      "loss": 0.175,
      "step": 4780
    },
    {
      "step": 4780,
      "train_flow_matching_loss": 0.158203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 4780,
      "train_flow_matching_loss": 0.0947265625,
      "train_physics_loss": 0.9649098515510559
    },
    {
      "step": 4780,
      "train_flow_matching_loss": 0.09375,
      "train_physics_loss": 1.0118006467819214
    },
    {
      "step": 4780,
      "train_flow_matching_loss": 0.291015625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08512526005506516,
      "learning_rate": 5e-05,
      "loss": 0.176,
      "step": 4790
    },
    {
      "step": 4790,
      "train_flow_matching_loss": 0.1318359375,
      "train_physics_loss": 0.0
    },
    {
      "step": 4790,
      "train_flow_matching_loss": 0.050048828125,
      "train_physics_loss": 0.8589378595352173
    },
    {
      "step": 4790,
      "train_flow_matching_loss": 0.037109375,
      "train_physics_loss": 0.8178135752677917
    },
    {
      "step": 4790,
      "train_flow_matching_loss": 0.1494140625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.13395415246486664,
      "learning_rate": 5e-05,
      "loss": 0.1735,
      "step": 4800
    },
    {
      "step": 4800,
      "train_flow_matching_loss": 0.1220703125,
      "train_physics_loss": 0.0
    },
    {
      "step": 4800,
      "train_flow_matching_loss": 0.1376953125,
      "train_physics_loss": 0.9439569711685181
    },
    {
      "step": 4800,
      "train_flow_matching_loss": 0.1630859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 4800,
      "train_flow_matching_loss": 0.1708984375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09878990054130554,
      "learning_rate": 5e-05,
      "loss": 0.1719,
      "step": 4810
    },
    {
      "step": 4810,
      "train_flow_matching_loss": 0.1318359375,
      "train_physics_loss": 0.0
    },
    {
      "step": 4810,
      "train_flow_matching_loss": 0.06103515625,
      "train_physics_loss": 0.8038758039474487
    },
    {
      "step": 4810,
      "train_flow_matching_loss": 0.0966796875,
      "train_physics_loss": 0.0
    },
    {
      "step": 4810,
      "train_flow_matching_loss": 0.259765625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.07039748132228851,
      "learning_rate": 5e-05,
      "loss": 0.1738,
      "step": 4820
    },
    {
      "step": 4820,
      "train_flow_matching_loss": 0.177734375,
      "train_physics_loss": 0.0
    },
    {
      "step": 4820,
      "train_flow_matching_loss": 0.05810546875,
      "train_physics_loss": 0.8422843813896179
    },
    {
      "step": 4820,
      "train_flow_matching_loss": 0.146484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 4820,
      "train_flow_matching_loss": 0.291015625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.13256920874118805,
      "learning_rate": 5e-05,
      "loss": 0.1765,
      "step": 4830
    },
    {
      "step": 4830,
      "train_flow_matching_loss": 0.158203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 4830,
      "train_flow_matching_loss": 0.0849609375,
      "train_physics_loss": 0.8743494153022766
    },
    {
      "step": 4830,
      "train_flow_matching_loss": 0.349609375,
      "train_physics_loss": 0.0
    },
    {
      "step": 4830,
      "train_flow_matching_loss": 0.2021484375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11857827752828598,
      "learning_rate": 5e-05,
      "loss": 0.1867,
      "step": 4840
    },
    {
      "step": 4840,
      "train_flow_matching_loss": 0.236328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 4840,
      "train_flow_matching_loss": 0.0810546875,
      "train_physics_loss": 1.00312077999115
    },
    {
      "step": 4840,
      "train_flow_matching_loss": 0.1328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 4840,
      "train_flow_matching_loss": 0.263671875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.7493497729301453,
      "learning_rate": 5e-05,
      "loss": 0.1745,
      "step": 4850
    },
    {
      "step": 4850,
      "train_flow_matching_loss": 0.1484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 4850,
      "train_flow_matching_loss": 0.048828125,
      "train_physics_loss": 0.8581119179725647
    },
    {
      "step": 4850,
      "train_flow_matching_loss": 0.169921875,
      "train_physics_loss": 0.0
    },
    {
      "step": 4850,
      "train_flow_matching_loss": 0.2060546875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.16080324351787567,
      "learning_rate": 5e-05,
      "loss": 0.1738,
      "step": 4860
    },
    {
      "step": 4860,
      "train_flow_matching_loss": 0.12109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 4860,
      "train_flow_matching_loss": 0.150390625,
      "train_physics_loss": 0.9822685718536377
    },
    {
      "step": 4860,
      "train_flow_matching_loss": 0.1806640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 4860,
      "train_flow_matching_loss": 0.12158203125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10107829421758652,
      "learning_rate": 5e-05,
      "loss": 0.1755,
      "step": 4870
    },
    {
      "step": 4870,
      "train_flow_matching_loss": 0.0771484375,
      "train_physics_loss": 0.8421527743339539
    },
    {
      "step": 4870,
      "train_flow_matching_loss": 0.10888671875,
      "train_physics_loss": 0.8516462445259094
    },
    {
      "step": 4870,
      "train_flow_matching_loss": 0.173828125,
      "train_physics_loss": 0.0
    },
    {
      "step": 4870,
      "train_flow_matching_loss": 0.232421875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.1564408391714096,
      "learning_rate": 5e-05,
      "loss": 0.1755,
      "step": 4880
    },
    {
      "step": 4880,
      "train_flow_matching_loss": 0.12451171875,
      "train_physics_loss": 0.9268143177032471
    },
    {
      "step": 4880,
      "train_flow_matching_loss": 0.0927734375,
      "train_physics_loss": 0.8768603205680847
    },
    {
      "step": 4880,
      "train_flow_matching_loss": 0.1494140625,
      "train_physics_loss": 0.0
    },
    {
      "step": 4880,
      "train_flow_matching_loss": 0.2353515625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09872664511203766,
      "learning_rate": 5e-05,
      "loss": 0.1779,
      "step": 4890
    },
    {
      "step": 4890,
      "train_flow_matching_loss": 0.1103515625,
      "train_physics_loss": 0.8447512984275818
    },
    {
      "step": 4890,
      "train_flow_matching_loss": 0.06103515625,
      "train_physics_loss": 0.9322717785835266
    },
    {
      "step": 4890,
      "train_flow_matching_loss": 0.1298828125,
      "train_physics_loss": 0.0
    },
    {
      "step": 4890,
      "train_flow_matching_loss": 0.1845703125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.17707699537277222,
      "learning_rate": 5e-05,
      "loss": 0.1808,
      "step": 4900
    },
    {
      "step": 4900,
      "train_flow_matching_loss": 0.08642578125,
      "train_physics_loss": 0.8157546520233154
    },
    {
      "step": 4900,
      "train_flow_matching_loss": 0.1298828125,
      "train_physics_loss": 0.9088171124458313
    },
    {
      "step": 4900,
      "train_flow_matching_loss": 0.27734375,
      "train_physics_loss": 0.0
    },
    {
      "step": 4900,
      "train_flow_matching_loss": 0.15625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10600326955318451,
      "learning_rate": 5e-05,
      "loss": 0.1738,
      "step": 4910
    },
    {
      "step": 4910,
      "train_flow_matching_loss": 0.05419921875,
      "train_physics_loss": 0.8129286766052246
    },
    {
      "step": 4910,
      "train_flow_matching_loss": 0.0888671875,
      "train_physics_loss": 0.9129925966262817
    },
    {
      "step": 4910,
      "train_flow_matching_loss": 0.212890625,
      "train_physics_loss": 0.0
    },
    {
      "step": 4910,
      "train_flow_matching_loss": 0.1650390625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.210341677069664,
      "learning_rate": 5e-05,
      "loss": 0.1698,
      "step": 4920
    },
    {
      "step": 4920,
      "train_flow_matching_loss": 0.052490234375,
      "train_physics_loss": 0.8069385290145874
    },
    {
      "step": 4920,
      "train_flow_matching_loss": 0.1923828125,
      "train_physics_loss": 0.0
    },
    {
      "step": 4920,
      "train_flow_matching_loss": 0.2275390625,
      "train_physics_loss": 0.0
    },
    {
      "step": 4920,
      "train_flow_matching_loss": 0.1796875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.20419791340827942,
      "learning_rate": 5e-05,
      "loss": 0.1769,
      "step": 4930
    },
    {
      "step": 4930,
      "train_flow_matching_loss": 0.1005859375,
      "train_physics_loss": 0.8989657759666443
    },
    {
      "step": 4930,
      "train_flow_matching_loss": 0.2060546875,
      "train_physics_loss": 0.0
    },
    {
      "step": 4930,
      "train_flow_matching_loss": 0.1591796875,
      "train_physics_loss": 0.0
    },
    {
      "step": 4930,
      "train_flow_matching_loss": 0.1416015625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09865016490221024,
      "learning_rate": 5e-05,
      "loss": 0.171,
      "step": 4940
    },
    {
      "step": 4940,
      "train_flow_matching_loss": 0.1298828125,
      "train_physics_loss": 0.935141921043396
    },
    {
      "step": 4940,
      "train_flow_matching_loss": 0.220703125,
      "train_physics_loss": 0.0
    },
    {
      "step": 4940,
      "train_flow_matching_loss": 0.12451171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 4940,
      "train_flow_matching_loss": 0.1923828125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11308129876852036,
      "learning_rate": 5e-05,
      "loss": 0.1759,
      "step": 4950
    },
    {
      "step": 4950,
      "train_flow_matching_loss": 0.09326171875,
      "train_physics_loss": 0.9914759993553162
    },
    {
      "step": 4950,
      "train_flow_matching_loss": 0.2265625,
      "train_physics_loss": 0.0
    },
    {
      "step": 4950,
      "train_flow_matching_loss": 0.1572265625,
      "train_physics_loss": 0.0
    },
    {
      "step": 4950,
      "train_flow_matching_loss": 0.1943359375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11859776079654694,
      "learning_rate": 5e-05,
      "loss": 0.1758,
      "step": 4960
    },
    {
      "step": 4960,
      "train_flow_matching_loss": 0.08251953125,
      "train_physics_loss": 0.9038950204849243
    },
    {
      "step": 4960,
      "train_flow_matching_loss": 0.11279296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 4960,
      "train_flow_matching_loss": 0.2333984375,
      "train_physics_loss": 0.0
    },
    {
      "step": 4960,
      "train_flow_matching_loss": 0.1767578125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09544996172189713,
      "learning_rate": 5e-05,
      "loss": 0.1787,
      "step": 4970
    },
    {
      "step": 4970,
      "train_flow_matching_loss": 0.0810546875,
      "train_physics_loss": 1.0133819580078125
    },
    {
      "step": 4970,
      "train_flow_matching_loss": 0.1298828125,
      "train_physics_loss": 0.0
    },
    {
      "step": 4970,
      "train_flow_matching_loss": 0.1484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 4970,
      "train_flow_matching_loss": 0.1240234375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.1212007999420166,
      "learning_rate": 5e-05,
      "loss": 0.1786,
      "step": 4980
    },
    {
      "step": 4980,
      "train_flow_matching_loss": 0.12451171875,
      "train_physics_loss": 0.9701960682868958
    },
    {
      "step": 4980,
      "train_flow_matching_loss": 0.1640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 4980,
      "train_flow_matching_loss": 0.1982421875,
      "train_physics_loss": 0.0
    },
    {
      "step": 4980,
      "train_flow_matching_loss": 0.2060546875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.12070580571889877,
      "learning_rate": 5e-05,
      "loss": 0.1782,
      "step": 4990
    },
    {
      "step": 4990,
      "train_flow_matching_loss": 0.12255859375,
      "train_physics_loss": 0.8219832181930542
    },
    {
      "step": 4990,
      "train_flow_matching_loss": 0.1337890625,
      "train_physics_loss": 0.0
    },
    {
      "step": 4990,
      "train_flow_matching_loss": 0.1865234375,
      "train_physics_loss": 0.0
    },
    {
      "step": 4990,
      "train_flow_matching_loss": 0.1337890625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11851227283477783,
      "learning_rate": 5e-05,
      "loss": 0.1762,
      "step": 5000
    },
    {
      "step": 5000,
      "train_flow_matching_loss": 0.07080078125,
      "train_physics_loss": 0.8504664897918701
    },
    {
      "step": 5000,
      "train_flow_matching_loss": 0.1357421875,
      "train_physics_loss": 0.0
    },
    {
      "step": 5000,
      "train_flow_matching_loss": 0.1533203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 5000,
      "train_flow_matching_loss": 0.173828125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.14276137948036194,
      "learning_rate": 5e-05,
      "loss": 0.1778,
      "step": 5010
    },
    {
      "step": 5010,
      "train_flow_matching_loss": 0.1513671875,
      "train_physics_loss": 0.9712202548980713
    },
    {
      "step": 5010,
      "train_flow_matching_loss": 0.193359375,
      "train_physics_loss": 0.0
    },
    {
      "step": 5010,
      "train_flow_matching_loss": 0.201171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 5010,
      "train_flow_matching_loss": 0.2353515625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.1263362467288971,
      "learning_rate": 5e-05,
      "loss": 0.1778,
      "step": 5020
    },
    {
      "step": 5020,
      "train_flow_matching_loss": 0.0849609375,
      "train_physics_loss": 0.8484971523284912
    },
    {
      "step": 5020,
      "train_flow_matching_loss": 0.1904296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 5020,
      "train_flow_matching_loss": 0.25,
      "train_physics_loss": 0.0
    },
    {
      "step": 5020,
      "train_flow_matching_loss": 0.1591796875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.5530884265899658,
      "learning_rate": 5e-05,
      "loss": 0.1757,
      "step": 5030
    },
    {
      "step": 5030,
      "train_flow_matching_loss": 0.054931640625,
      "train_physics_loss": 0.8422961831092834
    },
    {
      "step": 5030,
      "train_flow_matching_loss": 0.146484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 5030,
      "train_flow_matching_loss": 0.2236328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 5030,
      "train_flow_matching_loss": 0.2275390625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.1331142634153366,
      "learning_rate": 5e-05,
      "loss": 0.1734,
      "step": 5040
    },
    {
      "step": 5040,
      "train_flow_matching_loss": 0.11865234375,
      "train_physics_loss": 0.908068835735321
    },
    {
      "step": 5040,
      "train_flow_matching_loss": 0.1650390625,
      "train_physics_loss": 0.0
    },
    {
      "step": 5040,
      "train_flow_matching_loss": 0.140625,
      "train_physics_loss": 0.0
    },
    {
      "step": 5040,
      "train_flow_matching_loss": 0.1875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10928265005350113,
      "learning_rate": 5e-05,
      "loss": 0.1763,
      "step": 5050
    },
    {
      "step": 5050,
      "train_flow_matching_loss": 0.10595703125,
      "train_physics_loss": 0.8927035927772522
    },
    {
      "step": 5050,
      "train_flow_matching_loss": 0.1650390625,
      "train_physics_loss": 0.0
    },
    {
      "step": 5050,
      "train_flow_matching_loss": 0.2373046875,
      "train_physics_loss": 0.0
    },
    {
      "step": 5050,
      "train_flow_matching_loss": 0.177734375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11625370383262634,
      "learning_rate": 5e-05,
      "loss": 0.1719,
      "step": 5060
    },
    {
      "step": 5060,
      "train_flow_matching_loss": 0.06884765625,
      "train_physics_loss": 0.8731463551521301
    },
    {
      "step": 5060,
      "train_flow_matching_loss": 0.21875,
      "train_physics_loss": 0.0
    },
    {
      "step": 5060,
      "train_flow_matching_loss": 0.1962890625,
      "train_physics_loss": 0.0
    },
    {
      "step": 5060,
      "train_flow_matching_loss": 0.12451171875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10504218935966492,
      "learning_rate": 5e-05,
      "loss": 0.1716,
      "step": 5070
    },
    {
      "step": 5070,
      "train_flow_matching_loss": 0.09033203125,
      "train_physics_loss": 1.0390069484710693
    },
    {
      "step": 5070,
      "train_flow_matching_loss": 0.1376953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 5070,
      "train_flow_matching_loss": 0.1484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 5070,
      "train_flow_matching_loss": 0.2333984375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.13677962124347687,
      "learning_rate": 5e-05,
      "loss": 0.1681,
      "step": 5080
    },
    {
      "step": 5080,
      "train_flow_matching_loss": 0.08203125,
      "train_physics_loss": 0.8768457174301147
    },
    {
      "step": 5080,
      "train_flow_matching_loss": 0.09716796875,
      "train_physics_loss": 0.0
    },
    {
      "step": 5080,
      "train_flow_matching_loss": 0.1318359375,
      "train_physics_loss": 0.0
    },
    {
      "step": 5080,
      "train_flow_matching_loss": 0.09375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11550258845090866,
      "learning_rate": 5e-05,
      "loss": 0.1746,
      "step": 5090
    },
    {
      "step": 5090,
      "train_flow_matching_loss": 0.0966796875,
      "train_physics_loss": 0.8734221458435059
    },
    {
      "step": 5090,
      "train_flow_matching_loss": 0.1611328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 5090,
      "train_flow_matching_loss": 0.1962890625,
      "train_physics_loss": 0.0
    },
    {
      "step": 5090,
      "train_flow_matching_loss": 0.13671875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11650916188955307,
      "learning_rate": 5e-05,
      "loss": 0.1743,
      "step": 5100
    },
    {
      "step": 5100,
      "train_flow_matching_loss": 0.1376953125,
      "train_physics_loss": 0.9054964184761047
    },
    {
      "step": 5100,
      "train_flow_matching_loss": 0.142578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 5100,
      "train_flow_matching_loss": 0.1865234375,
      "train_physics_loss": 0.0
    },
    {
      "step": 5100,
      "train_flow_matching_loss": 0.1806640625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.14095832407474518,
      "learning_rate": 5e-05,
      "loss": 0.1721,
      "step": 5110
    },
    {
      "step": 5110,
      "train_flow_matching_loss": 0.06591796875,
      "train_physics_loss": 0.8528409600257874
    },
    {
      "step": 5110,
      "train_flow_matching_loss": 0.11328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 5110,
      "train_flow_matching_loss": 0.1806640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 5110,
      "train_flow_matching_loss": 0.1923828125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10239876061677933,
      "learning_rate": 5e-05,
      "loss": 0.1755,
      "step": 5120
    },
    {
      "step": 5120,
      "train_flow_matching_loss": 0.1923828125,
      "train_physics_loss": 0.0
    },
    {
      "step": 5120,
      "train_flow_matching_loss": 0.1357421875,
      "train_physics_loss": 0.0
    },
    {
      "step": 5120,
      "train_flow_matching_loss": 0.13671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 5120,
      "train_flow_matching_loss": 0.1015625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.1375982016324997,
      "learning_rate": 5e-05,
      "loss": 0.1709,
      "step": 5130
    },
    {
      "step": 5130,
      "train_flow_matching_loss": 0.1728515625,
      "train_physics_loss": 0.0
    },
    {
      "step": 5130,
      "train_flow_matching_loss": 0.10986328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 5130,
      "train_flow_matching_loss": 0.1806640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 5130,
      "train_flow_matching_loss": 0.2080078125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11994414031505585,
      "learning_rate": 5e-05,
      "loss": 0.169,
      "step": 5140
    },
    {
      "step": 5140,
      "train_flow_matching_loss": 0.2177734375,
      "train_physics_loss": 0.0
    },
    {
      "step": 5140,
      "train_flow_matching_loss": 0.1708984375,
      "train_physics_loss": 0.0
    },
    {
      "step": 5140,
      "train_flow_matching_loss": 0.1904296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 5140,
      "train_flow_matching_loss": 0.1787109375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.13046064972877502,
      "learning_rate": 5e-05,
      "loss": 0.1742,
      "step": 5150
    },
    {
      "step": 5150,
      "train_flow_matching_loss": 0.0673828125,
      "train_physics_loss": 0.0
    },
    {
      "step": 5150,
      "train_flow_matching_loss": 0.1796875,
      "train_physics_loss": 0.0
    },
    {
      "step": 5150,
      "train_flow_matching_loss": 0.12353515625,
      "train_physics_loss": 0.0
    },
    {
      "step": 5150,
      "train_flow_matching_loss": 0.0947265625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10641631484031677,
      "learning_rate": 5e-05,
      "loss": 0.1749,
      "step": 5160
    },
    {
      "step": 5160,
      "train_flow_matching_loss": 0.255859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 5160,
      "train_flow_matching_loss": 0.1416015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 5160,
      "train_flow_matching_loss": 0.197265625,
      "train_physics_loss": 0.0
    },
    {
      "step": 5160,
      "train_flow_matching_loss": 0.2109375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11539460718631744,
      "learning_rate": 5e-05,
      "loss": 0.167,
      "step": 5170
    },
    {
      "step": 5170,
      "train_flow_matching_loss": 0.1494140625,
      "train_physics_loss": 0.0
    },
    {
      "step": 5170,
      "train_flow_matching_loss": 0.2001953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 5170,
      "train_flow_matching_loss": 0.10498046875,
      "train_physics_loss": 0.0
    },
    {
      "step": 5170,
      "train_flow_matching_loss": 0.2333984375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.13264645636081696,
      "learning_rate": 5e-05,
      "loss": 0.1712,
      "step": 5180
    },
    {
      "step": 5180,
      "train_flow_matching_loss": 0.1298828125,
      "train_physics_loss": 0.0
    },
    {
      "step": 5180,
      "train_flow_matching_loss": 0.189453125,
      "train_physics_loss": 0.0
    },
    {
      "step": 5180,
      "train_flow_matching_loss": 0.1015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 5180,
      "train_flow_matching_loss": 0.12353515625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.1195286437869072,
      "learning_rate": 5e-05,
      "loss": 0.1735,
      "step": 5190
    },
    {
      "step": 5190,
      "train_flow_matching_loss": 0.1474609375,
      "train_physics_loss": 0.0
    },
    {
      "step": 5190,
      "train_flow_matching_loss": 0.125,
      "train_physics_loss": 0.0
    },
    {
      "step": 5190,
      "train_flow_matching_loss": 0.26953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 5190,
      "train_flow_matching_loss": 0.240234375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10646269470453262,
      "learning_rate": 5e-05,
      "loss": 0.1718,
      "step": 5200
    },
    {
      "step": 5200,
      "train_flow_matching_loss": 0.203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 5200,
      "train_flow_matching_loss": 0.2119140625,
      "train_physics_loss": 0.0
    },
    {
      "step": 5200,
      "train_flow_matching_loss": 0.07080078125,
      "train_physics_loss": 0.0
    },
    {
      "step": 5200,
      "train_flow_matching_loss": 0.15625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10650698095560074,
      "learning_rate": 5e-05,
      "loss": 0.1726,
      "step": 5210
    },
    {
      "step": 5210,
      "train_flow_matching_loss": 0.1708984375,
      "train_physics_loss": 0.0
    },
    {
      "step": 5210,
      "train_flow_matching_loss": 0.1767578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 5210,
      "train_flow_matching_loss": 0.267578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 5210,
      "train_flow_matching_loss": 0.142578125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11343096941709518,
      "learning_rate": 5e-05,
      "loss": 0.1734,
      "step": 5220
    },
    {
      "step": 5220,
      "train_flow_matching_loss": 0.2158203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 5220,
      "train_flow_matching_loss": 0.1748046875,
      "train_physics_loss": 0.0
    },
    {
      "step": 5220,
      "train_flow_matching_loss": 0.1572265625,
      "train_physics_loss": 0.0
    },
    {
      "step": 5220,
      "train_flow_matching_loss": 0.1533203125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09641677886247635,
      "learning_rate": 5e-05,
      "loss": 0.1705,
      "step": 5230
    },
    {
      "step": 5230,
      "train_flow_matching_loss": 0.236328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 5230,
      "train_flow_matching_loss": 0.140625,
      "train_physics_loss": 0.0
    },
    {
      "step": 5230,
      "train_flow_matching_loss": 0.1328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 5230,
      "train_flow_matching_loss": 0.1484375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10119319707155228,
      "learning_rate": 5e-05,
      "loss": 0.17,
      "step": 5240
    },
    {
      "step": 5240,
      "train_flow_matching_loss": 0.1669921875,
      "train_physics_loss": 0.0
    },
    {
      "step": 5240,
      "train_flow_matching_loss": 0.1708984375,
      "train_physics_loss": 0.0
    },
    {
      "step": 5240,
      "train_flow_matching_loss": 0.134765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 5240,
      "train_flow_matching_loss": 0.2060546875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09225160628557205,
      "learning_rate": 5e-05,
      "loss": 0.1693,
      "step": 5250
    },
    {
      "step": 5250,
      "train_flow_matching_loss": 0.08447265625,
      "train_physics_loss": 0.0
    },
    {
      "step": 5250,
      "train_flow_matching_loss": 0.1630859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 5250,
      "train_flow_matching_loss": 0.1630859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 5250,
      "train_flow_matching_loss": 0.2275390625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.12059245258569717,
      "learning_rate": 5e-05,
      "loss": 0.1743,
      "step": 5260
    },
    {
      "step": 5260,
      "train_flow_matching_loss": 0.1806640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 5260,
      "train_flow_matching_loss": 0.1533203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 5260,
      "train_flow_matching_loss": 0.1943359375,
      "train_physics_loss": 0.0
    },
    {
      "step": 5260,
      "train_flow_matching_loss": 0.197265625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.07703400403261185,
      "learning_rate": 5e-05,
      "loss": 0.1712,
      "step": 5270
    },
    {
      "step": 5270,
      "train_flow_matching_loss": 0.1220703125,
      "train_physics_loss": 0.0
    },
    {
      "step": 5270,
      "train_flow_matching_loss": 0.1201171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 5270,
      "train_flow_matching_loss": 0.2119140625,
      "train_physics_loss": 0.0
    },
    {
      "step": 5270,
      "train_flow_matching_loss": 0.2431640625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11151710897684097,
      "learning_rate": 5e-05,
      "loss": 0.1776,
      "step": 5280
    },
    {
      "step": 5280,
      "train_flow_matching_loss": 0.2021484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 5280,
      "train_flow_matching_loss": 0.1708984375,
      "train_physics_loss": 0.0
    },
    {
      "step": 5280,
      "train_flow_matching_loss": 0.185546875,
      "train_physics_loss": 0.0
    },
    {
      "step": 5280,
      "train_flow_matching_loss": 0.10546875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.1058875024318695,
      "learning_rate": 5e-05,
      "loss": 0.1727,
      "step": 5290
    },
    {
      "step": 5290,
      "train_flow_matching_loss": 0.2119140625,
      "train_physics_loss": 0.0
    },
    {
      "step": 5290,
      "train_flow_matching_loss": 0.2041015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 5290,
      "train_flow_matching_loss": 0.1396484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 5290,
      "train_flow_matching_loss": 0.1416015625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.12365060299634933,
      "learning_rate": 5e-05,
      "loss": 0.1687,
      "step": 5300
    },
    {
      "step": 5300,
      "train_flow_matching_loss": 0.1796875,
      "train_physics_loss": 0.0
    },
    {
      "step": 5300,
      "train_flow_matching_loss": 0.162109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 5300,
      "train_flow_matching_loss": 0.1259765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 5300,
      "train_flow_matching_loss": 0.220703125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08205143362283707,
      "learning_rate": 5e-05,
      "loss": 0.1735,
      "step": 5310
    },
    {
      "step": 5310,
      "train_flow_matching_loss": 0.1455078125,
      "train_physics_loss": 0.0
    },
    {
      "step": 5310,
      "train_flow_matching_loss": 0.1865234375,
      "train_physics_loss": 0.0
    },
    {
      "step": 5310,
      "train_flow_matching_loss": 0.185546875,
      "train_physics_loss": 0.0
    },
    {
      "step": 5310,
      "train_flow_matching_loss": 0.1416015625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.13346022367477417,
      "learning_rate": 5e-05,
      "loss": 0.1775,
      "step": 5320
    },
    {
      "step": 5320,
      "train_flow_matching_loss": 0.255859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 5320,
      "train_flow_matching_loss": 0.10791015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 5320,
      "train_flow_matching_loss": 0.1376953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 5320,
      "train_flow_matching_loss": 0.224609375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09951679408550262,
      "learning_rate": 5e-05,
      "loss": 0.1724,
      "step": 5330
    },
    {
      "step": 5330,
      "train_flow_matching_loss": 0.1640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 5330,
      "train_flow_matching_loss": 0.185546875,
      "train_physics_loss": 0.0
    },
    {
      "step": 5330,
      "train_flow_matching_loss": 0.1318359375,
      "train_physics_loss": 0.0
    },
    {
      "step": 5330,
      "train_flow_matching_loss": 0.0771484375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11810962855815887,
      "learning_rate": 5e-05,
      "loss": 0.1727,
      "step": 5340
    },
    {
      "step": 5340,
      "train_flow_matching_loss": 0.150390625,
      "train_physics_loss": 0.0
    },
    {
      "step": 5340,
      "train_flow_matching_loss": 0.2109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 5340,
      "train_flow_matching_loss": 0.1240234375,
      "train_physics_loss": 0.0
    },
    {
      "step": 5340,
      "train_flow_matching_loss": 0.2373046875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.0827426016330719,
      "learning_rate": 5e-05,
      "loss": 0.1732,
      "step": 5350
    },
    {
      "step": 5350,
      "train_flow_matching_loss": 0.150390625,
      "train_physics_loss": 0.0
    },
    {
      "step": 5350,
      "train_flow_matching_loss": 0.12890625,
      "train_physics_loss": 0.0
    },
    {
      "step": 5350,
      "train_flow_matching_loss": 0.162109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 5350,
      "train_flow_matching_loss": 0.171875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08776440471410751,
      "learning_rate": 5e-05,
      "loss": 0.1707,
      "step": 5360
    },
    {
      "step": 5360,
      "train_flow_matching_loss": 0.1669921875,
      "train_physics_loss": 0.0
    },
    {
      "step": 5360,
      "train_flow_matching_loss": 0.1845703125,
      "train_physics_loss": 0.0
    },
    {
      "step": 5360,
      "train_flow_matching_loss": 0.11279296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 5360,
      "train_flow_matching_loss": 0.255859375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.16031433641910553,
      "learning_rate": 5e-05,
      "loss": 0.1747,
      "step": 5370
    },
    {
      "step": 5370,
      "train_flow_matching_loss": 0.10546875,
      "train_physics_loss": 0.0
    },
    {
      "step": 5370,
      "train_flow_matching_loss": 0.2294921875,
      "train_physics_loss": 0.0
    },
    {
      "step": 5370,
      "train_flow_matching_loss": 0.1435546875,
      "train_physics_loss": 0.0
    },
    {
      "step": 5370,
      "train_flow_matching_loss": 0.1689453125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.07551825791597366,
      "learning_rate": 5e-05,
      "loss": 0.1718,
      "step": 5380
    },
    {
      "step": 5380,
      "train_flow_matching_loss": 0.158203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 5380,
      "train_flow_matching_loss": 0.1611328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 5380,
      "train_flow_matching_loss": 0.1298828125,
      "train_physics_loss": 0.0
    },
    {
      "step": 5380,
      "train_flow_matching_loss": 0.2138671875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.15122759342193604,
      "learning_rate": 5e-05,
      "loss": 0.1686,
      "step": 5390
    },
    {
      "step": 5390,
      "train_flow_matching_loss": 0.181640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 5390,
      "train_flow_matching_loss": 0.1201171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 5390,
      "train_flow_matching_loss": 0.07666015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 5390,
      "train_flow_matching_loss": 0.1787109375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.13944824039936066,
      "learning_rate": 5e-05,
      "loss": 0.1725,
      "step": 5400
    },
    {
      "step": 5400,
      "train_flow_matching_loss": 0.205078125,
      "train_physics_loss": 0.0
    },
    {
      "step": 5400,
      "train_flow_matching_loss": 0.1474609375,
      "train_physics_loss": 0.0
    },
    {
      "step": 5400,
      "train_flow_matching_loss": 0.1640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 5400,
      "train_flow_matching_loss": 0.1943359375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11438823491334915,
      "learning_rate": 5e-05,
      "loss": 0.1743,
      "step": 5410
    },
    {
      "step": 5410,
      "train_flow_matching_loss": 0.1484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 5410,
      "train_flow_matching_loss": 0.150390625,
      "train_physics_loss": 0.0
    },
    {
      "step": 5410,
      "train_flow_matching_loss": 0.1416015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 5410,
      "train_flow_matching_loss": 0.1572265625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.12123692035675049,
      "learning_rate": 5e-05,
      "loss": 0.1709,
      "step": 5420
    },
    {
      "step": 5420,
      "train_flow_matching_loss": 0.10546875,
      "train_physics_loss": 0.0
    },
    {
      "step": 5420,
      "train_flow_matching_loss": 0.11962890625,
      "train_physics_loss": 0.0
    },
    {
      "step": 5420,
      "train_flow_matching_loss": 0.2001953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 5420,
      "train_flow_matching_loss": 0.158203125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.1304289549589157,
      "learning_rate": 5e-05,
      "loss": 0.1734,
      "step": 5430
    },
    {
      "step": 5430,
      "train_flow_matching_loss": 0.1298828125,
      "train_physics_loss": 0.0
    },
    {
      "step": 5430,
      "train_flow_matching_loss": 0.1572265625,
      "train_physics_loss": 0.0
    },
    {
      "step": 5430,
      "train_flow_matching_loss": 0.2314453125,
      "train_physics_loss": 0.0
    },
    {
      "step": 5430,
      "train_flow_matching_loss": 0.1337890625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.1352962702512741,
      "learning_rate": 5e-05,
      "loss": 0.171,
      "step": 5440
    },
    {
      "step": 5440,
      "train_flow_matching_loss": 0.203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 5440,
      "train_flow_matching_loss": 0.11376953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 5440,
      "train_flow_matching_loss": 0.1953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 5440,
      "train_flow_matching_loss": 0.244140625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.16267172992229462,
      "learning_rate": 5e-05,
      "loss": 0.1741,
      "step": 5450
    },
    {
      "step": 5450,
      "train_flow_matching_loss": 0.166015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 5450,
      "train_flow_matching_loss": 0.234375,
      "train_physics_loss": 0.0
    },
    {
      "step": 5450,
      "train_flow_matching_loss": 0.1787109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 5450,
      "train_flow_matching_loss": 0.126953125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10607738047838211,
      "learning_rate": 5e-05,
      "loss": 0.1659,
      "step": 5460
    },
    {
      "step": 5460,
      "train_flow_matching_loss": 0.1103515625,
      "train_physics_loss": 0.0
    },
    {
      "step": 5460,
      "train_flow_matching_loss": 0.2001953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 5460,
      "train_flow_matching_loss": 0.10986328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 5460,
      "train_flow_matching_loss": 0.1826171875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.15906518697738647,
      "learning_rate": 5e-05,
      "loss": 0.1707,
      "step": 5470
    },
    {
      "step": 5470,
      "train_flow_matching_loss": 0.1865234375,
      "train_physics_loss": 0.0
    },
    {
      "step": 5470,
      "train_flow_matching_loss": 0.1064453125,
      "train_physics_loss": 0.0
    },
    {
      "step": 5470,
      "train_flow_matching_loss": 0.2119140625,
      "train_physics_loss": 0.0
    },
    {
      "step": 5470,
      "train_flow_matching_loss": 0.162109375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.13168145716190338,
      "learning_rate": 5e-05,
      "loss": 0.1716,
      "step": 5480
    },
    {
      "step": 5480,
      "train_flow_matching_loss": 0.12109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 5480,
      "train_flow_matching_loss": 0.138671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 5480,
      "train_flow_matching_loss": 0.158203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 5480,
      "train_flow_matching_loss": 0.1494140625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.33618104457855225,
      "learning_rate": 5e-05,
      "loss": 0.1671,
      "step": 5490
    },
    {
      "step": 5490,
      "train_flow_matching_loss": 0.1728515625,
      "train_physics_loss": 0.0
    },
    {
      "step": 5490,
      "train_flow_matching_loss": 0.2451171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 5490,
      "train_flow_matching_loss": 0.1572265625,
      "train_physics_loss": 0.0
    },
    {
      "step": 5490,
      "train_flow_matching_loss": 0.2490234375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09537000209093094,
      "learning_rate": 5e-05,
      "loss": 0.1661,
      "step": 5500
    },
    {
      "step": 5500,
      "train_flow_matching_loss": 0.1396484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 5500,
      "train_flow_matching_loss": 0.1591796875,
      "train_physics_loss": 0.0
    },
    {
      "step": 5500,
      "train_flow_matching_loss": 0.0654296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 5500,
      "train_flow_matching_loss": 0.2412109375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11736695468425751,
      "learning_rate": 5e-05,
      "loss": 0.1717,
      "step": 5510
    },
    {
      "step": 5510,
      "train_flow_matching_loss": 0.1591796875,
      "train_physics_loss": 0.0
    },
    {
      "step": 5510,
      "train_flow_matching_loss": 0.1728515625,
      "train_physics_loss": 0.0
    },
    {
      "step": 5510,
      "train_flow_matching_loss": 0.2001953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 5510,
      "train_flow_matching_loss": 0.2265625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.13306023180484772,
      "learning_rate": 5e-05,
      "loss": 0.169,
      "step": 5520
    },
    {
      "step": 5520,
      "train_flow_matching_loss": 0.177734375,
      "train_physics_loss": 0.0
    },
    {
      "step": 5520,
      "train_flow_matching_loss": 0.1728515625,
      "train_physics_loss": 0.0
    },
    {
      "step": 5520,
      "train_flow_matching_loss": 0.1279296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 5520,
      "train_flow_matching_loss": 0.2431640625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.13037818670272827,
      "learning_rate": 5e-05,
      "loss": 0.1801,
      "step": 5530
    },
    {
      "step": 5530,
      "train_flow_matching_loss": 0.1181640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 5530,
      "train_flow_matching_loss": 0.1201171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 5530,
      "train_flow_matching_loss": 0.234375,
      "train_physics_loss": 0.0
    },
    {
      "step": 5530,
      "train_flow_matching_loss": 0.1669921875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.14366580545902252,
      "learning_rate": 5e-05,
      "loss": 0.1694,
      "step": 5540
    },
    {
      "step": 5540,
      "train_flow_matching_loss": 0.1552734375,
      "train_physics_loss": 0.0
    },
    {
      "step": 5540,
      "train_flow_matching_loss": 0.166015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 5540,
      "train_flow_matching_loss": 0.1767578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 5540,
      "train_flow_matching_loss": 0.11083984375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10160145908594131,
      "learning_rate": 5e-05,
      "loss": 0.1701,
      "step": 5550
    },
    {
      "step": 5550,
      "train_flow_matching_loss": 0.09375,
      "train_physics_loss": 0.0
    },
    {
      "step": 5550,
      "train_flow_matching_loss": 0.1484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 5550,
      "train_flow_matching_loss": 0.181640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 5550,
      "train_flow_matching_loss": 0.1728515625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.12626753747463226,
      "learning_rate": 5e-05,
      "loss": 0.1729,
      "step": 5560
    },
    {
      "step": 5560,
      "train_flow_matching_loss": 0.2197265625,
      "train_physics_loss": 0.0
    },
    {
      "step": 5560,
      "train_flow_matching_loss": 0.10595703125,
      "train_physics_loss": 0.0
    },
    {
      "step": 5560,
      "train_flow_matching_loss": 0.10986328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 5560,
      "train_flow_matching_loss": 0.1455078125,
      "train_physics_loss": 0.8466217517852783
    },
    {
      "grad_norm": 0.13891510665416718,
      "learning_rate": 5e-05,
      "loss": 0.1655,
      "step": 5570
    },
    {
      "step": 5570,
      "train_flow_matching_loss": 0.1669921875,
      "train_physics_loss": 0.0
    },
    {
      "step": 5570,
      "train_flow_matching_loss": 0.1298828125,
      "train_physics_loss": 0.0
    },
    {
      "step": 5570,
      "train_flow_matching_loss": 0.1494140625,
      "train_physics_loss": 0.0
    },
    {
      "step": 5570,
      "train_flow_matching_loss": 0.09130859375,
      "train_physics_loss": 0.81119704246521
    },
    {
      "grad_norm": 0.13326311111450195,
      "learning_rate": 5e-05,
      "loss": 0.1732,
      "step": 5580
    },
    {
      "step": 5580,
      "train_flow_matching_loss": 0.10791015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 5580,
      "train_flow_matching_loss": 0.2041015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 5580,
      "train_flow_matching_loss": 0.13671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 5580,
      "train_flow_matching_loss": 0.078125,
      "train_physics_loss": 0.9737961292266846
    },
    {
      "grad_norm": 0.15996326506137848,
      "learning_rate": 5e-05,
      "loss": 0.1693,
      "step": 5590
    },
    {
      "step": 5590,
      "train_flow_matching_loss": 0.212890625,
      "train_physics_loss": 0.0
    },
    {
      "step": 5590,
      "train_flow_matching_loss": 0.130859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 5590,
      "train_flow_matching_loss": 0.1591796875,
      "train_physics_loss": 0.0
    },
    {
      "step": 5590,
      "train_flow_matching_loss": 0.06298828125,
      "train_physics_loss": 0.8269401788711548
    },
    {
      "grad_norm": 0.25981754064559937,
      "learning_rate": 5e-05,
      "loss": 0.1717,
      "step": 5600
    },
    {
      "step": 5600,
      "train_flow_matching_loss": 0.2255859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 5600,
      "train_flow_matching_loss": 0.1796875,
      "train_physics_loss": 0.0
    },
    {
      "step": 5600,
      "train_flow_matching_loss": 0.1630859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 5600,
      "train_flow_matching_loss": 0.07080078125,
      "train_physics_loss": 0.8278355002403259
    },
    {
      "grad_norm": 0.10542238503694534,
      "learning_rate": 5e-05,
      "loss": 0.1761,
      "step": 5610
    },
    {
      "step": 5610,
      "train_flow_matching_loss": 0.1357421875,
      "train_physics_loss": 0.0
    },
    {
      "step": 5610,
      "train_flow_matching_loss": 0.1767578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 5610,
      "train_flow_matching_loss": 0.181640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 5610,
      "train_flow_matching_loss": 0.181640625,
      "train_physics_loss": 1.0113894939422607
    },
    {
      "grad_norm": 0.13496427237987518,
      "learning_rate": 5e-05,
      "loss": 0.1734,
      "step": 5620
    },
    {
      "step": 5620,
      "train_flow_matching_loss": 0.1630859375,
      "train_physics_loss": 0.9322342872619629
    },
    {
      "step": 5620,
      "train_flow_matching_loss": 0.10888671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 5620,
      "train_flow_matching_loss": 0.1904296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 5620,
      "train_flow_matching_loss": 0.09375,
      "train_physics_loss": 0.8883784413337708
    },
    {
      "grad_norm": 0.11005838215351105,
      "learning_rate": 5e-05,
      "loss": 0.17,
      "step": 5630
    },
    {
      "step": 5630,
      "train_flow_matching_loss": 0.059326171875,
      "train_physics_loss": 0.8655458092689514
    },
    {
      "step": 5630,
      "train_flow_matching_loss": 0.171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 5630,
      "train_flow_matching_loss": 0.1875,
      "train_physics_loss": 0.0
    },
    {
      "step": 5630,
      "train_flow_matching_loss": 0.06982421875,
      "train_physics_loss": 0.818389356136322
    },
    {
      "grad_norm": 0.11499279737472534,
      "learning_rate": 5e-05,
      "loss": 0.1728,
      "step": 5640
    },
    {
      "step": 5640,
      "train_flow_matching_loss": 0.06201171875,
      "train_physics_loss": 0.7998926639556885
    },
    {
      "step": 5640,
      "train_flow_matching_loss": 0.1845703125,
      "train_physics_loss": 0.0
    },
    {
      "step": 5640,
      "train_flow_matching_loss": 0.2060546875,
      "train_physics_loss": 0.0
    },
    {
      "step": 5640,
      "train_flow_matching_loss": 0.1162109375,
      "train_physics_loss": 0.9080595970153809
    },
    {
      "grad_norm": 0.3187234103679657,
      "learning_rate": 5e-05,
      "loss": 0.1696,
      "step": 5650
    },
    {
      "step": 5650,
      "train_flow_matching_loss": 0.046630859375,
      "train_physics_loss": 0.8310559988021851
    },
    {
      "step": 5650,
      "train_flow_matching_loss": 0.2216796875,
      "train_physics_loss": 0.0
    },
    {
      "step": 5650,
      "train_flow_matching_loss": 0.1357421875,
      "train_physics_loss": 0.0
    },
    {
      "step": 5650,
      "train_flow_matching_loss": 0.0242919921875,
      "train_physics_loss": 0.8065208792686462
    },
    {
      "grad_norm": 0.10705998539924622,
      "learning_rate": 5e-05,
      "loss": 0.1735,
      "step": 5660
    },
    {
      "step": 5660,
      "train_flow_matching_loss": 0.08056640625,
      "train_physics_loss": 0.8818162679672241
    },
    {
      "step": 5660,
      "train_flow_matching_loss": 0.11279296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 5660,
      "train_flow_matching_loss": 0.224609375,
      "train_physics_loss": 0.0
    },
    {
      "step": 5660,
      "train_flow_matching_loss": 0.04638671875,
      "train_physics_loss": 0.8303053975105286
    },
    {
      "grad_norm": 0.11989390105009079,
      "learning_rate": 5e-05,
      "loss": 0.1725,
      "step": 5670
    },
    {
      "step": 5670,
      "train_flow_matching_loss": 0.0888671875,
      "train_physics_loss": 0.8104991316795349
    },
    {
      "step": 5670,
      "train_flow_matching_loss": 0.259765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 5670,
      "train_flow_matching_loss": 0.1962890625,
      "train_physics_loss": 0.0
    },
    {
      "step": 5670,
      "train_flow_matching_loss": 0.040283203125,
      "train_physics_loss": 0.8277183771133423
    },
    {
      "grad_norm": 0.11871691048145294,
      "learning_rate": 5e-05,
      "loss": 0.1707,
      "step": 5680
    },
    {
      "step": 5680,
      "train_flow_matching_loss": 0.07470703125,
      "train_physics_loss": 0.8885637521743774
    },
    {
      "step": 5680,
      "train_flow_matching_loss": 0.057373046875,
      "train_physics_loss": 0.7880875468254089
    },
    {
      "step": 5680,
      "train_flow_matching_loss": 0.26171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 5680,
      "train_flow_matching_loss": 0.1357421875,
      "train_physics_loss": 0.8563991189002991
    },
    {
      "grad_norm": 0.1262085884809494,
      "learning_rate": 5e-05,
      "loss": 0.1761,
      "step": 5690
    },
    {
      "step": 5690,
      "train_flow_matching_loss": 0.06787109375,
      "train_physics_loss": 0.8362040519714355
    },
    {
      "step": 5690,
      "train_flow_matching_loss": 0.07763671875,
      "train_physics_loss": 0.8670029640197754
    },
    {
      "step": 5690,
      "train_flow_matching_loss": 0.2138671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 5690,
      "train_flow_matching_loss": 0.09912109375,
      "train_physics_loss": 0.8851903676986694
    },
    {
      "grad_norm": 0.3235500156879425,
      "learning_rate": 5e-05,
      "loss": 0.1679,
      "step": 5700
    },
    {
      "step": 5700,
      "train_flow_matching_loss": 0.06787109375,
      "train_physics_loss": 0.8765276670455933
    },
    {
      "step": 5700,
      "train_flow_matching_loss": 0.07275390625,
      "train_physics_loss": 0.8899156451225281
    },
    {
      "step": 5700,
      "train_flow_matching_loss": 0.185546875,
      "train_physics_loss": 0.0
    },
    {
      "step": 5700,
      "train_flow_matching_loss": 0.06982421875,
      "train_physics_loss": 0.9341322183609009
    },
    {
      "grad_norm": 0.12344250828027725,
      "learning_rate": 5e-05,
      "loss": 0.1688,
      "step": 5710
    },
    {
      "step": 5710,
      "train_flow_matching_loss": 0.0673828125,
      "train_physics_loss": 0.8712029457092285
    },
    {
      "step": 5710,
      "train_flow_matching_loss": 0.08740234375,
      "train_physics_loss": 0.8353633880615234
    },
    {
      "step": 5710,
      "train_flow_matching_loss": 0.1123046875,
      "train_physics_loss": 0.0
    },
    {
      "step": 5710,
      "train_flow_matching_loss": 0.0869140625,
      "train_physics_loss": 0.8441979289054871
    },
    {
      "grad_norm": 0.18127982318401337,
      "learning_rate": 5e-05,
      "loss": 0.1733,
      "step": 5720
    },
    {
      "step": 5720,
      "train_flow_matching_loss": 0.051025390625,
      "train_physics_loss": 0.851629376411438
    },
    {
      "step": 5720,
      "train_flow_matching_loss": 0.1103515625,
      "train_physics_loss": 0.7715796232223511
    },
    {
      "step": 5720,
      "train_flow_matching_loss": 0.10595703125,
      "train_physics_loss": 0.0
    },
    {
      "step": 5720,
      "train_flow_matching_loss": 0.051025390625,
      "train_physics_loss": 0.8162336349487305
    },
    {
      "grad_norm": 0.11781666427850723,
      "learning_rate": 5e-05,
      "loss": 0.1712,
      "step": 5730
    },
    {
      "step": 5730,
      "train_flow_matching_loss": 0.09716796875,
      "train_physics_loss": 0.9231441020965576
    },
    {
      "step": 5730,
      "train_flow_matching_loss": 0.042236328125,
      "train_physics_loss": 0.9104220271110535
    },
    {
      "step": 5730,
      "train_flow_matching_loss": 0.26171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 5730,
      "train_flow_matching_loss": 0.05419921875,
      "train_physics_loss": 0.8160030245780945
    },
    {
      "grad_norm": 0.14124009013175964,
      "learning_rate": 5e-05,
      "loss": 0.1735,
      "step": 5740
    },
    {
      "step": 5740,
      "train_flow_matching_loss": 0.03759765625,
      "train_physics_loss": 0.8241657018661499
    },
    {
      "step": 5740,
      "train_flow_matching_loss": 0.047119140625,
      "train_physics_loss": 0.8409740328788757
    },
    {
      "step": 5740,
      "train_flow_matching_loss": 0.2236328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 5740,
      "train_flow_matching_loss": 0.0262451171875,
      "train_physics_loss": 0.7905383706092834
    },
    {
      "grad_norm": 0.09575387835502625,
      "learning_rate": 5e-05,
      "loss": 0.1725,
      "step": 5750
    },
    {
      "step": 5750,
      "train_flow_matching_loss": 0.0849609375,
      "train_physics_loss": 0.8885869383811951
    },
    {
      "step": 5750,
      "train_flow_matching_loss": 0.0908203125,
      "train_physics_loss": 0.8015846014022827
    },
    {
      "step": 5750,
      "train_flow_matching_loss": 0.220703125,
      "train_physics_loss": 0.0
    },
    {
      "step": 5750,
      "train_flow_matching_loss": 0.08642578125,
      "train_physics_loss": 0.8123838901519775
    },
    {
      "grad_norm": 0.45134708285331726,
      "learning_rate": 5e-05,
      "loss": 0.1715,
      "step": 5760
    },
    {
      "step": 5760,
      "train_flow_matching_loss": 0.09716796875,
      "train_physics_loss": 0.8783779144287109
    },
    {
      "step": 5760,
      "train_flow_matching_loss": 0.05126953125,
      "train_physics_loss": 0.8140982389450073
    },
    {
      "step": 5760,
      "train_flow_matching_loss": 0.25390625,
      "train_physics_loss": 0.0
    },
    {
      "step": 5760,
      "train_flow_matching_loss": 0.130859375,
      "train_physics_loss": 0.7492160201072693
    },
    {
      "grad_norm": 0.17524565756320953,
      "learning_rate": 5e-05,
      "loss": 0.1726,
      "step": 5770
    },
    {
      "step": 5770,
      "train_flow_matching_loss": 0.07470703125,
      "train_physics_loss": 0.8584541082382202
    },
    {
      "step": 5770,
      "train_flow_matching_loss": 0.07373046875,
      "train_physics_loss": 0.8435750007629395
    },
    {
      "step": 5770,
      "train_flow_matching_loss": 0.232421875,
      "train_physics_loss": 0.0
    },
    {
      "step": 5770,
      "train_flow_matching_loss": 0.08349609375,
      "train_physics_loss": 0.8855262398719788
    },
    {
      "grad_norm": 0.10583832859992981,
      "learning_rate": 5e-05,
      "loss": 0.1757,
      "step": 5780
    },
    {
      "step": 5780,
      "train_flow_matching_loss": 0.08154296875,
      "train_physics_loss": 0.8304222226142883
    },
    {
      "step": 5780,
      "train_flow_matching_loss": 0.07177734375,
      "train_physics_loss": 0.914093554019928
    },
    {
      "step": 5780,
      "train_flow_matching_loss": 0.240234375,
      "train_physics_loss": 0.0
    },
    {
      "step": 5780,
      "train_flow_matching_loss": 0.08251953125,
      "train_physics_loss": 0.8373814821243286
    },
    {
      "grad_norm": 0.10244619846343994,
      "learning_rate": 5e-05,
      "loss": 0.171,
      "step": 5790
    },
    {
      "step": 5790,
      "train_flow_matching_loss": 0.04248046875,
      "train_physics_loss": 0.8673127293586731
    },
    {
      "step": 5790,
      "train_flow_matching_loss": 0.0869140625,
      "train_physics_loss": 0.94512540102005
    },
    {
      "step": 5790,
      "train_flow_matching_loss": 0.1875,
      "train_physics_loss": 0.0
    },
    {
      "step": 5790,
      "train_flow_matching_loss": 0.0751953125,
      "train_physics_loss": 0.7970343828201294
    },
    {
      "grad_norm": 0.17320005595684052,
      "learning_rate": 5e-05,
      "loss": 0.169,
      "step": 5800
    },
    {
      "step": 5800,
      "train_flow_matching_loss": 0.054443359375,
      "train_physics_loss": 0.8273761868476868
    },
    {
      "step": 5800,
      "train_flow_matching_loss": 0.0546875,
      "train_physics_loss": 0.7673800587654114
    },
    {
      "step": 5800,
      "train_flow_matching_loss": 0.130859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 5800,
      "train_flow_matching_loss": 0.052490234375,
      "train_physics_loss": 0.874937891960144
    },
    {
      "grad_norm": 0.24139410257339478,
      "learning_rate": 5e-05,
      "loss": 0.1698,
      "step": 5810
    },
    {
      "step": 5810,
      "train_flow_matching_loss": 0.0947265625,
      "train_physics_loss": 0.8798376321792603
    },
    {
      "step": 5810,
      "train_flow_matching_loss": 0.08203125,
      "train_physics_loss": 0.8383052349090576
    },
    {
      "step": 5810,
      "train_flow_matching_loss": 0.1904296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 5810,
      "train_flow_matching_loss": 0.0625,
      "train_physics_loss": 0.8743511438369751
    },
    {
      "grad_norm": 0.11442209780216217,
      "learning_rate": 5e-05,
      "loss": 0.1658,
      "step": 5820
    },
    {
      "step": 5820,
      "train_flow_matching_loss": 0.0830078125,
      "train_physics_loss": 0.0
    },
    {
      "step": 5820,
      "train_flow_matching_loss": 0.142578125,
      "train_physics_loss": 0.9727907180786133
    },
    {
      "step": 5820,
      "train_flow_matching_loss": 0.1591796875,
      "train_physics_loss": 0.0
    },
    {
      "step": 5820,
      "train_flow_matching_loss": 0.1474609375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.0914728119969368,
      "learning_rate": 5e-05,
      "loss": 0.174,
      "step": 5830
    },
    {
      "step": 5830,
      "train_flow_matching_loss": 0.05126953125,
      "train_physics_loss": 0.9130123853683472
    },
    {
      "step": 5830,
      "train_flow_matching_loss": 0.05908203125,
      "train_physics_loss": 0.8450006246566772
    },
    {
      "step": 5830,
      "train_flow_matching_loss": 0.171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 5830,
      "train_flow_matching_loss": 0.1943359375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.0994994267821312,
      "learning_rate": 5e-05,
      "loss": 0.1706,
      "step": 5840
    },
    {
      "step": 5840,
      "train_flow_matching_loss": 0.1455078125,
      "train_physics_loss": 0.9474177360534668
    },
    {
      "step": 5840,
      "train_flow_matching_loss": 0.04443359375,
      "train_physics_loss": 0.823641300201416
    },
    {
      "step": 5840,
      "train_flow_matching_loss": 0.177734375,
      "train_physics_loss": 0.0
    },
    {
      "step": 5840,
      "train_flow_matching_loss": 0.1611328125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.130210280418396,
      "learning_rate": 5e-05,
      "loss": 0.1757,
      "step": 5850
    },
    {
      "step": 5850,
      "train_flow_matching_loss": 0.0712890625,
      "train_physics_loss": 0.8684374094009399
    },
    {
      "step": 5850,
      "train_flow_matching_loss": 0.189453125,
      "train_physics_loss": 0.9651980996131897
    },
    {
      "step": 5850,
      "train_flow_matching_loss": 0.11962890625,
      "train_physics_loss": 0.0
    },
    {
      "step": 5850,
      "train_flow_matching_loss": 0.1650390625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.14911793172359467,
      "learning_rate": 5e-05,
      "loss": 0.1707,
      "step": 5860
    },
    {
      "step": 5860,
      "train_flow_matching_loss": 0.06689453125,
      "train_physics_loss": 0.778348982334137
    },
    {
      "step": 5860,
      "train_flow_matching_loss": 0.0908203125,
      "train_physics_loss": 0.8936958312988281
    },
    {
      "step": 5860,
      "train_flow_matching_loss": 0.171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 5860,
      "train_flow_matching_loss": 0.154296875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.1995197832584381,
      "learning_rate": 5e-05,
      "loss": 0.1691,
      "step": 5870
    },
    {
      "step": 5870,
      "train_flow_matching_loss": 0.05859375,
      "train_physics_loss": 0.8814212679862976
    },
    {
      "step": 5870,
      "train_flow_matching_loss": 0.0849609375,
      "train_physics_loss": 0.8628324866294861
    },
    {
      "step": 5870,
      "train_flow_matching_loss": 0.208984375,
      "train_physics_loss": 0.0
    },
    {
      "step": 5870,
      "train_flow_matching_loss": 0.09326171875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.1429707109928131,
      "learning_rate": 5e-05,
      "loss": 0.1742,
      "step": 5880
    },
    {
      "step": 5880,
      "train_flow_matching_loss": 0.169921875,
      "train_physics_loss": 0.0
    },
    {
      "step": 5880,
      "train_flow_matching_loss": 0.05712890625,
      "train_physics_loss": 0.8505545854568481
    },
    {
      "step": 5880,
      "train_flow_matching_loss": 0.1953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 5880,
      "train_flow_matching_loss": 0.146484375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11686275899410248,
      "learning_rate": 5e-05,
      "loss": 0.1733,
      "step": 5890
    },
    {
      "step": 5890,
      "train_flow_matching_loss": 0.1533203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 5890,
      "train_flow_matching_loss": 0.09814453125,
      "train_physics_loss": 0.8248562216758728
    },
    {
      "step": 5890,
      "train_flow_matching_loss": 0.177734375,
      "train_physics_loss": 0.0
    },
    {
      "step": 5890,
      "train_flow_matching_loss": 0.12060546875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08272276073694229,
      "learning_rate": 5e-05,
      "loss": 0.1704,
      "step": 5900
    },
    {
      "step": 5900,
      "train_flow_matching_loss": 0.1611328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 5900,
      "train_flow_matching_loss": 0.1259765625,
      "train_physics_loss": 0.8691132664680481
    },
    {
      "step": 5900,
      "train_flow_matching_loss": 0.1611328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 5900,
      "train_flow_matching_loss": 0.1904296875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08575811982154846,
      "learning_rate": 5e-05,
      "loss": 0.1691,
      "step": 5910
    },
    {
      "step": 5910,
      "train_flow_matching_loss": 0.1435546875,
      "train_physics_loss": 0.0
    },
    {
      "step": 5910,
      "train_flow_matching_loss": 0.0888671875,
      "train_physics_loss": 0.8750696778297424
    },
    {
      "step": 5910,
      "train_flow_matching_loss": 0.2138671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 5910,
      "train_flow_matching_loss": 0.10693359375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08492664247751236,
      "learning_rate": 5e-05,
      "loss": 0.1722,
      "step": 5920
    },
    {
      "step": 5920,
      "train_flow_matching_loss": 0.12255859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 5920,
      "train_flow_matching_loss": 0.07080078125,
      "train_physics_loss": 0.8366697430610657
    },
    {
      "step": 5920,
      "train_flow_matching_loss": 0.2431640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 5920,
      "train_flow_matching_loss": 0.2060546875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.16150499880313873,
      "learning_rate": 5e-05,
      "loss": 0.1659,
      "step": 5930
    },
    {
      "step": 5930,
      "train_flow_matching_loss": 0.1728515625,
      "train_physics_loss": 0.0
    },
    {
      "step": 5930,
      "train_flow_matching_loss": 0.197265625,
      "train_physics_loss": 0.0
    },
    {
      "step": 5930,
      "train_flow_matching_loss": 0.2109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 5930,
      "train_flow_matching_loss": 0.2275390625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10629814118146896,
      "learning_rate": 5e-05,
      "loss": 0.1719,
      "step": 5940
    },
    {
      "step": 5940,
      "train_flow_matching_loss": 0.1328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 5940,
      "train_flow_matching_loss": 0.203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 5940,
      "train_flow_matching_loss": 0.271484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 5940,
      "train_flow_matching_loss": 0.1640625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.13102605938911438,
      "learning_rate": 5e-05,
      "loss": 0.1692,
      "step": 5950
    },
    {
      "step": 5950,
      "train_flow_matching_loss": 0.1513671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 5950,
      "train_flow_matching_loss": 0.11669921875,
      "train_physics_loss": 0.0
    },
    {
      "step": 5950,
      "train_flow_matching_loss": 0.134765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 5950,
      "train_flow_matching_loss": 0.1904296875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.07540751248598099,
      "learning_rate": 5e-05,
      "loss": 0.1698,
      "step": 5960
    },
    {
      "step": 5960,
      "train_flow_matching_loss": 0.138671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 5960,
      "train_flow_matching_loss": 0.1865234375,
      "train_physics_loss": 0.0
    },
    {
      "step": 5960,
      "train_flow_matching_loss": 0.177734375,
      "train_physics_loss": 0.0
    },
    {
      "step": 5960,
      "train_flow_matching_loss": 0.158203125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11316760629415512,
      "learning_rate": 5e-05,
      "loss": 0.1684,
      "step": 5970
    },
    {
      "step": 5970,
      "train_flow_matching_loss": 0.2236328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 5970,
      "train_flow_matching_loss": 0.140625,
      "train_physics_loss": 0.0
    },
    {
      "step": 5970,
      "train_flow_matching_loss": 0.2177734375,
      "train_physics_loss": 0.0
    },
    {
      "step": 5970,
      "train_flow_matching_loss": 0.1376953125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.13241122663021088,
      "learning_rate": 5e-05,
      "loss": 0.1732,
      "step": 5980
    },
    {
      "step": 5980,
      "train_flow_matching_loss": 0.2001953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 5980,
      "train_flow_matching_loss": 0.12353515625,
      "train_physics_loss": 0.0
    },
    {
      "step": 5980,
      "train_flow_matching_loss": 0.25,
      "train_physics_loss": 0.0
    },
    {
      "step": 5980,
      "train_flow_matching_loss": 0.1484375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09994414448738098,
      "learning_rate": 5e-05,
      "loss": 0.1668,
      "step": 5990
    },
    {
      "step": 5990,
      "train_flow_matching_loss": 0.1962890625,
      "train_physics_loss": 0.0
    },
    {
      "step": 5990,
      "train_flow_matching_loss": 0.08203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 5990,
      "train_flow_matching_loss": 0.1962890625,
      "train_physics_loss": 0.0
    },
    {
      "step": 5990,
      "train_flow_matching_loss": 0.1376953125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11552978307008743,
      "learning_rate": 5e-05,
      "loss": 0.1712,
      "step": 6000
    },
    {
      "step": 6000,
      "train_flow_matching_loss": 0.11181640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 6000,
      "train_flow_matching_loss": 0.171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 6000,
      "train_flow_matching_loss": 0.2373046875,
      "train_physics_loss": 0.0
    },
    {
      "step": 6000,
      "train_flow_matching_loss": 0.08349609375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11953350156545639,
      "learning_rate": 5e-05,
      "loss": 0.1685,
      "step": 6010
    },
    {
      "step": 6010,
      "train_flow_matching_loss": 0.2041015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 6010,
      "train_flow_matching_loss": 0.11181640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 6010,
      "train_flow_matching_loss": 0.1044921875,
      "train_physics_loss": 0.0
    },
    {
      "step": 6010,
      "train_flow_matching_loss": 0.11572265625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.12936735153198242,
      "learning_rate": 5e-05,
      "loss": 0.1701,
      "step": 6020
    },
    {
      "step": 6020,
      "train_flow_matching_loss": 0.12060546875,
      "train_physics_loss": 0.0
    },
    {
      "step": 6020,
      "train_flow_matching_loss": 0.1025390625,
      "train_physics_loss": 0.0
    },
    {
      "step": 6020,
      "train_flow_matching_loss": 0.2060546875,
      "train_physics_loss": 0.0
    },
    {
      "step": 6020,
      "train_flow_matching_loss": 0.1708984375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09525711089372635,
      "learning_rate": 5e-05,
      "loss": 0.1663,
      "step": 6030
    },
    {
      "step": 6030,
      "train_flow_matching_loss": 0.134765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 6030,
      "train_flow_matching_loss": 0.1396484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 6030,
      "train_flow_matching_loss": 0.2490234375,
      "train_physics_loss": 0.0
    },
    {
      "step": 6030,
      "train_flow_matching_loss": 0.2158203125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10425060987472534,
      "learning_rate": 5e-05,
      "loss": 0.1705,
      "step": 6040
    },
    {
      "step": 6040,
      "train_flow_matching_loss": 0.1630859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 6040,
      "train_flow_matching_loss": 0.2392578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 6040,
      "train_flow_matching_loss": 0.275390625,
      "train_physics_loss": 0.0
    },
    {
      "step": 6040,
      "train_flow_matching_loss": 0.126953125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09038829803466797,
      "learning_rate": 5e-05,
      "loss": 0.1653,
      "step": 6050
    },
    {
      "step": 6050,
      "train_flow_matching_loss": 0.29296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 6050,
      "train_flow_matching_loss": 0.1630859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 6050,
      "train_flow_matching_loss": 0.2197265625,
      "train_physics_loss": 0.0
    },
    {
      "step": 6050,
      "train_flow_matching_loss": 0.1630859375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11444208025932312,
      "learning_rate": 5e-05,
      "loss": 0.1671,
      "step": 6060
    },
    {
      "step": 6060,
      "train_flow_matching_loss": 0.142578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 6060,
      "train_flow_matching_loss": 0.158203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 6060,
      "train_flow_matching_loss": 0.11572265625,
      "train_physics_loss": 0.8936569690704346
    },
    {
      "step": 6060,
      "train_flow_matching_loss": 0.1787109375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.14980517327785492,
      "learning_rate": 5e-05,
      "loss": 0.1669,
      "step": 6070
    },
    {
      "step": 6070,
      "train_flow_matching_loss": 0.1357421875,
      "train_physics_loss": 0.0
    },
    {
      "step": 6070,
      "train_flow_matching_loss": 0.1728515625,
      "train_physics_loss": 0.0
    },
    {
      "step": 6070,
      "train_flow_matching_loss": 0.12890625,
      "train_physics_loss": 0.9109205007553101
    },
    {
      "step": 6070,
      "train_flow_matching_loss": 0.26953125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.14881306886672974,
      "learning_rate": 5e-05,
      "loss": 0.1765,
      "step": 6080
    },
    {
      "step": 6080,
      "train_flow_matching_loss": 0.169921875,
      "train_physics_loss": 0.0
    },
    {
      "step": 6080,
      "train_flow_matching_loss": 0.0888671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 6080,
      "train_flow_matching_loss": 0.052978515625,
      "train_physics_loss": 0.9029006958007812
    },
    {
      "step": 6080,
      "train_flow_matching_loss": 0.2041015625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.1169731393456459,
      "learning_rate": 5e-05,
      "loss": 0.1705,
      "step": 6090
    },
    {
      "step": 6090,
      "train_flow_matching_loss": 0.1611328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 6090,
      "train_flow_matching_loss": 0.173828125,
      "train_physics_loss": 0.0
    },
    {
      "step": 6090,
      "train_flow_matching_loss": 0.1513671875,
      "train_physics_loss": 0.8040655255317688
    },
    {
      "step": 6090,
      "train_flow_matching_loss": 0.205078125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10078093409538269,
      "learning_rate": 5e-05,
      "loss": 0.1676,
      "step": 6100
    },
    {
      "step": 6100,
      "train_flow_matching_loss": 0.154296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 6100,
      "train_flow_matching_loss": 0.11376953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 6100,
      "train_flow_matching_loss": 0.09423828125,
      "train_physics_loss": 0.8047821521759033
    },
    {
      "step": 6100,
      "train_flow_matching_loss": 0.11865234375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11153939366340637,
      "learning_rate": 5e-05,
      "loss": 0.1705,
      "step": 6110
    },
    {
      "step": 6110,
      "train_flow_matching_loss": 0.1064453125,
      "train_physics_loss": 0.0
    },
    {
      "step": 6110,
      "train_flow_matching_loss": 0.1357421875,
      "train_physics_loss": 0.0
    },
    {
      "step": 6110,
      "train_flow_matching_loss": 0.1484375,
      "train_physics_loss": 0.8665374517440796
    },
    {
      "step": 6110,
      "train_flow_matching_loss": 0.1513671875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.06968551874160767,
      "learning_rate": 5e-05,
      "loss": 0.1651,
      "step": 6120
    },
    {
      "step": 6120,
      "train_flow_matching_loss": 0.220703125,
      "train_physics_loss": 0.0
    },
    {
      "step": 6120,
      "train_flow_matching_loss": 0.1337890625,
      "train_physics_loss": 0.0
    },
    {
      "step": 6120,
      "train_flow_matching_loss": 0.130859375,
      "train_physics_loss": 0.9278265237808228
    },
    {
      "step": 6120,
      "train_flow_matching_loss": 0.12353515625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.12722909450531006,
      "learning_rate": 5e-05,
      "loss": 0.1724,
      "step": 6130
    },
    {
      "step": 6130,
      "train_flow_matching_loss": 0.1474609375,
      "train_physics_loss": 0.0
    },
    {
      "step": 6130,
      "train_flow_matching_loss": 0.1630859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 6130,
      "train_flow_matching_loss": 0.1015625,
      "train_physics_loss": 0.8786088824272156
    },
    {
      "step": 6130,
      "train_flow_matching_loss": 0.12060546875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10804591327905655,
      "learning_rate": 5e-05,
      "loss": 0.167,
      "step": 6140
    },
    {
      "step": 6140,
      "train_flow_matching_loss": 0.1845703125,
      "train_physics_loss": 0.0
    },
    {
      "step": 6140,
      "train_flow_matching_loss": 0.2001953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 6140,
      "train_flow_matching_loss": 0.07763671875,
      "train_physics_loss": 0.7974428534507751
    },
    {
      "step": 6140,
      "train_flow_matching_loss": 0.126953125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09672603011131287,
      "learning_rate": 5e-05,
      "loss": 0.165,
      "step": 6150
    },
    {
      "step": 6150,
      "train_flow_matching_loss": 0.12890625,
      "train_physics_loss": 0.0
    },
    {
      "step": 6150,
      "train_flow_matching_loss": 0.2421875,
      "train_physics_loss": 0.0
    },
    {
      "step": 6150,
      "train_flow_matching_loss": 0.189453125,
      "train_physics_loss": 0.9032256603240967
    },
    {
      "step": 6150,
      "train_flow_matching_loss": 0.1328125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.12735669314861298,
      "learning_rate": 5e-05,
      "loss": 0.1723,
      "step": 6160
    },
    {
      "step": 6160,
      "train_flow_matching_loss": 0.158203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 6160,
      "train_flow_matching_loss": 0.2353515625,
      "train_physics_loss": 0.0
    },
    {
      "step": 6160,
      "train_flow_matching_loss": 0.05224609375,
      "train_physics_loss": 0.8963804244995117
    },
    {
      "step": 6160,
      "train_flow_matching_loss": 0.1484375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08251035213470459,
      "learning_rate": 5e-05,
      "loss": 0.1689,
      "step": 6170
    },
    {
      "step": 6170,
      "train_flow_matching_loss": 0.1630859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 6170,
      "train_flow_matching_loss": 0.208984375,
      "train_physics_loss": 0.0
    },
    {
      "step": 6170,
      "train_flow_matching_loss": 0.11279296875,
      "train_physics_loss": 0.8771185874938965
    },
    {
      "step": 6170,
      "train_flow_matching_loss": 0.134765625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.0805521085858345,
      "learning_rate": 5e-05,
      "loss": 0.1692,
      "step": 6180
    },
    {
      "step": 6180,
      "train_flow_matching_loss": 0.28515625,
      "train_physics_loss": 0.0
    },
    {
      "step": 6180,
      "train_flow_matching_loss": 0.1708984375,
      "train_physics_loss": 0.0
    },
    {
      "step": 6180,
      "train_flow_matching_loss": 0.1318359375,
      "train_physics_loss": 0.8883423805236816
    },
    {
      "step": 6180,
      "train_flow_matching_loss": 0.203125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.15268702805042267,
      "learning_rate": 5e-05,
      "loss": 0.1713,
      "step": 6190
    },
    {
      "step": 6190,
      "train_flow_matching_loss": 0.1689453125,
      "train_physics_loss": 0.0
    },
    {
      "step": 6190,
      "train_flow_matching_loss": 0.1787109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 6190,
      "train_flow_matching_loss": 0.07763671875,
      "train_physics_loss": 0.8875340819358826
    },
    {
      "step": 6190,
      "train_flow_matching_loss": 0.12890625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.13847659528255463,
      "learning_rate": 5e-05,
      "loss": 0.1707,
      "step": 6200
    },
    {
      "step": 6200,
      "train_flow_matching_loss": 0.1953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 6200,
      "train_flow_matching_loss": 0.1103515625,
      "train_physics_loss": 0.0
    },
    {
      "step": 6200,
      "train_flow_matching_loss": 0.15625,
      "train_physics_loss": 1.1037756204605103
    },
    {
      "step": 6200,
      "train_flow_matching_loss": 0.1572265625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.19430643320083618,
      "learning_rate": 5e-05,
      "loss": 0.1726,
      "step": 6210
    },
    {
      "step": 6210,
      "train_flow_matching_loss": 0.1513671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 6210,
      "train_flow_matching_loss": 0.162109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 6210,
      "train_flow_matching_loss": 0.10107421875,
      "train_physics_loss": 1.003231167793274
    },
    {
      "step": 6210,
      "train_flow_matching_loss": 0.10791015625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08760684728622437,
      "learning_rate": 5e-05,
      "loss": 0.1675,
      "step": 6220
    },
    {
      "step": 6220,
      "train_flow_matching_loss": 0.1484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 6220,
      "train_flow_matching_loss": 0.244140625,
      "train_physics_loss": 0.0
    },
    {
      "step": 6220,
      "train_flow_matching_loss": 0.09326171875,
      "train_physics_loss": 0.7659672498703003
    },
    {
      "step": 6220,
      "train_flow_matching_loss": 0.169921875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.07998929172754288,
      "learning_rate": 5e-05,
      "loss": 0.1686,
      "step": 6230
    },
    {
      "step": 6230,
      "train_flow_matching_loss": 0.181640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 6230,
      "train_flow_matching_loss": 0.173828125,
      "train_physics_loss": 0.0
    },
    {
      "step": 6230,
      "train_flow_matching_loss": 0.061279296875,
      "train_physics_loss": 1.0129896402359009
    },
    {
      "step": 6230,
      "train_flow_matching_loss": 0.1787109375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08651034533977509,
      "learning_rate": 5e-05,
      "loss": 0.1731,
      "step": 6240
    },
    {
      "step": 6240,
      "train_flow_matching_loss": 0.2197265625,
      "train_physics_loss": 0.0
    },
    {
      "step": 6240,
      "train_flow_matching_loss": 0.1337890625,
      "train_physics_loss": 0.0
    },
    {
      "step": 6240,
      "train_flow_matching_loss": 0.05126953125,
      "train_physics_loss": 0.824759304523468
    },
    {
      "step": 6240,
      "train_flow_matching_loss": 0.140625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.13344338536262512,
      "learning_rate": 5e-05,
      "loss": 0.1731,
      "step": 6250
    },
    {
      "step": 6250,
      "train_flow_matching_loss": 0.1865234375,
      "train_physics_loss": 0.0
    },
    {
      "step": 6250,
      "train_flow_matching_loss": 0.11865234375,
      "train_physics_loss": 0.0
    },
    {
      "step": 6250,
      "train_flow_matching_loss": 0.08740234375,
      "train_physics_loss": 0.0
    },
    {
      "step": 6250,
      "train_flow_matching_loss": 0.185546875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.49037086963653564,
      "learning_rate": 5e-05,
      "loss": 0.1753,
      "step": 6260
    },
    {
      "step": 6260,
      "train_flow_matching_loss": 0.1943359375,
      "train_physics_loss": 0.0
    },
    {
      "step": 6260,
      "train_flow_matching_loss": 0.189453125,
      "train_physics_loss": 0.0
    },
    {
      "step": 6260,
      "train_flow_matching_loss": 0.0888671875,
      "train_physics_loss": 0.7942041158676147
    },
    {
      "step": 6260,
      "train_flow_matching_loss": 0.26171875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.2350030243396759,
      "learning_rate": 5e-05,
      "loss": 0.1686,
      "step": 6270
    },
    {
      "step": 6270,
      "train_flow_matching_loss": 0.142578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 6270,
      "train_flow_matching_loss": 0.08447265625,
      "train_physics_loss": 0.0
    },
    {
      "step": 6270,
      "train_flow_matching_loss": 0.1357421875,
      "train_physics_loss": 1.07256019115448
    },
    {
      "step": 6270,
      "train_flow_matching_loss": 0.2041015625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10206209868192673,
      "learning_rate": 5e-05,
      "loss": 0.1703,
      "step": 6280
    },
    {
      "step": 6280,
      "train_flow_matching_loss": 0.1728515625,
      "train_physics_loss": 0.0
    },
    {
      "step": 6280,
      "train_flow_matching_loss": 0.1533203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 6280,
      "train_flow_matching_loss": 0.07421875,
      "train_physics_loss": 0.924303412437439
    },
    {
      "step": 6280,
      "train_flow_matching_loss": 0.1826171875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09163948148488998,
      "learning_rate": 5e-05,
      "loss": 0.1741,
      "step": 6290
    },
    {
      "step": 6290,
      "train_flow_matching_loss": 0.1884765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 6290,
      "train_flow_matching_loss": 0.2236328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 6290,
      "train_flow_matching_loss": 0.0712890625,
      "train_physics_loss": 0.8564404249191284
    },
    {
      "step": 6290,
      "train_flow_matching_loss": 0.1728515625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.7749484777450562,
      "learning_rate": 5e-05,
      "loss": 0.1702,
      "step": 6300
    },
    {
      "step": 6300,
      "train_flow_matching_loss": 0.234375,
      "train_physics_loss": 0.0
    },
    {
      "step": 6300,
      "train_flow_matching_loss": 0.08544921875,
      "train_physics_loss": 0.0
    },
    {
      "step": 6300,
      "train_flow_matching_loss": 0.1416015625,
      "train_physics_loss": 0.9255414605140686
    },
    {
      "step": 6300,
      "train_flow_matching_loss": 0.1494140625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.1523500382900238,
      "learning_rate": 5e-05,
      "loss": 0.1726,
      "step": 6310
    },
    {
      "step": 6310,
      "train_flow_matching_loss": 0.1884765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 6310,
      "train_flow_matching_loss": 0.1396484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 6310,
      "train_flow_matching_loss": 0.078125,
      "train_physics_loss": 0.8282768130302429
    },
    {
      "step": 6310,
      "train_flow_matching_loss": 0.1484375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.1271776258945465,
      "learning_rate": 5e-05,
      "loss": 0.1688,
      "step": 6320
    },
    {
      "step": 6320,
      "train_flow_matching_loss": 0.1904296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 6320,
      "train_flow_matching_loss": 0.1044921875,
      "train_physics_loss": 0.0
    },
    {
      "step": 6320,
      "train_flow_matching_loss": 0.0859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 6320,
      "train_flow_matching_loss": 0.1787109375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10713176429271698,
      "learning_rate": 5e-05,
      "loss": 0.168,
      "step": 6330
    },
    {
      "step": 6330,
      "train_flow_matching_loss": 0.2109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 6330,
      "train_flow_matching_loss": 0.150390625,
      "train_physics_loss": 0.0
    },
    {
      "step": 6330,
      "train_flow_matching_loss": 0.169921875,
      "train_physics_loss": 0.0
    },
    {
      "step": 6330,
      "train_flow_matching_loss": 0.1875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10408598184585571,
      "learning_rate": 5e-05,
      "loss": 0.1666,
      "step": 6340
    },
    {
      "step": 6340,
      "train_flow_matching_loss": 0.197265625,
      "train_physics_loss": 0.0
    },
    {
      "step": 6340,
      "train_flow_matching_loss": 0.208984375,
      "train_physics_loss": 0.0
    },
    {
      "step": 6340,
      "train_flow_matching_loss": 0.1572265625,
      "train_physics_loss": 0.0
    },
    {
      "step": 6340,
      "train_flow_matching_loss": 0.1494140625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11862245947122574,
      "learning_rate": 5e-05,
      "loss": 0.1691,
      "step": 6350
    },
    {
      "step": 6350,
      "train_flow_matching_loss": 0.1376953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 6350,
      "train_flow_matching_loss": 0.1552734375,
      "train_physics_loss": 0.0
    },
    {
      "step": 6350,
      "train_flow_matching_loss": 0.1455078125,
      "train_physics_loss": 0.0
    },
    {
      "step": 6350,
      "train_flow_matching_loss": 0.1748046875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.32328474521636963,
      "learning_rate": 5e-05,
      "loss": 0.1723,
      "step": 6360
    },
    {
      "step": 6360,
      "train_flow_matching_loss": 0.279296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 6360,
      "train_flow_matching_loss": 0.08984375,
      "train_physics_loss": 0.0
    },
    {
      "step": 6360,
      "train_flow_matching_loss": 0.1376953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 6360,
      "train_flow_matching_loss": 0.162109375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.12779057025909424,
      "learning_rate": 5e-05,
      "loss": 0.1686,
      "step": 6370
    },
    {
      "step": 6370,
      "train_flow_matching_loss": 0.2275390625,
      "train_physics_loss": 0.0
    },
    {
      "step": 6370,
      "train_flow_matching_loss": 0.10986328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 6370,
      "train_flow_matching_loss": 0.09326171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 6370,
      "train_flow_matching_loss": 0.130859375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.079154834151268,
      "learning_rate": 5e-05,
      "loss": 0.1688,
      "step": 6380
    },
    {
      "step": 6380,
      "train_flow_matching_loss": 0.1455078125,
      "train_physics_loss": 0.0
    },
    {
      "step": 6380,
      "train_flow_matching_loss": 0.13671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 6380,
      "train_flow_matching_loss": 0.1396484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 6380,
      "train_flow_matching_loss": 0.1875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09951560199260712,
      "learning_rate": 5e-05,
      "loss": 0.1742,
      "step": 6390
    },
    {
      "step": 6390,
      "train_flow_matching_loss": 0.1494140625,
      "train_physics_loss": 0.0
    },
    {
      "step": 6390,
      "train_flow_matching_loss": 0.142578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 6390,
      "train_flow_matching_loss": 0.1259765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 6390,
      "train_flow_matching_loss": 0.126953125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.1179545447230339,
      "learning_rate": 5e-05,
      "loss": 0.1672,
      "step": 6400
    },
    {
      "step": 6400,
      "train_flow_matching_loss": 0.1259765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 6400,
      "train_flow_matching_loss": 0.12353515625,
      "train_physics_loss": 0.0
    },
    {
      "step": 6400,
      "train_flow_matching_loss": 0.138671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 6400,
      "train_flow_matching_loss": 0.166015625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.14587950706481934,
      "learning_rate": 5e-05,
      "loss": 0.1732,
      "step": 6410
    },
    {
      "step": 6410,
      "train_flow_matching_loss": 0.150390625,
      "train_physics_loss": 0.0
    },
    {
      "step": 6410,
      "train_flow_matching_loss": 0.1474609375,
      "train_physics_loss": 0.0
    },
    {
      "step": 6410,
      "train_flow_matching_loss": 0.09814453125,
      "train_physics_loss": 0.0
    },
    {
      "step": 6410,
      "train_flow_matching_loss": 0.1552734375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10389851778745651,
      "learning_rate": 5e-05,
      "loss": 0.1695,
      "step": 6420
    },
    {
      "step": 6420,
      "train_flow_matching_loss": 0.1318359375,
      "train_physics_loss": 0.0
    },
    {
      "step": 6420,
      "train_flow_matching_loss": 0.12353515625,
      "train_physics_loss": 0.0
    },
    {
      "step": 6420,
      "train_flow_matching_loss": 0.1240234375,
      "train_physics_loss": 0.0
    },
    {
      "step": 6420,
      "train_flow_matching_loss": 0.1953125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.12009403109550476,
      "learning_rate": 5e-05,
      "loss": 0.1678,
      "step": 6430
    },
    {
      "step": 6430,
      "train_flow_matching_loss": 0.2412109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 6430,
      "train_flow_matching_loss": 0.171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 6430,
      "train_flow_matching_loss": 0.1826171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 6430,
      "train_flow_matching_loss": 0.126953125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.1063004583120346,
      "learning_rate": 5e-05,
      "loss": 0.1698,
      "step": 6440
    },
    {
      "step": 6440,
      "train_flow_matching_loss": 0.15625,
      "train_physics_loss": 0.0
    },
    {
      "step": 6440,
      "train_flow_matching_loss": 0.171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 6440,
      "train_flow_matching_loss": 0.12890625,
      "train_physics_loss": 0.0
    },
    {
      "step": 6440,
      "train_flow_matching_loss": 0.146484375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.07011321932077408,
      "learning_rate": 5e-05,
      "loss": 0.1648,
      "step": 6450
    },
    {
      "step": 6450,
      "train_flow_matching_loss": 0.2001953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 6450,
      "train_flow_matching_loss": 0.1435546875,
      "train_physics_loss": 0.0
    },
    {
      "step": 6450,
      "train_flow_matching_loss": 0.173828125,
      "train_physics_loss": 0.0
    },
    {
      "step": 6450,
      "train_flow_matching_loss": 0.1318359375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.1054430603981018,
      "learning_rate": 5e-05,
      "loss": 0.168,
      "step": 6460
    },
    {
      "step": 6460,
      "train_flow_matching_loss": 0.1953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 6460,
      "train_flow_matching_loss": 0.10400390625,
      "train_physics_loss": 0.0
    },
    {
      "step": 6460,
      "train_flow_matching_loss": 0.11474609375,
      "train_physics_loss": 0.0
    },
    {
      "step": 6460,
      "train_flow_matching_loss": 0.1279296875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09722769260406494,
      "learning_rate": 5e-05,
      "loss": 0.1726,
      "step": 6470
    },
    {
      "step": 6470,
      "train_flow_matching_loss": 0.2490234375,
      "train_physics_loss": 0.0
    },
    {
      "step": 6470,
      "train_flow_matching_loss": 0.177734375,
      "train_physics_loss": 0.0
    },
    {
      "step": 6470,
      "train_flow_matching_loss": 0.130859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 6470,
      "train_flow_matching_loss": 0.1494140625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.1552644520998001,
      "learning_rate": 5e-05,
      "loss": 0.1651,
      "step": 6480
    },
    {
      "step": 6480,
      "train_flow_matching_loss": 0.154296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 6480,
      "train_flow_matching_loss": 0.134765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 6480,
      "train_flow_matching_loss": 0.1494140625,
      "train_physics_loss": 0.0
    },
    {
      "step": 6480,
      "train_flow_matching_loss": 0.10888671875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.0756092220544815,
      "learning_rate": 5e-05,
      "loss": 0.1678,
      "step": 6490
    },
    {
      "step": 6490,
      "train_flow_matching_loss": 0.1884765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 6490,
      "train_flow_matching_loss": 0.1787109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 6490,
      "train_flow_matching_loss": 0.142578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 6490,
      "train_flow_matching_loss": 0.1728515625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.17515069246292114,
      "learning_rate": 5e-05,
      "loss": 0.1679,
      "step": 6500
    },
    {
      "step": 6500,
      "train_flow_matching_loss": 0.1279296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 6500,
      "train_flow_matching_loss": 0.11669921875,
      "train_physics_loss": 0.0
    },
    {
      "step": 6500,
      "train_flow_matching_loss": 0.1767578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 6500,
      "train_flow_matching_loss": 0.11669921875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11451427638530731,
      "learning_rate": 5e-05,
      "loss": 0.1688,
      "step": 6510
    },
    {
      "step": 6510,
      "train_flow_matching_loss": 0.2216796875,
      "train_physics_loss": 0.0
    },
    {
      "step": 6510,
      "train_flow_matching_loss": 0.224609375,
      "train_physics_loss": 0.0
    },
    {
      "step": 6510,
      "train_flow_matching_loss": 0.21875,
      "train_physics_loss": 0.0
    },
    {
      "step": 6510,
      "train_flow_matching_loss": 0.1435546875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.13607119023799896,
      "learning_rate": 5e-05,
      "loss": 0.1711,
      "step": 6520
    },
    {
      "step": 6520,
      "train_flow_matching_loss": 0.1884765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 6520,
      "train_flow_matching_loss": 0.07763671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 6520,
      "train_flow_matching_loss": 0.1787109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 6520,
      "train_flow_matching_loss": 0.162109375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09111930429935455,
      "learning_rate": 5e-05,
      "loss": 0.1703,
      "step": 6530
    },
    {
      "step": 6530,
      "train_flow_matching_loss": 0.140625,
      "train_physics_loss": 0.0
    },
    {
      "step": 6530,
      "train_flow_matching_loss": 0.173828125,
      "train_physics_loss": 0.0
    },
    {
      "step": 6530,
      "train_flow_matching_loss": 0.228515625,
      "train_physics_loss": 0.0
    },
    {
      "step": 6530,
      "train_flow_matching_loss": 0.162109375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.36029762029647827,
      "learning_rate": 5e-05,
      "loss": 0.1684,
      "step": 6540
    },
    {
      "step": 6540,
      "train_flow_matching_loss": 0.162109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 6540,
      "train_flow_matching_loss": 0.1689453125,
      "train_physics_loss": 0.0
    },
    {
      "step": 6540,
      "train_flow_matching_loss": 0.171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 6540,
      "train_flow_matching_loss": 0.1142578125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.114288330078125,
      "learning_rate": 5e-05,
      "loss": 0.164,
      "step": 6550
    },
    {
      "step": 6550,
      "train_flow_matching_loss": 0.1357421875,
      "train_physics_loss": 0.0
    },
    {
      "step": 6550,
      "train_flow_matching_loss": 0.1357421875,
      "train_physics_loss": 0.0
    },
    {
      "step": 6550,
      "train_flow_matching_loss": 0.1611328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 6550,
      "train_flow_matching_loss": 0.189453125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11846859753131866,
      "learning_rate": 5e-05,
      "loss": 0.168,
      "step": 6560
    },
    {
      "step": 6560,
      "train_flow_matching_loss": 0.1962890625,
      "train_physics_loss": 0.0
    },
    {
      "step": 6560,
      "train_flow_matching_loss": 0.146484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 6560,
      "train_flow_matching_loss": 0.1142578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 6560,
      "train_flow_matching_loss": 0.142578125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09667627513408661,
      "learning_rate": 5e-05,
      "loss": 0.1725,
      "step": 6570
    },
    {
      "step": 6570,
      "train_flow_matching_loss": 0.1552734375,
      "train_physics_loss": 0.0
    },
    {
      "step": 6570,
      "train_flow_matching_loss": 0.2451171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 6570,
      "train_flow_matching_loss": 0.1826171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 6570,
      "train_flow_matching_loss": 0.1953125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10650286078453064,
      "learning_rate": 5e-05,
      "loss": 0.1742,
      "step": 6580
    },
    {
      "step": 6580,
      "train_flow_matching_loss": 0.11474609375,
      "train_physics_loss": 0.0
    },
    {
      "step": 6580,
      "train_flow_matching_loss": 0.2431640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 6580,
      "train_flow_matching_loss": 0.2119140625,
      "train_physics_loss": 0.0
    },
    {
      "step": 6580,
      "train_flow_matching_loss": 0.1748046875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.12461835891008377,
      "learning_rate": 5e-05,
      "loss": 0.1689,
      "step": 6590
    },
    {
      "step": 6590,
      "train_flow_matching_loss": 0.10498046875,
      "train_physics_loss": 0.0
    },
    {
      "step": 6590,
      "train_flow_matching_loss": 0.189453125,
      "train_physics_loss": 0.0
    },
    {
      "step": 6590,
      "train_flow_matching_loss": 0.11669921875,
      "train_physics_loss": 0.0
    },
    {
      "step": 6590,
      "train_flow_matching_loss": 0.1826171875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11295779049396515,
      "learning_rate": 5e-05,
      "loss": 0.165,
      "step": 6600
    },
    {
      "step": 6600,
      "train_flow_matching_loss": 0.134765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 6600,
      "train_flow_matching_loss": 0.1494140625,
      "train_physics_loss": 0.0
    },
    {
      "step": 6600,
      "train_flow_matching_loss": 0.1953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 6600,
      "train_flow_matching_loss": 0.2216796875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11578629165887833,
      "learning_rate": 5e-05,
      "loss": 0.1639,
      "step": 6610
    },
    {
      "step": 6610,
      "train_flow_matching_loss": 0.189453125,
      "train_physics_loss": 0.0
    },
    {
      "step": 6610,
      "train_flow_matching_loss": 0.1357421875,
      "train_physics_loss": 0.0
    },
    {
      "step": 6610,
      "train_flow_matching_loss": 0.08447265625,
      "train_physics_loss": 0.0
    },
    {
      "step": 6610,
      "train_flow_matching_loss": 0.224609375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.0971355065703392,
      "learning_rate": 5e-05,
      "loss": 0.1659,
      "step": 6620
    },
    {
      "step": 6620,
      "train_flow_matching_loss": 0.1845703125,
      "train_physics_loss": 0.0
    },
    {
      "step": 6620,
      "train_flow_matching_loss": 0.228515625,
      "train_physics_loss": 0.0
    },
    {
      "step": 6620,
      "train_flow_matching_loss": 0.2001953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 6620,
      "train_flow_matching_loss": 0.1396484375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09487245231866837,
      "learning_rate": 5e-05,
      "loss": 0.1689,
      "step": 6630
    },
    {
      "step": 6630,
      "train_flow_matching_loss": 0.0771484375,
      "train_physics_loss": 0.921662449836731
    },
    {
      "step": 6630,
      "train_flow_matching_loss": 0.201171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 6630,
      "train_flow_matching_loss": 0.2177734375,
      "train_physics_loss": 0.0
    },
    {
      "step": 6630,
      "train_flow_matching_loss": 0.1982421875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.1022564023733139,
      "learning_rate": 5e-05,
      "loss": 0.1699,
      "step": 6640
    },
    {
      "step": 6640,
      "train_flow_matching_loss": 0.09033203125,
      "train_physics_loss": 0.872527539730072
    },
    {
      "step": 6640,
      "train_flow_matching_loss": 0.1806640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 6640,
      "train_flow_matching_loss": 0.2021484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 6640,
      "train_flow_matching_loss": 0.1708984375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09990014135837555,
      "learning_rate": 5e-05,
      "loss": 0.1678,
      "step": 6650
    },
    {
      "step": 6650,
      "train_flow_matching_loss": 0.0625,
      "train_physics_loss": 0.8270888924598694
    },
    {
      "step": 6650,
      "train_flow_matching_loss": 0.1376953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 6650,
      "train_flow_matching_loss": 0.1171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 6650,
      "train_flow_matching_loss": 0.1474609375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.14252930879592896,
      "learning_rate": 5e-05,
      "loss": 0.1713,
      "step": 6660
    },
    {
      "step": 6660,
      "train_flow_matching_loss": 0.09326171875,
      "train_physics_loss": 0.8769527077674866
    },
    {
      "step": 6660,
      "train_flow_matching_loss": 0.10986328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 6660,
      "train_flow_matching_loss": 0.2001953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 6660,
      "train_flow_matching_loss": 0.1396484375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08128003031015396,
      "learning_rate": 5e-05,
      "loss": 0.1676,
      "step": 6670
    },
    {
      "step": 6670,
      "train_flow_matching_loss": 0.060302734375,
      "train_physics_loss": 0.8267886638641357
    },
    {
      "step": 6670,
      "train_flow_matching_loss": 0.1494140625,
      "train_physics_loss": 0.0
    },
    {
      "step": 6670,
      "train_flow_matching_loss": 0.193359375,
      "train_physics_loss": 0.0
    },
    {
      "step": 6670,
      "train_flow_matching_loss": 0.1865234375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09043054282665253,
      "learning_rate": 5e-05,
      "loss": 0.1708,
      "step": 6680
    },
    {
      "step": 6680,
      "train_flow_matching_loss": 0.08935546875,
      "train_physics_loss": 0.8561639189720154
    },
    {
      "step": 6680,
      "train_flow_matching_loss": 0.197265625,
      "train_physics_loss": 0.0
    },
    {
      "step": 6680,
      "train_flow_matching_loss": 0.1357421875,
      "train_physics_loss": 0.0
    },
    {
      "step": 6680,
      "train_flow_matching_loss": 0.1591796875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.12681539356708527,
      "learning_rate": 5e-05,
      "loss": 0.1669,
      "step": 6690
    },
    {
      "step": 6690,
      "train_flow_matching_loss": 0.061279296875,
      "train_physics_loss": 0.8108660578727722
    },
    {
      "step": 6690,
      "train_flow_matching_loss": 0.1513671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 6690,
      "train_flow_matching_loss": 0.185546875,
      "train_physics_loss": 0.0
    },
    {
      "step": 6690,
      "train_flow_matching_loss": 0.1005859375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08331726491451263,
      "learning_rate": 5e-05,
      "loss": 0.168,
      "step": 6700
    },
    {
      "step": 6700,
      "train_flow_matching_loss": 0.09765625,
      "train_physics_loss": 0.8881831169128418
    },
    {
      "step": 6700,
      "train_flow_matching_loss": 0.1318359375,
      "train_physics_loss": 0.0
    },
    {
      "step": 6700,
      "train_flow_matching_loss": 0.130859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 6700,
      "train_flow_matching_loss": 0.08837890625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.16373421251773834,
      "learning_rate": 5e-05,
      "loss": 0.1651,
      "step": 6710
    },
    {
      "step": 6710,
      "train_flow_matching_loss": 0.09375,
      "train_physics_loss": 0.8398433923721313
    },
    {
      "step": 6710,
      "train_flow_matching_loss": 0.1650390625,
      "train_physics_loss": 0.0
    },
    {
      "step": 6710,
      "train_flow_matching_loss": 0.1591796875,
      "train_physics_loss": 0.0
    },
    {
      "step": 6710,
      "train_flow_matching_loss": 0.10400390625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.13488933444023132,
      "learning_rate": 5e-05,
      "loss": 0.1637,
      "step": 6720
    },
    {
      "step": 6720,
      "train_flow_matching_loss": 0.1416015625,
      "train_physics_loss": 0.8919281959533691
    },
    {
      "step": 6720,
      "train_flow_matching_loss": 0.228515625,
      "train_physics_loss": 0.0
    },
    {
      "step": 6720,
      "train_flow_matching_loss": 0.2294921875,
      "train_physics_loss": 0.0
    },
    {
      "step": 6720,
      "train_flow_matching_loss": 0.09716796875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.12345168739557266,
      "learning_rate": 5e-05,
      "loss": 0.1665,
      "step": 6730
    },
    {
      "step": 6730,
      "train_flow_matching_loss": 0.1181640625,
      "train_physics_loss": 0.7976353168487549
    },
    {
      "step": 6730,
      "train_flow_matching_loss": 0.1572265625,
      "train_physics_loss": 0.0
    },
    {
      "step": 6730,
      "train_flow_matching_loss": 0.10107421875,
      "train_physics_loss": 0.0
    },
    {
      "step": 6730,
      "train_flow_matching_loss": 0.171875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08671899884939194,
      "learning_rate": 5e-05,
      "loss": 0.1745,
      "step": 6740
    },
    {
      "step": 6740,
      "train_flow_matching_loss": 0.037353515625,
      "train_physics_loss": 0.9316241145133972
    },
    {
      "step": 6740,
      "train_flow_matching_loss": 0.171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 6740,
      "train_flow_matching_loss": 0.189453125,
      "train_physics_loss": 0.0
    },
    {
      "step": 6740,
      "train_flow_matching_loss": 0.181640625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10001932084560394,
      "learning_rate": 5e-05,
      "loss": 0.1674,
      "step": 6750
    },
    {
      "step": 6750,
      "train_flow_matching_loss": 0.1025390625,
      "train_physics_loss": 0.9193806648254395
    },
    {
      "step": 6750,
      "train_flow_matching_loss": 0.185546875,
      "train_physics_loss": 0.0
    },
    {
      "step": 6750,
      "train_flow_matching_loss": 0.146484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 6750,
      "train_flow_matching_loss": 0.12255859375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.1014678105711937,
      "learning_rate": 5e-05,
      "loss": 0.164,
      "step": 6760
    },
    {
      "step": 6760,
      "train_flow_matching_loss": 0.13671875,
      "train_physics_loss": 0.8881444931030273
    },
    {
      "step": 6760,
      "train_flow_matching_loss": 0.10791015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 6760,
      "train_flow_matching_loss": 0.173828125,
      "train_physics_loss": 0.0
    },
    {
      "step": 6760,
      "train_flow_matching_loss": 0.1748046875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10172766447067261,
      "learning_rate": 5e-05,
      "loss": 0.1633,
      "step": 6770
    },
    {
      "step": 6770,
      "train_flow_matching_loss": 0.0908203125,
      "train_physics_loss": 0.7781020402908325
    },
    {
      "step": 6770,
      "train_flow_matching_loss": 0.138671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 6770,
      "train_flow_matching_loss": 0.1484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 6770,
      "train_flow_matching_loss": 0.1279296875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.17788496613502502,
      "learning_rate": 5e-05,
      "loss": 0.1627,
      "step": 6780
    },
    {
      "step": 6780,
      "train_flow_matching_loss": 0.08935546875,
      "train_physics_loss": 0.8432785868644714
    },
    {
      "step": 6780,
      "train_flow_matching_loss": 0.189453125,
      "train_physics_loss": 0.0
    },
    {
      "step": 6780,
      "train_flow_matching_loss": 0.11181640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 6780,
      "train_flow_matching_loss": 0.1298828125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.16504645347595215,
      "learning_rate": 5e-05,
      "loss": 0.1669,
      "step": 6790
    },
    {
      "step": 6790,
      "train_flow_matching_loss": 0.08740234375,
      "train_physics_loss": 0.8565552830696106
    },
    {
      "step": 6790,
      "train_flow_matching_loss": 0.298828125,
      "train_physics_loss": 0.0
    },
    {
      "step": 6790,
      "train_flow_matching_loss": 0.10888671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 6790,
      "train_flow_matching_loss": 0.1123046875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10346049815416336,
      "learning_rate": 5e-05,
      "loss": 0.17,
      "step": 6800
    },
    {
      "step": 6800,
      "train_flow_matching_loss": 0.1083984375,
      "train_physics_loss": 0.7892056703567505
    },
    {
      "step": 6800,
      "train_flow_matching_loss": 0.1083984375,
      "train_physics_loss": 0.0
    },
    {
      "step": 6800,
      "train_flow_matching_loss": 0.1259765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 6800,
      "train_flow_matching_loss": 0.25,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10044530034065247,
      "learning_rate": 5e-05,
      "loss": 0.164,
      "step": 6810
    },
    {
      "step": 6810,
      "train_flow_matching_loss": 0.09375,
      "train_physics_loss": 0.8943554759025574
    },
    {
      "step": 6810,
      "train_flow_matching_loss": 0.1640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 6810,
      "train_flow_matching_loss": 0.201171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 6810,
      "train_flow_matching_loss": 0.1611328125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.22797714173793793,
      "learning_rate": 5e-05,
      "loss": 0.1634,
      "step": 6820
    },
    {
      "step": 6820,
      "train_flow_matching_loss": 0.10595703125,
      "train_physics_loss": 0.8847098350524902
    },
    {
      "step": 6820,
      "train_flow_matching_loss": 0.2265625,
      "train_physics_loss": 0.0
    },
    {
      "step": 6820,
      "train_flow_matching_loss": 0.212890625,
      "train_physics_loss": 0.0
    },
    {
      "step": 6820,
      "train_flow_matching_loss": 0.130859375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.1060519590973854,
      "learning_rate": 5e-05,
      "loss": 0.162,
      "step": 6830
    },
    {
      "step": 6830,
      "train_flow_matching_loss": 0.08740234375,
      "train_physics_loss": 0.96514892578125
    },
    {
      "step": 6830,
      "train_flow_matching_loss": 0.11181640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 6830,
      "train_flow_matching_loss": 0.11669921875,
      "train_physics_loss": 0.0
    },
    {
      "step": 6830,
      "train_flow_matching_loss": 0.1865234375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11975228786468506,
      "learning_rate": 5e-05,
      "loss": 0.1645,
      "step": 6840
    },
    {
      "step": 6840,
      "train_flow_matching_loss": 0.1357421875,
      "train_physics_loss": 0.8851432800292969
    },
    {
      "step": 6840,
      "train_flow_matching_loss": 0.130859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 6840,
      "train_flow_matching_loss": 0.21875,
      "train_physics_loss": 0.0
    },
    {
      "step": 6840,
      "train_flow_matching_loss": 0.1953125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08311069756746292,
      "learning_rate": 5e-05,
      "loss": 0.1661,
      "step": 6850
    },
    {
      "step": 6850,
      "train_flow_matching_loss": 0.0693359375,
      "train_physics_loss": 0.8938043713569641
    },
    {
      "step": 6850,
      "train_flow_matching_loss": 0.171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 6850,
      "train_flow_matching_loss": 0.146484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 6850,
      "train_flow_matching_loss": 0.1533203125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.19032146036624908,
      "learning_rate": 5e-05,
      "loss": 0.1732,
      "step": 6860
    },
    {
      "step": 6860,
      "train_flow_matching_loss": 0.11962890625,
      "train_physics_loss": 0.8368831872940063
    },
    {
      "step": 6860,
      "train_flow_matching_loss": 0.1572265625,
      "train_physics_loss": 0.0
    },
    {
      "step": 6860,
      "train_flow_matching_loss": 0.2734375,
      "train_physics_loss": 0.0
    },
    {
      "step": 6860,
      "train_flow_matching_loss": 0.1494140625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.16358543932437897,
      "learning_rate": 5e-05,
      "loss": 0.1696,
      "step": 6870
    },
    {
      "step": 6870,
      "train_flow_matching_loss": 0.1484375,
      "train_physics_loss": 0.8299540877342224
    },
    {
      "step": 6870,
      "train_flow_matching_loss": 0.2421875,
      "train_physics_loss": 0.0
    },
    {
      "step": 6870,
      "train_flow_matching_loss": 0.126953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 6870,
      "train_flow_matching_loss": 0.1123046875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10679058730602264,
      "learning_rate": 5e-05,
      "loss": 0.17,
      "step": 6880
    },
    {
      "step": 6880,
      "train_flow_matching_loss": 0.042724609375,
      "train_physics_loss": 0.8255634903907776
    },
    {
      "step": 6880,
      "train_flow_matching_loss": 0.2197265625,
      "train_physics_loss": 0.0
    },
    {
      "step": 6880,
      "train_flow_matching_loss": 0.2080078125,
      "train_physics_loss": 0.0
    },
    {
      "step": 6880,
      "train_flow_matching_loss": 0.1484375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11592579632997513,
      "learning_rate": 5e-05,
      "loss": 0.1644,
      "step": 6890
    },
    {
      "step": 6890,
      "train_flow_matching_loss": 0.1689453125,
      "train_physics_loss": 0.0
    },
    {
      "step": 6890,
      "train_flow_matching_loss": 0.2099609375,
      "train_physics_loss": 0.0
    },
    {
      "step": 6890,
      "train_flow_matching_loss": 0.201171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 6890,
      "train_flow_matching_loss": 0.15625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11402201652526855,
      "learning_rate": 5e-05,
      "loss": 0.1642,
      "step": 6900
    },
    {
      "step": 6900,
      "train_flow_matching_loss": 0.130859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 6900,
      "train_flow_matching_loss": 0.294921875,
      "train_physics_loss": 0.0
    },
    {
      "step": 6900,
      "train_flow_matching_loss": 0.259765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 6900,
      "train_flow_matching_loss": 0.1005859375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.1222715899348259,
      "learning_rate": 5e-05,
      "loss": 0.17,
      "step": 6910
    },
    {
      "step": 6910,
      "train_flow_matching_loss": 0.2431640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 6910,
      "train_flow_matching_loss": 0.267578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 6910,
      "train_flow_matching_loss": 0.1572265625,
      "train_physics_loss": 0.0
    },
    {
      "step": 6910,
      "train_flow_matching_loss": 0.1484375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.21492038667201996,
      "learning_rate": 5e-05,
      "loss": 0.1741,
      "step": 6920
    },
    {
      "step": 6920,
      "train_flow_matching_loss": 0.142578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 6920,
      "train_flow_matching_loss": 0.1884765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 6920,
      "train_flow_matching_loss": 0.1376953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 6920,
      "train_flow_matching_loss": 0.162109375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.12087912112474442,
      "learning_rate": 5e-05,
      "loss": 0.1616,
      "step": 6930
    },
    {
      "step": 6930,
      "train_flow_matching_loss": 0.1806640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 6930,
      "train_flow_matching_loss": 0.130859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 6930,
      "train_flow_matching_loss": 0.2138671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 6930,
      "train_flow_matching_loss": 0.1376953125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.14078634977340698,
      "learning_rate": 5e-05,
      "loss": 0.1674,
      "step": 6940
    },
    {
      "step": 6940,
      "train_flow_matching_loss": 0.2177734375,
      "train_physics_loss": 0.0
    },
    {
      "step": 6940,
      "train_flow_matching_loss": 0.283203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 6940,
      "train_flow_matching_loss": 0.26171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 6940,
      "train_flow_matching_loss": 0.1494140625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10601939260959625,
      "learning_rate": 5e-05,
      "loss": 0.1724,
      "step": 6950
    },
    {
      "step": 6950,
      "train_flow_matching_loss": 0.1328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 6950,
      "train_flow_matching_loss": 0.0927734375,
      "train_physics_loss": 0.0
    },
    {
      "step": 6950,
      "train_flow_matching_loss": 0.1640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 6950,
      "train_flow_matching_loss": 0.11962890625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.15533944964408875,
      "learning_rate": 5e-05,
      "loss": 0.1675,
      "step": 6960
    },
    {
      "step": 6960,
      "train_flow_matching_loss": 0.205078125,
      "train_physics_loss": 0.0
    },
    {
      "step": 6960,
      "train_flow_matching_loss": 0.142578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 6960,
      "train_flow_matching_loss": 0.201171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 6960,
      "train_flow_matching_loss": 0.1884765625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.16506865620613098,
      "learning_rate": 5e-05,
      "loss": 0.1646,
      "step": 6970
    },
    {
      "step": 6970,
      "train_flow_matching_loss": 0.154296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 6970,
      "train_flow_matching_loss": 0.2158203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 6970,
      "train_flow_matching_loss": 0.2177734375,
      "train_physics_loss": 0.0
    },
    {
      "step": 6970,
      "train_flow_matching_loss": 0.1171875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.19654954969882965,
      "learning_rate": 5e-05,
      "loss": 0.1651,
      "step": 6980
    },
    {
      "step": 6980,
      "train_flow_matching_loss": 0.1396484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 6980,
      "train_flow_matching_loss": 0.140625,
      "train_physics_loss": 0.0
    },
    {
      "step": 6980,
      "train_flow_matching_loss": 0.2236328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 6980,
      "train_flow_matching_loss": 0.1806640625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10434846580028534,
      "learning_rate": 5e-05,
      "loss": 0.1699,
      "step": 6990
    },
    {
      "step": 6990,
      "train_flow_matching_loss": 0.2060546875,
      "train_physics_loss": 0.0
    },
    {
      "step": 6990,
      "train_flow_matching_loss": 0.1484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 6990,
      "train_flow_matching_loss": 0.1875,
      "train_physics_loss": 0.0
    },
    {
      "step": 6990,
      "train_flow_matching_loss": 0.1572265625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11646996438503265,
      "learning_rate": 5e-05,
      "loss": 0.1625,
      "step": 7000
    },
    {
      "step": 7000,
      "train_flow_matching_loss": 0.2001953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 7000,
      "train_flow_matching_loss": 0.220703125,
      "train_physics_loss": 0.0
    },
    {
      "step": 7000,
      "train_flow_matching_loss": 0.25,
      "train_physics_loss": 0.0
    },
    {
      "step": 7000,
      "train_flow_matching_loss": 0.1767578125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.4637978672981262,
      "learning_rate": 5e-05,
      "loss": 0.1697,
      "step": 7010
    },
    {
      "step": 7010,
      "train_flow_matching_loss": 0.10400390625,
      "train_physics_loss": 0.0
    },
    {
      "step": 7010,
      "train_flow_matching_loss": 0.1728515625,
      "train_physics_loss": 0.0
    },
    {
      "step": 7010,
      "train_flow_matching_loss": 0.2021484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 7010,
      "train_flow_matching_loss": 0.251953125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.1310480833053589,
      "learning_rate": 5e-05,
      "loss": 0.16,
      "step": 7020
    },
    {
      "step": 7020,
      "train_flow_matching_loss": 0.1181640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 7020,
      "train_flow_matching_loss": 0.1396484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 7020,
      "train_flow_matching_loss": 0.267578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 7020,
      "train_flow_matching_loss": 0.1787109375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.13973784446716309,
      "learning_rate": 5e-05,
      "loss": 0.1696,
      "step": 7030
    },
    {
      "step": 7030,
      "train_flow_matching_loss": 0.134765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 7030,
      "train_flow_matching_loss": 0.1201171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 7030,
      "train_flow_matching_loss": 0.287109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 7030,
      "train_flow_matching_loss": 0.2265625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10788028687238693,
      "learning_rate": 5e-05,
      "loss": 0.1672,
      "step": 7040
    },
    {
      "step": 7040,
      "train_flow_matching_loss": 0.11474609375,
      "train_physics_loss": 0.0
    },
    {
      "step": 7040,
      "train_flow_matching_loss": 0.1591796875,
      "train_physics_loss": 0.0
    },
    {
      "step": 7040,
      "train_flow_matching_loss": 0.2470703125,
      "train_physics_loss": 0.0
    },
    {
      "step": 7040,
      "train_flow_matching_loss": 0.12451171875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10466364771127701,
      "learning_rate": 5e-05,
      "loss": 0.1742,
      "step": 7050
    },
    {
      "step": 7050,
      "train_flow_matching_loss": 0.150390625,
      "train_physics_loss": 0.0
    },
    {
      "step": 7050,
      "train_flow_matching_loss": 0.1806640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 7050,
      "train_flow_matching_loss": 0.2041015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 7050,
      "train_flow_matching_loss": 0.21875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10564779490232468,
      "learning_rate": 5e-05,
      "loss": 0.1671,
      "step": 7060
    },
    {
      "step": 7060,
      "train_flow_matching_loss": 0.1904296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 7060,
      "train_flow_matching_loss": 0.2275390625,
      "train_physics_loss": 0.0
    },
    {
      "step": 7060,
      "train_flow_matching_loss": 0.1748046875,
      "train_physics_loss": 0.0
    },
    {
      "step": 7060,
      "train_flow_matching_loss": 0.10107421875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.1471744179725647,
      "learning_rate": 5e-05,
      "loss": 0.1666,
      "step": 7070
    },
    {
      "step": 7070,
      "train_flow_matching_loss": 0.203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 7070,
      "train_flow_matching_loss": 0.11767578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 7070,
      "train_flow_matching_loss": 0.25390625,
      "train_physics_loss": 0.0
    },
    {
      "step": 7070,
      "train_flow_matching_loss": 0.2080078125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.24367089569568634,
      "learning_rate": 5e-05,
      "loss": 0.1628,
      "step": 7080
    },
    {
      "step": 7080,
      "train_flow_matching_loss": 0.1533203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 7080,
      "train_flow_matching_loss": 0.1640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 7080,
      "train_flow_matching_loss": 0.212890625,
      "train_physics_loss": 0.0
    },
    {
      "step": 7080,
      "train_flow_matching_loss": 0.142578125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.16318480670452118,
      "learning_rate": 5e-05,
      "loss": 0.164,
      "step": 7090
    },
    {
      "step": 7090,
      "train_flow_matching_loss": 0.21875,
      "train_physics_loss": 0.0
    },
    {
      "step": 7090,
      "train_flow_matching_loss": 0.1689453125,
      "train_physics_loss": 0.0
    },
    {
      "step": 7090,
      "train_flow_matching_loss": 0.166015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 7090,
      "train_flow_matching_loss": 0.197265625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09553211182355881,
      "learning_rate": 5e-05,
      "loss": 0.1646,
      "step": 7100
    },
    {
      "step": 7100,
      "train_flow_matching_loss": 0.134765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 7100,
      "train_flow_matching_loss": 0.1455078125,
      "train_physics_loss": 0.0
    },
    {
      "step": 7100,
      "train_flow_matching_loss": 0.10888671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 7100,
      "train_flow_matching_loss": 0.1728515625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09852882474660873,
      "learning_rate": 5e-05,
      "loss": 0.1707,
      "step": 7110
    },
    {
      "step": 7110,
      "train_flow_matching_loss": 0.1708984375,
      "train_physics_loss": 0.0
    },
    {
      "step": 7110,
      "train_flow_matching_loss": 0.2236328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 7110,
      "train_flow_matching_loss": 0.1630859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 7110,
      "train_flow_matching_loss": 0.1376953125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.13432539999485016,
      "learning_rate": 5e-05,
      "loss": 0.1678,
      "step": 7120
    },
    {
      "step": 7120,
      "train_flow_matching_loss": 0.1708984375,
      "train_physics_loss": 0.0
    },
    {
      "step": 7120,
      "train_flow_matching_loss": 0.0966796875,
      "train_physics_loss": 0.0
    },
    {
      "step": 7120,
      "train_flow_matching_loss": 0.1083984375,
      "train_physics_loss": 0.0
    },
    {
      "step": 7120,
      "train_flow_matching_loss": 0.1552734375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10377608239650726,
      "learning_rate": 5e-05,
      "loss": 0.1643,
      "step": 7130
    },
    {
      "step": 7130,
      "train_flow_matching_loss": 0.1953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 7130,
      "train_flow_matching_loss": 0.0732421875,
      "train_physics_loss": 0.0
    },
    {
      "step": 7130,
      "train_flow_matching_loss": 0.1669921875,
      "train_physics_loss": 0.0
    },
    {
      "step": 7130,
      "train_flow_matching_loss": 0.1640625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.12627653777599335,
      "learning_rate": 5e-05,
      "loss": 0.1684,
      "step": 7140
    },
    {
      "step": 7140,
      "train_flow_matching_loss": 0.15625,
      "train_physics_loss": 0.0
    },
    {
      "step": 7140,
      "train_flow_matching_loss": 0.1826171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 7140,
      "train_flow_matching_loss": 0.11376953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 7140,
      "train_flow_matching_loss": 0.2197265625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.2643689811229706,
      "learning_rate": 5e-05,
      "loss": 0.1709,
      "step": 7150
    },
    {
      "step": 7150,
      "train_flow_matching_loss": 0.220703125,
      "train_physics_loss": 0.0
    },
    {
      "step": 7150,
      "train_flow_matching_loss": 0.1748046875,
      "train_physics_loss": 0.0
    },
    {
      "step": 7150,
      "train_flow_matching_loss": 0.142578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 7150,
      "train_flow_matching_loss": 0.1845703125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.3031417429447174,
      "learning_rate": 5e-05,
      "loss": 0.1684,
      "step": 7160
    },
    {
      "step": 7160,
      "train_flow_matching_loss": 0.181640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 7160,
      "train_flow_matching_loss": 0.1474609375,
      "train_physics_loss": 0.0
    },
    {
      "step": 7160,
      "train_flow_matching_loss": 0.1640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 7160,
      "train_flow_matching_loss": 0.1611328125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.12960447371006012,
      "learning_rate": 5e-05,
      "loss": 0.1693,
      "step": 7170
    },
    {
      "step": 7170,
      "train_flow_matching_loss": 0.09619140625,
      "train_physics_loss": 0.0
    },
    {
      "step": 7170,
      "train_flow_matching_loss": 0.1787109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 7170,
      "train_flow_matching_loss": 0.1357421875,
      "train_physics_loss": 0.0
    },
    {
      "step": 7170,
      "train_flow_matching_loss": 0.181640625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.1508447229862213,
      "learning_rate": 5e-05,
      "loss": 0.169,
      "step": 7180
    },
    {
      "step": 7180,
      "train_flow_matching_loss": 0.166015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 7180,
      "train_flow_matching_loss": 0.2265625,
      "train_physics_loss": 0.0
    },
    {
      "step": 7180,
      "train_flow_matching_loss": 0.11181640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 7180,
      "train_flow_matching_loss": 0.1416015625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10767516493797302,
      "learning_rate": 5e-05,
      "loss": 0.1727,
      "step": 7190
    },
    {
      "step": 7190,
      "train_flow_matching_loss": 0.2060546875,
      "train_physics_loss": 0.0
    },
    {
      "step": 7190,
      "train_flow_matching_loss": 0.1826171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 7190,
      "train_flow_matching_loss": 0.134765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 7190,
      "train_flow_matching_loss": 0.1669921875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11930953711271286,
      "learning_rate": 5e-05,
      "loss": 0.1689,
      "step": 7200
    },
    {
      "step": 7200,
      "train_flow_matching_loss": 0.1591796875,
      "train_physics_loss": 0.0
    },
    {
      "step": 7200,
      "train_flow_matching_loss": 0.138671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 7200,
      "train_flow_matching_loss": 0.126953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 7200,
      "train_flow_matching_loss": 0.1005859375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.1163916066288948,
      "learning_rate": 5e-05,
      "loss": 0.1669,
      "step": 7210
    },
    {
      "step": 7210,
      "train_flow_matching_loss": 0.1376953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 7210,
      "train_flow_matching_loss": 0.193359375,
      "train_physics_loss": 0.0
    },
    {
      "step": 7210,
      "train_flow_matching_loss": 0.1572265625,
      "train_physics_loss": 0.0
    },
    {
      "step": 7210,
      "train_flow_matching_loss": 0.1484375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10769419372081757,
      "learning_rate": 5e-05,
      "loss": 0.1699,
      "step": 7220
    },
    {
      "step": 7220,
      "train_flow_matching_loss": 0.232421875,
      "train_physics_loss": 0.0
    },
    {
      "step": 7220,
      "train_flow_matching_loss": 0.1513671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 7220,
      "train_flow_matching_loss": 0.169921875,
      "train_physics_loss": 0.0
    },
    {
      "step": 7220,
      "train_flow_matching_loss": 0.177734375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.07801498472690582,
      "learning_rate": 5e-05,
      "loss": 0.1631,
      "step": 7230
    },
    {
      "step": 7230,
      "train_flow_matching_loss": 0.232421875,
      "train_physics_loss": 0.0
    },
    {
      "step": 7230,
      "train_flow_matching_loss": 0.1328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 7230,
      "train_flow_matching_loss": 0.1259765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 7230,
      "train_flow_matching_loss": 0.12109375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09600752592086792,
      "learning_rate": 5e-05,
      "loss": 0.1685,
      "step": 7240
    },
    {
      "step": 7240,
      "train_flow_matching_loss": 0.1748046875,
      "train_physics_loss": 0.0
    },
    {
      "step": 7240,
      "train_flow_matching_loss": 0.150390625,
      "train_physics_loss": 0.0
    },
    {
      "step": 7240,
      "train_flow_matching_loss": 0.130859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 7240,
      "train_flow_matching_loss": 0.1416015625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11120712012052536,
      "learning_rate": 5e-05,
      "loss": 0.1616,
      "step": 7250
    },
    {
      "step": 7250,
      "train_flow_matching_loss": 0.236328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 7250,
      "train_flow_matching_loss": 0.2265625,
      "train_physics_loss": 0.0
    },
    {
      "step": 7250,
      "train_flow_matching_loss": 0.134765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 7250,
      "train_flow_matching_loss": 0.1875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.1434060037136078,
      "learning_rate": 5e-05,
      "loss": 0.1653,
      "step": 7260
    },
    {
      "step": 7260,
      "train_flow_matching_loss": 0.1591796875,
      "train_physics_loss": 0.0
    },
    {
      "step": 7260,
      "train_flow_matching_loss": 0.1435546875,
      "train_physics_loss": 0.0
    },
    {
      "step": 7260,
      "train_flow_matching_loss": 0.11865234375,
      "train_physics_loss": 0.0
    },
    {
      "step": 7260,
      "train_flow_matching_loss": 0.1513671875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.14023251831531525,
      "learning_rate": 5e-05,
      "loss": 0.1708,
      "step": 7270
    },
    {
      "step": 7270,
      "train_flow_matching_loss": 0.1953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 7270,
      "train_flow_matching_loss": 0.142578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 7270,
      "train_flow_matching_loss": 0.08935546875,
      "train_physics_loss": 0.0
    },
    {
      "step": 7270,
      "train_flow_matching_loss": 0.1220703125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.12827743589878082,
      "learning_rate": 5e-05,
      "loss": 0.1632,
      "step": 7280
    },
    {
      "step": 7280,
      "train_flow_matching_loss": 0.1845703125,
      "train_physics_loss": 0.0
    },
    {
      "step": 7280,
      "train_flow_matching_loss": 0.1640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 7280,
      "train_flow_matching_loss": 0.0654296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 7280,
      "train_flow_matching_loss": 0.158203125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09306617826223373,
      "learning_rate": 5e-05,
      "loss": 0.1659,
      "step": 7290
    },
    {
      "step": 7290,
      "train_flow_matching_loss": 0.2236328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 7290,
      "train_flow_matching_loss": 0.2080078125,
      "train_physics_loss": 0.0
    },
    {
      "step": 7290,
      "train_flow_matching_loss": 0.11474609375,
      "train_physics_loss": 0.0
    },
    {
      "step": 7290,
      "train_flow_matching_loss": 0.1748046875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.13189813494682312,
      "learning_rate": 5e-05,
      "loss": 0.172,
      "step": 7300
    },
    {
      "step": 7300,
      "train_flow_matching_loss": 0.1396484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 7300,
      "train_flow_matching_loss": 0.1337890625,
      "train_physics_loss": 0.0
    },
    {
      "step": 7300,
      "train_flow_matching_loss": 0.1640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 7300,
      "train_flow_matching_loss": 0.126953125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11972419172525406,
      "learning_rate": 5e-05,
      "loss": 0.1667,
      "step": 7310
    },
    {
      "step": 7310,
      "train_flow_matching_loss": 0.1767578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 7310,
      "train_flow_matching_loss": 0.2158203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 7310,
      "train_flow_matching_loss": 0.154296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 7310,
      "train_flow_matching_loss": 0.0830078125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08814510703086853,
      "learning_rate": 5e-05,
      "loss": 0.1628,
      "step": 7320
    },
    {
      "step": 7320,
      "train_flow_matching_loss": 0.1533203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 7320,
      "train_flow_matching_loss": 0.1748046875,
      "train_physics_loss": 0.0
    },
    {
      "step": 7320,
      "train_flow_matching_loss": 0.10791015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 7320,
      "train_flow_matching_loss": 0.11376953125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.12125426530838013,
      "learning_rate": 5e-05,
      "loss": 0.1722,
      "step": 7330
    },
    {
      "step": 7330,
      "train_flow_matching_loss": 0.154296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 7330,
      "train_flow_matching_loss": 0.2001953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 7330,
      "train_flow_matching_loss": 0.1484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 7330,
      "train_flow_matching_loss": 0.1259765625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.13110551238059998,
      "learning_rate": 5e-05,
      "loss": 0.1627,
      "step": 7340
    },
    {
      "step": 7340,
      "train_flow_matching_loss": 0.177734375,
      "train_physics_loss": 0.0
    },
    {
      "step": 7340,
      "train_flow_matching_loss": 0.1484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 7340,
      "train_flow_matching_loss": 0.1533203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 7340,
      "train_flow_matching_loss": 0.142578125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08048176020383835,
      "learning_rate": 5e-05,
      "loss": 0.1664,
      "step": 7350
    },
    {
      "step": 7350,
      "train_flow_matching_loss": 0.1484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 7350,
      "train_flow_matching_loss": 0.1875,
      "train_physics_loss": 0.0
    },
    {
      "step": 7350,
      "train_flow_matching_loss": 0.173828125,
      "train_physics_loss": 0.0
    },
    {
      "step": 7350,
      "train_flow_matching_loss": 0.12060546875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11299837380647659,
      "learning_rate": 5e-05,
      "loss": 0.1655,
      "step": 7360
    },
    {
      "step": 7360,
      "train_flow_matching_loss": 0.171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 7360,
      "train_flow_matching_loss": 0.1298828125,
      "train_physics_loss": 0.0
    },
    {
      "step": 7360,
      "train_flow_matching_loss": 0.11865234375,
      "train_physics_loss": 0.0
    },
    {
      "step": 7360,
      "train_flow_matching_loss": 0.11572265625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.1400432139635086,
      "learning_rate": 5e-05,
      "loss": 0.1653,
      "step": 7370
    },
    {
      "step": 7370,
      "train_flow_matching_loss": 0.1484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 7370,
      "train_flow_matching_loss": 0.150390625,
      "train_physics_loss": 0.0
    },
    {
      "step": 7370,
      "train_flow_matching_loss": 0.1826171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 7370,
      "train_flow_matching_loss": 0.12353515625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.1447344869375229,
      "learning_rate": 5e-05,
      "loss": 0.1694,
      "step": 7380
    },
    {
      "step": 7380,
      "train_flow_matching_loss": 0.10205078125,
      "train_physics_loss": 0.0
    },
    {
      "step": 7380,
      "train_flow_matching_loss": 0.181640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 7380,
      "train_flow_matching_loss": 0.1728515625,
      "train_physics_loss": 0.0
    },
    {
      "step": 7380,
      "train_flow_matching_loss": 0.1806640625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.1426522135734558,
      "learning_rate": 5e-05,
      "loss": 0.1691,
      "step": 7390
    },
    {
      "step": 7390,
      "train_flow_matching_loss": 0.150390625,
      "train_physics_loss": 0.0
    },
    {
      "step": 7390,
      "train_flow_matching_loss": 0.1318359375,
      "train_physics_loss": 0.0
    },
    {
      "step": 7390,
      "train_flow_matching_loss": 0.1640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 7390,
      "train_flow_matching_loss": 0.171875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08806011080741882,
      "learning_rate": 5e-05,
      "loss": 0.1652,
      "step": 7400
    },
    {
      "step": 7400,
      "train_flow_matching_loss": 0.1220703125,
      "train_physics_loss": 0.0
    },
    {
      "step": 7400,
      "train_flow_matching_loss": 0.21875,
      "train_physics_loss": 0.0
    },
    {
      "step": 7400,
      "train_flow_matching_loss": 0.1318359375,
      "train_physics_loss": 0.0
    },
    {
      "step": 7400,
      "train_flow_matching_loss": 0.1796875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08208870142698288,
      "learning_rate": 5e-05,
      "loss": 0.1612,
      "step": 7410
    },
    {
      "step": 7410,
      "train_flow_matching_loss": 0.1572265625,
      "train_physics_loss": 0.0
    },
    {
      "step": 7410,
      "train_flow_matching_loss": 0.2236328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 7410,
      "train_flow_matching_loss": 0.10888671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 7410,
      "train_flow_matching_loss": 0.21875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.15059933066368103,
      "learning_rate": 5e-05,
      "loss": 0.17,
      "step": 7420
    },
    {
      "step": 7420,
      "train_flow_matching_loss": 0.150390625,
      "train_physics_loss": 0.0
    },
    {
      "step": 7420,
      "train_flow_matching_loss": 0.1591796875,
      "train_physics_loss": 0.0
    },
    {
      "step": 7420,
      "train_flow_matching_loss": 0.1591796875,
      "train_physics_loss": 0.0
    },
    {
      "step": 7420,
      "train_flow_matching_loss": 0.11767578125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08149541914463043,
      "learning_rate": 5e-05,
      "loss": 0.1598,
      "step": 7430
    },
    {
      "step": 7430,
      "train_flow_matching_loss": 0.11181640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 7430,
      "train_flow_matching_loss": 0.1689453125,
      "train_physics_loss": 0.0
    },
    {
      "step": 7430,
      "train_flow_matching_loss": 0.1611328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 7430,
      "train_flow_matching_loss": 0.2138671875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10684700310230255,
      "learning_rate": 5e-05,
      "loss": 0.1691,
      "step": 7440
    },
    {
      "step": 7440,
      "train_flow_matching_loss": 0.11181640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 7440,
      "train_flow_matching_loss": 0.1630859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 7440,
      "train_flow_matching_loss": 0.11328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 7440,
      "train_flow_matching_loss": 0.1611328125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.1181335300207138,
      "learning_rate": 5e-05,
      "loss": 0.1628,
      "step": 7450
    },
    {
      "step": 7450,
      "train_flow_matching_loss": 0.2470703125,
      "train_physics_loss": 0.0
    },
    {
      "step": 7450,
      "train_flow_matching_loss": 0.0693359375,
      "train_physics_loss": 0.0
    },
    {
      "step": 7450,
      "train_flow_matching_loss": 0.177734375,
      "train_physics_loss": 0.0
    },
    {
      "step": 7450,
      "train_flow_matching_loss": 0.1796875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.07566478848457336,
      "learning_rate": 5e-05,
      "loss": 0.1661,
      "step": 7460
    },
    {
      "step": 7460,
      "train_flow_matching_loss": 0.166015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 7460,
      "train_flow_matching_loss": 0.189453125,
      "train_physics_loss": 0.0
    },
    {
      "step": 7460,
      "train_flow_matching_loss": 0.224609375,
      "train_physics_loss": 0.0
    },
    {
      "step": 7460,
      "train_flow_matching_loss": 0.220703125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10023114830255508,
      "learning_rate": 5e-05,
      "loss": 0.1648,
      "step": 7470
    },
    {
      "step": 7470,
      "train_flow_matching_loss": 0.2412109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 7470,
      "train_flow_matching_loss": 0.146484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 7470,
      "train_flow_matching_loss": 0.1953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 7470,
      "train_flow_matching_loss": 0.10791015625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.1001996174454689,
      "learning_rate": 5e-05,
      "loss": 0.1638,
      "step": 7480
    },
    {
      "step": 7480,
      "train_flow_matching_loss": 0.2490234375,
      "train_physics_loss": 0.0
    },
    {
      "step": 7480,
      "train_flow_matching_loss": 0.130859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 7480,
      "train_flow_matching_loss": 0.09326171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 7480,
      "train_flow_matching_loss": 0.2392578125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09767232090234756,
      "learning_rate": 5e-05,
      "loss": 0.1675,
      "step": 7490
    },
    {
      "step": 7490,
      "train_flow_matching_loss": 0.169921875,
      "train_physics_loss": 0.0
    },
    {
      "step": 7490,
      "train_flow_matching_loss": 0.10986328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 7490,
      "train_flow_matching_loss": 0.1337890625,
      "train_physics_loss": 0.0
    },
    {
      "step": 7490,
      "train_flow_matching_loss": 0.1640625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.19973334670066833,
      "learning_rate": 5e-05,
      "loss": 0.1653,
      "step": 7500
    },
    {
      "step": 7500,
      "train_flow_matching_loss": 0.1787109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 7500,
      "train_flow_matching_loss": 0.1591796875,
      "train_physics_loss": 0.0
    },
    {
      "step": 7500,
      "train_flow_matching_loss": 0.1611328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 7500,
      "train_flow_matching_loss": 0.1953125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11502013355493546,
      "learning_rate": 5e-05,
      "loss": 0.1633,
      "step": 7510
    },
    {
      "step": 7510,
      "train_flow_matching_loss": 0.1708984375,
      "train_physics_loss": 0.0
    },
    {
      "step": 7510,
      "train_flow_matching_loss": 0.1826171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 7510,
      "train_flow_matching_loss": 0.1064453125,
      "train_physics_loss": 0.0
    },
    {
      "step": 7510,
      "train_flow_matching_loss": 0.08154296875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.9112717509269714,
      "learning_rate": 5e-05,
      "loss": 0.1655,
      "step": 7520
    },
    {
      "step": 7520,
      "train_flow_matching_loss": 0.1337890625,
      "train_physics_loss": 0.0
    },
    {
      "step": 7520,
      "train_flow_matching_loss": 0.1171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 7520,
      "train_flow_matching_loss": 0.1328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 7520,
      "train_flow_matching_loss": 0.150390625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.090570367872715,
      "learning_rate": 5e-05,
      "loss": 0.1673,
      "step": 7530
    },
    {
      "step": 7530,
      "train_flow_matching_loss": 0.21875,
      "train_physics_loss": 0.0
    },
    {
      "step": 7530,
      "train_flow_matching_loss": 0.130859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 7530,
      "train_flow_matching_loss": 0.1552734375,
      "train_physics_loss": 0.0
    },
    {
      "step": 7530,
      "train_flow_matching_loss": 0.125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09858183562755585,
      "learning_rate": 5e-05,
      "loss": 0.1593,
      "step": 7540
    },
    {
      "step": 7540,
      "train_flow_matching_loss": 0.1376953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 7540,
      "train_flow_matching_loss": 0.203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 7540,
      "train_flow_matching_loss": 0.126953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 7540,
      "train_flow_matching_loss": 0.1748046875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10727089643478394,
      "learning_rate": 5e-05,
      "loss": 0.1659,
      "step": 7550
    },
    {
      "step": 7550,
      "train_flow_matching_loss": 0.1865234375,
      "train_physics_loss": 0.0
    },
    {
      "step": 7550,
      "train_flow_matching_loss": 0.142578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 7550,
      "train_flow_matching_loss": 0.1455078125,
      "train_physics_loss": 0.0
    },
    {
      "step": 7550,
      "train_flow_matching_loss": 0.09326171875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.14391350746154785,
      "learning_rate": 5e-05,
      "loss": 0.1629,
      "step": 7560
    },
    {
      "step": 7560,
      "train_flow_matching_loss": 0.1337890625,
      "train_physics_loss": 0.0
    },
    {
      "step": 7560,
      "train_flow_matching_loss": 0.1611328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 7560,
      "train_flow_matching_loss": 0.060791015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 7560,
      "train_flow_matching_loss": 0.1669921875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.158908873796463,
      "learning_rate": 5e-05,
      "loss": 0.1611,
      "step": 7570
    },
    {
      "step": 7570,
      "train_flow_matching_loss": 0.1611328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 7570,
      "train_flow_matching_loss": 0.10400390625,
      "train_physics_loss": 0.0
    },
    {
      "step": 7570,
      "train_flow_matching_loss": 0.142578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 7570,
      "train_flow_matching_loss": 0.1513671875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11116967350244522,
      "learning_rate": 5e-05,
      "loss": 0.1671,
      "step": 7580
    },
    {
      "step": 7580,
      "train_flow_matching_loss": 0.1318359375,
      "train_physics_loss": 0.0
    },
    {
      "step": 7580,
      "train_flow_matching_loss": 0.1455078125,
      "train_physics_loss": 0.0
    },
    {
      "step": 7580,
      "train_flow_matching_loss": 0.197265625,
      "train_physics_loss": 0.0
    },
    {
      "step": 7580,
      "train_flow_matching_loss": 0.11181640625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10071544349193573,
      "learning_rate": 5e-05,
      "loss": 0.1645,
      "step": 7590
    },
    {
      "step": 7590,
      "train_flow_matching_loss": 0.177734375,
      "train_physics_loss": 0.0
    },
    {
      "step": 7590,
      "train_flow_matching_loss": 0.1083984375,
      "train_physics_loss": 0.0
    },
    {
      "step": 7590,
      "train_flow_matching_loss": 0.162109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 7590,
      "train_flow_matching_loss": 0.1552734375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11494274437427521,
      "learning_rate": 5e-05,
      "loss": 0.1686,
      "step": 7600
    },
    {
      "step": 7600,
      "train_flow_matching_loss": 0.11767578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 7600,
      "train_flow_matching_loss": 0.158203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 7600,
      "train_flow_matching_loss": 0.154296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 7600,
      "train_flow_matching_loss": 0.10888671875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.1434922069311142,
      "learning_rate": 5e-05,
      "loss": 0.1631,
      "step": 7610
    },
    {
      "step": 7610,
      "train_flow_matching_loss": 0.10302734375,
      "train_physics_loss": 0.0
    },
    {
      "step": 7610,
      "train_flow_matching_loss": 0.12353515625,
      "train_physics_loss": 0.0
    },
    {
      "step": 7610,
      "train_flow_matching_loss": 0.1884765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 7610,
      "train_flow_matching_loss": 0.1806640625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.1671060174703598,
      "learning_rate": 5e-05,
      "loss": 0.1641,
      "step": 7620
    },
    {
      "step": 7620,
      "train_flow_matching_loss": 0.1474609375,
      "train_physics_loss": 0.0
    },
    {
      "step": 7620,
      "train_flow_matching_loss": 0.154296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 7620,
      "train_flow_matching_loss": 0.10888671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 7620,
      "train_flow_matching_loss": 0.142578125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11077111959457397,
      "learning_rate": 5e-05,
      "loss": 0.1695,
      "step": 7630
    },
    {
      "step": 7630,
      "train_flow_matching_loss": 0.181640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 7630,
      "train_flow_matching_loss": 0.2138671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 7630,
      "train_flow_matching_loss": 0.1982421875,
      "train_physics_loss": 0.0
    },
    {
      "step": 7630,
      "train_flow_matching_loss": 0.078125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10986579954624176,
      "learning_rate": 5e-05,
      "loss": 0.1683,
      "step": 7640
    },
    {
      "step": 7640,
      "train_flow_matching_loss": 0.1904296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 7640,
      "train_flow_matching_loss": 0.1982421875,
      "train_physics_loss": 0.0
    },
    {
      "step": 7640,
      "train_flow_matching_loss": 0.09521484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 7640,
      "train_flow_matching_loss": 0.1748046875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.12524913251399994,
      "learning_rate": 5e-05,
      "loss": 0.1682,
      "step": 7650
    },
    {
      "step": 7650,
      "train_flow_matching_loss": 0.1591796875,
      "train_physics_loss": 0.0
    },
    {
      "step": 7650,
      "train_flow_matching_loss": 0.171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 7650,
      "train_flow_matching_loss": 0.220703125,
      "train_physics_loss": 0.0
    },
    {
      "step": 7650,
      "train_flow_matching_loss": 0.1728515625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.104213647544384,
      "learning_rate": 5e-05,
      "loss": 0.1661,
      "step": 7660
    },
    {
      "step": 7660,
      "train_flow_matching_loss": 0.12060546875,
      "train_physics_loss": 0.0
    },
    {
      "step": 7660,
      "train_flow_matching_loss": 0.08056640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 7660,
      "train_flow_matching_loss": 0.2265625,
      "train_physics_loss": 0.0
    },
    {
      "step": 7660,
      "train_flow_matching_loss": 0.2294921875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.13805639743804932,
      "learning_rate": 5e-05,
      "loss": 0.1687,
      "step": 7670
    },
    {
      "step": 7670,
      "train_flow_matching_loss": 0.1640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 7670,
      "train_flow_matching_loss": 0.2412109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 7670,
      "train_flow_matching_loss": 0.138671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 7670,
      "train_flow_matching_loss": 0.1884765625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.13798180222511292,
      "learning_rate": 5e-05,
      "loss": 0.1638,
      "step": 7680
    },
    {
      "step": 7680,
      "train_flow_matching_loss": 0.142578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 7680,
      "train_flow_matching_loss": 0.1923828125,
      "train_physics_loss": 0.0
    },
    {
      "step": 7680,
      "train_flow_matching_loss": 0.138671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 7680,
      "train_flow_matching_loss": 0.1572265625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.12281744182109833,
      "learning_rate": 5e-05,
      "loss": 0.1669,
      "step": 7690
    },
    {
      "step": 7690,
      "train_flow_matching_loss": 0.1396484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 7690,
      "train_flow_matching_loss": 0.2119140625,
      "train_physics_loss": 0.0
    },
    {
      "step": 7690,
      "train_flow_matching_loss": 0.2216796875,
      "train_physics_loss": 0.0
    },
    {
      "step": 7690,
      "train_flow_matching_loss": 0.11962890625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.37624233961105347,
      "learning_rate": 5e-05,
      "loss": 0.1678,
      "step": 7700
    },
    {
      "step": 7700,
      "train_flow_matching_loss": 0.1650390625,
      "train_physics_loss": 0.0
    },
    {
      "step": 7700,
      "train_flow_matching_loss": 0.1630859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 7700,
      "train_flow_matching_loss": 0.275390625,
      "train_physics_loss": 0.0
    },
    {
      "step": 7700,
      "train_flow_matching_loss": 0.125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10671185702085495,
      "learning_rate": 5e-05,
      "loss": 0.1662,
      "step": 7710
    },
    {
      "step": 7710,
      "train_flow_matching_loss": 0.1845703125,
      "train_physics_loss": 0.0
    },
    {
      "step": 7710,
      "train_flow_matching_loss": 0.11767578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 7710,
      "train_flow_matching_loss": 0.09765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 7710,
      "train_flow_matching_loss": 0.07958984375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.257860004901886,
      "learning_rate": 5e-05,
      "loss": 0.1673,
      "step": 7720
    },
    {
      "step": 7720,
      "train_flow_matching_loss": 0.2216796875,
      "train_physics_loss": 0.0
    },
    {
      "step": 7720,
      "train_flow_matching_loss": 0.1611328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 7720,
      "train_flow_matching_loss": 0.150390625,
      "train_physics_loss": 0.0
    },
    {
      "step": 7720,
      "train_flow_matching_loss": 0.1025390625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10086796432733536,
      "learning_rate": 5e-05,
      "loss": 0.1732,
      "step": 7730
    },
    {
      "step": 7730,
      "train_flow_matching_loss": 0.15625,
      "train_physics_loss": 0.0
    },
    {
      "step": 7730,
      "train_flow_matching_loss": 0.1611328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 7730,
      "train_flow_matching_loss": 0.1533203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 7730,
      "train_flow_matching_loss": 0.1767578125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.13401272892951965,
      "learning_rate": 5e-05,
      "loss": 0.1639,
      "step": 7740
    },
    {
      "step": 7740,
      "train_flow_matching_loss": 0.208984375,
      "train_physics_loss": 0.0
    },
    {
      "step": 7740,
      "train_flow_matching_loss": 0.169921875,
      "train_physics_loss": 0.0
    },
    {
      "step": 7740,
      "train_flow_matching_loss": 0.2080078125,
      "train_physics_loss": 0.0
    },
    {
      "step": 7740,
      "train_flow_matching_loss": 0.10498046875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10817185044288635,
      "learning_rate": 5e-05,
      "loss": 0.1673,
      "step": 7750
    },
    {
      "step": 7750,
      "train_flow_matching_loss": 0.1875,
      "train_physics_loss": 0.0
    },
    {
      "step": 7750,
      "train_flow_matching_loss": 0.146484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 7750,
      "train_flow_matching_loss": 0.1484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 7750,
      "train_flow_matching_loss": 0.181640625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08969826251268387,
      "learning_rate": 5e-05,
      "loss": 0.1657,
      "step": 7760
    },
    {
      "step": 7760,
      "train_flow_matching_loss": 0.1494140625,
      "train_physics_loss": 0.0
    },
    {
      "step": 7760,
      "train_flow_matching_loss": 0.1416015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 7760,
      "train_flow_matching_loss": 0.10009765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 7760,
      "train_flow_matching_loss": 0.1279296875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.13743901252746582,
      "learning_rate": 5e-05,
      "loss": 0.1566,
      "step": 7770
    },
    {
      "step": 7770,
      "train_flow_matching_loss": 0.1396484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 7770,
      "train_flow_matching_loss": 0.1611328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 7770,
      "train_flow_matching_loss": 0.1845703125,
      "train_physics_loss": 0.0
    },
    {
      "step": 7770,
      "train_flow_matching_loss": 0.193359375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.13110017776489258,
      "learning_rate": 5e-05,
      "loss": 0.1672,
      "step": 7780
    },
    {
      "step": 7780,
      "train_flow_matching_loss": 0.11962890625,
      "train_physics_loss": 0.0
    },
    {
      "step": 7780,
      "train_flow_matching_loss": 0.11669921875,
      "train_physics_loss": 0.0
    },
    {
      "step": 7780,
      "train_flow_matching_loss": 0.1201171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 7780,
      "train_flow_matching_loss": 0.1337890625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08286643028259277,
      "learning_rate": 5e-05,
      "loss": 0.1582,
      "step": 7790
    },
    {
      "step": 7790,
      "train_flow_matching_loss": 0.1591796875,
      "train_physics_loss": 0.0
    },
    {
      "step": 7790,
      "train_flow_matching_loss": 0.2216796875,
      "train_physics_loss": 0.0
    },
    {
      "step": 7790,
      "train_flow_matching_loss": 0.1767578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 7790,
      "train_flow_matching_loss": 0.1083984375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.12040673941373825,
      "learning_rate": 5e-05,
      "loss": 0.1629,
      "step": 7800
    },
    {
      "step": 7800,
      "train_flow_matching_loss": 0.205078125,
      "train_physics_loss": 0.0
    },
    {
      "step": 7800,
      "train_flow_matching_loss": 0.1943359375,
      "train_physics_loss": 0.0
    },
    {
      "step": 7800,
      "train_flow_matching_loss": 0.1630859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 7800,
      "train_flow_matching_loss": 0.0830078125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09983223676681519,
      "learning_rate": 5e-05,
      "loss": 0.1659,
      "step": 7810
    },
    {
      "step": 7810,
      "train_flow_matching_loss": 0.251953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 7810,
      "train_flow_matching_loss": 0.2236328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 7810,
      "train_flow_matching_loss": 0.220703125,
      "train_physics_loss": 0.0
    },
    {
      "step": 7810,
      "train_flow_matching_loss": 0.1513671875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09877630323171616,
      "learning_rate": 5e-05,
      "loss": 0.1666,
      "step": 7820
    },
    {
      "step": 7820,
      "train_flow_matching_loss": 0.21875,
      "train_physics_loss": 0.0
    },
    {
      "step": 7820,
      "train_flow_matching_loss": 0.09814453125,
      "train_physics_loss": 0.0
    },
    {
      "step": 7820,
      "train_flow_matching_loss": 0.25390625,
      "train_physics_loss": 0.0
    },
    {
      "step": 7820,
      "train_flow_matching_loss": 0.12158203125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11541001498699188,
      "learning_rate": 5e-05,
      "loss": 0.1703,
      "step": 7830
    },
    {
      "step": 7830,
      "train_flow_matching_loss": 0.1669921875,
      "train_physics_loss": 0.0
    },
    {
      "step": 7830,
      "train_flow_matching_loss": 0.109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 7830,
      "train_flow_matching_loss": 0.1611328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 7830,
      "train_flow_matching_loss": 0.138671875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.13368120789527893,
      "learning_rate": 5e-05,
      "loss": 0.1684,
      "step": 7840
    },
    {
      "step": 7840,
      "train_flow_matching_loss": 0.181640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 7840,
      "train_flow_matching_loss": 0.169921875,
      "train_physics_loss": 0.0
    },
    {
      "step": 7840,
      "train_flow_matching_loss": 0.1435546875,
      "train_physics_loss": 0.0
    },
    {
      "step": 7840,
      "train_flow_matching_loss": 0.1376953125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11541292816400528,
      "learning_rate": 5e-05,
      "loss": 0.1642,
      "step": 7850
    },
    {
      "step": 7850,
      "train_flow_matching_loss": 0.1435546875,
      "train_physics_loss": 0.0
    },
    {
      "step": 7850,
      "train_flow_matching_loss": 0.1982421875,
      "train_physics_loss": 0.0
    },
    {
      "step": 7850,
      "train_flow_matching_loss": 0.1943359375,
      "train_physics_loss": 0.0
    },
    {
      "step": 7850,
      "train_flow_matching_loss": 0.11572265625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09854160249233246,
      "learning_rate": 5e-05,
      "loss": 0.16,
      "step": 7860
    },
    {
      "step": 7860,
      "train_flow_matching_loss": 0.11328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 7860,
      "train_flow_matching_loss": 0.287109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 7860,
      "train_flow_matching_loss": 0.25390625,
      "train_physics_loss": 0.0
    },
    {
      "step": 7860,
      "train_flow_matching_loss": 0.048583984375,
      "train_physics_loss": 0.763329267501831
    },
    {
      "grad_norm": 0.10024703294038773,
      "learning_rate": 5e-05,
      "loss": 0.1611,
      "step": 7870
    },
    {
      "step": 7870,
      "train_flow_matching_loss": 0.1875,
      "train_physics_loss": 0.0
    },
    {
      "step": 7870,
      "train_flow_matching_loss": 0.2080078125,
      "train_physics_loss": 0.0
    },
    {
      "step": 7870,
      "train_flow_matching_loss": 0.1484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 7870,
      "train_flow_matching_loss": 0.09716796875,
      "train_physics_loss": 0.9467118978500366
    },
    {
      "grad_norm": 0.1375938206911087,
      "learning_rate": 5e-05,
      "loss": 0.162,
      "step": 7880
    },
    {
      "step": 7880,
      "train_flow_matching_loss": 0.185546875,
      "train_physics_loss": 0.0
    },
    {
      "step": 7880,
      "train_flow_matching_loss": 0.234375,
      "train_physics_loss": 0.0
    },
    {
      "step": 7880,
      "train_flow_matching_loss": 0.1806640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 7880,
      "train_flow_matching_loss": 0.06591796875,
      "train_physics_loss": 0.8492067456245422
    },
    {
      "grad_norm": 0.07943107187747955,
      "learning_rate": 5e-05,
      "loss": 0.1625,
      "step": 7890
    },
    {
      "step": 7890,
      "train_flow_matching_loss": 0.1044921875,
      "train_physics_loss": 0.0
    },
    {
      "step": 7890,
      "train_flow_matching_loss": 0.34375,
      "train_physics_loss": 0.0
    },
    {
      "step": 7890,
      "train_flow_matching_loss": 0.177734375,
      "train_physics_loss": 0.0
    },
    {
      "step": 7890,
      "train_flow_matching_loss": 0.1357421875,
      "train_physics_loss": 0.8674699664115906
    },
    {
      "grad_norm": 0.12040968239307404,
      "learning_rate": 5e-05,
      "loss": 0.1623,
      "step": 7900
    },
    {
      "step": 7900,
      "train_flow_matching_loss": 0.1630859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 7900,
      "train_flow_matching_loss": 0.2041015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 7900,
      "train_flow_matching_loss": 0.1630859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 7900,
      "train_flow_matching_loss": 0.05810546875,
      "train_physics_loss": 0.791098952293396
    },
    {
      "grad_norm": 0.10772067308425903,
      "learning_rate": 5e-05,
      "loss": 0.1711,
      "step": 7910
    },
    {
      "step": 7910,
      "train_flow_matching_loss": 0.1455078125,
      "train_physics_loss": 0.0
    },
    {
      "step": 7910,
      "train_flow_matching_loss": 0.1591796875,
      "train_physics_loss": 0.0
    },
    {
      "step": 7910,
      "train_flow_matching_loss": 0.134765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 7910,
      "train_flow_matching_loss": 0.0732421875,
      "train_physics_loss": 0.816150963306427
    },
    {
      "grad_norm": 0.17658792436122894,
      "learning_rate": 5e-05,
      "loss": 0.1646,
      "step": 7920
    },
    {
      "step": 7920,
      "train_flow_matching_loss": 0.1630859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 7920,
      "train_flow_matching_loss": 0.1455078125,
      "train_physics_loss": 0.0
    },
    {
      "step": 7920,
      "train_flow_matching_loss": 0.1640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 7920,
      "train_flow_matching_loss": 0.0849609375,
      "train_physics_loss": 0.7499893307685852
    },
    {
      "grad_norm": 0.1219070702791214,
      "learning_rate": 5e-05,
      "loss": 0.1626,
      "step": 7930
    },
    {
      "step": 7930,
      "train_flow_matching_loss": 0.1201171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 7930,
      "train_flow_matching_loss": 0.1513671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 7930,
      "train_flow_matching_loss": 0.1259765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 7930,
      "train_flow_matching_loss": 0.0712890625,
      "train_physics_loss": 0.8175150156021118
    },
    {
      "grad_norm": 0.5647398233413696,
      "learning_rate": 5e-05,
      "loss": 0.1638,
      "step": 7940
    },
    {
      "step": 7940,
      "train_flow_matching_loss": 0.244140625,
      "train_physics_loss": 0.0
    },
    {
      "step": 7940,
      "train_flow_matching_loss": 0.146484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 7940,
      "train_flow_matching_loss": 0.1923828125,
      "train_physics_loss": 0.0
    },
    {
      "step": 7940,
      "train_flow_matching_loss": 0.150390625,
      "train_physics_loss": 0.7866424322128296
    },
    {
      "grad_norm": 0.1627054065465927,
      "learning_rate": 5e-05,
      "loss": 0.168,
      "step": 7950
    },
    {
      "step": 7950,
      "train_flow_matching_loss": 0.1357421875,
      "train_physics_loss": 0.0
    },
    {
      "step": 7950,
      "train_flow_matching_loss": 0.248046875,
      "train_physics_loss": 0.0
    },
    {
      "step": 7950,
      "train_flow_matching_loss": 0.1376953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 7950,
      "train_flow_matching_loss": 0.13671875,
      "train_physics_loss": 0.8171056509017944
    },
    {
      "grad_norm": 0.1208002120256424,
      "learning_rate": 5e-05,
      "loss": 0.1672,
      "step": 7960
    },
    {
      "step": 7960,
      "train_flow_matching_loss": 0.2177734375,
      "train_physics_loss": 0.0
    },
    {
      "step": 7960,
      "train_flow_matching_loss": 0.150390625,
      "train_physics_loss": 0.0
    },
    {
      "step": 7960,
      "train_flow_matching_loss": 0.1640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 7960,
      "train_flow_matching_loss": 0.056884765625,
      "train_physics_loss": 0.774406373500824
    },
    {
      "grad_norm": 0.08792964369058609,
      "learning_rate": 5e-05,
      "loss": 0.1657,
      "step": 7970
    },
    {
      "step": 7970,
      "train_flow_matching_loss": 0.11572265625,
      "train_physics_loss": 0.0
    },
    {
      "step": 7970,
      "train_flow_matching_loss": 0.09814453125,
      "train_physics_loss": 0.0
    },
    {
      "step": 7970,
      "train_flow_matching_loss": 0.1474609375,
      "train_physics_loss": 0.0
    },
    {
      "step": 7970,
      "train_flow_matching_loss": 0.0869140625,
      "train_physics_loss": 0.8562120199203491
    },
    {
      "grad_norm": 0.5257554650306702,
      "learning_rate": 5e-05,
      "loss": 0.1712,
      "step": 7980
    },
    {
      "step": 7980,
      "train_flow_matching_loss": 0.1591796875,
      "train_physics_loss": 0.0
    },
    {
      "step": 7980,
      "train_flow_matching_loss": 0.1396484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 7980,
      "train_flow_matching_loss": 0.064453125,
      "train_physics_loss": 0.0
    },
    {
      "step": 7980,
      "train_flow_matching_loss": 0.07177734375,
      "train_physics_loss": 0.8490859270095825
    },
    {
      "grad_norm": 0.1056375801563263,
      "learning_rate": 5e-05,
      "loss": 0.1651,
      "step": 7990
    },
    {
      "step": 7990,
      "train_flow_matching_loss": 0.1474609375,
      "train_physics_loss": 0.0
    },
    {
      "step": 7990,
      "train_flow_matching_loss": 0.181640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 7990,
      "train_flow_matching_loss": 0.10498046875,
      "train_physics_loss": 0.0
    },
    {
      "step": 7990,
      "train_flow_matching_loss": 0.08740234375,
      "train_physics_loss": 0.9123825430870056
    },
    {
      "grad_norm": 0.11378657072782516,
      "learning_rate": 5e-05,
      "loss": 0.1666,
      "step": 8000
    },
    {
      "step": 8000,
      "train_flow_matching_loss": 0.208984375,
      "train_physics_loss": 0.0
    },
    {
      "step": 8000,
      "train_flow_matching_loss": 0.208984375,
      "train_physics_loss": 0.0
    },
    {
      "step": 8000,
      "train_flow_matching_loss": 0.1142578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 8000,
      "train_flow_matching_loss": 0.078125,
      "train_physics_loss": 0.8667581081390381
    },
    {
      "grad_norm": 0.351975679397583,
      "learning_rate": 5e-05,
      "loss": 0.1666,
      "step": 8010
    },
    {
      "step": 8010,
      "train_flow_matching_loss": 0.1572265625,
      "train_physics_loss": 0.0
    },
    {
      "step": 8010,
      "train_flow_matching_loss": 0.1669921875,
      "train_physics_loss": 0.0
    },
    {
      "step": 8010,
      "train_flow_matching_loss": 0.142578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 8010,
      "train_flow_matching_loss": 0.0771484375,
      "train_physics_loss": 0.8353311419487
    },
    {
      "grad_norm": 0.19388993084430695,
      "learning_rate": 5e-05,
      "loss": 0.1615,
      "step": 8020
    },
    {
      "step": 8020,
      "train_flow_matching_loss": 0.271484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 8020,
      "train_flow_matching_loss": 0.1650390625,
      "train_physics_loss": 0.0
    },
    {
      "step": 8020,
      "train_flow_matching_loss": 0.11669921875,
      "train_physics_loss": 0.0
    },
    {
      "step": 8020,
      "train_flow_matching_loss": 0.07080078125,
      "train_physics_loss": 0.8191248178482056
    },
    {
      "grad_norm": 0.09389719367027283,
      "learning_rate": 5e-05,
      "loss": 0.1642,
      "step": 8030
    },
    {
      "step": 8030,
      "train_flow_matching_loss": 0.1181640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 8030,
      "train_flow_matching_loss": 0.1435546875,
      "train_physics_loss": 0.0
    },
    {
      "step": 8030,
      "train_flow_matching_loss": 0.1787109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 8030,
      "train_flow_matching_loss": 0.042724609375,
      "train_physics_loss": 0.8611627817153931
    },
    {
      "grad_norm": 0.1331472098827362,
      "learning_rate": 5e-05,
      "loss": 0.1642,
      "step": 8040
    },
    {
      "step": 8040,
      "train_flow_matching_loss": 0.09228515625,
      "train_physics_loss": 0.0
    },
    {
      "step": 8040,
      "train_flow_matching_loss": 0.1279296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 8040,
      "train_flow_matching_loss": 0.0849609375,
      "train_physics_loss": 0.0
    },
    {
      "step": 8040,
      "train_flow_matching_loss": 0.041259765625,
      "train_physics_loss": 0.8044790625572205
    },
    {
      "grad_norm": 0.12988893687725067,
      "learning_rate": 5e-05,
      "loss": 0.1579,
      "step": 8050
    },
    {
      "step": 8050,
      "train_flow_matching_loss": 0.1015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 8050,
      "train_flow_matching_loss": 0.1611328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 8050,
      "train_flow_matching_loss": 0.1494140625,
      "train_physics_loss": 0.0
    },
    {
      "step": 8050,
      "train_flow_matching_loss": 0.11181640625,
      "train_physics_loss": 0.794396698474884
    },
    {
      "grad_norm": 0.09903950244188309,
      "learning_rate": 5e-05,
      "loss": 0.1631,
      "step": 8060
    },
    {
      "step": 8060,
      "train_flow_matching_loss": 0.1572265625,
      "train_physics_loss": 0.0
    },
    {
      "step": 8060,
      "train_flow_matching_loss": 0.1572265625,
      "train_physics_loss": 0.0
    },
    {
      "step": 8060,
      "train_flow_matching_loss": 0.1533203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 8060,
      "train_flow_matching_loss": 0.03271484375,
      "train_physics_loss": 0.7324172854423523
    },
    {
      "grad_norm": 0.1310536116361618,
      "learning_rate": 5e-05,
      "loss": 0.1613,
      "step": 8070
    },
    {
      "step": 8070,
      "train_flow_matching_loss": 0.13671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 8070,
      "train_flow_matching_loss": 0.1484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 8070,
      "train_flow_matching_loss": 0.1630859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 8070,
      "train_flow_matching_loss": 0.09912109375,
      "train_physics_loss": 0.8281338810920715
    },
    {
      "grad_norm": 0.1385609656572342,
      "learning_rate": 5e-05,
      "loss": 0.1596,
      "step": 8080
    },
    {
      "step": 8080,
      "train_flow_matching_loss": 0.142578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 8080,
      "train_flow_matching_loss": 0.1376953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 8080,
      "train_flow_matching_loss": 0.166015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 8080,
      "train_flow_matching_loss": 0.052734375,
      "train_physics_loss": 0.8783278465270996
    },
    {
      "grad_norm": 0.0986965149641037,
      "learning_rate": 5e-05,
      "loss": 0.1573,
      "step": 8090
    },
    {
      "step": 8090,
      "train_flow_matching_loss": 0.09619140625,
      "train_physics_loss": 0.0
    },
    {
      "step": 8090,
      "train_flow_matching_loss": 0.08984375,
      "train_physics_loss": 0.0
    },
    {
      "step": 8090,
      "train_flow_matching_loss": 0.0966796875,
      "train_physics_loss": 0.0
    },
    {
      "step": 8090,
      "train_flow_matching_loss": 0.068359375,
      "train_physics_loss": 0.8153338432312012
    },
    {
      "grad_norm": 0.10078234225511551,
      "learning_rate": 5e-05,
      "loss": 0.1577,
      "step": 8100
    },
    {
      "step": 8100,
      "train_flow_matching_loss": 0.419921875,
      "train_physics_loss": 0.0
    },
    {
      "step": 8100,
      "train_flow_matching_loss": 0.146484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 8100,
      "train_flow_matching_loss": 0.166015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 8100,
      "train_flow_matching_loss": 0.06494140625,
      "train_physics_loss": 0.8166027069091797
    },
    {
      "grad_norm": 0.08824418485164642,
      "learning_rate": 5e-05,
      "loss": 0.1643,
      "step": 8110
    },
    {
      "step": 8110,
      "train_flow_matching_loss": 0.1318359375,
      "train_physics_loss": 0.0
    },
    {
      "step": 8110,
      "train_flow_matching_loss": 0.12255859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 8110,
      "train_flow_matching_loss": 0.1904296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 8110,
      "train_flow_matching_loss": 0.07275390625,
      "train_physics_loss": 0.8559013605117798
    },
    {
      "grad_norm": 0.11728664487600327,
      "learning_rate": 5e-05,
      "loss": 0.1662,
      "step": 8120
    },
    {
      "step": 8120,
      "train_flow_matching_loss": 0.10595703125,
      "train_physics_loss": 0.0
    },
    {
      "step": 8120,
      "train_flow_matching_loss": 0.2255859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 8120,
      "train_flow_matching_loss": 0.146484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 8120,
      "train_flow_matching_loss": 0.03759765625,
      "train_physics_loss": 0.7646984457969666
    },
    {
      "grad_norm": 0.09797152876853943,
      "learning_rate": 5e-05,
      "loss": 0.1594,
      "step": 8130
    },
    {
      "step": 8130,
      "train_flow_matching_loss": 0.154296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 8130,
      "train_flow_matching_loss": 0.1630859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 8130,
      "train_flow_matching_loss": 0.185546875,
      "train_physics_loss": 0.0
    },
    {
      "step": 8130,
      "train_flow_matching_loss": 0.046142578125,
      "train_physics_loss": 0.783484160900116
    },
    {
      "grad_norm": 0.1227559894323349,
      "learning_rate": 5e-05,
      "loss": 0.1635,
      "step": 8140
    },
    {
      "step": 8140,
      "train_flow_matching_loss": 0.10595703125,
      "train_physics_loss": 0.0
    },
    {
      "step": 8140,
      "train_flow_matching_loss": 0.10595703125,
      "train_physics_loss": 0.0
    },
    {
      "step": 8140,
      "train_flow_matching_loss": 0.1279296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 8140,
      "train_flow_matching_loss": 0.0693359375,
      "train_physics_loss": 0.8348479866981506
    },
    {
      "grad_norm": 0.11222875118255615,
      "learning_rate": 5e-05,
      "loss": 0.1638,
      "step": 8150
    },
    {
      "step": 8150,
      "train_flow_matching_loss": 0.1845703125,
      "train_physics_loss": 0.0
    },
    {
      "step": 8150,
      "train_flow_matching_loss": 0.1533203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 8150,
      "train_flow_matching_loss": 0.166015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 8150,
      "train_flow_matching_loss": 0.08154296875,
      "train_physics_loss": 0.8589270114898682
    },
    {
      "grad_norm": 0.08907824754714966,
      "learning_rate": 5e-05,
      "loss": 0.1638,
      "step": 8160
    },
    {
      "step": 8160,
      "train_flow_matching_loss": 0.1875,
      "train_physics_loss": 0.0
    },
    {
      "step": 8160,
      "train_flow_matching_loss": 0.1474609375,
      "train_physics_loss": 0.0
    },
    {
      "step": 8160,
      "train_flow_matching_loss": 0.130859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 8160,
      "train_flow_matching_loss": 0.0966796875,
      "train_physics_loss": 0.7879788279533386
    },
    {
      "grad_norm": 0.10858204960823059,
      "learning_rate": 5e-05,
      "loss": 0.1628,
      "step": 8170
    },
    {
      "step": 8170,
      "train_flow_matching_loss": 0.1640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 8170,
      "train_flow_matching_loss": 0.2099609375,
      "train_physics_loss": 0.0
    },
    {
      "step": 8170,
      "train_flow_matching_loss": 0.1494140625,
      "train_physics_loss": 0.0
    },
    {
      "step": 8170,
      "train_flow_matching_loss": 0.04345703125,
      "train_physics_loss": 0.7897172570228577
    },
    {
      "grad_norm": 0.10234246402978897,
      "learning_rate": 5e-05,
      "loss": 0.1623,
      "step": 8180
    },
    {
      "step": 8180,
      "train_flow_matching_loss": 0.1337890625,
      "train_physics_loss": 0.0
    },
    {
      "step": 8180,
      "train_flow_matching_loss": 0.1650390625,
      "train_physics_loss": 0.0
    },
    {
      "step": 8180,
      "train_flow_matching_loss": 0.1513671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 8180,
      "train_flow_matching_loss": 0.042724609375,
      "train_physics_loss": 0.8154028058052063
    },
    {
      "grad_norm": 0.11090386658906937,
      "learning_rate": 5e-05,
      "loss": 0.1592,
      "step": 8190
    },
    {
      "step": 8190,
      "train_flow_matching_loss": 0.1630859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 8190,
      "train_flow_matching_loss": 0.1484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 8190,
      "train_flow_matching_loss": 0.189453125,
      "train_physics_loss": 0.0
    },
    {
      "step": 8190,
      "train_flow_matching_loss": 0.06298828125,
      "train_physics_loss": 0.7986930012702942
    },
    {
      "grad_norm": 0.20571640133857727,
      "learning_rate": 5e-05,
      "loss": 0.1648,
      "step": 8200
    },
    {
      "step": 8200,
      "train_flow_matching_loss": 0.171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 8200,
      "train_flow_matching_loss": 0.158203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 8200,
      "train_flow_matching_loss": 0.1435546875,
      "train_physics_loss": 0.0
    },
    {
      "step": 8200,
      "train_flow_matching_loss": 0.068359375,
      "train_physics_loss": 0.7467670440673828
    },
    {
      "grad_norm": 0.1650497168302536,
      "learning_rate": 5e-05,
      "loss": 0.1593,
      "step": 8210
    },
    {
      "step": 8210,
      "train_flow_matching_loss": 0.244140625,
      "train_physics_loss": 0.0
    },
    {
      "step": 8210,
      "train_flow_matching_loss": 0.1650390625,
      "train_physics_loss": 0.0
    },
    {
      "step": 8210,
      "train_flow_matching_loss": 0.1181640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 8210,
      "train_flow_matching_loss": 0.06201171875,
      "train_physics_loss": 0.8839035630226135
    },
    {
      "grad_norm": 0.07836677879095078,
      "learning_rate": 5e-05,
      "loss": 0.1666,
      "step": 8220
    },
    {
      "step": 8220,
      "train_flow_matching_loss": 0.1904296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 8220,
      "train_flow_matching_loss": 0.12353515625,
      "train_physics_loss": 0.0
    },
    {
      "step": 8220,
      "train_flow_matching_loss": 0.0859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 8220,
      "train_flow_matching_loss": 0.057861328125,
      "train_physics_loss": 0.811739444732666
    },
    {
      "grad_norm": 0.11268146336078644,
      "learning_rate": 5e-05,
      "loss": 0.1628,
      "step": 8230
    },
    {
      "step": 8230,
      "train_flow_matching_loss": 0.1708984375,
      "train_physics_loss": 0.0
    },
    {
      "step": 8230,
      "train_flow_matching_loss": 0.162109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 8230,
      "train_flow_matching_loss": 0.109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 8230,
      "train_flow_matching_loss": 0.07470703125,
      "train_physics_loss": 0.7848135232925415
    },
    {
      "grad_norm": 0.07531463354825974,
      "learning_rate": 5e-05,
      "loss": 0.1605,
      "step": 8240
    },
    {
      "step": 8240,
      "train_flow_matching_loss": 0.09521484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 8240,
      "train_flow_matching_loss": 0.1455078125,
      "train_physics_loss": 0.0
    },
    {
      "step": 8240,
      "train_flow_matching_loss": 0.1630859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 8240,
      "train_flow_matching_loss": 0.05126953125,
      "train_physics_loss": 0.8367385268211365
    },
    {
      "grad_norm": 0.15315793454647064,
      "learning_rate": 5e-05,
      "loss": 0.1612,
      "step": 8250
    },
    {
      "step": 8250,
      "train_flow_matching_loss": 0.1572265625,
      "train_physics_loss": 0.0
    },
    {
      "step": 8250,
      "train_flow_matching_loss": 0.2099609375,
      "train_physics_loss": 0.0
    },
    {
      "step": 8250,
      "train_flow_matching_loss": 0.1416015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 8250,
      "train_flow_matching_loss": 0.1337890625,
      "train_physics_loss": 0.8459014296531677
    },
    {
      "grad_norm": 0.07044717669487,
      "learning_rate": 5e-05,
      "loss": 0.1624,
      "step": 8260
    },
    {
      "step": 8260,
      "train_flow_matching_loss": 0.2119140625,
      "train_physics_loss": 0.0
    },
    {
      "step": 8260,
      "train_flow_matching_loss": 0.2119140625,
      "train_physics_loss": 0.0
    },
    {
      "step": 8260,
      "train_flow_matching_loss": 0.166015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 8260,
      "train_flow_matching_loss": 0.0947265625,
      "train_physics_loss": 0.8596821427345276
    },
    {
      "grad_norm": 0.10279734432697296,
      "learning_rate": 5e-05,
      "loss": 0.1685,
      "step": 8270
    },
    {
      "step": 8270,
      "train_flow_matching_loss": 0.203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 8270,
      "train_flow_matching_loss": 0.1728515625,
      "train_physics_loss": 0.0
    },
    {
      "step": 8270,
      "train_flow_matching_loss": 0.1572265625,
      "train_physics_loss": 0.0
    },
    {
      "step": 8270,
      "train_flow_matching_loss": 0.0966796875,
      "train_physics_loss": 0.8863744735717773
    },
    {
      "grad_norm": 0.10211660712957382,
      "learning_rate": 5e-05,
      "loss": 0.1657,
      "step": 8280
    },
    {
      "step": 8280,
      "train_flow_matching_loss": 0.1748046875,
      "train_physics_loss": 0.0
    },
    {
      "step": 8280,
      "train_flow_matching_loss": 0.1015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 8280,
      "train_flow_matching_loss": 0.1748046875,
      "train_physics_loss": 0.0
    },
    {
      "step": 8280,
      "train_flow_matching_loss": 0.049072265625,
      "train_physics_loss": 0.7703425288200378
    },
    {
      "grad_norm": 0.10906993597745895,
      "learning_rate": 5e-05,
      "loss": 0.1634,
      "step": 8290
    },
    {
      "step": 8290,
      "train_flow_matching_loss": 0.10107421875,
      "train_physics_loss": 0.0
    },
    {
      "step": 8290,
      "train_flow_matching_loss": 0.1513671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 8290,
      "train_flow_matching_loss": 0.1494140625,
      "train_physics_loss": 0.0
    },
    {
      "step": 8290,
      "train_flow_matching_loss": 0.026611328125,
      "train_physics_loss": 0.8723745346069336
    },
    {
      "grad_norm": 0.09566630423069,
      "learning_rate": 5e-05,
      "loss": 0.1587,
      "step": 8300
    },
    {
      "step": 8300,
      "train_flow_matching_loss": 0.1591796875,
      "train_physics_loss": 0.0
    },
    {
      "step": 8300,
      "train_flow_matching_loss": 0.1611328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 8300,
      "train_flow_matching_loss": 0.142578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 8300,
      "train_flow_matching_loss": 0.0869140625,
      "train_physics_loss": 0.8100084066390991
    },
    {
      "grad_norm": 0.31684646010398865,
      "learning_rate": 5e-05,
      "loss": 0.1685,
      "step": 8310
    },
    {
      "step": 8310,
      "train_flow_matching_loss": 0.2099609375,
      "train_physics_loss": 0.0
    },
    {
      "step": 8310,
      "train_flow_matching_loss": 0.201171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 8310,
      "train_flow_matching_loss": 0.142578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 8310,
      "train_flow_matching_loss": 0.036376953125,
      "train_physics_loss": 0.7959607839584351
    },
    {
      "grad_norm": 0.11078432947397232,
      "learning_rate": 5e-05,
      "loss": 0.1659,
      "step": 8320
    },
    {
      "step": 8320,
      "train_flow_matching_loss": 0.1103515625,
      "train_physics_loss": 0.0
    },
    {
      "step": 8320,
      "train_flow_matching_loss": 0.1630859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 8320,
      "train_flow_matching_loss": 0.21875,
      "train_physics_loss": 0.0
    },
    {
      "step": 8320,
      "train_flow_matching_loss": 0.045654296875,
      "train_physics_loss": 0.7873620986938477
    },
    {
      "grad_norm": 0.07066893577575684,
      "learning_rate": 5e-05,
      "loss": 0.1639,
      "step": 8330
    },
    {
      "step": 8330,
      "train_flow_matching_loss": 0.1201171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 8330,
      "train_flow_matching_loss": 0.1552734375,
      "train_physics_loss": 0.0
    },
    {
      "step": 8330,
      "train_flow_matching_loss": 0.1357421875,
      "train_physics_loss": 0.0
    },
    {
      "step": 8330,
      "train_flow_matching_loss": 0.09228515625,
      "train_physics_loss": 0.8785071969032288
    },
    {
      "grad_norm": 0.10669203102588654,
      "learning_rate": 5e-05,
      "loss": 0.1639,
      "step": 8340
    },
    {
      "step": 8340,
      "train_flow_matching_loss": 0.197265625,
      "train_physics_loss": 0.0
    },
    {
      "step": 8340,
      "train_flow_matching_loss": 0.169921875,
      "train_physics_loss": 0.0
    },
    {
      "step": 8340,
      "train_flow_matching_loss": 0.169921875,
      "train_physics_loss": 0.0
    },
    {
      "step": 8340,
      "train_flow_matching_loss": 0.08251953125,
      "train_physics_loss": 0.775063157081604
    },
    {
      "grad_norm": 0.12615607678890228,
      "learning_rate": 5e-05,
      "loss": 0.163,
      "step": 8350
    },
    {
      "step": 8350,
      "train_flow_matching_loss": 0.1484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 8350,
      "train_flow_matching_loss": 0.1484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 8350,
      "train_flow_matching_loss": 0.166015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 8350,
      "train_flow_matching_loss": 0.05712890625,
      "train_physics_loss": 0.8547657132148743
    },
    {
      "grad_norm": 0.09619211405515671,
      "learning_rate": 5e-05,
      "loss": 0.163,
      "step": 8360
    },
    {
      "step": 8360,
      "train_flow_matching_loss": 0.2021484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 8360,
      "train_flow_matching_loss": 0.130859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 8360,
      "train_flow_matching_loss": 0.1875,
      "train_physics_loss": 0.0
    },
    {
      "step": 8360,
      "train_flow_matching_loss": 0.037841796875,
      "train_physics_loss": 0.757602870464325
    },
    {
      "grad_norm": 0.10240598767995834,
      "learning_rate": 5e-05,
      "loss": 0.1566,
      "step": 8370
    },
    {
      "step": 8370,
      "train_flow_matching_loss": 0.1728515625,
      "train_physics_loss": 0.0
    },
    {
      "step": 8370,
      "train_flow_matching_loss": 0.1630859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 8370,
      "train_flow_matching_loss": 0.12060546875,
      "train_physics_loss": 0.0
    },
    {
      "step": 8370,
      "train_flow_matching_loss": 0.1669921875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.13537293672561646,
      "learning_rate": 5e-05,
      "loss": 0.1652,
      "step": 8380
    },
    {
      "step": 8380,
      "train_flow_matching_loss": 0.1650390625,
      "train_physics_loss": 0.0
    },
    {
      "step": 8380,
      "train_flow_matching_loss": 0.158203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 8380,
      "train_flow_matching_loss": 0.1875,
      "train_physics_loss": 0.0
    },
    {
      "step": 8380,
      "train_flow_matching_loss": 0.1142578125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09444604068994522,
      "learning_rate": 5e-05,
      "loss": 0.1657,
      "step": 8390
    },
    {
      "step": 8390,
      "train_flow_matching_loss": 0.177734375,
      "train_physics_loss": 0.0
    },
    {
      "step": 8390,
      "train_flow_matching_loss": 0.2041015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 8390,
      "train_flow_matching_loss": 0.205078125,
      "train_physics_loss": 0.0
    },
    {
      "step": 8390,
      "train_flow_matching_loss": 0.2099609375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.279755562543869,
      "learning_rate": 5e-05,
      "loss": 0.156,
      "step": 8400
    },
    {
      "step": 8400,
      "train_flow_matching_loss": 0.177734375,
      "train_physics_loss": 0.0
    },
    {
      "step": 8400,
      "train_flow_matching_loss": 0.201171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 8400,
      "train_flow_matching_loss": 0.1474609375,
      "train_physics_loss": 0.0
    },
    {
      "step": 8400,
      "train_flow_matching_loss": 0.1513671875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.2261536717414856,
      "learning_rate": 5e-05,
      "loss": 0.1624,
      "step": 8410
    },
    {
      "step": 8410,
      "train_flow_matching_loss": 0.1650390625,
      "train_physics_loss": 0.0
    },
    {
      "step": 8410,
      "train_flow_matching_loss": 0.146484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 8410,
      "train_flow_matching_loss": 0.12109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 8410,
      "train_flow_matching_loss": 0.2314453125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.07716157287359238,
      "learning_rate": 5e-05,
      "loss": 0.1584,
      "step": 8420
    },
    {
      "step": 8420,
      "train_flow_matching_loss": 0.173828125,
      "train_physics_loss": 0.0
    },
    {
      "step": 8420,
      "train_flow_matching_loss": 0.1865234375,
      "train_physics_loss": 0.0
    },
    {
      "step": 8420,
      "train_flow_matching_loss": 0.1982421875,
      "train_physics_loss": 0.0
    },
    {
      "step": 8420,
      "train_flow_matching_loss": 0.1962890625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10698012262582779,
      "learning_rate": 5e-05,
      "loss": 0.1642,
      "step": 8430
    },
    {
      "step": 8430,
      "train_flow_matching_loss": 0.12890625,
      "train_physics_loss": 0.0
    },
    {
      "step": 8430,
      "train_flow_matching_loss": 0.1552734375,
      "train_physics_loss": 0.0
    },
    {
      "step": 8430,
      "train_flow_matching_loss": 0.1494140625,
      "train_physics_loss": 0.0
    },
    {
      "step": 8430,
      "train_flow_matching_loss": 0.1416015625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11612800508737564,
      "learning_rate": 5e-05,
      "loss": 0.159,
      "step": 8440
    },
    {
      "step": 8440,
      "train_flow_matching_loss": 0.11474609375,
      "train_physics_loss": 0.0
    },
    {
      "step": 8440,
      "train_flow_matching_loss": 0.224609375,
      "train_physics_loss": 0.0
    },
    {
      "step": 8440,
      "train_flow_matching_loss": 0.2001953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 8440,
      "train_flow_matching_loss": 0.1796875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09550846368074417,
      "learning_rate": 5e-05,
      "loss": 0.1626,
      "step": 8450
    },
    {
      "step": 8450,
      "train_flow_matching_loss": 0.13671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 8450,
      "train_flow_matching_loss": 0.1611328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 8450,
      "train_flow_matching_loss": 0.10693359375,
      "train_physics_loss": 0.0
    },
    {
      "step": 8450,
      "train_flow_matching_loss": 0.1201171875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.12036674469709396,
      "learning_rate": 5e-05,
      "loss": 0.1584,
      "step": 8460
    },
    {
      "step": 8460,
      "train_flow_matching_loss": 0.048583984375,
      "train_physics_loss": 0.8072558641433716
    },
    {
      "step": 8460,
      "train_flow_matching_loss": 0.1904296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 8460,
      "train_flow_matching_loss": 0.1865234375,
      "train_physics_loss": 0.0
    },
    {
      "step": 8460,
      "train_flow_matching_loss": 0.140625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.095166876912117,
      "learning_rate": 5e-05,
      "loss": 0.1586,
      "step": 8470
    },
    {
      "step": 8470,
      "train_flow_matching_loss": 0.0830078125,
      "train_physics_loss": 0.7855221629142761
    },
    {
      "step": 8470,
      "train_flow_matching_loss": 0.1337890625,
      "train_physics_loss": 0.0
    },
    {
      "step": 8470,
      "train_flow_matching_loss": 0.08935546875,
      "train_physics_loss": 0.0
    },
    {
      "step": 8470,
      "train_flow_matching_loss": 0.126953125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.17604109644889832,
      "learning_rate": 5e-05,
      "loss": 0.169,
      "step": 8480
    },
    {
      "step": 8480,
      "train_flow_matching_loss": 0.10693359375,
      "train_physics_loss": 0.8652119040489197
    },
    {
      "step": 8480,
      "train_flow_matching_loss": 0.0654296875,
      "train_physics_loss": 0.8882322311401367
    },
    {
      "step": 8480,
      "train_flow_matching_loss": 0.1240234375,
      "train_physics_loss": 0.0
    },
    {
      "step": 8480,
      "train_flow_matching_loss": 0.1728515625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11495635658502579,
      "learning_rate": 5e-05,
      "loss": 0.1611,
      "step": 8490
    },
    {
      "step": 8490,
      "train_flow_matching_loss": 0.0791015625,
      "train_physics_loss": 0.8199526071548462
    },
    {
      "step": 8490,
      "train_flow_matching_loss": 0.0869140625,
      "train_physics_loss": 0.8954217433929443
    },
    {
      "step": 8490,
      "train_flow_matching_loss": 0.12060546875,
      "train_physics_loss": 0.0
    },
    {
      "step": 8490,
      "train_flow_matching_loss": 0.208984375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.17720894515514374,
      "learning_rate": 5e-05,
      "loss": 0.1592,
      "step": 8500
    },
    {
      "step": 8500,
      "train_flow_matching_loss": 0.1181640625,
      "train_physics_loss": 0.8303738832473755
    },
    {
      "step": 8500,
      "train_flow_matching_loss": 0.03173828125,
      "train_physics_loss": 0.8029997944831848
    },
    {
      "step": 8500,
      "train_flow_matching_loss": 0.1474609375,
      "train_physics_loss": 0.0
    },
    {
      "step": 8500,
      "train_flow_matching_loss": 0.205078125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.1322515457868576,
      "learning_rate": 5e-05,
      "loss": 0.164,
      "step": 8510
    },
    {
      "step": 8510,
      "train_flow_matching_loss": 0.08154296875,
      "train_physics_loss": 0.7648589015007019
    },
    {
      "step": 8510,
      "train_flow_matching_loss": 0.078125,
      "train_physics_loss": 0.7924188375473022
    },
    {
      "step": 8510,
      "train_flow_matching_loss": 0.2177734375,
      "train_physics_loss": 0.0
    },
    {
      "step": 8510,
      "train_flow_matching_loss": 0.2041015625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.1984468698501587,
      "learning_rate": 5e-05,
      "loss": 0.1622,
      "step": 8520
    },
    {
      "step": 8520,
      "train_flow_matching_loss": 0.109375,
      "train_physics_loss": 0.8363748788833618
    },
    {
      "step": 8520,
      "train_flow_matching_loss": 0.05810546875,
      "train_physics_loss": 0.818169355392456
    },
    {
      "step": 8520,
      "train_flow_matching_loss": 0.205078125,
      "train_physics_loss": 0.0
    },
    {
      "step": 8520,
      "train_flow_matching_loss": 0.158203125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.1026853546500206,
      "learning_rate": 5e-05,
      "loss": 0.1593,
      "step": 8530
    },
    {
      "step": 8530,
      "train_flow_matching_loss": 0.09619140625,
      "train_physics_loss": 0.8115110397338867
    },
    {
      "step": 8530,
      "train_flow_matching_loss": 0.050537109375,
      "train_physics_loss": 0.8062587380409241
    },
    {
      "step": 8530,
      "train_flow_matching_loss": 0.1865234375,
      "train_physics_loss": 0.0
    },
    {
      "step": 8530,
      "train_flow_matching_loss": 0.2021484375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08298677206039429,
      "learning_rate": 5e-05,
      "loss": 0.1586,
      "step": 8540
    },
    {
      "step": 8540,
      "train_flow_matching_loss": 0.2470703125,
      "train_physics_loss": 0.7254905104637146
    },
    {
      "step": 8540,
      "train_flow_matching_loss": 0.0830078125,
      "train_physics_loss": 0.9614734053611755
    },
    {
      "step": 8540,
      "train_flow_matching_loss": 0.26171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 8540,
      "train_flow_matching_loss": 0.1689453125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.12949302792549133,
      "learning_rate": 5e-05,
      "loss": 0.1646,
      "step": 8550
    },
    {
      "step": 8550,
      "train_flow_matching_loss": 0.06787109375,
      "train_physics_loss": 0.7889395952224731
    },
    {
      "step": 8550,
      "train_flow_matching_loss": 0.06982421875,
      "train_physics_loss": 0.7883290648460388
    },
    {
      "step": 8550,
      "train_flow_matching_loss": 0.166015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 8550,
      "train_flow_matching_loss": 0.1796875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.13703908026218414,
      "learning_rate": 5e-05,
      "loss": 0.1598,
      "step": 8560
    },
    {
      "step": 8560,
      "train_flow_matching_loss": 0.10498046875,
      "train_physics_loss": 0.8016258478164673
    },
    {
      "step": 8560,
      "train_flow_matching_loss": 0.050537109375,
      "train_physics_loss": 0.7467831969261169
    },
    {
      "step": 8560,
      "train_flow_matching_loss": 0.1435546875,
      "train_physics_loss": 0.0
    },
    {
      "step": 8560,
      "train_flow_matching_loss": 0.10302734375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09158302843570709,
      "learning_rate": 5e-05,
      "loss": 0.1648,
      "step": 8570
    },
    {
      "step": 8570,
      "train_flow_matching_loss": 0.051513671875,
      "train_physics_loss": 0.7760276794433594
    },
    {
      "step": 8570,
      "train_flow_matching_loss": 0.08447265625,
      "train_physics_loss": 0.7353581786155701
    },
    {
      "step": 8570,
      "train_flow_matching_loss": 0.1689453125,
      "train_physics_loss": 0.0
    },
    {
      "step": 8570,
      "train_flow_matching_loss": 0.09619140625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08317150175571442,
      "learning_rate": 5e-05,
      "loss": 0.1569,
      "step": 8580
    },
    {
      "step": 8580,
      "train_flow_matching_loss": 0.059814453125,
      "train_physics_loss": 0.8244892954826355
    },
    {
      "step": 8580,
      "train_flow_matching_loss": 0.06396484375,
      "train_physics_loss": 0.7955383062362671
    },
    {
      "step": 8580,
      "train_flow_matching_loss": 0.1513671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 8580,
      "train_flow_matching_loss": 0.1259765625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.1013246476650238,
      "learning_rate": 5e-05,
      "loss": 0.1574,
      "step": 8590
    },
    {
      "step": 8590,
      "train_flow_matching_loss": 0.07275390625,
      "train_physics_loss": 0.896310567855835
    },
    {
      "step": 8590,
      "train_flow_matching_loss": 0.049072265625,
      "train_physics_loss": 0.7828596234321594
    },
    {
      "step": 8590,
      "train_flow_matching_loss": 0.1640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 8590,
      "train_flow_matching_loss": 0.2080078125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.144245907664299,
      "learning_rate": 5e-05,
      "loss": 0.1639,
      "step": 8600
    },
    {
      "step": 8600,
      "train_flow_matching_loss": 0.048583984375,
      "train_physics_loss": 0.9418439865112305
    },
    {
      "step": 8600,
      "train_flow_matching_loss": 0.043701171875,
      "train_physics_loss": 0.7841942310333252
    },
    {
      "step": 8600,
      "train_flow_matching_loss": 0.10205078125,
      "train_physics_loss": 0.0
    },
    {
      "step": 8600,
      "train_flow_matching_loss": 0.154296875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.163987398147583,
      "learning_rate": 5e-05,
      "loss": 0.1599,
      "step": 8610
    },
    {
      "step": 8610,
      "train_flow_matching_loss": 0.0791015625,
      "train_physics_loss": 0.8094195127487183
    },
    {
      "step": 8610,
      "train_flow_matching_loss": 0.04150390625,
      "train_physics_loss": 0.7630707621574402
    },
    {
      "step": 8610,
      "train_flow_matching_loss": 0.21875,
      "train_physics_loss": 0.0
    },
    {
      "step": 8610,
      "train_flow_matching_loss": 0.203125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.14016994833946228,
      "learning_rate": 5e-05,
      "loss": 0.1618,
      "step": 8620
    },
    {
      "step": 8620,
      "train_flow_matching_loss": 0.08740234375,
      "train_physics_loss": 0.8388767838478088
    },
    {
      "step": 8620,
      "train_flow_matching_loss": 0.05419921875,
      "train_physics_loss": 0.7593998312950134
    },
    {
      "step": 8620,
      "train_flow_matching_loss": 0.11865234375,
      "train_physics_loss": 0.0
    },
    {
      "step": 8620,
      "train_flow_matching_loss": 0.1259765625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.0961272343993187,
      "learning_rate": 5e-05,
      "loss": 0.1582,
      "step": 8630
    },
    {
      "step": 8630,
      "train_flow_matching_loss": 0.0859375,
      "train_physics_loss": 0.7903817892074585
    },
    {
      "step": 8630,
      "train_flow_matching_loss": 0.11962890625,
      "train_physics_loss": 0.7284271717071533
    },
    {
      "step": 8630,
      "train_flow_matching_loss": 0.1640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 8630,
      "train_flow_matching_loss": 0.1865234375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11635106056928635,
      "learning_rate": 5e-05,
      "loss": 0.1603,
      "step": 8640
    },
    {
      "step": 8640,
      "train_flow_matching_loss": 0.04248046875,
      "train_physics_loss": 0.8113228678703308
    },
    {
      "step": 8640,
      "train_flow_matching_loss": 0.09423828125,
      "train_physics_loss": 0.7607038617134094
    },
    {
      "step": 8640,
      "train_flow_matching_loss": 0.154296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 8640,
      "train_flow_matching_loss": 0.130859375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.13682624697685242,
      "learning_rate": 5e-05,
      "loss": 0.1608,
      "step": 8650
    },
    {
      "step": 8650,
      "train_flow_matching_loss": 0.07373046875,
      "train_physics_loss": 0.8936039805412292
    },
    {
      "step": 8650,
      "train_flow_matching_loss": 0.060791015625,
      "train_physics_loss": 0.7430676817893982
    },
    {
      "step": 8650,
      "train_flow_matching_loss": 0.09912109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 8650,
      "train_flow_matching_loss": 0.15625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.12306512147188187,
      "learning_rate": 5e-05,
      "loss": 0.1606,
      "step": 8660
    },
    {
      "step": 8660,
      "train_flow_matching_loss": 0.08642578125,
      "train_physics_loss": 0.7577756643295288
    },
    {
      "step": 8660,
      "train_flow_matching_loss": 0.060546875,
      "train_physics_loss": 0.7845762968063354
    },
    {
      "step": 8660,
      "train_flow_matching_loss": 0.10107421875,
      "train_physics_loss": 0.0
    },
    {
      "step": 8660,
      "train_flow_matching_loss": 0.134765625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09826861321926117,
      "learning_rate": 5e-05,
      "loss": 0.1598,
      "step": 8670
    },
    {
      "step": 8670,
      "train_flow_matching_loss": 0.1064453125,
      "train_physics_loss": 0.7707484364509583
    },
    {
      "step": 8670,
      "train_flow_matching_loss": 0.08642578125,
      "train_physics_loss": 0.8173661231994629
    },
    {
      "step": 8670,
      "train_flow_matching_loss": 0.169921875,
      "train_physics_loss": 0.0
    },
    {
      "step": 8670,
      "train_flow_matching_loss": 0.1630859375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 1.0793777704238892,
      "learning_rate": 5e-05,
      "loss": 0.165,
      "step": 8680
    },
    {
      "step": 8680,
      "train_flow_matching_loss": 0.0966796875,
      "train_physics_loss": 0.8014321327209473
    },
    {
      "step": 8680,
      "train_flow_matching_loss": 0.08349609375,
      "train_physics_loss": 0.797080397605896
    },
    {
      "step": 8680,
      "train_flow_matching_loss": 0.1982421875,
      "train_physics_loss": 0.0
    },
    {
      "step": 8680,
      "train_flow_matching_loss": 0.1005859375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.07500893622636795,
      "learning_rate": 5e-05,
      "loss": 0.1612,
      "step": 8690
    },
    {
      "step": 8690,
      "train_flow_matching_loss": 0.024658203125,
      "train_physics_loss": 0.7546182870864868
    },
    {
      "step": 8690,
      "train_flow_matching_loss": 0.08251953125,
      "train_physics_loss": 0.7619998455047607
    },
    {
      "step": 8690,
      "train_flow_matching_loss": 0.126953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 8690,
      "train_flow_matching_loss": 0.1337890625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.1455307900905609,
      "learning_rate": 5e-05,
      "loss": 0.1568,
      "step": 8700
    },
    {
      "step": 8700,
      "train_flow_matching_loss": 0.1220703125,
      "train_physics_loss": 0.813719630241394
    },
    {
      "step": 8700,
      "train_flow_matching_loss": 0.07763671875,
      "train_physics_loss": 0.8352037668228149
    },
    {
      "step": 8700,
      "train_flow_matching_loss": 0.1962890625,
      "train_physics_loss": 0.0
    },
    {
      "step": 8700,
      "train_flow_matching_loss": 0.1123046875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10029931366443634,
      "learning_rate": 5e-05,
      "loss": 0.1561,
      "step": 8710
    },
    {
      "step": 8710,
      "train_flow_matching_loss": 0.061279296875,
      "train_physics_loss": 0.8826774954795837
    },
    {
      "step": 8710,
      "train_flow_matching_loss": 0.055908203125,
      "train_physics_loss": 0.7824199795722961
    },
    {
      "step": 8710,
      "train_flow_matching_loss": 0.1923828125,
      "train_physics_loss": 0.0
    },
    {
      "step": 8710,
      "train_flow_matching_loss": 0.09375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10960599780082703,
      "learning_rate": 5e-05,
      "loss": 0.1624,
      "step": 8720
    },
    {
      "step": 8720,
      "train_flow_matching_loss": 0.1494140625,
      "train_physics_loss": 0.0
    },
    {
      "step": 8720,
      "train_flow_matching_loss": 0.11328125,
      "train_physics_loss": 0.8025091886520386
    },
    {
      "step": 8720,
      "train_flow_matching_loss": 0.2373046875,
      "train_physics_loss": 0.0
    },
    {
      "step": 8720,
      "train_flow_matching_loss": 0.10888671875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10720736533403397,
      "learning_rate": 5e-05,
      "loss": 0.1615,
      "step": 8730
    },
    {
      "step": 8730,
      "train_flow_matching_loss": 0.1953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 8730,
      "train_flow_matching_loss": 0.0654296875,
      "train_physics_loss": 0.7841397523880005
    },
    {
      "step": 8730,
      "train_flow_matching_loss": 0.2265625,
      "train_physics_loss": 0.0
    },
    {
      "step": 8730,
      "train_flow_matching_loss": 0.11181640625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.1574275642633438,
      "learning_rate": 5e-05,
      "loss": 0.1581,
      "step": 8740
    },
    {
      "step": 8740,
      "train_flow_matching_loss": 0.208984375,
      "train_physics_loss": 0.0
    },
    {
      "step": 8740,
      "train_flow_matching_loss": 0.205078125,
      "train_physics_loss": 0.0
    },
    {
      "step": 8740,
      "train_flow_matching_loss": 0.1318359375,
      "train_physics_loss": 0.0
    },
    {
      "step": 8740,
      "train_flow_matching_loss": 0.1484375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09554370492696762,
      "learning_rate": 5e-05,
      "loss": 0.1622,
      "step": 8750
    },
    {
      "step": 8750,
      "train_flow_matching_loss": 0.134765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 8750,
      "train_flow_matching_loss": 0.1025390625,
      "train_physics_loss": 0.0
    },
    {
      "step": 8750,
      "train_flow_matching_loss": 0.2080078125,
      "train_physics_loss": 0.0
    },
    {
      "step": 8750,
      "train_flow_matching_loss": 0.1044921875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.34934109449386597,
      "learning_rate": 5e-05,
      "loss": 0.1629,
      "step": 8760
    },
    {
      "step": 8760,
      "train_flow_matching_loss": 0.216796875,
      "train_physics_loss": 0.0
    },
    {
      "step": 8760,
      "train_flow_matching_loss": 0.158203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 8760,
      "train_flow_matching_loss": 0.1318359375,
      "train_physics_loss": 0.0
    },
    {
      "step": 8760,
      "train_flow_matching_loss": 0.1396484375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.408486008644104,
      "learning_rate": 5e-05,
      "loss": 0.1593,
      "step": 8770
    },
    {
      "step": 8770,
      "train_flow_matching_loss": 0.1298828125,
      "train_physics_loss": 0.0
    },
    {
      "step": 8770,
      "train_flow_matching_loss": 0.185546875,
      "train_physics_loss": 0.0
    },
    {
      "step": 8770,
      "train_flow_matching_loss": 0.208984375,
      "train_physics_loss": 0.0
    },
    {
      "step": 8770,
      "train_flow_matching_loss": 0.130859375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.12308097630739212,
      "learning_rate": 5e-05,
      "loss": 0.1614,
      "step": 8780
    },
    {
      "step": 8780,
      "train_flow_matching_loss": 0.212890625,
      "train_physics_loss": 0.0
    },
    {
      "step": 8780,
      "train_flow_matching_loss": 0.166015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 8780,
      "train_flow_matching_loss": 0.2099609375,
      "train_physics_loss": 0.0
    },
    {
      "step": 8780,
      "train_flow_matching_loss": 0.1572265625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.12506677210330963,
      "learning_rate": 5e-05,
      "loss": 0.1672,
      "step": 8790
    },
    {
      "step": 8790,
      "train_flow_matching_loss": 0.11083984375,
      "train_physics_loss": 0.0
    },
    {
      "step": 8790,
      "train_flow_matching_loss": 0.234375,
      "train_physics_loss": 0.0
    },
    {
      "step": 8790,
      "train_flow_matching_loss": 0.1884765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 8790,
      "train_flow_matching_loss": 0.1787109375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.111077219247818,
      "learning_rate": 5e-05,
      "loss": 0.1608,
      "step": 8800
    },
    {
      "step": 8800,
      "train_flow_matching_loss": 0.234375,
      "train_physics_loss": 0.0
    },
    {
      "step": 8800,
      "train_flow_matching_loss": 0.146484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 8800,
      "train_flow_matching_loss": 0.1376953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 8800,
      "train_flow_matching_loss": 0.08544921875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08572602272033691,
      "learning_rate": 5e-05,
      "loss": 0.1584,
      "step": 8810
    },
    {
      "step": 8810,
      "train_flow_matching_loss": 0.07958984375,
      "train_physics_loss": 0.0
    },
    {
      "step": 8810,
      "train_flow_matching_loss": 0.1611328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 8810,
      "train_flow_matching_loss": 0.1796875,
      "train_physics_loss": 0.0
    },
    {
      "step": 8810,
      "train_flow_matching_loss": 0.1513671875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10413847118616104,
      "learning_rate": 5e-05,
      "loss": 0.1646,
      "step": 8820
    },
    {
      "step": 8820,
      "train_flow_matching_loss": 0.126953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 8820,
      "train_flow_matching_loss": 0.169921875,
      "train_physics_loss": 0.0
    },
    {
      "step": 8820,
      "train_flow_matching_loss": 0.1689453125,
      "train_physics_loss": 0.0
    },
    {
      "step": 8820,
      "train_flow_matching_loss": 0.1875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10414993762969971,
      "learning_rate": 5e-05,
      "loss": 0.1571,
      "step": 8830
    },
    {
      "step": 8830,
      "train_flow_matching_loss": 0.1484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 8830,
      "train_flow_matching_loss": 0.1953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 8830,
      "train_flow_matching_loss": 0.11767578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 8830,
      "train_flow_matching_loss": 0.12255859375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09472321718931198,
      "learning_rate": 5e-05,
      "loss": 0.1628,
      "step": 8840
    },
    {
      "step": 8840,
      "train_flow_matching_loss": 0.1591796875,
      "train_physics_loss": 0.0
    },
    {
      "step": 8840,
      "train_flow_matching_loss": 0.1904296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 8840,
      "train_flow_matching_loss": 0.11474609375,
      "train_physics_loss": 0.0
    },
    {
      "step": 8840,
      "train_flow_matching_loss": 0.1376953125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11112593114376068,
      "learning_rate": 5e-05,
      "loss": 0.1636,
      "step": 8850
    },
    {
      "step": 8850,
      "train_flow_matching_loss": 0.166015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 8850,
      "train_flow_matching_loss": 0.2041015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 8850,
      "train_flow_matching_loss": 0.134765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 8850,
      "train_flow_matching_loss": 0.166015625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09138376265764236,
      "learning_rate": 5e-05,
      "loss": 0.1605,
      "step": 8860
    },
    {
      "step": 8860,
      "train_flow_matching_loss": 0.134765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 8860,
      "train_flow_matching_loss": 0.1552734375,
      "train_physics_loss": 0.0
    },
    {
      "step": 8860,
      "train_flow_matching_loss": 0.1767578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 8860,
      "train_flow_matching_loss": 0.1630859375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10251940786838531,
      "learning_rate": 5e-05,
      "loss": 0.1576,
      "step": 8870
    },
    {
      "step": 8870,
      "train_flow_matching_loss": 0.10302734375,
      "train_physics_loss": 0.0
    },
    {
      "step": 8870,
      "train_flow_matching_loss": 0.205078125,
      "train_physics_loss": 0.0
    },
    {
      "step": 8870,
      "train_flow_matching_loss": 0.126953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 8870,
      "train_flow_matching_loss": 0.130859375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11561401188373566,
      "learning_rate": 5e-05,
      "loss": 0.163,
      "step": 8880
    },
    {
      "step": 8880,
      "train_flow_matching_loss": 0.1806640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 8880,
      "train_flow_matching_loss": 0.1376953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 8880,
      "train_flow_matching_loss": 0.173828125,
      "train_physics_loss": 0.0
    },
    {
      "step": 8880,
      "train_flow_matching_loss": 0.1494140625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.0853806808590889,
      "learning_rate": 5e-05,
      "loss": 0.1565,
      "step": 8890
    },
    {
      "step": 8890,
      "train_flow_matching_loss": 0.2021484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 8890,
      "train_flow_matching_loss": 0.1767578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 8890,
      "train_flow_matching_loss": 0.1962890625,
      "train_physics_loss": 0.0
    },
    {
      "step": 8890,
      "train_flow_matching_loss": 0.171875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.07852333784103394,
      "learning_rate": 5e-05,
      "loss": 0.1615,
      "step": 8900
    },
    {
      "step": 8900,
      "train_flow_matching_loss": 0.10791015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 8900,
      "train_flow_matching_loss": 0.2216796875,
      "train_physics_loss": 0.0
    },
    {
      "step": 8900,
      "train_flow_matching_loss": 0.0830078125,
      "train_physics_loss": 0.0
    },
    {
      "step": 8900,
      "train_flow_matching_loss": 0.12353515625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10867798328399658,
      "learning_rate": 5e-05,
      "loss": 0.1599,
      "step": 8910
    },
    {
      "step": 8910,
      "train_flow_matching_loss": 0.2353515625,
      "train_physics_loss": 0.0
    },
    {
      "step": 8910,
      "train_flow_matching_loss": 0.228515625,
      "train_physics_loss": 0.0
    },
    {
      "step": 8910,
      "train_flow_matching_loss": 0.1875,
      "train_physics_loss": 0.0
    },
    {
      "step": 8910,
      "train_flow_matching_loss": 0.1455078125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11278471350669861,
      "learning_rate": 5e-05,
      "loss": 0.163,
      "step": 8920
    },
    {
      "step": 8920,
      "train_flow_matching_loss": 0.2080078125,
      "train_physics_loss": 0.0
    },
    {
      "step": 8920,
      "train_flow_matching_loss": 0.2109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 8920,
      "train_flow_matching_loss": 0.1630859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 8920,
      "train_flow_matching_loss": 0.1904296875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.35854393243789673,
      "learning_rate": 5e-05,
      "loss": 0.1594,
      "step": 8930
    },
    {
      "step": 8930,
      "train_flow_matching_loss": 0.181640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 8930,
      "train_flow_matching_loss": 0.1943359375,
      "train_physics_loss": 0.0
    },
    {
      "step": 8930,
      "train_flow_matching_loss": 0.1376953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 8930,
      "train_flow_matching_loss": 0.1513671875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.16128849983215332,
      "learning_rate": 5e-05,
      "loss": 0.1596,
      "step": 8940
    },
    {
      "step": 8940,
      "train_flow_matching_loss": 0.1962890625,
      "train_physics_loss": 0.0
    },
    {
      "step": 8940,
      "train_flow_matching_loss": 0.1630859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 8940,
      "train_flow_matching_loss": 0.10498046875,
      "train_physics_loss": 0.0
    },
    {
      "step": 8940,
      "train_flow_matching_loss": 0.130859375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09620051831007004,
      "learning_rate": 5e-05,
      "loss": 0.1559,
      "step": 8950
    },
    {
      "step": 8950,
      "train_flow_matching_loss": 0.1904296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 8950,
      "train_flow_matching_loss": 0.1572265625,
      "train_physics_loss": 0.0
    },
    {
      "step": 8950,
      "train_flow_matching_loss": 0.2197265625,
      "train_physics_loss": 0.0
    },
    {
      "step": 8950,
      "train_flow_matching_loss": 0.10791015625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.13224013149738312,
      "learning_rate": 5e-05,
      "loss": 0.1675,
      "step": 8960
    },
    {
      "step": 8960,
      "train_flow_matching_loss": 0.1044921875,
      "train_physics_loss": 0.0
    },
    {
      "step": 8960,
      "train_flow_matching_loss": 0.1533203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 8960,
      "train_flow_matching_loss": 0.181640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 8960,
      "train_flow_matching_loss": 0.1279296875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.15189257264137268,
      "learning_rate": 5e-05,
      "loss": 0.1555,
      "step": 8970
    },
    {
      "step": 8970,
      "train_flow_matching_loss": 0.171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 8970,
      "train_flow_matching_loss": 0.181640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 8970,
      "train_flow_matching_loss": 0.173828125,
      "train_physics_loss": 0.0
    },
    {
      "step": 8970,
      "train_flow_matching_loss": 0.1259765625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11213100701570511,
      "learning_rate": 5e-05,
      "loss": 0.1633,
      "step": 8980
    },
    {
      "step": 8980,
      "train_flow_matching_loss": 0.1279296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 8980,
      "train_flow_matching_loss": 0.1240234375,
      "train_physics_loss": 0.0
    },
    {
      "step": 8980,
      "train_flow_matching_loss": 0.08203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 8980,
      "train_flow_matching_loss": 0.138671875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10363752394914627,
      "learning_rate": 5e-05,
      "loss": 0.1627,
      "step": 8990
    },
    {
      "step": 8990,
      "train_flow_matching_loss": 0.11474609375,
      "train_physics_loss": 0.0
    },
    {
      "step": 8990,
      "train_flow_matching_loss": 0.1953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 8990,
      "train_flow_matching_loss": 0.1552734375,
      "train_physics_loss": 0.0
    },
    {
      "step": 8990,
      "train_flow_matching_loss": 0.1572265625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10402128100395203,
      "learning_rate": 5e-05,
      "loss": 0.1638,
      "step": 9000
    },
    {
      "step": 9000,
      "train_flow_matching_loss": 0.158203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 9000,
      "train_flow_matching_loss": 0.1435546875,
      "train_physics_loss": 0.0
    },
    {
      "step": 9000,
      "train_flow_matching_loss": 0.1708984375,
      "train_physics_loss": 0.0
    },
    {
      "step": 9000,
      "train_flow_matching_loss": 0.0986328125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.16737249493598938,
      "learning_rate": 5e-05,
      "loss": 0.1574,
      "step": 9010
    },
    {
      "step": 9010,
      "train_flow_matching_loss": 0.1796875,
      "train_physics_loss": 0.0
    },
    {
      "step": 9010,
      "train_flow_matching_loss": 0.126953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 9010,
      "train_flow_matching_loss": 0.2353515625,
      "train_physics_loss": 0.0
    },
    {
      "step": 9010,
      "train_flow_matching_loss": 0.126953125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09235397726297379,
      "learning_rate": 5e-05,
      "loss": 0.1604,
      "step": 9020
    },
    {
      "step": 9020,
      "train_flow_matching_loss": 0.1259765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 9020,
      "train_flow_matching_loss": 0.181640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 9020,
      "train_flow_matching_loss": 0.205078125,
      "train_physics_loss": 0.0
    },
    {
      "step": 9020,
      "train_flow_matching_loss": 0.1044921875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.07581480592489243,
      "learning_rate": 5e-05,
      "loss": 0.1543,
      "step": 9030
    },
    {
      "step": 9030,
      "train_flow_matching_loss": 0.126953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 9030,
      "train_flow_matching_loss": 0.1240234375,
      "train_physics_loss": 0.0
    },
    {
      "step": 9030,
      "train_flow_matching_loss": 0.12353515625,
      "train_physics_loss": 0.0
    },
    {
      "step": 9030,
      "train_flow_matching_loss": 0.1943359375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11997681111097336,
      "learning_rate": 5e-05,
      "loss": 0.1608,
      "step": 9040
    },
    {
      "step": 9040,
      "train_flow_matching_loss": 0.1787109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 9040,
      "train_flow_matching_loss": 0.1640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 9040,
      "train_flow_matching_loss": 0.0947265625,
      "train_physics_loss": 0.0
    },
    {
      "step": 9040,
      "train_flow_matching_loss": 0.10107421875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.1388426274061203,
      "learning_rate": 5e-05,
      "loss": 0.1572,
      "step": 9050
    },
    {
      "step": 9050,
      "train_flow_matching_loss": 0.1455078125,
      "train_physics_loss": 0.0
    },
    {
      "step": 9050,
      "train_flow_matching_loss": 0.1455078125,
      "train_physics_loss": 0.0
    },
    {
      "step": 9050,
      "train_flow_matching_loss": 0.09228515625,
      "train_physics_loss": 0.0
    },
    {
      "step": 9050,
      "train_flow_matching_loss": 0.126953125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.05907746031880379,
      "learning_rate": 5e-05,
      "loss": 0.158,
      "step": 9060
    },
    {
      "step": 9060,
      "train_flow_matching_loss": 0.10302734375,
      "train_physics_loss": 0.0
    },
    {
      "step": 9060,
      "train_flow_matching_loss": 0.1484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 9060,
      "train_flow_matching_loss": 0.126953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 9060,
      "train_flow_matching_loss": 0.140625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.13017235696315765,
      "learning_rate": 5e-05,
      "loss": 0.1615,
      "step": 9070
    },
    {
      "step": 9070,
      "train_flow_matching_loss": 0.1318359375,
      "train_physics_loss": 0.0
    },
    {
      "step": 9070,
      "train_flow_matching_loss": 0.15625,
      "train_physics_loss": 0.0
    },
    {
      "step": 9070,
      "train_flow_matching_loss": 0.10693359375,
      "train_physics_loss": 0.0
    },
    {
      "step": 9070,
      "train_flow_matching_loss": 0.1455078125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08665087819099426,
      "learning_rate": 5e-05,
      "loss": 0.1588,
      "step": 9080
    },
    {
      "step": 9080,
      "train_flow_matching_loss": 0.1181640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 9080,
      "train_flow_matching_loss": 0.2138671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 9080,
      "train_flow_matching_loss": 0.1591796875,
      "train_physics_loss": 0.0
    },
    {
      "step": 9080,
      "train_flow_matching_loss": 0.1005859375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09406599402427673,
      "learning_rate": 5e-05,
      "loss": 0.1595,
      "step": 9090
    },
    {
      "step": 9090,
      "train_flow_matching_loss": 0.125,
      "train_physics_loss": 0.0
    },
    {
      "step": 9090,
      "train_flow_matching_loss": 0.12353515625,
      "train_physics_loss": 0.0
    },
    {
      "step": 9090,
      "train_flow_matching_loss": 0.09130859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 9090,
      "train_flow_matching_loss": 0.12109375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09033878892660141,
      "learning_rate": 5e-05,
      "loss": 0.1593,
      "step": 9100
    },
    {
      "step": 9100,
      "train_flow_matching_loss": 0.208984375,
      "train_physics_loss": 0.0
    },
    {
      "step": 9100,
      "train_flow_matching_loss": 0.1181640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 9100,
      "train_flow_matching_loss": 0.1572265625,
      "train_physics_loss": 0.0
    },
    {
      "step": 9100,
      "train_flow_matching_loss": 0.1455078125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11407460272312164,
      "learning_rate": 5e-05,
      "loss": 0.1606,
      "step": 9110
    },
    {
      "step": 9110,
      "train_flow_matching_loss": 0.125,
      "train_physics_loss": 0.0
    },
    {
      "step": 9110,
      "train_flow_matching_loss": 0.1552734375,
      "train_physics_loss": 0.0
    },
    {
      "step": 9110,
      "train_flow_matching_loss": 0.1201171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 9110,
      "train_flow_matching_loss": 0.166015625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.20247983932495117,
      "learning_rate": 5e-05,
      "loss": 0.1599,
      "step": 9120
    },
    {
      "step": 9120,
      "train_flow_matching_loss": 0.177734375,
      "train_physics_loss": 0.0
    },
    {
      "step": 9120,
      "train_flow_matching_loss": 0.10498046875,
      "train_physics_loss": 0.0
    },
    {
      "step": 9120,
      "train_flow_matching_loss": 0.1845703125,
      "train_physics_loss": 0.0
    },
    {
      "step": 9120,
      "train_flow_matching_loss": 0.25390625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.1021827831864357,
      "learning_rate": 5e-05,
      "loss": 0.1529,
      "step": 9130
    },
    {
      "step": 9130,
      "train_flow_matching_loss": 0.125,
      "train_physics_loss": 0.0
    },
    {
      "step": 9130,
      "train_flow_matching_loss": 0.1025390625,
      "train_physics_loss": 0.0
    },
    {
      "step": 9130,
      "train_flow_matching_loss": 0.1591796875,
      "train_physics_loss": 0.0
    },
    {
      "step": 9130,
      "train_flow_matching_loss": 0.1650390625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10386350750923157,
      "learning_rate": 5e-05,
      "loss": 0.158,
      "step": 9140
    },
    {
      "step": 9140,
      "train_flow_matching_loss": 0.1240234375,
      "train_physics_loss": 0.0
    },
    {
      "step": 9140,
      "train_flow_matching_loss": 0.1533203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 9140,
      "train_flow_matching_loss": 0.1435546875,
      "train_physics_loss": 0.0
    },
    {
      "step": 9140,
      "train_flow_matching_loss": 0.068359375,
      "train_physics_loss": 0.703398585319519
    },
    {
      "grad_norm": 0.08835448324680328,
      "learning_rate": 5e-05,
      "loss": 0.161,
      "step": 9150
    },
    {
      "step": 9150,
      "train_flow_matching_loss": 0.1806640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 9150,
      "train_flow_matching_loss": 0.1396484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 9150,
      "train_flow_matching_loss": 0.0712890625,
      "train_physics_loss": 0.761110782623291
    },
    {
      "step": 9150,
      "train_flow_matching_loss": 0.1357421875,
      "train_physics_loss": 0.7744954228401184
    },
    {
      "grad_norm": 0.10603845119476318,
      "learning_rate": 5e-05,
      "loss": 0.1592,
      "step": 9160
    },
    {
      "step": 9160,
      "train_flow_matching_loss": 0.1611328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 9160,
      "train_flow_matching_loss": 0.1396484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 9160,
      "train_flow_matching_loss": 0.1103515625,
      "train_physics_loss": 0.8735278248786926
    },
    {
      "step": 9160,
      "train_flow_matching_loss": 0.0810546875,
      "train_physics_loss": 0.7606942653656006
    },
    {
      "grad_norm": 0.1029578447341919,
      "learning_rate": 5e-05,
      "loss": 0.1637,
      "step": 9170
    },
    {
      "step": 9170,
      "train_flow_matching_loss": 0.2099609375,
      "train_physics_loss": 0.0
    },
    {
      "step": 9170,
      "train_flow_matching_loss": 0.26953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 9170,
      "train_flow_matching_loss": 0.11962890625,
      "train_physics_loss": 0.8068167567253113
    },
    {
      "step": 9170,
      "train_flow_matching_loss": 0.09228515625,
      "train_physics_loss": 0.7405682802200317
    },
    {
      "grad_norm": 0.08763290196657181,
      "learning_rate": 5e-05,
      "loss": 0.1621,
      "step": 9180
    },
    {
      "step": 9180,
      "train_flow_matching_loss": 0.138671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 9180,
      "train_flow_matching_loss": 0.1396484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 9180,
      "train_flow_matching_loss": 0.09423828125,
      "train_physics_loss": 0.7543764114379883
    },
    {
      "step": 9180,
      "train_flow_matching_loss": 0.07763671875,
      "train_physics_loss": 0.7102823257446289
    },
    {
      "grad_norm": 0.1335349678993225,
      "learning_rate": 5e-05,
      "loss": 0.1568,
      "step": 9190
    },
    {
      "step": 9190,
      "train_flow_matching_loss": 0.10595703125,
      "train_physics_loss": 0.0
    },
    {
      "step": 9190,
      "train_flow_matching_loss": 0.1298828125,
      "train_physics_loss": 0.0
    },
    {
      "step": 9190,
      "train_flow_matching_loss": 0.0625,
      "train_physics_loss": 0.7370773553848267
    },
    {
      "step": 9190,
      "train_flow_matching_loss": 0.11083984375,
      "train_physics_loss": 0.7507750391960144
    },
    {
      "grad_norm": 0.22086788713932037,
      "learning_rate": 5e-05,
      "loss": 0.1616,
      "step": 9200
    },
    {
      "step": 9200,
      "train_flow_matching_loss": 0.1181640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 9200,
      "train_flow_matching_loss": 0.1181640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 9200,
      "train_flow_matching_loss": 0.08837890625,
      "train_physics_loss": 0.7456842660903931
    },
    {
      "step": 9200,
      "train_flow_matching_loss": 0.056884765625,
      "train_physics_loss": 0.736181378364563
    },
    {
      "grad_norm": 0.17513392865657806,
      "learning_rate": 5e-05,
      "loss": 0.1609,
      "step": 9210
    },
    {
      "step": 9210,
      "train_flow_matching_loss": 0.1572265625,
      "train_physics_loss": 0.0
    },
    {
      "step": 9210,
      "train_flow_matching_loss": 0.154296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 9210,
      "train_flow_matching_loss": 0.04052734375,
      "train_physics_loss": 0.7669681310653687
    },
    {
      "step": 9210,
      "train_flow_matching_loss": 0.09521484375,
      "train_physics_loss": 0.811478316783905
    },
    {
      "grad_norm": 0.0889938473701477,
      "learning_rate": 5e-05,
      "loss": 0.1629,
      "step": 9220
    },
    {
      "step": 9220,
      "train_flow_matching_loss": 0.1572265625,
      "train_physics_loss": 0.0
    },
    {
      "step": 9220,
      "train_flow_matching_loss": 0.1318359375,
      "train_physics_loss": 0.0
    },
    {
      "step": 9220,
      "train_flow_matching_loss": 0.09716796875,
      "train_physics_loss": 0.7033873796463013
    },
    {
      "step": 9220,
      "train_flow_matching_loss": 0.109375,
      "train_physics_loss": 0.8739922046661377
    },
    {
      "grad_norm": 0.09659211337566376,
      "learning_rate": 5e-05,
      "loss": 0.1608,
      "step": 9230
    },
    {
      "step": 9230,
      "train_flow_matching_loss": 0.07373046875,
      "train_physics_loss": 0.7698211073875427
    },
    {
      "step": 9230,
      "train_flow_matching_loss": 0.1572265625,
      "train_physics_loss": 0.0
    },
    {
      "step": 9230,
      "train_flow_matching_loss": 0.10693359375,
      "train_physics_loss": 0.7404171228408813
    },
    {
      "step": 9230,
      "train_flow_matching_loss": 0.07763671875,
      "train_physics_loss": 0.7164875864982605
    },
    {
      "grad_norm": 0.1300290822982788,
      "learning_rate": 5e-05,
      "loss": 0.1581,
      "step": 9240
    },
    {
      "step": 9240,
      "train_flow_matching_loss": 0.11328125,
      "train_physics_loss": 0.74704509973526
    },
    {
      "step": 9240,
      "train_flow_matching_loss": 0.1728515625,
      "train_physics_loss": 0.0
    },
    {
      "step": 9240,
      "train_flow_matching_loss": 0.05078125,
      "train_physics_loss": 0.7793575525283813
    },
    {
      "step": 9240,
      "train_flow_matching_loss": 0.08349609375,
      "train_physics_loss": 0.8193185925483704
    },
    {
      "grad_norm": 0.13964618742465973,
      "learning_rate": 5e-05,
      "loss": 0.1625,
      "step": 9250
    },
    {
      "step": 9250,
      "train_flow_matching_loss": 0.10400390625,
      "train_physics_loss": 0.7259439826011658
    },
    {
      "step": 9250,
      "train_flow_matching_loss": 0.1279296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 9250,
      "train_flow_matching_loss": 0.04638671875,
      "train_physics_loss": 0.7148886919021606
    },
    {
      "step": 9250,
      "train_flow_matching_loss": 0.12255859375,
      "train_physics_loss": 0.8880577683448792
    },
    {
      "grad_norm": 0.5044543147087097,
      "learning_rate": 5e-05,
      "loss": 0.1592,
      "step": 9260
    },
    {
      "step": 9260,
      "train_flow_matching_loss": 0.09423828125,
      "train_physics_loss": 0.7993042469024658
    },
    {
      "step": 9260,
      "train_flow_matching_loss": 0.1875,
      "train_physics_loss": 0.0
    },
    {
      "step": 9260,
      "train_flow_matching_loss": 0.0654296875,
      "train_physics_loss": 0.7389227747917175
    },
    {
      "step": 9260,
      "train_flow_matching_loss": 0.05517578125,
      "train_physics_loss": 0.7754034996032715
    },
    {
      "grad_norm": 0.20816151797771454,
      "learning_rate": 5e-05,
      "loss": 0.1601,
      "step": 9270
    },
    {
      "step": 9270,
      "train_flow_matching_loss": 0.09423828125,
      "train_physics_loss": 0.8224921822547913
    },
    {
      "step": 9270,
      "train_flow_matching_loss": 0.1279296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 9270,
      "train_flow_matching_loss": 0.055908203125,
      "train_physics_loss": 0.7137913107872009
    },
    {
      "step": 9270,
      "train_flow_matching_loss": 0.045166015625,
      "train_physics_loss": 0.7141024470329285
    },
    {
      "grad_norm": 0.08280481398105621,
      "learning_rate": 5e-05,
      "loss": 0.1584,
      "step": 9280
    },
    {
      "step": 9280,
      "train_flow_matching_loss": 0.09521484375,
      "train_physics_loss": 0.727252185344696
    },
    {
      "step": 9280,
      "train_flow_matching_loss": 0.1396484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 9280,
      "train_flow_matching_loss": 0.052001953125,
      "train_physics_loss": 0.756649374961853
    },
    {
      "step": 9280,
      "train_flow_matching_loss": 0.047607421875,
      "train_physics_loss": 0.7473219633102417
    },
    {
      "grad_norm": 0.12319713085889816,
      "learning_rate": 5e-05,
      "loss": 0.1586,
      "step": 9290
    },
    {
      "step": 9290,
      "train_flow_matching_loss": 0.05712890625,
      "train_physics_loss": 0.7379494905471802
    },
    {
      "step": 9290,
      "train_flow_matching_loss": 0.09130859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 9290,
      "train_flow_matching_loss": 0.0712890625,
      "train_physics_loss": 0.7137587070465088
    },
    {
      "step": 9290,
      "train_flow_matching_loss": 0.1015625,
      "train_physics_loss": 0.7893387079238892
    },
    {
      "grad_norm": 0.12853305041790009,
      "learning_rate": 5e-05,
      "loss": 0.1626,
      "step": 9300
    },
    {
      "step": 9300,
      "train_flow_matching_loss": 0.10009765625,
      "train_physics_loss": 0.9504919052124023
    },
    {
      "step": 9300,
      "train_flow_matching_loss": 0.1669921875,
      "train_physics_loss": 0.0
    },
    {
      "step": 9300,
      "train_flow_matching_loss": 0.0849609375,
      "train_physics_loss": 0.6798545122146606
    },
    {
      "step": 9300,
      "train_flow_matching_loss": 0.1103515625,
      "train_physics_loss": 0.852247416973114
    },
    {
      "grad_norm": 0.13082978129386902,
      "learning_rate": 5e-05,
      "loss": 0.1613,
      "step": 9310
    },
    {
      "step": 9310,
      "train_flow_matching_loss": 0.1494140625,
      "train_physics_loss": 0.7606300115585327
    },
    {
      "step": 9310,
      "train_flow_matching_loss": 0.1640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 9310,
      "train_flow_matching_loss": 0.056884765625,
      "train_physics_loss": 0.8057335615158081
    },
    {
      "step": 9310,
      "train_flow_matching_loss": 0.142578125,
      "train_physics_loss": 0.7705537676811218
    },
    {
      "grad_norm": 0.4879722595214844,
      "learning_rate": 5e-05,
      "loss": 0.1603,
      "step": 9320
    },
    {
      "step": 9320,
      "train_flow_matching_loss": 0.04443359375,
      "train_physics_loss": 0.7130759954452515
    },
    {
      "step": 9320,
      "train_flow_matching_loss": 0.1748046875,
      "train_physics_loss": 0.0
    },
    {
      "step": 9320,
      "train_flow_matching_loss": 0.06298828125,
      "train_physics_loss": 0.7857612371444702
    },
    {
      "step": 9320,
      "train_flow_matching_loss": 0.11328125,
      "train_physics_loss": 0.7172865867614746
    },
    {
      "grad_norm": 0.13232861459255219,
      "learning_rate": 5e-05,
      "loss": 0.1602,
      "step": 9330
    },
    {
      "step": 9330,
      "train_flow_matching_loss": 0.03564453125,
      "train_physics_loss": 0.7315791249275208
    },
    {
      "step": 9330,
      "train_flow_matching_loss": 0.244140625,
      "train_physics_loss": 0.0
    },
    {
      "step": 9330,
      "train_flow_matching_loss": 0.08642578125,
      "train_physics_loss": 0.7635853886604309
    },
    {
      "step": 9330,
      "train_flow_matching_loss": 0.09912109375,
      "train_physics_loss": 0.7336889505386353
    },
    {
      "grad_norm": 0.09179417043924332,
      "learning_rate": 5e-05,
      "loss": 0.159,
      "step": 9340
    },
    {
      "step": 9340,
      "train_flow_matching_loss": 0.062255859375,
      "train_physics_loss": 0.7165908813476562
    },
    {
      "step": 9340,
      "train_flow_matching_loss": 0.09423828125,
      "train_physics_loss": 0.0
    },
    {
      "step": 9340,
      "train_flow_matching_loss": 0.06640625,
      "train_physics_loss": 0.741042971611023
    },
    {
      "step": 9340,
      "train_flow_matching_loss": 0.13671875,
      "train_physics_loss": 0.8089035749435425
    },
    {
      "grad_norm": 0.1016167476773262,
      "learning_rate": 5e-05,
      "loss": 0.1536,
      "step": 9350
    },
    {
      "step": 9350,
      "train_flow_matching_loss": 0.10986328125,
      "train_physics_loss": 0.7788077592849731
    },
    {
      "step": 9350,
      "train_flow_matching_loss": 0.1787109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 9350,
      "train_flow_matching_loss": 0.07080078125,
      "train_physics_loss": 0.7469981908798218
    },
    {
      "step": 9350,
      "train_flow_matching_loss": 0.08837890625,
      "train_physics_loss": 0.8164000511169434
    },
    {
      "grad_norm": 0.09147607535123825,
      "learning_rate": 5e-05,
      "loss": 0.1615,
      "step": 9360
    },
    {
      "step": 9360,
      "train_flow_matching_loss": 0.07275390625,
      "train_physics_loss": 0.8583841323852539
    },
    {
      "step": 9360,
      "train_flow_matching_loss": 0.2353515625,
      "train_physics_loss": 0.0
    },
    {
      "step": 9360,
      "train_flow_matching_loss": 0.08447265625,
      "train_physics_loss": 0.7986686825752258
    },
    {
      "step": 9360,
      "train_flow_matching_loss": 0.037353515625,
      "train_physics_loss": 0.7546172142028809
    },
    {
      "grad_norm": 0.10624420642852783,
      "learning_rate": 5e-05,
      "loss": 0.1624,
      "step": 9370
    },
    {
      "step": 9370,
      "train_flow_matching_loss": 0.07080078125,
      "train_physics_loss": 0.6781212091445923
    },
    {
      "step": 9370,
      "train_flow_matching_loss": 0.09912109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 9370,
      "train_flow_matching_loss": 0.1171875,
      "train_physics_loss": 0.7366958260536194
    },
    {
      "step": 9370,
      "train_flow_matching_loss": 0.032470703125,
      "train_physics_loss": 0.6801437735557556
    },
    {
      "grad_norm": 0.1321457326412201,
      "learning_rate": 5e-05,
      "loss": 0.1614,
      "step": 9380
    },
    {
      "step": 9380,
      "train_flow_matching_loss": 0.08544921875,
      "train_physics_loss": 0.7867214679718018
    },
    {
      "step": 9380,
      "train_flow_matching_loss": 0.150390625,
      "train_physics_loss": 0.0
    },
    {
      "step": 9380,
      "train_flow_matching_loss": 0.05712890625,
      "train_physics_loss": 0.7318757176399231
    },
    {
      "step": 9380,
      "train_flow_matching_loss": 0.06787109375,
      "train_physics_loss": 0.7134580016136169
    },
    {
      "grad_norm": 0.09035155922174454,
      "learning_rate": 5e-05,
      "loss": 0.1572,
      "step": 9390
    },
    {
      "step": 9390,
      "train_flow_matching_loss": 0.1533203125,
      "train_physics_loss": 0.8184448480606079
    },
    {
      "step": 9390,
      "train_flow_matching_loss": 0.1416015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 9390,
      "train_flow_matching_loss": 0.09228515625,
      "train_physics_loss": 0.7579543590545654
    },
    {
      "step": 9390,
      "train_flow_matching_loss": 0.06591796875,
      "train_physics_loss": 0.6782029271125793
    },
    {
      "grad_norm": 0.0913672000169754,
      "learning_rate": 5e-05,
      "loss": 0.1566,
      "step": 9400
    },
    {
      "step": 9400,
      "train_flow_matching_loss": 0.052734375,
      "train_physics_loss": 0.7739683389663696
    },
    {
      "step": 9400,
      "train_flow_matching_loss": 0.11376953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 9400,
      "train_flow_matching_loss": 0.224609375,
      "train_physics_loss": 0.0
    },
    {
      "step": 9400,
      "train_flow_matching_loss": 0.0654296875,
      "train_physics_loss": 0.7278730869293213
    },
    {
      "grad_norm": 0.1256059855222702,
      "learning_rate": 5e-05,
      "loss": 0.1563,
      "step": 9410
    },
    {
      "step": 9410,
      "train_flow_matching_loss": 0.08251953125,
      "train_physics_loss": 0.7786605358123779
    },
    {
      "step": 9410,
      "train_flow_matching_loss": 0.1396484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 9410,
      "train_flow_matching_loss": 0.1796875,
      "train_physics_loss": 0.0
    },
    {
      "step": 9410,
      "train_flow_matching_loss": 0.058837890625,
      "train_physics_loss": 0.7066689729690552
    },
    {
      "grad_norm": 0.09143273532390594,
      "learning_rate": 5e-05,
      "loss": 0.1595,
      "step": 9420
    },
    {
      "step": 9420,
      "train_flow_matching_loss": 0.058837890625,
      "train_physics_loss": 0.7266599535942078
    },
    {
      "step": 9420,
      "train_flow_matching_loss": 0.1279296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 9420,
      "train_flow_matching_loss": 0.193359375,
      "train_physics_loss": 0.0
    },
    {
      "step": 9420,
      "train_flow_matching_loss": 0.1201171875,
      "train_physics_loss": 0.7131859064102173
    },
    {
      "grad_norm": 0.1127856969833374,
      "learning_rate": 5e-05,
      "loss": 0.1602,
      "step": 9430
    },
    {
      "step": 9430,
      "train_flow_matching_loss": 0.11328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 9430,
      "train_flow_matching_loss": 0.1259765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 9430,
      "train_flow_matching_loss": 0.1953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 9430,
      "train_flow_matching_loss": 0.07080078125,
      "train_physics_loss": 0.6965629458427429
    },
    {
      "grad_norm": 0.08566690236330032,
      "learning_rate": 5e-05,
      "loss": 0.1611,
      "step": 9440
    },
    {
      "step": 9440,
      "train_flow_matching_loss": 0.1318359375,
      "train_physics_loss": 0.8503362536430359
    },
    {
      "step": 9440,
      "train_flow_matching_loss": 0.1669921875,
      "train_physics_loss": 0.0
    },
    {
      "step": 9440,
      "train_flow_matching_loss": 0.15625,
      "train_physics_loss": 0.0
    },
    {
      "step": 9440,
      "train_flow_matching_loss": 0.052001953125,
      "train_physics_loss": 0.7293465733528137
    },
    {
      "grad_norm": 0.08285161107778549,
      "learning_rate": 5e-05,
      "loss": 0.1593,
      "step": 9450
    },
    {
      "step": 9450,
      "train_flow_matching_loss": 0.07275390625,
      "train_physics_loss": 0.6819099187850952
    },
    {
      "step": 9450,
      "train_flow_matching_loss": 0.11181640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 9450,
      "train_flow_matching_loss": 0.1259765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 9450,
      "train_flow_matching_loss": 0.08984375,
      "train_physics_loss": 0.7138503789901733
    },
    {
      "grad_norm": 0.08976971358060837,
      "learning_rate": 5e-05,
      "loss": 0.1621,
      "step": 9460
    },
    {
      "step": 9460,
      "train_flow_matching_loss": 0.053955078125,
      "train_physics_loss": 0.6976413130760193
    },
    {
      "step": 9460,
      "train_flow_matching_loss": 0.244140625,
      "train_physics_loss": 0.0
    },
    {
      "step": 9460,
      "train_flow_matching_loss": 0.2421875,
      "train_physics_loss": 0.0
    },
    {
      "step": 9460,
      "train_flow_matching_loss": 0.15625,
      "train_physics_loss": 0.8340611457824707
    },
    {
      "grad_norm": 0.10743208229541779,
      "learning_rate": 5e-05,
      "loss": 0.1581,
      "step": 9470
    },
    {
      "step": 9470,
      "train_flow_matching_loss": 0.08349609375,
      "train_physics_loss": 0.7675508856773376
    },
    {
      "step": 9470,
      "train_flow_matching_loss": 0.1484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 9470,
      "train_flow_matching_loss": 0.203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 9470,
      "train_flow_matching_loss": 0.12353515625,
      "train_physics_loss": 0.7710567116737366
    },
    {
      "grad_norm": 0.11021403223276138,
      "learning_rate": 5e-05,
      "loss": 0.1584,
      "step": 9480
    },
    {
      "step": 9480,
      "train_flow_matching_loss": 0.162109375,
      "train_physics_loss": 0.7355179786682129
    },
    {
      "step": 9480,
      "train_flow_matching_loss": 0.1591796875,
      "train_physics_loss": 0.0
    },
    {
      "step": 9480,
      "train_flow_matching_loss": 0.1689453125,
      "train_physics_loss": 0.0
    },
    {
      "step": 9480,
      "train_flow_matching_loss": 0.09326171875,
      "train_physics_loss": 0.7157162427902222
    },
    {
      "grad_norm": 0.4321560561656952,
      "learning_rate": 5e-05,
      "loss": 0.1679,
      "step": 9490
    },
    {
      "step": 9490,
      "train_flow_matching_loss": 0.1416015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 9490,
      "train_flow_matching_loss": 0.162109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 9490,
      "train_flow_matching_loss": 0.1552734375,
      "train_physics_loss": 0.0
    },
    {
      "step": 9490,
      "train_flow_matching_loss": 0.07666015625,
      "train_physics_loss": 0.6319864392280579
    },
    {
      "grad_norm": 0.10088860988616943,
      "learning_rate": 5e-05,
      "loss": 0.1596,
      "step": 9500
    },
    {
      "step": 9500,
      "train_flow_matching_loss": 0.181640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 9500,
      "train_flow_matching_loss": 0.166015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 9500,
      "train_flow_matching_loss": 0.177734375,
      "train_physics_loss": 0.0
    },
    {
      "step": 9500,
      "train_flow_matching_loss": 0.10107421875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09414558857679367,
      "learning_rate": 5e-05,
      "loss": 0.1558,
      "step": 9510
    },
    {
      "step": 9510,
      "train_flow_matching_loss": 0.203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 9510,
      "train_flow_matching_loss": 0.2109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 9510,
      "train_flow_matching_loss": 0.10302734375,
      "train_physics_loss": 0.0
    },
    {
      "step": 9510,
      "train_flow_matching_loss": 0.1259765625,
      "train_physics_loss": 0.6424440741539001
    },
    {
      "grad_norm": 0.1490015983581543,
      "learning_rate": 5e-05,
      "loss": 0.1599,
      "step": 9520
    },
    {
      "step": 9520,
      "train_flow_matching_loss": 0.1787109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 9520,
      "train_flow_matching_loss": 0.185546875,
      "train_physics_loss": 0.0
    },
    {
      "step": 9520,
      "train_flow_matching_loss": 0.1455078125,
      "train_physics_loss": 0.0
    },
    {
      "step": 9520,
      "train_flow_matching_loss": 0.08251953125,
      "train_physics_loss": 0.7660413980484009
    },
    {
      "grad_norm": 0.10759488493204117,
      "learning_rate": 5e-05,
      "loss": 0.1597,
      "step": 9530
    },
    {
      "step": 9530,
      "train_flow_matching_loss": 0.201171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 9530,
      "train_flow_matching_loss": 0.1357421875,
      "train_physics_loss": 0.0
    },
    {
      "step": 9530,
      "train_flow_matching_loss": 0.13671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 9530,
      "train_flow_matching_loss": 0.11767578125,
      "train_physics_loss": 0.7901870012283325
    },
    {
      "grad_norm": 0.05926850065588951,
      "learning_rate": 5e-05,
      "loss": 0.1615,
      "step": 9540
    },
    {
      "step": 9540,
      "train_flow_matching_loss": 0.2041015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 9540,
      "train_flow_matching_loss": 0.177734375,
      "train_physics_loss": 0.0
    },
    {
      "step": 9540,
      "train_flow_matching_loss": 0.1435546875,
      "train_physics_loss": 0.0
    },
    {
      "step": 9540,
      "train_flow_matching_loss": 0.07080078125,
      "train_physics_loss": 0.7122834324836731
    },
    {
      "grad_norm": 0.12131095677614212,
      "learning_rate": 5e-05,
      "loss": 0.1584,
      "step": 9550
    },
    {
      "step": 9550,
      "train_flow_matching_loss": 0.12158203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 9550,
      "train_flow_matching_loss": 0.1748046875,
      "train_physics_loss": 0.0
    },
    {
      "step": 9550,
      "train_flow_matching_loss": 0.2060546875,
      "train_physics_loss": 0.0
    },
    {
      "step": 9550,
      "train_flow_matching_loss": 0.09326171875,
      "train_physics_loss": 0.6826133728027344
    },
    {
      "grad_norm": 0.0801767110824585,
      "learning_rate": 5e-05,
      "loss": 0.159,
      "step": 9560
    },
    {
      "step": 9560,
      "train_flow_matching_loss": 0.2373046875,
      "train_physics_loss": 0.0
    },
    {
      "step": 9560,
      "train_flow_matching_loss": 0.26953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 9560,
      "train_flow_matching_loss": 0.1806640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 9560,
      "train_flow_matching_loss": 0.11962890625,
      "train_physics_loss": 0.7698678374290466
    },
    {
      "grad_norm": 0.06965447217226028,
      "learning_rate": 5e-05,
      "loss": 0.1595,
      "step": 9570
    },
    {
      "step": 9570,
      "train_flow_matching_loss": 0.154296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 9570,
      "train_flow_matching_loss": 0.1376953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 9570,
      "train_flow_matching_loss": 0.275390625,
      "train_physics_loss": 0.0
    },
    {
      "step": 9570,
      "train_flow_matching_loss": 0.060302734375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11819278448820114,
      "learning_rate": 5e-05,
      "loss": 0.1608,
      "step": 9580
    },
    {
      "step": 9580,
      "train_flow_matching_loss": 0.185546875,
      "train_physics_loss": 0.0
    },
    {
      "step": 9580,
      "train_flow_matching_loss": 0.1611328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 9580,
      "train_flow_matching_loss": 0.189453125,
      "train_physics_loss": 0.0
    },
    {
      "step": 9580,
      "train_flow_matching_loss": 0.12353515625,
      "train_physics_loss": 0.6780657172203064
    },
    {
      "grad_norm": 0.07101361453533173,
      "learning_rate": 5e-05,
      "loss": 0.154,
      "step": 9590
    },
    {
      "step": 9590,
      "train_flow_matching_loss": 0.173828125,
      "train_physics_loss": 0.0
    },
    {
      "step": 9590,
      "train_flow_matching_loss": 0.166015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 9590,
      "train_flow_matching_loss": 0.169921875,
      "train_physics_loss": 0.0
    },
    {
      "step": 9590,
      "train_flow_matching_loss": 0.0732421875,
      "train_physics_loss": 0.7592797875404358
    },
    {
      "grad_norm": 0.13443027436733246,
      "learning_rate": 5e-05,
      "loss": 0.1554,
      "step": 9600
    },
    {
      "step": 9600,
      "train_flow_matching_loss": 0.203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 9600,
      "train_flow_matching_loss": 0.1376953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 9600,
      "train_flow_matching_loss": 0.2373046875,
      "train_physics_loss": 0.0
    },
    {
      "step": 9600,
      "train_flow_matching_loss": 0.0986328125,
      "train_physics_loss": 0.6931596994400024
    },
    {
      "grad_norm": 0.11131393164396286,
      "learning_rate": 5e-05,
      "loss": 0.1628,
      "step": 9610
    },
    {
      "step": 9610,
      "train_flow_matching_loss": 0.1318359375,
      "train_physics_loss": 0.0
    },
    {
      "step": 9610,
      "train_flow_matching_loss": 0.1376953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 9610,
      "train_flow_matching_loss": 0.10205078125,
      "train_physics_loss": 0.0
    },
    {
      "step": 9610,
      "train_flow_matching_loss": 0.07666015625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.15218842029571533,
      "learning_rate": 5e-05,
      "loss": 0.1589,
      "step": 9620
    },
    {
      "step": 9620,
      "train_flow_matching_loss": 0.2197265625,
      "train_physics_loss": 0.0
    },
    {
      "step": 9620,
      "train_flow_matching_loss": 0.162109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 9620,
      "train_flow_matching_loss": 0.1513671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 9620,
      "train_flow_matching_loss": 0.142578125,
      "train_physics_loss": 0.7902889847755432
    },
    {
      "grad_norm": 0.0914798229932785,
      "learning_rate": 5e-05,
      "loss": 0.1598,
      "step": 9630
    },
    {
      "step": 9630,
      "train_flow_matching_loss": 0.205078125,
      "train_physics_loss": 0.0
    },
    {
      "step": 9630,
      "train_flow_matching_loss": 0.1875,
      "train_physics_loss": 0.0
    },
    {
      "step": 9630,
      "train_flow_matching_loss": 0.169921875,
      "train_physics_loss": 0.0
    },
    {
      "step": 9630,
      "train_flow_matching_loss": 0.0830078125,
      "train_physics_loss": 0.7624704241752625
    },
    {
      "grad_norm": 0.11142514646053314,
      "learning_rate": 5e-05,
      "loss": 0.1569,
      "step": 9640
    },
    {
      "step": 9640,
      "train_flow_matching_loss": 0.1533203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 9640,
      "train_flow_matching_loss": 0.185546875,
      "train_physics_loss": 0.0
    },
    {
      "step": 9640,
      "train_flow_matching_loss": 0.1533203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 9640,
      "train_flow_matching_loss": 0.09130859375,
      "train_physics_loss": 0.7590856552124023
    },
    {
      "grad_norm": 0.1118946447968483,
      "learning_rate": 5e-05,
      "loss": 0.1603,
      "step": 9650
    },
    {
      "step": 9650,
      "train_flow_matching_loss": 0.1689453125,
      "train_physics_loss": 0.0
    },
    {
      "step": 9650,
      "train_flow_matching_loss": 0.2236328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 9650,
      "train_flow_matching_loss": 0.2119140625,
      "train_physics_loss": 0.0
    },
    {
      "step": 9650,
      "train_flow_matching_loss": 0.08740234375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11235976219177246,
      "learning_rate": 5e-05,
      "loss": 0.1565,
      "step": 9660
    },
    {
      "step": 9660,
      "train_flow_matching_loss": 0.169921875,
      "train_physics_loss": 0.0
    },
    {
      "step": 9660,
      "train_flow_matching_loss": 0.11962890625,
      "train_physics_loss": 0.0
    },
    {
      "step": 9660,
      "train_flow_matching_loss": 0.12060546875,
      "train_physics_loss": 0.0
    },
    {
      "step": 9660,
      "train_flow_matching_loss": 0.126953125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.12509585916996002,
      "learning_rate": 5e-05,
      "loss": 0.157,
      "step": 9670
    },
    {
      "step": 9670,
      "train_flow_matching_loss": 0.2236328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 9670,
      "train_flow_matching_loss": 0.11767578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 9670,
      "train_flow_matching_loss": 0.2041015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 9670,
      "train_flow_matching_loss": 0.1767578125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09457810968160629,
      "learning_rate": 5e-05,
      "loss": 0.157,
      "step": 9680
    },
    {
      "step": 9680,
      "train_flow_matching_loss": 0.1669921875,
      "train_physics_loss": 0.0
    },
    {
      "step": 9680,
      "train_flow_matching_loss": 0.12158203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 9680,
      "train_flow_matching_loss": 0.1416015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 9680,
      "train_flow_matching_loss": 0.0986328125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11789695173501968,
      "learning_rate": 5e-05,
      "loss": 0.1545,
      "step": 9690
    },
    {
      "step": 9690,
      "train_flow_matching_loss": 0.11865234375,
      "train_physics_loss": 0.0
    },
    {
      "step": 9690,
      "train_flow_matching_loss": 0.10400390625,
      "train_physics_loss": 0.0
    },
    {
      "step": 9690,
      "train_flow_matching_loss": 0.1142578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 9690,
      "train_flow_matching_loss": 0.26171875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.1216481402516365,
      "learning_rate": 5e-05,
      "loss": 0.1562,
      "step": 9700
    },
    {
      "step": 9700,
      "train_flow_matching_loss": 0.259765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 9700,
      "train_flow_matching_loss": 0.1748046875,
      "train_physics_loss": 0.0
    },
    {
      "step": 9700,
      "train_flow_matching_loss": 0.15625,
      "train_physics_loss": 0.0
    },
    {
      "step": 9700,
      "train_flow_matching_loss": 0.12890625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09023042768239975,
      "learning_rate": 5e-05,
      "loss": 0.1599,
      "step": 9710
    },
    {
      "step": 9710,
      "train_flow_matching_loss": 0.2265625,
      "train_physics_loss": 0.0
    },
    {
      "step": 9710,
      "train_flow_matching_loss": 0.205078125,
      "train_physics_loss": 0.0
    },
    {
      "step": 9710,
      "train_flow_matching_loss": 0.2255859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 9710,
      "train_flow_matching_loss": 0.1982421875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11593042314052582,
      "learning_rate": 5e-05,
      "loss": 0.1565,
      "step": 9720
    },
    {
      "step": 9720,
      "train_flow_matching_loss": 0.10009765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 9720,
      "train_flow_matching_loss": 0.228515625,
      "train_physics_loss": 0.0
    },
    {
      "step": 9720,
      "train_flow_matching_loss": 0.09521484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 9720,
      "train_flow_matching_loss": 0.1416015625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10137110948562622,
      "learning_rate": 5e-05,
      "loss": 0.1591,
      "step": 9730
    },
    {
      "step": 9730,
      "train_flow_matching_loss": 0.1923828125,
      "train_physics_loss": 0.0
    },
    {
      "step": 9730,
      "train_flow_matching_loss": 0.154296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 9730,
      "train_flow_matching_loss": 0.1923828125,
      "train_physics_loss": 0.0
    },
    {
      "step": 9730,
      "train_flow_matching_loss": 0.1943359375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09228608012199402,
      "learning_rate": 5e-05,
      "loss": 0.1518,
      "step": 9740
    },
    {
      "step": 9740,
      "train_flow_matching_loss": 0.1337890625,
      "train_physics_loss": 0.0
    },
    {
      "step": 9740,
      "train_flow_matching_loss": 0.2255859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 9740,
      "train_flow_matching_loss": 0.1220703125,
      "train_physics_loss": 0.0
    },
    {
      "step": 9740,
      "train_flow_matching_loss": 0.08642578125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09927590191364288,
      "learning_rate": 5e-05,
      "loss": 0.1584,
      "step": 9750
    },
    {
      "step": 9750,
      "train_flow_matching_loss": 0.1298828125,
      "train_physics_loss": 0.0
    },
    {
      "step": 9750,
      "train_flow_matching_loss": 0.109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 9750,
      "train_flow_matching_loss": 0.1240234375,
      "train_physics_loss": 0.0
    },
    {
      "step": 9750,
      "train_flow_matching_loss": 0.15625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.3835512101650238,
      "learning_rate": 5e-05,
      "loss": 0.1528,
      "step": 9760
    },
    {
      "step": 9760,
      "train_flow_matching_loss": 0.1787109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 9760,
      "train_flow_matching_loss": 0.1767578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 9760,
      "train_flow_matching_loss": 0.1552734375,
      "train_physics_loss": 0.0
    },
    {
      "step": 9760,
      "train_flow_matching_loss": 0.142578125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09602285921573639,
      "learning_rate": 5e-05,
      "loss": 0.1589,
      "step": 9770
    },
    {
      "step": 9770,
      "train_flow_matching_loss": 0.1533203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 9770,
      "train_flow_matching_loss": 0.1923828125,
      "train_physics_loss": 0.0
    },
    {
      "step": 9770,
      "train_flow_matching_loss": 0.1298828125,
      "train_physics_loss": 0.0
    },
    {
      "step": 9770,
      "train_flow_matching_loss": 0.11572265625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.1139197051525116,
      "learning_rate": 5e-05,
      "loss": 0.1533,
      "step": 9780
    },
    {
      "step": 9780,
      "train_flow_matching_loss": 0.1396484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 9780,
      "train_flow_matching_loss": 0.173828125,
      "train_physics_loss": 0.0
    },
    {
      "step": 9780,
      "train_flow_matching_loss": 0.1748046875,
      "train_physics_loss": 0.0
    },
    {
      "step": 9780,
      "train_flow_matching_loss": 0.1640625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.13272979855537415,
      "learning_rate": 5e-05,
      "loss": 0.1553,
      "step": 9790
    },
    {
      "step": 9790,
      "train_flow_matching_loss": 0.2080078125,
      "train_physics_loss": 0.0
    },
    {
      "step": 9790,
      "train_flow_matching_loss": 0.1640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 9790,
      "train_flow_matching_loss": 0.150390625,
      "train_physics_loss": 0.0
    },
    {
      "step": 9790,
      "train_flow_matching_loss": 0.1376953125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10049784928560257,
      "learning_rate": 5e-05,
      "loss": 0.1554,
      "step": 9800
    },
    {
      "step": 9800,
      "train_flow_matching_loss": 0.279296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 9800,
      "train_flow_matching_loss": 0.1142578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 9800,
      "train_flow_matching_loss": 0.10107421875,
      "train_physics_loss": 0.0
    },
    {
      "step": 9800,
      "train_flow_matching_loss": 0.1220703125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.12109503895044327,
      "learning_rate": 5e-05,
      "loss": 0.1587,
      "step": 9810
    },
    {
      "step": 9810,
      "train_flow_matching_loss": 0.1376953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 9810,
      "train_flow_matching_loss": 0.05419921875,
      "train_physics_loss": 0.7597852945327759
    },
    {
      "step": 9810,
      "train_flow_matching_loss": 0.109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 9810,
      "train_flow_matching_loss": 0.208984375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.13978976011276245,
      "learning_rate": 5e-05,
      "loss": 0.1536,
      "step": 9820
    },
    {
      "step": 9820,
      "train_flow_matching_loss": 0.2109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 9820,
      "train_flow_matching_loss": 0.07177734375,
      "train_physics_loss": 0.6884956359863281
    },
    {
      "step": 9820,
      "train_flow_matching_loss": 0.1806640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 9820,
      "train_flow_matching_loss": 0.1748046875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09414032101631165,
      "learning_rate": 5e-05,
      "loss": 0.1582,
      "step": 9830
    },
    {
      "step": 9830,
      "train_flow_matching_loss": 0.1796875,
      "train_physics_loss": 0.0
    },
    {
      "step": 9830,
      "train_flow_matching_loss": 0.030517578125,
      "train_physics_loss": 0.6642864942550659
    },
    {
      "step": 9830,
      "train_flow_matching_loss": 0.197265625,
      "train_physics_loss": 0.0
    },
    {
      "step": 9830,
      "train_flow_matching_loss": 0.1416015625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.12701082229614258,
      "learning_rate": 5e-05,
      "loss": 0.1569,
      "step": 9840
    },
    {
      "step": 9840,
      "train_flow_matching_loss": 0.1201171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 9840,
      "train_flow_matching_loss": 0.030517578125,
      "train_physics_loss": 0.7004157900810242
    },
    {
      "step": 9840,
      "train_flow_matching_loss": 0.10888671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 9840,
      "train_flow_matching_loss": 0.07470703125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08997088670730591,
      "learning_rate": 5e-05,
      "loss": 0.1634,
      "step": 9850
    },
    {
      "step": 9850,
      "train_flow_matching_loss": 0.1826171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 9850,
      "train_flow_matching_loss": 0.1162109375,
      "train_physics_loss": 0.682485044002533
    },
    {
      "step": 9850,
      "train_flow_matching_loss": 0.1611328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 9850,
      "train_flow_matching_loss": 0.126953125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10927104949951172,
      "learning_rate": 5e-05,
      "loss": 0.1547,
      "step": 9860
    },
    {
      "step": 9860,
      "train_flow_matching_loss": 0.1455078125,
      "train_physics_loss": 0.0
    },
    {
      "step": 9860,
      "train_flow_matching_loss": 0.08447265625,
      "train_physics_loss": 0.6953850984573364
    },
    {
      "step": 9860,
      "train_flow_matching_loss": 0.1650390625,
      "train_physics_loss": 0.0
    },
    {
      "step": 9860,
      "train_flow_matching_loss": 0.1552734375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10366059839725494,
      "learning_rate": 5e-05,
      "loss": 0.1504,
      "step": 9870
    },
    {
      "step": 9870,
      "train_flow_matching_loss": 0.1572265625,
      "train_physics_loss": 0.0
    },
    {
      "step": 9870,
      "train_flow_matching_loss": 0.0634765625,
      "train_physics_loss": 0.685246467590332
    },
    {
      "step": 9870,
      "train_flow_matching_loss": 0.1611328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 9870,
      "train_flow_matching_loss": 0.1298828125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.1030498743057251,
      "learning_rate": 5e-05,
      "loss": 0.1541,
      "step": 9880
    },
    {
      "step": 9880,
      "train_flow_matching_loss": 0.1435546875,
      "train_physics_loss": 0.0
    },
    {
      "step": 9880,
      "train_flow_matching_loss": 0.06787109375,
      "train_physics_loss": 0.7121525406837463
    },
    {
      "step": 9880,
      "train_flow_matching_loss": 0.1826171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 9880,
      "train_flow_matching_loss": 0.1689453125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.13122855126857758,
      "learning_rate": 5e-05,
      "loss": 0.1557,
      "step": 9890
    },
    {
      "step": 9890,
      "train_flow_matching_loss": 0.2060546875,
      "train_physics_loss": 0.0
    },
    {
      "step": 9890,
      "train_flow_matching_loss": 0.0673828125,
      "train_physics_loss": 0.6981884241104126
    },
    {
      "step": 9890,
      "train_flow_matching_loss": 0.1416015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 9890,
      "train_flow_matching_loss": 0.1533203125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09165263175964355,
      "learning_rate": 5e-05,
      "loss": 0.1565,
      "step": 9900
    },
    {
      "step": 9900,
      "train_flow_matching_loss": 0.1953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 9900,
      "train_flow_matching_loss": 0.057373046875,
      "train_physics_loss": 0.7074612975120544
    },
    {
      "step": 9900,
      "train_flow_matching_loss": 0.10986328125,
      "train_physics_loss": 0.7768580317497253
    },
    {
      "step": 9900,
      "train_flow_matching_loss": 0.09521484375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10506664216518402,
      "learning_rate": 5e-05,
      "loss": 0.1555,
      "step": 9910
    },
    {
      "step": 9910,
      "train_flow_matching_loss": 0.181640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 9910,
      "train_flow_matching_loss": 0.06982421875,
      "train_physics_loss": 0.6610649824142456
    },
    {
      "step": 9910,
      "train_flow_matching_loss": 0.162109375,
      "train_physics_loss": 0.7645537853240967
    },
    {
      "step": 9910,
      "train_flow_matching_loss": 0.166015625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.12491442263126373,
      "learning_rate": 5e-05,
      "loss": 0.1603,
      "step": 9920
    },
    {
      "step": 9920,
      "train_flow_matching_loss": 0.1396484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 9920,
      "train_flow_matching_loss": 0.07568359375,
      "train_physics_loss": 0.7337055802345276
    },
    {
      "step": 9920,
      "train_flow_matching_loss": 0.05029296875,
      "train_physics_loss": 0.6296170353889465
    },
    {
      "step": 9920,
      "train_flow_matching_loss": 0.1748046875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09821788221597672,
      "learning_rate": 5e-05,
      "loss": 0.1539,
      "step": 9930
    },
    {
      "step": 9930,
      "train_flow_matching_loss": 0.1513671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 9930,
      "train_flow_matching_loss": 0.0859375,
      "train_physics_loss": 0.7080762982368469
    },
    {
      "step": 9930,
      "train_flow_matching_loss": 0.06884765625,
      "train_physics_loss": 0.661867618560791
    },
    {
      "step": 9930,
      "train_flow_matching_loss": 0.208984375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09716085344552994,
      "learning_rate": 5e-05,
      "loss": 0.1632,
      "step": 9940
    },
    {
      "step": 9940,
      "train_flow_matching_loss": 0.1708984375,
      "train_physics_loss": 0.0
    },
    {
      "step": 9940,
      "train_flow_matching_loss": 0.060302734375,
      "train_physics_loss": 0.7150586247444153
    },
    {
      "step": 9940,
      "train_flow_matching_loss": 0.03955078125,
      "train_physics_loss": 0.6456136703491211
    },
    {
      "step": 9940,
      "train_flow_matching_loss": 0.1865234375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08999879658222198,
      "learning_rate": 5e-05,
      "loss": 0.1597,
      "step": 9950
    },
    {
      "step": 9950,
      "train_flow_matching_loss": 0.1865234375,
      "train_physics_loss": 0.0
    },
    {
      "step": 9950,
      "train_flow_matching_loss": 0.08935546875,
      "train_physics_loss": 0.6261997818946838
    },
    {
      "step": 9950,
      "train_flow_matching_loss": 0.030517578125,
      "train_physics_loss": 0.7090020775794983
    },
    {
      "step": 9950,
      "train_flow_matching_loss": 0.1953125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08589662611484528,
      "learning_rate": 5e-05,
      "loss": 0.1554,
      "step": 9960
    },
    {
      "step": 9960,
      "train_flow_matching_loss": 0.1865234375,
      "train_physics_loss": 0.0
    },
    {
      "step": 9960,
      "train_flow_matching_loss": 0.08935546875,
      "train_physics_loss": 0.6117346286773682
    },
    {
      "step": 9960,
      "train_flow_matching_loss": 0.04443359375,
      "train_physics_loss": 0.8006846308708191
    },
    {
      "step": 9960,
      "train_flow_matching_loss": 0.11279296875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.0989043265581131,
      "learning_rate": 5e-05,
      "loss": 0.1521,
      "step": 9970
    },
    {
      "step": 9970,
      "train_flow_matching_loss": 0.1044921875,
      "train_physics_loss": 0.0
    },
    {
      "step": 9970,
      "train_flow_matching_loss": 0.08447265625,
      "train_physics_loss": 0.8241479992866516
    },
    {
      "step": 9970,
      "train_flow_matching_loss": 0.07763671875,
      "train_physics_loss": 0.5649765729904175
    },
    {
      "step": 9970,
      "train_flow_matching_loss": 0.11279296875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.1618122160434723,
      "learning_rate": 5e-05,
      "loss": 0.1495,
      "step": 9980
    },
    {
      "step": 9980,
      "train_flow_matching_loss": 0.2197265625,
      "train_physics_loss": 0.0
    },
    {
      "step": 9980,
      "train_flow_matching_loss": 0.053466796875,
      "train_physics_loss": 0.7671714425086975
    },
    {
      "step": 9980,
      "train_flow_matching_loss": 0.056884765625,
      "train_physics_loss": 0.680380642414093
    },
    {
      "step": 9980,
      "train_flow_matching_loss": 0.150390625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11153916269540787,
      "learning_rate": 5e-05,
      "loss": 0.1572,
      "step": 9990
    },
    {
      "step": 9990,
      "train_flow_matching_loss": 0.12158203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 9990,
      "train_flow_matching_loss": 0.1328125,
      "train_physics_loss": 0.7288480401039124
    },
    {
      "step": 9990,
      "train_flow_matching_loss": 0.0296630859375,
      "train_physics_loss": 0.7150623202323914
    },
    {
      "step": 9990,
      "train_flow_matching_loss": 0.1826171875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11827754229307175,
      "learning_rate": 5e-05,
      "loss": 0.1576,
      "step": 10000
    },
    {
      "step": 10000,
      "train_flow_matching_loss": 0.166015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 10000,
      "train_flow_matching_loss": 0.1259765625,
      "train_physics_loss": 0.6912809014320374
    },
    {
      "step": 10000,
      "train_flow_matching_loss": 0.10400390625,
      "train_physics_loss": 0.6328630447387695
    },
    {
      "step": 10000,
      "train_flow_matching_loss": 0.1669921875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09402474015951157,
      "learning_rate": 5e-05,
      "loss": 0.1586,
      "step": 10010
    },
    {
      "step": 10010,
      "train_flow_matching_loss": 0.09326171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 10010,
      "train_flow_matching_loss": 0.07666015625,
      "train_physics_loss": 0.8642578125
    },
    {
      "step": 10010,
      "train_flow_matching_loss": 0.056640625,
      "train_physics_loss": 0.7497150301933289
    },
    {
      "step": 10010,
      "train_flow_matching_loss": 0.142578125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.1352216601371765,
      "learning_rate": 5e-05,
      "loss": 0.154,
      "step": 10020
    },
    {
      "step": 10020,
      "train_flow_matching_loss": 0.173828125,
      "train_physics_loss": 0.0
    },
    {
      "step": 10020,
      "train_flow_matching_loss": 0.045166015625,
      "train_physics_loss": 0.7514991164207458
    },
    {
      "step": 10020,
      "train_flow_matching_loss": 0.09130859375,
      "train_physics_loss": 0.7378132939338684
    },
    {
      "step": 10020,
      "train_flow_matching_loss": 0.1923828125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08776354044675827,
      "learning_rate": 5e-05,
      "loss": 0.153,
      "step": 10030
    },
    {
      "step": 10030,
      "train_flow_matching_loss": 0.162109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 10030,
      "train_flow_matching_loss": 0.06787109375,
      "train_physics_loss": 0.6894251108169556
    },
    {
      "step": 10030,
      "train_flow_matching_loss": 0.06494140625,
      "train_physics_loss": 0.7340836524963379
    },
    {
      "step": 10030,
      "train_flow_matching_loss": 0.140625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.07523960620164871,
      "learning_rate": 5e-05,
      "loss": 0.1568,
      "step": 10040
    },
    {
      "step": 10040,
      "train_flow_matching_loss": 0.1435546875,
      "train_physics_loss": 0.0
    },
    {
      "step": 10040,
      "train_flow_matching_loss": 0.08251953125,
      "train_physics_loss": 0.7251020669937134
    },
    {
      "step": 10040,
      "train_flow_matching_loss": 0.09228515625,
      "train_physics_loss": 0.7148847579956055
    },
    {
      "step": 10040,
      "train_flow_matching_loss": 0.1796875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.1010768935084343,
      "learning_rate": 5e-05,
      "loss": 0.1538,
      "step": 10050
    },
    {
      "step": 10050,
      "train_flow_matching_loss": 0.1552734375,
      "train_physics_loss": 0.0
    },
    {
      "step": 10050,
      "train_flow_matching_loss": 0.0859375,
      "train_physics_loss": 0.7237282395362854
    },
    {
      "step": 10050,
      "train_flow_matching_loss": 0.11376953125,
      "train_physics_loss": 0.6744272112846375
    },
    {
      "step": 10050,
      "train_flow_matching_loss": 0.1328125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09929423779249191,
      "learning_rate": 5e-05,
      "loss": 0.1543,
      "step": 10060
    },
    {
      "step": 10060,
      "train_flow_matching_loss": 0.1923828125,
      "train_physics_loss": 0.0
    },
    {
      "step": 10060,
      "train_flow_matching_loss": 0.15625,
      "train_physics_loss": 0.0
    },
    {
      "step": 10060,
      "train_flow_matching_loss": 0.033447265625,
      "train_physics_loss": 0.6465128660202026
    },
    {
      "step": 10060,
      "train_flow_matching_loss": 0.1796875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11591658741235733,
      "learning_rate": 5e-05,
      "loss": 0.1565,
      "step": 10070
    },
    {
      "step": 10070,
      "train_flow_matching_loss": 0.1650390625,
      "train_physics_loss": 0.0
    },
    {
      "step": 10070,
      "train_flow_matching_loss": 0.259765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 10070,
      "train_flow_matching_loss": 0.048583984375,
      "train_physics_loss": 0.7283834218978882
    },
    {
      "step": 10070,
      "train_flow_matching_loss": 0.1806640625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09212633967399597,
      "learning_rate": 5e-05,
      "loss": 0.1544,
      "step": 10080
    },
    {
      "step": 10080,
      "train_flow_matching_loss": 0.248046875,
      "train_physics_loss": 0.0
    },
    {
      "step": 10080,
      "train_flow_matching_loss": 0.1572265625,
      "train_physics_loss": 0.0
    },
    {
      "step": 10080,
      "train_flow_matching_loss": 0.055419921875,
      "train_physics_loss": 0.7088516354560852
    },
    {
      "step": 10080,
      "train_flow_matching_loss": 0.1787109375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.6117150783538818,
      "learning_rate": 5e-05,
      "loss": 0.1537,
      "step": 10090
    },
    {
      "step": 10090,
      "train_flow_matching_loss": 0.1357421875,
      "train_physics_loss": 0.0
    },
    {
      "step": 10090,
      "train_flow_matching_loss": 0.11279296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 10090,
      "train_flow_matching_loss": 0.052001953125,
      "train_physics_loss": 0.7823362350463867
    },
    {
      "step": 10090,
      "train_flow_matching_loss": 0.1376953125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08353642374277115,
      "learning_rate": 5e-05,
      "loss": 0.1557,
      "step": 10100
    },
    {
      "step": 10100,
      "train_flow_matching_loss": 0.287109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 10100,
      "train_flow_matching_loss": 0.1484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 10100,
      "train_flow_matching_loss": 0.06787109375,
      "train_physics_loss": 0.7608223557472229
    },
    {
      "step": 10100,
      "train_flow_matching_loss": 0.1103515625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11561872065067291,
      "learning_rate": 5e-05,
      "loss": 0.1557,
      "step": 10110
    },
    {
      "step": 10110,
      "train_flow_matching_loss": 0.12451171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 10110,
      "train_flow_matching_loss": 0.193359375,
      "train_physics_loss": 0.0
    },
    {
      "step": 10110,
      "train_flow_matching_loss": 0.07763671875,
      "train_physics_loss": 0.6385045051574707
    },
    {
      "step": 10110,
      "train_flow_matching_loss": 0.1669921875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11595352739095688,
      "learning_rate": 5e-05,
      "loss": 0.1547,
      "step": 10120
    },
    {
      "step": 10120,
      "train_flow_matching_loss": 0.1337890625,
      "train_physics_loss": 0.0
    },
    {
      "step": 10120,
      "train_flow_matching_loss": 0.162109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 10120,
      "train_flow_matching_loss": 0.058349609375,
      "train_physics_loss": 0.8117828369140625
    },
    {
      "step": 10120,
      "train_flow_matching_loss": 0.1298828125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08193749189376831,
      "learning_rate": 5e-05,
      "loss": 0.1526,
      "step": 10130
    },
    {
      "step": 10130,
      "train_flow_matching_loss": 0.1640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 10130,
      "train_flow_matching_loss": 0.1669921875,
      "train_physics_loss": 0.0
    },
    {
      "step": 10130,
      "train_flow_matching_loss": 0.062255859375,
      "train_physics_loss": 0.7379646301269531
    },
    {
      "step": 10130,
      "train_flow_matching_loss": 0.1376953125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11951009184122086,
      "learning_rate": 5e-05,
      "loss": 0.152,
      "step": 10140
    },
    {
      "step": 10140,
      "train_flow_matching_loss": 0.08740234375,
      "train_physics_loss": 0.0
    },
    {
      "step": 10140,
      "train_flow_matching_loss": 0.1728515625,
      "train_physics_loss": 0.0
    },
    {
      "step": 10140,
      "train_flow_matching_loss": 0.06298828125,
      "train_physics_loss": 0.6760740280151367
    },
    {
      "step": 10140,
      "train_flow_matching_loss": 0.177734375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08073021471500397,
      "learning_rate": 5e-05,
      "loss": 0.1529,
      "step": 10150
    },
    {
      "step": 10150,
      "train_flow_matching_loss": 0.177734375,
      "train_physics_loss": 0.0
    },
    {
      "step": 10150,
      "train_flow_matching_loss": 0.1279296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 10150,
      "train_flow_matching_loss": 0.06396484375,
      "train_physics_loss": 0.7633816599845886
    },
    {
      "step": 10150,
      "train_flow_matching_loss": 0.236328125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08498774468898773,
      "learning_rate": 5e-05,
      "loss": 0.1512,
      "step": 10160
    },
    {
      "step": 10160,
      "train_flow_matching_loss": 0.2080078125,
      "train_physics_loss": 0.0
    },
    {
      "step": 10160,
      "train_flow_matching_loss": 0.1484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 10160,
      "train_flow_matching_loss": 0.1787109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 10160,
      "train_flow_matching_loss": 0.07080078125,
      "train_physics_loss": 0.7454479932785034
    },
    {
      "grad_norm": 0.12304313480854034,
      "learning_rate": 5e-05,
      "loss": 0.1534,
      "step": 10170
    },
    {
      "step": 10170,
      "train_flow_matching_loss": 0.1337890625,
      "train_physics_loss": 0.0
    },
    {
      "step": 10170,
      "train_flow_matching_loss": 0.1552734375,
      "train_physics_loss": 0.0
    },
    {
      "step": 10170,
      "train_flow_matching_loss": 0.1298828125,
      "train_physics_loss": 0.0
    },
    {
      "step": 10170,
      "train_flow_matching_loss": 0.06884765625,
      "train_physics_loss": 0.7065111398696899
    },
    {
      "grad_norm": 0.10807237029075623,
      "learning_rate": 5e-05,
      "loss": 0.1546,
      "step": 10180
    },
    {
      "step": 10180,
      "train_flow_matching_loss": 0.1728515625,
      "train_physics_loss": 0.0
    },
    {
      "step": 10180,
      "train_flow_matching_loss": 0.142578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 10180,
      "train_flow_matching_loss": 0.1455078125,
      "train_physics_loss": 0.0
    },
    {
      "step": 10180,
      "train_flow_matching_loss": 0.044189453125,
      "train_physics_loss": 0.62296062707901
    },
    {
      "grad_norm": 0.17246127128601074,
      "learning_rate": 5e-05,
      "loss": 0.1575,
      "step": 10190
    },
    {
      "step": 10190,
      "train_flow_matching_loss": 0.1337890625,
      "train_physics_loss": 0.0
    },
    {
      "step": 10190,
      "train_flow_matching_loss": 0.1611328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 10190,
      "train_flow_matching_loss": 0.2080078125,
      "train_physics_loss": 0.0
    },
    {
      "step": 10190,
      "train_flow_matching_loss": 0.07861328125,
      "train_physics_loss": 0.7289263606071472
    },
    {
      "grad_norm": 0.14037837088108063,
      "learning_rate": 5e-05,
      "loss": 0.1521,
      "step": 10200
    },
    {
      "step": 10200,
      "train_flow_matching_loss": 0.171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 10200,
      "train_flow_matching_loss": 0.11669921875,
      "train_physics_loss": 0.0
    },
    {
      "step": 10200,
      "train_flow_matching_loss": 0.1982421875,
      "train_physics_loss": 0.0
    },
    {
      "step": 10200,
      "train_flow_matching_loss": 0.06396484375,
      "train_physics_loss": 0.6890622973442078
    },
    {
      "grad_norm": 0.14972245693206787,
      "learning_rate": 5e-05,
      "loss": 0.1562,
      "step": 10210
    },
    {
      "step": 10210,
      "train_flow_matching_loss": 0.173828125,
      "train_physics_loss": 0.0
    },
    {
      "step": 10210,
      "train_flow_matching_loss": 0.1416015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 10210,
      "train_flow_matching_loss": 0.1875,
      "train_physics_loss": 0.0
    },
    {
      "step": 10210,
      "train_flow_matching_loss": 0.052001953125,
      "train_physics_loss": 0.7576966881752014
    },
    {
      "grad_norm": 0.09901165962219238,
      "learning_rate": 5e-05,
      "loss": 0.1544,
      "step": 10220
    },
    {
      "step": 10220,
      "train_flow_matching_loss": 0.1875,
      "train_physics_loss": 0.0
    },
    {
      "step": 10220,
      "train_flow_matching_loss": 0.1865234375,
      "train_physics_loss": 0.0
    },
    {
      "step": 10220,
      "train_flow_matching_loss": 0.10107421875,
      "train_physics_loss": 0.0
    },
    {
      "step": 10220,
      "train_flow_matching_loss": 0.0673828125,
      "train_physics_loss": 0.6786337494850159
    },
    {
      "grad_norm": 0.4943019449710846,
      "learning_rate": 5e-05,
      "loss": 0.1596,
      "step": 10230
    },
    {
      "step": 10230,
      "train_flow_matching_loss": 0.1845703125,
      "train_physics_loss": 0.0
    },
    {
      "step": 10230,
      "train_flow_matching_loss": 0.09228515625,
      "train_physics_loss": 0.0
    },
    {
      "step": 10230,
      "train_flow_matching_loss": 0.1318359375,
      "train_physics_loss": 0.0
    },
    {
      "step": 10230,
      "train_flow_matching_loss": 0.0791015625,
      "train_physics_loss": 0.6143622994422913
    },
    {
      "grad_norm": 0.11657460033893585,
      "learning_rate": 5e-05,
      "loss": 0.1557,
      "step": 10240
    },
    {
      "step": 10240,
      "train_flow_matching_loss": 0.142578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 10240,
      "train_flow_matching_loss": 0.1083984375,
      "train_physics_loss": 0.0
    },
    {
      "step": 10240,
      "train_flow_matching_loss": 0.2255859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 10240,
      "train_flow_matching_loss": 0.05615234375,
      "train_physics_loss": 0.7574293613433838
    },
    {
      "grad_norm": 0.09428001940250397,
      "learning_rate": 5e-05,
      "loss": 0.1515,
      "step": 10250
    },
    {
      "step": 10250,
      "train_flow_matching_loss": 0.1455078125,
      "train_physics_loss": 0.0
    },
    {
      "step": 10250,
      "train_flow_matching_loss": 0.1201171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 10250,
      "train_flow_matching_loss": 0.2041015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 10250,
      "train_flow_matching_loss": 0.06298828125,
      "train_physics_loss": 0.6985786557197571
    },
    {
      "grad_norm": 0.09930926561355591,
      "learning_rate": 5e-05,
      "loss": 0.1579,
      "step": 10260
    },
    {
      "step": 10260,
      "train_flow_matching_loss": 0.2119140625,
      "train_physics_loss": 0.0
    },
    {
      "step": 10260,
      "train_flow_matching_loss": 0.2001953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 10260,
      "train_flow_matching_loss": 0.1982421875,
      "train_physics_loss": 0.0
    },
    {
      "step": 10260,
      "train_flow_matching_loss": 0.09033203125,
      "train_physics_loss": 0.7321757674217224
    },
    {
      "grad_norm": 0.10198968648910522,
      "learning_rate": 5e-05,
      "loss": 0.1526,
      "step": 10270
    },
    {
      "step": 10270,
      "train_flow_matching_loss": 0.1591796875,
      "train_physics_loss": 0.0
    },
    {
      "step": 10270,
      "train_flow_matching_loss": 0.1611328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 10270,
      "train_flow_matching_loss": 0.1650390625,
      "train_physics_loss": 0.0
    },
    {
      "step": 10270,
      "train_flow_matching_loss": 0.1142578125,
      "train_physics_loss": 0.7650066614151001
    },
    {
      "grad_norm": 0.15474222600460052,
      "learning_rate": 5e-05,
      "loss": 0.164,
      "step": 10280
    },
    {
      "step": 10280,
      "train_flow_matching_loss": 0.1455078125,
      "train_physics_loss": 0.0
    },
    {
      "step": 10280,
      "train_flow_matching_loss": 0.1474609375,
      "train_physics_loss": 0.0
    },
    {
      "step": 10280,
      "train_flow_matching_loss": 0.1689453125,
      "train_physics_loss": 0.0
    },
    {
      "step": 10280,
      "train_flow_matching_loss": 0.07373046875,
      "train_physics_loss": 0.7483633756637573
    },
    {
      "grad_norm": 0.11868631094694138,
      "learning_rate": 5e-05,
      "loss": 0.1598,
      "step": 10290
    },
    {
      "step": 10290,
      "train_flow_matching_loss": 0.220703125,
      "train_physics_loss": 0.0
    },
    {
      "step": 10290,
      "train_flow_matching_loss": 0.1181640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 10290,
      "train_flow_matching_loss": 0.1748046875,
      "train_physics_loss": 0.0
    },
    {
      "step": 10290,
      "train_flow_matching_loss": 0.08740234375,
      "train_physics_loss": 0.671788215637207
    },
    {
      "grad_norm": 0.0912327691912651,
      "learning_rate": 5e-05,
      "loss": 0.1521,
      "step": 10300
    },
    {
      "step": 10300,
      "train_flow_matching_loss": 0.09716796875,
      "train_physics_loss": 0.0
    },
    {
      "step": 10300,
      "train_flow_matching_loss": 0.162109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 10300,
      "train_flow_matching_loss": 0.1494140625,
      "train_physics_loss": 0.0
    },
    {
      "step": 10300,
      "train_flow_matching_loss": 0.056640625,
      "train_physics_loss": 0.7625473737716675
    },
    {
      "grad_norm": 0.11266638338565826,
      "learning_rate": 5e-05,
      "loss": 0.1517,
      "step": 10310
    },
    {
      "step": 10310,
      "train_flow_matching_loss": 0.1318359375,
      "train_physics_loss": 0.0
    },
    {
      "step": 10310,
      "train_flow_matching_loss": 0.08349609375,
      "train_physics_loss": 0.0
    },
    {
      "step": 10310,
      "train_flow_matching_loss": 0.1943359375,
      "train_physics_loss": 0.0
    },
    {
      "step": 10310,
      "train_flow_matching_loss": 0.06689453125,
      "train_physics_loss": 0.671381413936615
    },
    {
      "grad_norm": 0.06162134185433388,
      "learning_rate": 5e-05,
      "loss": 0.1537,
      "step": 10320
    },
    {
      "step": 10320,
      "train_flow_matching_loss": 0.10400390625,
      "train_physics_loss": 0.0
    },
    {
      "step": 10320,
      "train_flow_matching_loss": 0.1884765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 10320,
      "train_flow_matching_loss": 0.2216796875,
      "train_physics_loss": 0.0
    },
    {
      "step": 10320,
      "train_flow_matching_loss": 0.05419921875,
      "train_physics_loss": 0.6811701655387878
    },
    {
      "grad_norm": 0.5055870413780212,
      "learning_rate": 5e-05,
      "loss": 0.1555,
      "step": 10330
    },
    {
      "step": 10330,
      "train_flow_matching_loss": 0.11572265625,
      "train_physics_loss": 0.0
    },
    {
      "step": 10330,
      "train_flow_matching_loss": 0.0751953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 10330,
      "train_flow_matching_loss": 0.11767578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 10330,
      "train_flow_matching_loss": 0.0634765625,
      "train_physics_loss": 0.7132017612457275
    },
    {
      "grad_norm": 0.0986408069729805,
      "learning_rate": 5e-05,
      "loss": 0.1513,
      "step": 10340
    },
    {
      "step": 10340,
      "train_flow_matching_loss": 0.181640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 10340,
      "train_flow_matching_loss": 0.09814453125,
      "train_physics_loss": 0.0
    },
    {
      "step": 10340,
      "train_flow_matching_loss": 0.1767578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 10340,
      "train_flow_matching_loss": 0.10107421875,
      "train_physics_loss": 0.6768725514411926
    },
    {
      "grad_norm": 0.14288528263568878,
      "learning_rate": 5e-05,
      "loss": 0.1542,
      "step": 10350
    },
    {
      "step": 10350,
      "train_flow_matching_loss": 0.1484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 10350,
      "train_flow_matching_loss": 0.1181640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 10350,
      "train_flow_matching_loss": 0.2109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 10350,
      "train_flow_matching_loss": 0.0654296875,
      "train_physics_loss": 0.6592787504196167
    },
    {
      "grad_norm": 0.10748599469661713,
      "learning_rate": 5e-05,
      "loss": 0.1483,
      "step": 10360
    },
    {
      "step": 10360,
      "train_flow_matching_loss": 0.134765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 10360,
      "train_flow_matching_loss": 0.203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 10360,
      "train_flow_matching_loss": 0.126953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 10360,
      "train_flow_matching_loss": 0.09228515625,
      "train_physics_loss": 0.697324812412262
    },
    {
      "grad_norm": 0.09555666148662567,
      "learning_rate": 5e-05,
      "loss": 0.1568,
      "step": 10370
    },
    {
      "step": 10370,
      "train_flow_matching_loss": 0.1328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 10370,
      "train_flow_matching_loss": 0.1630859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 10370,
      "train_flow_matching_loss": 0.2431640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 10370,
      "train_flow_matching_loss": 0.059326171875,
      "train_physics_loss": 0.7112282514572144
    },
    {
      "grad_norm": 0.12341047823429108,
      "learning_rate": 5e-05,
      "loss": 0.1549,
      "step": 10380
    },
    {
      "step": 10380,
      "train_flow_matching_loss": 0.177734375,
      "train_physics_loss": 0.0
    },
    {
      "step": 10380,
      "train_flow_matching_loss": 0.185546875,
      "train_physics_loss": 0.0
    },
    {
      "step": 10380,
      "train_flow_matching_loss": 0.193359375,
      "train_physics_loss": 0.0
    },
    {
      "step": 10380,
      "train_flow_matching_loss": 0.050537109375,
      "train_physics_loss": 0.6746622323989868
    },
    {
      "grad_norm": 0.07854972034692764,
      "learning_rate": 5e-05,
      "loss": 0.1544,
      "step": 10390
    },
    {
      "step": 10390,
      "train_flow_matching_loss": 0.1767578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 10390,
      "train_flow_matching_loss": 0.166015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 10390,
      "train_flow_matching_loss": 0.203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 10390,
      "train_flow_matching_loss": 0.06103515625,
      "train_physics_loss": 0.649101734161377
    },
    {
      "grad_norm": 0.0893130749464035,
      "learning_rate": 5e-05,
      "loss": 0.1544,
      "step": 10400
    },
    {
      "step": 10400,
      "train_flow_matching_loss": 0.10595703125,
      "train_physics_loss": 0.0
    },
    {
      "step": 10400,
      "train_flow_matching_loss": 0.1552734375,
      "train_physics_loss": 0.0
    },
    {
      "step": 10400,
      "train_flow_matching_loss": 0.1416015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 10400,
      "train_flow_matching_loss": 0.142578125,
      "train_physics_loss": 0.7403524518013
    },
    {
      "grad_norm": 0.09192292392253876,
      "learning_rate": 5e-05,
      "loss": 0.1537,
      "step": 10410
    },
    {
      "step": 10410,
      "train_flow_matching_loss": 0.1591796875,
      "train_physics_loss": 0.0
    },
    {
      "step": 10410,
      "train_flow_matching_loss": 0.1298828125,
      "train_physics_loss": 0.0
    },
    {
      "step": 10410,
      "train_flow_matching_loss": 0.203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 10410,
      "train_flow_matching_loss": 0.0634765625,
      "train_physics_loss": 0.6980850696563721
    },
    {
      "grad_norm": 0.11878643184900284,
      "learning_rate": 5e-05,
      "loss": 0.1544,
      "step": 10420
    },
    {
      "step": 10420,
      "train_flow_matching_loss": 0.1279296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 10420,
      "train_flow_matching_loss": 0.1044921875,
      "train_physics_loss": 0.0
    },
    {
      "step": 10420,
      "train_flow_matching_loss": 0.1513671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 10420,
      "train_flow_matching_loss": 0.11767578125,
      "train_physics_loss": 0.6560471057891846
    },
    {
      "grad_norm": 0.13796134293079376,
      "learning_rate": 5e-05,
      "loss": 0.1506,
      "step": 10430
    },
    {
      "step": 10430,
      "train_flow_matching_loss": 0.1279296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 10430,
      "train_flow_matching_loss": 0.09326171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 10430,
      "train_flow_matching_loss": 0.2216796875,
      "train_physics_loss": 0.0
    },
    {
      "step": 10430,
      "train_flow_matching_loss": 0.1865234375,
      "train_physics_loss": 0.738429844379425
    },
    {
      "grad_norm": 0.1408456414937973,
      "learning_rate": 5e-05,
      "loss": 0.1566,
      "step": 10440
    },
    {
      "step": 10440,
      "train_flow_matching_loss": 0.1201171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 10440,
      "train_flow_matching_loss": 0.1806640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 10440,
      "train_flow_matching_loss": 0.10498046875,
      "train_physics_loss": 0.0
    },
    {
      "step": 10440,
      "train_flow_matching_loss": 0.10009765625,
      "train_physics_loss": 0.6939684152603149
    },
    {
      "grad_norm": 0.08226025849580765,
      "learning_rate": 5e-05,
      "loss": 0.155,
      "step": 10450
    },
    {
      "step": 10450,
      "train_flow_matching_loss": 0.1279296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 10450,
      "train_flow_matching_loss": 0.1728515625,
      "train_physics_loss": 0.0
    },
    {
      "step": 10450,
      "train_flow_matching_loss": 0.1796875,
      "train_physics_loss": 0.0
    },
    {
      "step": 10450,
      "train_flow_matching_loss": 0.1171875,
      "train_physics_loss": 0.6588352918624878
    },
    {
      "grad_norm": 0.09827858954668045,
      "learning_rate": 5e-05,
      "loss": 0.1528,
      "step": 10460
    },
    {
      "step": 10460,
      "train_flow_matching_loss": 0.185546875,
      "train_physics_loss": 0.0
    },
    {
      "step": 10460,
      "train_flow_matching_loss": 0.1572265625,
      "train_physics_loss": 0.0
    },
    {
      "step": 10460,
      "train_flow_matching_loss": 0.134765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 10460,
      "train_flow_matching_loss": 0.0634765625,
      "train_physics_loss": 0.6408481001853943
    },
    {
      "grad_norm": 0.2071797400712967,
      "learning_rate": 5e-05,
      "loss": 0.1526,
      "step": 10470
    },
    {
      "step": 10470,
      "train_flow_matching_loss": 0.154296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 10470,
      "train_flow_matching_loss": 0.11376953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 10470,
      "train_flow_matching_loss": 0.1611328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 10470,
      "train_flow_matching_loss": 0.049560546875,
      "train_physics_loss": 0.6366487741470337
    },
    {
      "grad_norm": 0.09729577600955963,
      "learning_rate": 5e-05,
      "loss": 0.1515,
      "step": 10480
    },
    {
      "step": 10480,
      "train_flow_matching_loss": 0.10791015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 10480,
      "train_flow_matching_loss": 0.0732421875,
      "train_physics_loss": 0.0
    },
    {
      "step": 10480,
      "train_flow_matching_loss": 0.0986328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 10480,
      "train_flow_matching_loss": 0.07373046875,
      "train_physics_loss": 0.6797226667404175
    },
    {
      "grad_norm": 0.10217031836509705,
      "learning_rate": 5e-05,
      "loss": 0.1549,
      "step": 10490
    },
    {
      "step": 10490,
      "train_flow_matching_loss": 0.1640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 10490,
      "train_flow_matching_loss": 0.1220703125,
      "train_physics_loss": 0.0
    },
    {
      "step": 10490,
      "train_flow_matching_loss": 0.11669921875,
      "train_physics_loss": 0.0
    },
    {
      "step": 10490,
      "train_flow_matching_loss": 0.05419921875,
      "train_physics_loss": 0.7092652320861816
    },
    {
      "grad_norm": 0.09898381680250168,
      "learning_rate": 5e-05,
      "loss": 0.1543,
      "step": 10500
    },
    {
      "step": 10500,
      "train_flow_matching_loss": 0.0830078125,
      "train_physics_loss": 0.0
    },
    {
      "step": 10500,
      "train_flow_matching_loss": 0.09716796875,
      "train_physics_loss": 0.0
    },
    {
      "step": 10500,
      "train_flow_matching_loss": 0.205078125,
      "train_physics_loss": 0.0
    },
    {
      "step": 10500,
      "train_flow_matching_loss": 0.08642578125,
      "train_physics_loss": 0.6846373081207275
    },
    {
      "grad_norm": 0.11705233156681061,
      "learning_rate": 5e-05,
      "loss": 0.1564,
      "step": 10510
    },
    {
      "step": 10510,
      "train_flow_matching_loss": 0.1240234375,
      "train_physics_loss": 0.0
    },
    {
      "step": 10510,
      "train_flow_matching_loss": 0.10791015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 10510,
      "train_flow_matching_loss": 0.1015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 10510,
      "train_flow_matching_loss": 0.1044921875,
      "train_physics_loss": 0.6391971707344055
    },
    {
      "grad_norm": 0.1175551638007164,
      "learning_rate": 5e-05,
      "loss": 0.1573,
      "step": 10520
    },
    {
      "step": 10520,
      "train_flow_matching_loss": 0.1748046875,
      "train_physics_loss": 0.0
    },
    {
      "step": 10520,
      "train_flow_matching_loss": 0.142578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 10520,
      "train_flow_matching_loss": 0.1396484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 10520,
      "train_flow_matching_loss": 0.05712890625,
      "train_physics_loss": 0.6786211729049683
    },
    {
      "grad_norm": 0.10026626288890839,
      "learning_rate": 5e-05,
      "loss": 0.1591,
      "step": 10530
    },
    {
      "step": 10530,
      "train_flow_matching_loss": 0.0830078125,
      "train_physics_loss": 0.0
    },
    {
      "step": 10530,
      "train_flow_matching_loss": 0.1435546875,
      "train_physics_loss": 0.0
    },
    {
      "step": 10530,
      "train_flow_matching_loss": 0.08544921875,
      "train_physics_loss": 0.0
    },
    {
      "step": 10530,
      "train_flow_matching_loss": 0.054443359375,
      "train_physics_loss": 0.6567326784133911
    },
    {
      "grad_norm": 0.11013572663068771,
      "learning_rate": 5e-05,
      "loss": 0.149,
      "step": 10540
    },
    {
      "step": 10540,
      "train_flow_matching_loss": 0.138671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 10540,
      "train_flow_matching_loss": 0.1572265625,
      "train_physics_loss": 0.0
    },
    {
      "step": 10540,
      "train_flow_matching_loss": 0.2158203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 10540,
      "train_flow_matching_loss": 0.06884765625,
      "train_physics_loss": 0.6704771518707275
    },
    {
      "grad_norm": 0.1045113056898117,
      "learning_rate": 5e-05,
      "loss": 0.1491,
      "step": 10550
    },
    {
      "step": 10550,
      "train_flow_matching_loss": 0.1953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 10550,
      "train_flow_matching_loss": 0.240234375,
      "train_physics_loss": 0.0
    },
    {
      "step": 10550,
      "train_flow_matching_loss": 0.1591796875,
      "train_physics_loss": 0.0
    },
    {
      "step": 10550,
      "train_flow_matching_loss": 0.0693359375,
      "train_physics_loss": 0.7332534193992615
    },
    {
      "grad_norm": 0.09295078366994858,
      "learning_rate": 5e-05,
      "loss": 0.15,
      "step": 10560
    },
    {
      "step": 10560,
      "train_flow_matching_loss": 0.1455078125,
      "train_physics_loss": 0.0
    },
    {
      "step": 10560,
      "train_flow_matching_loss": 0.10791015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 10560,
      "train_flow_matching_loss": 0.1103515625,
      "train_physics_loss": 0.0
    },
    {
      "step": 10560,
      "train_flow_matching_loss": 0.09912109375,
      "train_physics_loss": 0.6530518531799316
    },
    {
      "grad_norm": 0.1268429011106491,
      "learning_rate": 5e-05,
      "loss": 0.1518,
      "step": 10570
    },
    {
      "step": 10570,
      "train_flow_matching_loss": 0.15625,
      "train_physics_loss": 0.0
    },
    {
      "step": 10570,
      "train_flow_matching_loss": 0.09130859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 10570,
      "train_flow_matching_loss": 0.177734375,
      "train_physics_loss": 0.0
    },
    {
      "step": 10570,
      "train_flow_matching_loss": 0.0673828125,
      "train_physics_loss": 0.7724848985671997
    },
    {
      "grad_norm": 0.37192583084106445,
      "learning_rate": 5e-05,
      "loss": 0.1493,
      "step": 10580
    },
    {
      "step": 10580,
      "train_flow_matching_loss": 0.11474609375,
      "train_physics_loss": 0.0
    },
    {
      "step": 10580,
      "train_flow_matching_loss": 0.1572265625,
      "train_physics_loss": 0.0
    },
    {
      "step": 10580,
      "train_flow_matching_loss": 0.12890625,
      "train_physics_loss": 0.0
    },
    {
      "step": 10580,
      "train_flow_matching_loss": 0.0791015625,
      "train_physics_loss": 0.6657469868659973
    },
    {
      "grad_norm": 0.1295796036720276,
      "learning_rate": 5e-05,
      "loss": 0.1563,
      "step": 10590
    },
    {
      "step": 10590,
      "train_flow_matching_loss": 0.1494140625,
      "train_physics_loss": 0.0
    },
    {
      "step": 10590,
      "train_flow_matching_loss": 0.08056640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 10590,
      "train_flow_matching_loss": 0.126953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 10590,
      "train_flow_matching_loss": 0.10400390625,
      "train_physics_loss": 0.7738023996353149
    },
    {
      "grad_norm": 0.10455682873725891,
      "learning_rate": 5e-05,
      "loss": 0.1539,
      "step": 10600
    },
    {
      "step": 10600,
      "train_flow_matching_loss": 0.1357421875,
      "train_physics_loss": 0.0
    },
    {
      "step": 10600,
      "train_flow_matching_loss": 0.2021484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 10600,
      "train_flow_matching_loss": 0.1943359375,
      "train_physics_loss": 0.0
    },
    {
      "step": 10600,
      "train_flow_matching_loss": 0.0546875,
      "train_physics_loss": 0.6376603841781616
    },
    {
      "grad_norm": 0.12855491042137146,
      "learning_rate": 5e-05,
      "loss": 0.1536,
      "step": 10610
    },
    {
      "step": 10610,
      "train_flow_matching_loss": 0.1435546875,
      "train_physics_loss": 0.0
    },
    {
      "step": 10610,
      "train_flow_matching_loss": 0.0888671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 10610,
      "train_flow_matching_loss": 0.169921875,
      "train_physics_loss": 0.0
    },
    {
      "step": 10610,
      "train_flow_matching_loss": 0.08056640625,
      "train_physics_loss": 0.5845230221748352
    },
    {
      "grad_norm": 0.10182201862335205,
      "learning_rate": 5e-05,
      "loss": 0.1508,
      "step": 10620
    },
    {
      "step": 10620,
      "train_flow_matching_loss": 0.10595703125,
      "train_physics_loss": 0.0
    },
    {
      "step": 10620,
      "train_flow_matching_loss": 0.166015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 10620,
      "train_flow_matching_loss": 0.1689453125,
      "train_physics_loss": 0.0
    },
    {
      "step": 10620,
      "train_flow_matching_loss": 0.07861328125,
      "train_physics_loss": 0.7086322903633118
    },
    {
      "grad_norm": 0.12066491693258286,
      "learning_rate": 5e-05,
      "loss": 0.1571,
      "step": 10630
    },
    {
      "step": 10630,
      "train_flow_matching_loss": 0.1220703125,
      "train_physics_loss": 0.0
    },
    {
      "step": 10630,
      "train_flow_matching_loss": 0.1845703125,
      "train_physics_loss": 0.0
    },
    {
      "step": 10630,
      "train_flow_matching_loss": 0.09814453125,
      "train_physics_loss": 0.0
    },
    {
      "step": 10630,
      "train_flow_matching_loss": 0.09423828125,
      "train_physics_loss": 0.6907011866569519
    },
    {
      "grad_norm": 0.10639655590057373,
      "learning_rate": 5e-05,
      "loss": 0.1526,
      "step": 10640
    },
    {
      "step": 10640,
      "train_flow_matching_loss": 0.1572265625,
      "train_physics_loss": 0.0
    },
    {
      "step": 10640,
      "train_flow_matching_loss": 0.0986328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 10640,
      "train_flow_matching_loss": 0.1103515625,
      "train_physics_loss": 0.0
    },
    {
      "step": 10640,
      "train_flow_matching_loss": 0.058349609375,
      "train_physics_loss": 0.6554334759712219
    },
    {
      "grad_norm": 0.12207173556089401,
      "learning_rate": 5e-05,
      "loss": 0.1522,
      "step": 10650
    },
    {
      "step": 10650,
      "train_flow_matching_loss": 0.1318359375,
      "train_physics_loss": 0.0
    },
    {
      "step": 10650,
      "train_flow_matching_loss": 0.08349609375,
      "train_physics_loss": 0.0
    },
    {
      "step": 10650,
      "train_flow_matching_loss": 0.146484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 10650,
      "train_flow_matching_loss": 0.09033203125,
      "train_physics_loss": 0.6722708940505981
    },
    {
      "grad_norm": 0.12047018110752106,
      "learning_rate": 5e-05,
      "loss": 0.1526,
      "step": 10660
    },
    {
      "step": 10660,
      "train_flow_matching_loss": 0.11572265625,
      "train_physics_loss": 0.0
    },
    {
      "step": 10660,
      "train_flow_matching_loss": 0.2333984375,
      "train_physics_loss": 0.0
    },
    {
      "step": 10660,
      "train_flow_matching_loss": 0.134765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 10660,
      "train_flow_matching_loss": 0.05615234375,
      "train_physics_loss": 0.6809768080711365
    },
    {
      "grad_norm": 0.12335681170225143,
      "learning_rate": 5e-05,
      "loss": 0.1557,
      "step": 10670
    },
    {
      "step": 10670,
      "train_flow_matching_loss": 0.154296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 10670,
      "train_flow_matching_loss": 0.134765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 10670,
      "train_flow_matching_loss": 0.1416015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 10670,
      "train_flow_matching_loss": 0.2158203125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11665091663599014,
      "learning_rate": 5e-05,
      "loss": 0.1567,
      "step": 10680
    },
    {
      "step": 10680,
      "train_flow_matching_loss": 0.1181640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 10680,
      "train_flow_matching_loss": 0.0947265625,
      "train_physics_loss": 0.0
    },
    {
      "step": 10680,
      "train_flow_matching_loss": 0.11572265625,
      "train_physics_loss": 0.0
    },
    {
      "step": 10680,
      "train_flow_matching_loss": 0.1572265625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11929816752672195,
      "learning_rate": 5e-05,
      "loss": 0.1524,
      "step": 10690
    },
    {
      "step": 10690,
      "train_flow_matching_loss": 0.130859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 10690,
      "train_flow_matching_loss": 0.173828125,
      "train_physics_loss": 0.0
    },
    {
      "step": 10690,
      "train_flow_matching_loss": 0.1787109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 10690,
      "train_flow_matching_loss": 0.2060546875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.16224639117717743,
      "learning_rate": 5e-05,
      "loss": 0.1536,
      "step": 10700
    },
    {
      "step": 10700,
      "train_flow_matching_loss": 0.1845703125,
      "train_physics_loss": 0.0
    },
    {
      "step": 10700,
      "train_flow_matching_loss": 0.1123046875,
      "train_physics_loss": 0.0
    },
    {
      "step": 10700,
      "train_flow_matching_loss": 0.1474609375,
      "train_physics_loss": 0.0
    },
    {
      "step": 10700,
      "train_flow_matching_loss": 0.1591796875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08269358426332474,
      "learning_rate": 5e-05,
      "loss": 0.159,
      "step": 10710
    },
    {
      "step": 10710,
      "train_flow_matching_loss": 0.13671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 10710,
      "train_flow_matching_loss": 0.1513671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 10710,
      "train_flow_matching_loss": 0.07421875,
      "train_physics_loss": 0.0
    },
    {
      "step": 10710,
      "train_flow_matching_loss": 0.10009765625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.13056600093841553,
      "learning_rate": 5e-05,
      "loss": 0.149,
      "step": 10720
    },
    {
      "step": 10720,
      "train_flow_matching_loss": 0.1435546875,
      "train_physics_loss": 0.0
    },
    {
      "step": 10720,
      "train_flow_matching_loss": 0.1357421875,
      "train_physics_loss": 0.0
    },
    {
      "step": 10720,
      "train_flow_matching_loss": 0.10888671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 10720,
      "train_flow_matching_loss": 0.2119140625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.07354921847581863,
      "learning_rate": 5e-05,
      "loss": 0.1514,
      "step": 10730
    },
    {
      "step": 10730,
      "train_flow_matching_loss": 0.134765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 10730,
      "train_flow_matching_loss": 0.1455078125,
      "train_physics_loss": 0.0
    },
    {
      "step": 10730,
      "train_flow_matching_loss": 0.205078125,
      "train_physics_loss": 0.0
    },
    {
      "step": 10730,
      "train_flow_matching_loss": 0.1826171875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.407306045293808,
      "learning_rate": 5e-05,
      "loss": 0.1592,
      "step": 10740
    },
    {
      "step": 10740,
      "train_flow_matching_loss": 0.1318359375,
      "train_physics_loss": 0.0
    },
    {
      "step": 10740,
      "train_flow_matching_loss": 0.169921875,
      "train_physics_loss": 0.0
    },
    {
      "step": 10740,
      "train_flow_matching_loss": 0.1376953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 10740,
      "train_flow_matching_loss": 0.13671875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09071262925863266,
      "learning_rate": 5e-05,
      "loss": 0.1517,
      "step": 10750
    },
    {
      "step": 10750,
      "train_flow_matching_loss": 0.09912109375,
      "train_physics_loss": 0.6871161460876465
    },
    {
      "step": 10750,
      "train_flow_matching_loss": 0.1259765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 10750,
      "train_flow_matching_loss": 0.275390625,
      "train_physics_loss": 0.0
    },
    {
      "step": 10750,
      "train_flow_matching_loss": 0.130859375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.12506774067878723,
      "learning_rate": 5e-05,
      "loss": 0.1554,
      "step": 10760
    },
    {
      "step": 10760,
      "train_flow_matching_loss": 0.09228515625,
      "train_physics_loss": 0.6162019968032837
    },
    {
      "step": 10760,
      "train_flow_matching_loss": 0.1806640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 10760,
      "train_flow_matching_loss": 0.2001953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 10760,
      "train_flow_matching_loss": 0.1650390625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.34432560205459595,
      "learning_rate": 5e-05,
      "loss": 0.1532,
      "step": 10770
    },
    {
      "step": 10770,
      "train_flow_matching_loss": 0.06103515625,
      "train_physics_loss": 0.7097808122634888
    },
    {
      "step": 10770,
      "train_flow_matching_loss": 0.1982421875,
      "train_physics_loss": 0.0
    },
    {
      "step": 10770,
      "train_flow_matching_loss": 0.1396484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 10770,
      "train_flow_matching_loss": 0.1591796875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.06784665584564209,
      "learning_rate": 5e-05,
      "loss": 0.1522,
      "step": 10780
    },
    {
      "step": 10780,
      "train_flow_matching_loss": 0.1513671875,
      "train_physics_loss": 0.6311445236206055
    },
    {
      "step": 10780,
      "train_flow_matching_loss": 0.171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 10780,
      "train_flow_matching_loss": 0.1787109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 10780,
      "train_flow_matching_loss": 0.2294921875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.0821816697716713,
      "learning_rate": 5e-05,
      "loss": 0.1546,
      "step": 10790
    },
    {
      "step": 10790,
      "train_flow_matching_loss": 0.13671875,
      "train_physics_loss": 0.6738661527633667
    },
    {
      "step": 10790,
      "train_flow_matching_loss": 0.08447265625,
      "train_physics_loss": 0.0
    },
    {
      "step": 10790,
      "train_flow_matching_loss": 0.09423828125,
      "train_physics_loss": 0.0
    },
    {
      "step": 10790,
      "train_flow_matching_loss": 0.126953125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10272758454084396,
      "learning_rate": 5e-05,
      "loss": 0.154,
      "step": 10800
    },
    {
      "step": 10800,
      "train_flow_matching_loss": 0.05712890625,
      "train_physics_loss": 0.6349512338638306
    },
    {
      "step": 10800,
      "train_flow_matching_loss": 0.1513671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 10800,
      "train_flow_matching_loss": 0.138671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 10800,
      "train_flow_matching_loss": 0.10498046875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.13415364921092987,
      "learning_rate": 5e-05,
      "loss": 0.1573,
      "step": 10810
    },
    {
      "step": 10810,
      "train_flow_matching_loss": 0.07470703125,
      "train_physics_loss": 0.7445074319839478
    },
    {
      "step": 10810,
      "train_flow_matching_loss": 0.181640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 10810,
      "train_flow_matching_loss": 0.169921875,
      "train_physics_loss": 0.0
    },
    {
      "step": 10810,
      "train_flow_matching_loss": 0.11767578125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.0850294902920723,
      "learning_rate": 5e-05,
      "loss": 0.1587,
      "step": 10820
    },
    {
      "step": 10820,
      "train_flow_matching_loss": 0.08642578125,
      "train_physics_loss": 0.643181324005127
    },
    {
      "step": 10820,
      "train_flow_matching_loss": 0.07568359375,
      "train_physics_loss": 0.0
    },
    {
      "step": 10820,
      "train_flow_matching_loss": 0.318359375,
      "train_physics_loss": 0.0
    },
    {
      "step": 10820,
      "train_flow_matching_loss": 0.1298828125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.06863053888082504,
      "learning_rate": 5e-05,
      "loss": 0.1554,
      "step": 10830
    },
    {
      "step": 10830,
      "train_flow_matching_loss": 0.061279296875,
      "train_physics_loss": 0.6245204210281372
    },
    {
      "step": 10830,
      "train_flow_matching_loss": 0.169921875,
      "train_physics_loss": 0.0
    },
    {
      "step": 10830,
      "train_flow_matching_loss": 0.1181640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 10830,
      "train_flow_matching_loss": 0.1943359375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.40820562839508057,
      "learning_rate": 5e-05,
      "loss": 0.1568,
      "step": 10840
    },
    {
      "step": 10840,
      "train_flow_matching_loss": 0.162109375,
      "train_physics_loss": 0.7467575669288635
    },
    {
      "step": 10840,
      "train_flow_matching_loss": 0.11572265625,
      "train_physics_loss": 0.0
    },
    {
      "step": 10840,
      "train_flow_matching_loss": 0.1376953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 10840,
      "train_flow_matching_loss": 0.2041015625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.519730269908905,
      "learning_rate": 5e-05,
      "loss": 0.1505,
      "step": 10850
    },
    {
      "step": 10850,
      "train_flow_matching_loss": 0.03271484375,
      "train_physics_loss": 0.6600480079650879
    },
    {
      "step": 10850,
      "train_flow_matching_loss": 0.11572265625,
      "train_physics_loss": 0.0
    },
    {
      "step": 10850,
      "train_flow_matching_loss": 0.12109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 10850,
      "train_flow_matching_loss": 0.1298828125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08643652498722076,
      "learning_rate": 5e-05,
      "loss": 0.1517,
      "step": 10860
    },
    {
      "step": 10860,
      "train_flow_matching_loss": 0.0849609375,
      "train_physics_loss": 0.7093921303749084
    },
    {
      "step": 10860,
      "train_flow_matching_loss": 0.1171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 10860,
      "train_flow_matching_loss": 0.171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 10860,
      "train_flow_matching_loss": 0.10498046875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.0998479351401329,
      "learning_rate": 5e-05,
      "loss": 0.1506,
      "step": 10870
    },
    {
      "step": 10870,
      "train_flow_matching_loss": 0.050048828125,
      "train_physics_loss": 0.6701446771621704
    },
    {
      "step": 10870,
      "train_flow_matching_loss": 0.1474609375,
      "train_physics_loss": 0.0
    },
    {
      "step": 10870,
      "train_flow_matching_loss": 0.1357421875,
      "train_physics_loss": 0.0
    },
    {
      "step": 10870,
      "train_flow_matching_loss": 0.173828125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.0921853631734848,
      "learning_rate": 5e-05,
      "loss": 0.1499,
      "step": 10880
    },
    {
      "step": 10880,
      "train_flow_matching_loss": 0.056396484375,
      "train_physics_loss": 0.5826440453529358
    },
    {
      "step": 10880,
      "train_flow_matching_loss": 0.166015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 10880,
      "train_flow_matching_loss": 0.1923828125,
      "train_physics_loss": 0.0
    },
    {
      "step": 10880,
      "train_flow_matching_loss": 0.12353515625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10983406007289886,
      "learning_rate": 5e-05,
      "loss": 0.1565,
      "step": 10890
    },
    {
      "step": 10890,
      "train_flow_matching_loss": 0.0296630859375,
      "train_physics_loss": 0.6131038069725037
    },
    {
      "step": 10890,
      "train_flow_matching_loss": 0.10107421875,
      "train_physics_loss": 0.0
    },
    {
      "step": 10890,
      "train_flow_matching_loss": 0.109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 10890,
      "train_flow_matching_loss": 0.1259765625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.1414598524570465,
      "learning_rate": 5e-05,
      "loss": 0.1506,
      "step": 10900
    },
    {
      "step": 10900,
      "train_flow_matching_loss": 0.08642578125,
      "train_physics_loss": 0.6969965100288391
    },
    {
      "step": 10900,
      "train_flow_matching_loss": 0.2119140625,
      "train_physics_loss": 0.0
    },
    {
      "step": 10900,
      "train_flow_matching_loss": 0.1806640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 10900,
      "train_flow_matching_loss": 0.109375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10042062401771545,
      "learning_rate": 5e-05,
      "loss": 0.1539,
      "step": 10910
    },
    {
      "step": 10910,
      "train_flow_matching_loss": 0.0810546875,
      "train_physics_loss": 0.6186632513999939
    },
    {
      "step": 10910,
      "train_flow_matching_loss": 0.08447265625,
      "train_physics_loss": 0.0
    },
    {
      "step": 10910,
      "train_flow_matching_loss": 0.109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 10910,
      "train_flow_matching_loss": 0.1005859375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09365078061819077,
      "learning_rate": 5e-05,
      "loss": 0.1513,
      "step": 10920
    },
    {
      "step": 10920,
      "train_flow_matching_loss": 0.1083984375,
      "train_physics_loss": 0.7836071848869324
    },
    {
      "step": 10920,
      "train_flow_matching_loss": 0.2080078125,
      "train_physics_loss": 0.0
    },
    {
      "step": 10920,
      "train_flow_matching_loss": 0.06884765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 10920,
      "train_flow_matching_loss": 0.27734375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08846587687730789,
      "learning_rate": 5e-05,
      "loss": 0.1586,
      "step": 10930
    },
    {
      "step": 10930,
      "train_flow_matching_loss": 0.08349609375,
      "train_physics_loss": 0.6024545431137085
    },
    {
      "step": 10930,
      "train_flow_matching_loss": 0.12158203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 10930,
      "train_flow_matching_loss": 0.146484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 10930,
      "train_flow_matching_loss": 0.1552734375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09696148335933685,
      "learning_rate": 5e-05,
      "loss": 0.15,
      "step": 10940
    },
    {
      "step": 10940,
      "train_flow_matching_loss": 0.045654296875,
      "train_physics_loss": 0.6655479669570923
    },
    {
      "step": 10940,
      "train_flow_matching_loss": 0.1708984375,
      "train_physics_loss": 0.0
    },
    {
      "step": 10940,
      "train_flow_matching_loss": 0.189453125,
      "train_physics_loss": 0.0
    },
    {
      "step": 10940,
      "train_flow_matching_loss": 0.1396484375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.1009211540222168,
      "learning_rate": 5e-05,
      "loss": 0.1517,
      "step": 10950
    },
    {
      "step": 10950,
      "train_flow_matching_loss": 0.0654296875,
      "train_physics_loss": 0.6005021929740906
    },
    {
      "step": 10950,
      "train_flow_matching_loss": 0.0986328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 10950,
      "train_flow_matching_loss": 0.1689453125,
      "train_physics_loss": 0.0
    },
    {
      "step": 10950,
      "train_flow_matching_loss": 0.1357421875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.07945967465639114,
      "learning_rate": 5e-05,
      "loss": 0.1524,
      "step": 10960
    },
    {
      "step": 10960,
      "train_flow_matching_loss": 0.04931640625,
      "train_physics_loss": 0.6266781687736511
    },
    {
      "step": 10960,
      "train_flow_matching_loss": 0.1572265625,
      "train_physics_loss": 0.0
    },
    {
      "step": 10960,
      "train_flow_matching_loss": 0.146484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 10960,
      "train_flow_matching_loss": 0.2353515625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.0827263742685318,
      "learning_rate": 5e-05,
      "loss": 0.1541,
      "step": 10970
    },
    {
      "step": 10970,
      "train_flow_matching_loss": 0.1025390625,
      "train_physics_loss": 0.7207057476043701
    },
    {
      "step": 10970,
      "train_flow_matching_loss": 0.1376953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 10970,
      "train_flow_matching_loss": 0.10888671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 10970,
      "train_flow_matching_loss": 0.1923828125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10926368087530136,
      "learning_rate": 5e-05,
      "loss": 0.155,
      "step": 10980
    },
    {
      "step": 10980,
      "train_flow_matching_loss": 0.044189453125,
      "train_physics_loss": 0.7044110894203186
    },
    {
      "step": 10980,
      "train_flow_matching_loss": 0.1484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 10980,
      "train_flow_matching_loss": 0.1630859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 10980,
      "train_flow_matching_loss": 0.1591796875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.07645805180072784,
      "learning_rate": 5e-05,
      "loss": 0.1522,
      "step": 10990
    },
    {
      "step": 10990,
      "train_flow_matching_loss": 0.051025390625,
      "train_physics_loss": 0.7412557601928711
    },
    {
      "step": 10990,
      "train_flow_matching_loss": 0.130859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 10990,
      "train_flow_matching_loss": 0.201171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 10990,
      "train_flow_matching_loss": 0.203125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.07400606572628021,
      "learning_rate": 5e-05,
      "loss": 0.1538,
      "step": 11000
    },
    {
      "step": 11000,
      "train_flow_matching_loss": 0.04443359375,
      "train_physics_loss": 0.7155929803848267
    },
    {
      "step": 11000,
      "train_flow_matching_loss": 0.11767578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 11000,
      "train_flow_matching_loss": 0.1884765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 11000,
      "train_flow_matching_loss": 0.2138671875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08982660621404648,
      "learning_rate": 5e-05,
      "loss": 0.1505,
      "step": 11010
    },
    {
      "step": 11010,
      "train_flow_matching_loss": 0.1513671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 11010,
      "train_flow_matching_loss": 0.1787109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 11010,
      "train_flow_matching_loss": 0.228515625,
      "train_physics_loss": 0.0
    },
    {
      "step": 11010,
      "train_flow_matching_loss": 0.2236328125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09441836178302765,
      "learning_rate": 5e-05,
      "loss": 0.154,
      "step": 11020
    },
    {
      "step": 11020,
      "train_flow_matching_loss": 0.126953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 11020,
      "train_flow_matching_loss": 0.138671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 11020,
      "train_flow_matching_loss": 0.173828125,
      "train_physics_loss": 0.0
    },
    {
      "step": 11020,
      "train_flow_matching_loss": 0.1669921875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.0861303061246872,
      "learning_rate": 5e-05,
      "loss": 0.1536,
      "step": 11030
    },
    {
      "step": 11030,
      "train_flow_matching_loss": 0.1259765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 11030,
      "train_flow_matching_loss": 0.06201171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 11030,
      "train_flow_matching_loss": 0.1962890625,
      "train_physics_loss": 0.0
    },
    {
      "step": 11030,
      "train_flow_matching_loss": 0.1318359375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09348321706056595,
      "learning_rate": 5e-05,
      "loss": 0.1476,
      "step": 11040
    },
    {
      "step": 11040,
      "train_flow_matching_loss": 0.1572265625,
      "train_physics_loss": 0.0
    },
    {
      "step": 11040,
      "train_flow_matching_loss": 0.08642578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 11040,
      "train_flow_matching_loss": 0.10546875,
      "train_physics_loss": 0.0
    },
    {
      "step": 11040,
      "train_flow_matching_loss": 0.15625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11867357045412064,
      "learning_rate": 5e-05,
      "loss": 0.1518,
      "step": 11050
    },
    {
      "step": 11050,
      "train_flow_matching_loss": 0.08544921875,
      "train_physics_loss": 0.0
    },
    {
      "step": 11050,
      "train_flow_matching_loss": 0.1328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 11050,
      "train_flow_matching_loss": 0.154296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 11050,
      "train_flow_matching_loss": 0.2060546875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.07739751785993576,
      "learning_rate": 5e-05,
      "loss": 0.1481,
      "step": 11060
    },
    {
      "step": 11060,
      "train_flow_matching_loss": 0.1865234375,
      "train_physics_loss": 0.0
    },
    {
      "step": 11060,
      "train_flow_matching_loss": 0.1572265625,
      "train_physics_loss": 0.0
    },
    {
      "step": 11060,
      "train_flow_matching_loss": 0.1435546875,
      "train_physics_loss": 0.0
    },
    {
      "step": 11060,
      "train_flow_matching_loss": 0.162109375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11725882440805435,
      "learning_rate": 5e-05,
      "loss": 0.1579,
      "step": 11070
    },
    {
      "step": 11070,
      "train_flow_matching_loss": 0.255859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 11070,
      "train_flow_matching_loss": 0.1689453125,
      "train_physics_loss": 0.0
    },
    {
      "step": 11070,
      "train_flow_matching_loss": 0.2060546875,
      "train_physics_loss": 0.0
    },
    {
      "step": 11070,
      "train_flow_matching_loss": 0.205078125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.13517750799655914,
      "learning_rate": 5e-05,
      "loss": 0.1538,
      "step": 11080
    },
    {
      "step": 11080,
      "train_flow_matching_loss": 0.1552734375,
      "train_physics_loss": 0.0
    },
    {
      "step": 11080,
      "train_flow_matching_loss": 0.173828125,
      "train_physics_loss": 0.0
    },
    {
      "step": 11080,
      "train_flow_matching_loss": 0.171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 11080,
      "train_flow_matching_loss": 0.142578125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.15423209965229034,
      "learning_rate": 5e-05,
      "loss": 0.1504,
      "step": 11090
    },
    {
      "step": 11090,
      "train_flow_matching_loss": 0.1533203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 11090,
      "train_flow_matching_loss": 0.1396484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 11090,
      "train_flow_matching_loss": 0.1357421875,
      "train_physics_loss": 0.0
    },
    {
      "step": 11090,
      "train_flow_matching_loss": 0.1181640625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08506607264280319,
      "learning_rate": 5e-05,
      "loss": 0.1535,
      "step": 11100
    },
    {
      "step": 11100,
      "train_flow_matching_loss": 0.162109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 11100,
      "train_flow_matching_loss": 0.2041015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 11100,
      "train_flow_matching_loss": 0.11083984375,
      "train_physics_loss": 0.0
    },
    {
      "step": 11100,
      "train_flow_matching_loss": 0.1826171875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11027487367391586,
      "learning_rate": 5e-05,
      "loss": 0.16,
      "step": 11110
    },
    {
      "step": 11110,
      "train_flow_matching_loss": 0.232421875,
      "train_physics_loss": 0.0
    },
    {
      "step": 11110,
      "train_flow_matching_loss": 0.1416015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 11110,
      "train_flow_matching_loss": 0.1181640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 11110,
      "train_flow_matching_loss": 0.1201171875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11268500983715057,
      "learning_rate": 5e-05,
      "loss": 0.1514,
      "step": 11120
    },
    {
      "step": 11120,
      "train_flow_matching_loss": 0.10302734375,
      "train_physics_loss": 0.0
    },
    {
      "step": 11120,
      "train_flow_matching_loss": 0.1005859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 11120,
      "train_flow_matching_loss": 0.1962890625,
      "train_physics_loss": 0.0
    },
    {
      "step": 11120,
      "train_flow_matching_loss": 0.146484375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09783663600683212,
      "learning_rate": 5e-05,
      "loss": 0.1501,
      "step": 11130
    },
    {
      "step": 11130,
      "train_flow_matching_loss": 0.1484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 11130,
      "train_flow_matching_loss": 0.162109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 11130,
      "train_flow_matching_loss": 0.09521484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 11130,
      "train_flow_matching_loss": 0.11376953125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.114181749522686,
      "learning_rate": 5e-05,
      "loss": 0.1535,
      "step": 11140
    },
    {
      "step": 11140,
      "train_flow_matching_loss": 0.138671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 11140,
      "train_flow_matching_loss": 0.171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 11140,
      "train_flow_matching_loss": 0.169921875,
      "train_physics_loss": 0.0
    },
    {
      "step": 11140,
      "train_flow_matching_loss": 0.12060546875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09250496327877045,
      "learning_rate": 5e-05,
      "loss": 0.1582,
      "step": 11150
    },
    {
      "step": 11150,
      "train_flow_matching_loss": 0.1923828125,
      "train_physics_loss": 0.0
    },
    {
      "step": 11150,
      "train_flow_matching_loss": 0.08740234375,
      "train_physics_loss": 0.0
    },
    {
      "step": 11150,
      "train_flow_matching_loss": 0.1630859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 11150,
      "train_flow_matching_loss": 0.1875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.4408174753189087,
      "learning_rate": 5e-05,
      "loss": 0.1564,
      "step": 11160
    },
    {
      "step": 11160,
      "train_flow_matching_loss": 0.126953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 11160,
      "train_flow_matching_loss": 0.13671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 11160,
      "train_flow_matching_loss": 0.1962890625,
      "train_physics_loss": 0.0
    },
    {
      "step": 11160,
      "train_flow_matching_loss": 0.11669921875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.124453604221344,
      "learning_rate": 5e-05,
      "loss": 0.1557,
      "step": 11170
    },
    {
      "step": 11170,
      "train_flow_matching_loss": 0.1796875,
      "train_physics_loss": 0.0
    },
    {
      "step": 11170,
      "train_flow_matching_loss": 0.2255859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 11170,
      "train_flow_matching_loss": 0.1044921875,
      "train_physics_loss": 0.0
    },
    {
      "step": 11170,
      "train_flow_matching_loss": 0.2041015625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11377842724323273,
      "learning_rate": 5e-05,
      "loss": 0.1509,
      "step": 11180
    },
    {
      "step": 11180,
      "train_flow_matching_loss": 0.1865234375,
      "train_physics_loss": 0.0
    },
    {
      "step": 11180,
      "train_flow_matching_loss": 0.130859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 11180,
      "train_flow_matching_loss": 0.166015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 11180,
      "train_flow_matching_loss": 0.189453125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.2486741989850998,
      "learning_rate": 5e-05,
      "loss": 0.1551,
      "step": 11190
    },
    {
      "step": 11190,
      "train_flow_matching_loss": 0.169921875,
      "train_physics_loss": 0.0
    },
    {
      "step": 11190,
      "train_flow_matching_loss": 0.1953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 11190,
      "train_flow_matching_loss": 0.201171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 11190,
      "train_flow_matching_loss": 0.1640625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08824627846479416,
      "learning_rate": 5e-05,
      "loss": 0.1545,
      "step": 11200
    },
    {
      "step": 11200,
      "train_flow_matching_loss": 0.166015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 11200,
      "train_flow_matching_loss": 0.11474609375,
      "train_physics_loss": 0.0
    },
    {
      "step": 11200,
      "train_flow_matching_loss": 0.1513671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 11200,
      "train_flow_matching_loss": 0.162109375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.1059727668762207,
      "learning_rate": 5e-05,
      "loss": 0.1549,
      "step": 11210
    },
    {
      "step": 11210,
      "train_flow_matching_loss": 0.134765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 11210,
      "train_flow_matching_loss": 0.1416015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 11210,
      "train_flow_matching_loss": 0.158203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 11210,
      "train_flow_matching_loss": 0.10498046875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10597974807024002,
      "learning_rate": 5e-05,
      "loss": 0.1528,
      "step": 11220
    },
    {
      "step": 11220,
      "train_flow_matching_loss": 0.1962890625,
      "train_physics_loss": 0.0
    },
    {
      "step": 11220,
      "train_flow_matching_loss": 0.1572265625,
      "train_physics_loss": 0.0
    },
    {
      "step": 11220,
      "train_flow_matching_loss": 0.08544921875,
      "train_physics_loss": 0.0
    },
    {
      "step": 11220,
      "train_flow_matching_loss": 0.1474609375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11324918270111084,
      "learning_rate": 5e-05,
      "loss": 0.1542,
      "step": 11230
    },
    {
      "step": 11230,
      "train_flow_matching_loss": 0.1796875,
      "train_physics_loss": 0.0
    },
    {
      "step": 11230,
      "train_flow_matching_loss": 0.1494140625,
      "train_physics_loss": 0.0
    },
    {
      "step": 11230,
      "train_flow_matching_loss": 0.126953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 11230,
      "train_flow_matching_loss": 0.134765625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.18985934555530548,
      "learning_rate": 5e-05,
      "loss": 0.1525,
      "step": 11240
    },
    {
      "step": 11240,
      "train_flow_matching_loss": 0.1884765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 11240,
      "train_flow_matching_loss": 0.212890625,
      "train_physics_loss": 0.0
    },
    {
      "step": 11240,
      "train_flow_matching_loss": 0.1640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 11240,
      "train_flow_matching_loss": 0.1416015625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10937516391277313,
      "learning_rate": 5e-05,
      "loss": 0.1574,
      "step": 11250
    },
    {
      "step": 11250,
      "train_flow_matching_loss": 0.15625,
      "train_physics_loss": 0.0
    },
    {
      "step": 11250,
      "train_flow_matching_loss": 0.10888671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 11250,
      "train_flow_matching_loss": 0.0966796875,
      "train_physics_loss": 0.0
    },
    {
      "step": 11250,
      "train_flow_matching_loss": 0.1328125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10803784430027008,
      "learning_rate": 5e-05,
      "loss": 0.1549,
      "step": 11260
    },
    {
      "step": 11260,
      "train_flow_matching_loss": 0.138671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 11260,
      "train_flow_matching_loss": 0.1572265625,
      "train_physics_loss": 0.0
    },
    {
      "step": 11260,
      "train_flow_matching_loss": 0.150390625,
      "train_physics_loss": 0.0
    },
    {
      "step": 11260,
      "train_flow_matching_loss": 0.15625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10562417656183243,
      "learning_rate": 5e-05,
      "loss": 0.1517,
      "step": 11270
    },
    {
      "step": 11270,
      "train_flow_matching_loss": 0.1435546875,
      "train_physics_loss": 0.0
    },
    {
      "step": 11270,
      "train_flow_matching_loss": 0.1748046875,
      "train_physics_loss": 0.0
    },
    {
      "step": 11270,
      "train_flow_matching_loss": 0.1318359375,
      "train_physics_loss": 0.0
    },
    {
      "step": 11270,
      "train_flow_matching_loss": 0.1748046875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.07261224091053009,
      "learning_rate": 5e-05,
      "loss": 0.1535,
      "step": 11280
    },
    {
      "step": 11280,
      "train_flow_matching_loss": 0.138671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 11280,
      "train_flow_matching_loss": 0.2158203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 11280,
      "train_flow_matching_loss": 0.11962890625,
      "train_physics_loss": 0.0
    },
    {
      "step": 11280,
      "train_flow_matching_loss": 0.1220703125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.12251531332731247,
      "learning_rate": 5e-05,
      "loss": 0.1567,
      "step": 11290
    },
    {
      "step": 11290,
      "train_flow_matching_loss": 0.1748046875,
      "train_physics_loss": 0.0
    },
    {
      "step": 11290,
      "train_flow_matching_loss": 0.169921875,
      "train_physics_loss": 0.0
    },
    {
      "step": 11290,
      "train_flow_matching_loss": 0.1513671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 11290,
      "train_flow_matching_loss": 0.10791015625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08185406029224396,
      "learning_rate": 5e-05,
      "loss": 0.1558,
      "step": 11300
    },
    {
      "step": 11300,
      "train_flow_matching_loss": 0.1240234375,
      "train_physics_loss": 0.0
    },
    {
      "step": 11300,
      "train_flow_matching_loss": 0.1435546875,
      "train_physics_loss": 0.0
    },
    {
      "step": 11300,
      "train_flow_matching_loss": 0.12060546875,
      "train_physics_loss": 0.0
    },
    {
      "step": 11300,
      "train_flow_matching_loss": 0.10107421875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09729006886482239,
      "learning_rate": 5e-05,
      "loss": 0.1502,
      "step": 11310
    },
    {
      "step": 11310,
      "train_flow_matching_loss": 0.1220703125,
      "train_physics_loss": 0.0
    },
    {
      "step": 11310,
      "train_flow_matching_loss": 0.10693359375,
      "train_physics_loss": 0.0
    },
    {
      "step": 11310,
      "train_flow_matching_loss": 0.10302734375,
      "train_physics_loss": 0.0
    },
    {
      "step": 11310,
      "train_flow_matching_loss": 0.1640625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08999990671873093,
      "learning_rate": 5e-05,
      "loss": 0.1483,
      "step": 11320
    },
    {
      "step": 11320,
      "train_flow_matching_loss": 0.126953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 11320,
      "train_flow_matching_loss": 0.1455078125,
      "train_physics_loss": 0.0
    },
    {
      "step": 11320,
      "train_flow_matching_loss": 0.1796875,
      "train_physics_loss": 0.0
    },
    {
      "step": 11320,
      "train_flow_matching_loss": 0.09375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.13621282577514648,
      "learning_rate": 5e-05,
      "loss": 0.1507,
      "step": 11330
    },
    {
      "step": 11330,
      "train_flow_matching_loss": 0.12255859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 11330,
      "train_flow_matching_loss": 0.275390625,
      "train_physics_loss": 0.0
    },
    {
      "step": 11330,
      "train_flow_matching_loss": 0.1552734375,
      "train_physics_loss": 0.0
    },
    {
      "step": 11330,
      "train_flow_matching_loss": 0.146484375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.1286398023366928,
      "learning_rate": 5e-05,
      "loss": 0.1533,
      "step": 11340
    },
    {
      "step": 11340,
      "train_flow_matching_loss": 0.1650390625,
      "train_physics_loss": 0.0
    },
    {
      "step": 11340,
      "train_flow_matching_loss": 0.169921875,
      "train_physics_loss": 0.0
    },
    {
      "step": 11340,
      "train_flow_matching_loss": 0.2333984375,
      "train_physics_loss": 0.0
    },
    {
      "step": 11340,
      "train_flow_matching_loss": 0.185546875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11673038452863693,
      "learning_rate": 5e-05,
      "loss": 0.1475,
      "step": 11350
    },
    {
      "step": 11350,
      "train_flow_matching_loss": 0.0947265625,
      "train_physics_loss": 0.0
    },
    {
      "step": 11350,
      "train_flow_matching_loss": 0.2109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 11350,
      "train_flow_matching_loss": 0.1279296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 11350,
      "train_flow_matching_loss": 0.138671875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.07946759462356567,
      "learning_rate": 5e-05,
      "loss": 0.1504,
      "step": 11360
    },
    {
      "step": 11360,
      "train_flow_matching_loss": 0.1259765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 11360,
      "train_flow_matching_loss": 0.1435546875,
      "train_physics_loss": 0.0
    },
    {
      "step": 11360,
      "train_flow_matching_loss": 0.1982421875,
      "train_physics_loss": 0.0
    },
    {
      "step": 11360,
      "train_flow_matching_loss": 0.1796875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09517654031515121,
      "learning_rate": 5e-05,
      "loss": 0.1512,
      "step": 11370
    },
    {
      "step": 11370,
      "train_flow_matching_loss": 0.2236328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 11370,
      "train_flow_matching_loss": 0.205078125,
      "train_physics_loss": 0.0
    },
    {
      "step": 11370,
      "train_flow_matching_loss": 0.267578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 11370,
      "train_flow_matching_loss": 0.09814453125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.13077959418296814,
      "learning_rate": 5e-05,
      "loss": 0.1499,
      "step": 11380
    },
    {
      "step": 11380,
      "train_flow_matching_loss": 0.1064453125,
      "train_physics_loss": 0.0
    },
    {
      "step": 11380,
      "train_flow_matching_loss": 0.1533203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 11380,
      "train_flow_matching_loss": 0.1005859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 11380,
      "train_flow_matching_loss": 0.171875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08671097457408905,
      "learning_rate": 5e-05,
      "loss": 0.1466,
      "step": 11390
    },
    {
      "step": 11390,
      "train_flow_matching_loss": 0.1396484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 11390,
      "train_flow_matching_loss": 0.09375,
      "train_physics_loss": 0.0
    },
    {
      "step": 11390,
      "train_flow_matching_loss": 0.1201171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 11390,
      "train_flow_matching_loss": 0.1650390625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10167763382196426,
      "learning_rate": 5e-05,
      "loss": 0.1433,
      "step": 11400
    },
    {
      "step": 11400,
      "train_flow_matching_loss": 0.0986328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 11400,
      "train_flow_matching_loss": 0.03466796875,
      "train_physics_loss": 0.6654384732246399
    },
    {
      "step": 11400,
      "train_flow_matching_loss": 0.1435546875,
      "train_physics_loss": 0.0
    },
    {
      "step": 11400,
      "train_flow_matching_loss": 0.1572265625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.704250156879425,
      "learning_rate": 5e-05,
      "loss": 0.1524,
      "step": 11410
    },
    {
      "step": 11410,
      "train_flow_matching_loss": 0.12060546875,
      "train_physics_loss": 0.0
    },
    {
      "step": 11410,
      "train_flow_matching_loss": 0.06591796875,
      "train_physics_loss": 0.684070348739624
    },
    {
      "step": 11410,
      "train_flow_matching_loss": 0.2197265625,
      "train_physics_loss": 0.0
    },
    {
      "step": 11410,
      "train_flow_matching_loss": 0.1552734375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10168607532978058,
      "learning_rate": 5e-05,
      "loss": 0.1537,
      "step": 11420
    },
    {
      "step": 11420,
      "train_flow_matching_loss": 0.0810546875,
      "train_physics_loss": 0.0
    },
    {
      "step": 11420,
      "train_flow_matching_loss": 0.07421875,
      "train_physics_loss": 0.686951756477356
    },
    {
      "step": 11420,
      "train_flow_matching_loss": 0.11474609375,
      "train_physics_loss": 0.0
    },
    {
      "step": 11420,
      "train_flow_matching_loss": 0.166015625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08063144981861115,
      "learning_rate": 5e-05,
      "loss": 0.1493,
      "step": 11430
    },
    {
      "step": 11430,
      "train_flow_matching_loss": 0.1494140625,
      "train_physics_loss": 0.0
    },
    {
      "step": 11430,
      "train_flow_matching_loss": 0.0771484375,
      "train_physics_loss": 0.8558065295219421
    },
    {
      "step": 11430,
      "train_flow_matching_loss": 0.1533203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 11430,
      "train_flow_matching_loss": 0.1416015625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09007474035024643,
      "learning_rate": 5e-05,
      "loss": 0.1508,
      "step": 11440
    },
    {
      "step": 11440,
      "train_flow_matching_loss": 0.11669921875,
      "train_physics_loss": 0.0
    },
    {
      "step": 11440,
      "train_flow_matching_loss": 0.09130859375,
      "train_physics_loss": 0.6737698912620544
    },
    {
      "step": 11440,
      "train_flow_matching_loss": 0.1669921875,
      "train_physics_loss": 0.0
    },
    {
      "step": 11440,
      "train_flow_matching_loss": 0.1484375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09921795129776001,
      "learning_rate": 5e-05,
      "loss": 0.1499,
      "step": 11450
    },
    {
      "step": 11450,
      "train_flow_matching_loss": 0.171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 11450,
      "train_flow_matching_loss": 0.12353515625,
      "train_physics_loss": 0.6408408880233765
    },
    {
      "step": 11450,
      "train_flow_matching_loss": 0.1953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 11450,
      "train_flow_matching_loss": 0.177734375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11307654529809952,
      "learning_rate": 5e-05,
      "loss": 0.1574,
      "step": 11460
    },
    {
      "step": 11460,
      "train_flow_matching_loss": 0.162109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 11460,
      "train_flow_matching_loss": 0.0849609375,
      "train_physics_loss": 0.6061710715293884
    },
    {
      "step": 11460,
      "train_flow_matching_loss": 0.10546875,
      "train_physics_loss": 0.0
    },
    {
      "step": 11460,
      "train_flow_matching_loss": 0.1767578125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08772291988134384,
      "learning_rate": 5e-05,
      "loss": 0.1525,
      "step": 11470
    },
    {
      "step": 11470,
      "train_flow_matching_loss": 0.1396484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 11470,
      "train_flow_matching_loss": 0.1591796875,
      "train_physics_loss": 0.7703222036361694
    },
    {
      "step": 11470,
      "train_flow_matching_loss": 0.2080078125,
      "train_physics_loss": 0.0
    },
    {
      "step": 11470,
      "train_flow_matching_loss": 0.171875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10002238303422928,
      "learning_rate": 5e-05,
      "loss": 0.1525,
      "step": 11480
    },
    {
      "step": 11480,
      "train_flow_matching_loss": 0.12451171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 11480,
      "train_flow_matching_loss": 0.078125,
      "train_physics_loss": 0.6207637190818787
    },
    {
      "step": 11480,
      "train_flow_matching_loss": 0.259765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 11480,
      "train_flow_matching_loss": 0.1572265625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.12618473172187805,
      "learning_rate": 5e-05,
      "loss": 0.1504,
      "step": 11490
    },
    {
      "step": 11490,
      "train_flow_matching_loss": 0.146484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 11490,
      "train_flow_matching_loss": 0.046630859375,
      "train_physics_loss": 0.6776151061058044
    },
    {
      "step": 11490,
      "train_flow_matching_loss": 0.1171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 11490,
      "train_flow_matching_loss": 0.2451171875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11008479446172714,
      "learning_rate": 5e-05,
      "loss": 0.1515,
      "step": 11500
    },
    {
      "step": 11500,
      "train_flow_matching_loss": 0.1796875,
      "train_physics_loss": 0.0
    },
    {
      "step": 11500,
      "train_flow_matching_loss": 0.0888671875,
      "train_physics_loss": 0.6289188265800476
    },
    {
      "step": 11500,
      "train_flow_matching_loss": 0.138671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 11500,
      "train_flow_matching_loss": 0.177734375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.055701907724142075,
      "learning_rate": 5e-05,
      "loss": 0.1486,
      "step": 11510
    },
    {
      "step": 11510,
      "train_flow_matching_loss": 0.13671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 11510,
      "train_flow_matching_loss": 0.08447265625,
      "train_physics_loss": 0.6220194697380066
    },
    {
      "step": 11510,
      "train_flow_matching_loss": 0.1728515625,
      "train_physics_loss": 0.0
    },
    {
      "step": 11510,
      "train_flow_matching_loss": 0.150390625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09985820949077606,
      "learning_rate": 5e-05,
      "loss": 0.1511,
      "step": 11520
    },
    {
      "step": 11520,
      "train_flow_matching_loss": 0.1982421875,
      "train_physics_loss": 0.0
    },
    {
      "step": 11520,
      "train_flow_matching_loss": 0.05810546875,
      "train_physics_loss": 0.6645147800445557
    },
    {
      "step": 11520,
      "train_flow_matching_loss": 0.1689453125,
      "train_physics_loss": 0.0
    },
    {
      "step": 11520,
      "train_flow_matching_loss": 0.1513671875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.1654149293899536,
      "learning_rate": 5e-05,
      "loss": 0.1539,
      "step": 11530
    },
    {
      "step": 11530,
      "train_flow_matching_loss": 0.11474609375,
      "train_physics_loss": 0.0
    },
    {
      "step": 11530,
      "train_flow_matching_loss": 0.09375,
      "train_physics_loss": 0.6994760632514954
    },
    {
      "step": 11530,
      "train_flow_matching_loss": 0.162109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 11530,
      "train_flow_matching_loss": 0.1552734375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.0760331079363823,
      "learning_rate": 5e-05,
      "loss": 0.1509,
      "step": 11540
    },
    {
      "step": 11540,
      "train_flow_matching_loss": 0.09912109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 11540,
      "train_flow_matching_loss": 0.052490234375,
      "train_physics_loss": 0.6872183084487915
    },
    {
      "step": 11540,
      "train_flow_matching_loss": 0.1767578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 11540,
      "train_flow_matching_loss": 0.1962890625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.13836224377155304,
      "learning_rate": 5e-05,
      "loss": 0.156,
      "step": 11550
    },
    {
      "step": 11550,
      "train_flow_matching_loss": 0.07373046875,
      "train_physics_loss": 0.0
    },
    {
      "step": 11550,
      "train_flow_matching_loss": 0.10888671875,
      "train_physics_loss": 0.7016816139221191
    },
    {
      "step": 11550,
      "train_flow_matching_loss": 0.25390625,
      "train_physics_loss": 0.0
    },
    {
      "step": 11550,
      "train_flow_matching_loss": 0.185546875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.07477396726608276,
      "learning_rate": 5e-05,
      "loss": 0.1503,
      "step": 11560
    },
    {
      "step": 11560,
      "train_flow_matching_loss": 0.09130859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 11560,
      "train_flow_matching_loss": 0.0703125,
      "train_physics_loss": 0.6852140426635742
    },
    {
      "step": 11560,
      "train_flow_matching_loss": 0.1591796875,
      "train_physics_loss": 0.0
    },
    {
      "step": 11560,
      "train_flow_matching_loss": 0.1494140625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.5968763828277588,
      "learning_rate": 5e-05,
      "loss": 0.1484,
      "step": 11570
    },
    {
      "step": 11570,
      "train_flow_matching_loss": 0.189453125,
      "train_physics_loss": 0.0
    },
    {
      "step": 11570,
      "train_flow_matching_loss": 0.043212890625,
      "train_physics_loss": 0.5796712040901184
    },
    {
      "step": 11570,
      "train_flow_matching_loss": 0.2138671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 11570,
      "train_flow_matching_loss": 0.138671875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10058198124170303,
      "learning_rate": 5e-05,
      "loss": 0.147,
      "step": 11580
    },
    {
      "step": 11580,
      "train_flow_matching_loss": 0.1357421875,
      "train_physics_loss": 0.0
    },
    {
      "step": 11580,
      "train_flow_matching_loss": 0.07470703125,
      "train_physics_loss": 0.6315310001373291
    },
    {
      "step": 11580,
      "train_flow_matching_loss": 0.1875,
      "train_physics_loss": 0.0
    },
    {
      "step": 11580,
      "train_flow_matching_loss": 0.146484375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08004732429981232,
      "learning_rate": 5e-05,
      "loss": 0.1526,
      "step": 11590
    },
    {
      "step": 11590,
      "train_flow_matching_loss": 0.173828125,
      "train_physics_loss": 0.0
    },
    {
      "step": 11590,
      "train_flow_matching_loss": 0.05078125,
      "train_physics_loss": 0.6444084644317627
    },
    {
      "step": 11590,
      "train_flow_matching_loss": 0.142578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 11590,
      "train_flow_matching_loss": 0.1259765625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08760813623666763,
      "learning_rate": 5e-05,
      "loss": 0.1536,
      "step": 11600
    },
    {
      "step": 11600,
      "train_flow_matching_loss": 0.1318359375,
      "train_physics_loss": 0.0
    },
    {
      "step": 11600,
      "train_flow_matching_loss": 0.049560546875,
      "train_physics_loss": 0.6063379049301147
    },
    {
      "step": 11600,
      "train_flow_matching_loss": 0.0888671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 11600,
      "train_flow_matching_loss": 0.1875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.35329657793045044,
      "learning_rate": 5e-05,
      "loss": 0.1473,
      "step": 11610
    },
    {
      "step": 11610,
      "train_flow_matching_loss": 0.173828125,
      "train_physics_loss": 0.0
    },
    {
      "step": 11610,
      "train_flow_matching_loss": 0.056396484375,
      "train_physics_loss": 0.7045332789421082
    },
    {
      "step": 11610,
      "train_flow_matching_loss": 0.1416015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 11610,
      "train_flow_matching_loss": 0.2099609375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.1692485362291336,
      "learning_rate": 5e-05,
      "loss": 0.1549,
      "step": 11620
    },
    {
      "step": 11620,
      "train_flow_matching_loss": 0.154296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 11620,
      "train_flow_matching_loss": 0.034423828125,
      "train_physics_loss": 0.6743664741516113
    },
    {
      "step": 11620,
      "train_flow_matching_loss": 0.2255859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 11620,
      "train_flow_matching_loss": 0.1416015625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.22391247749328613,
      "learning_rate": 5e-05,
      "loss": 0.1488,
      "step": 11630
    },
    {
      "step": 11630,
      "train_flow_matching_loss": 0.09912109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 11630,
      "train_flow_matching_loss": 0.08251953125,
      "train_physics_loss": 0.6394078135490417
    },
    {
      "step": 11630,
      "train_flow_matching_loss": 0.08349609375,
      "train_physics_loss": 0.0
    },
    {
      "step": 11630,
      "train_flow_matching_loss": 0.265625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10189703106880188,
      "learning_rate": 5e-05,
      "loss": 0.1534,
      "step": 11640
    },
    {
      "step": 11640,
      "train_flow_matching_loss": 0.1552734375,
      "train_physics_loss": 0.0
    },
    {
      "step": 11640,
      "train_flow_matching_loss": 0.1572265625,
      "train_physics_loss": 0.0
    },
    {
      "step": 11640,
      "train_flow_matching_loss": 0.1552734375,
      "train_physics_loss": 0.0
    },
    {
      "step": 11640,
      "train_flow_matching_loss": 0.11962890625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.31034454703330994,
      "learning_rate": 5e-05,
      "loss": 0.1495,
      "step": 11650
    },
    {
      "step": 11650,
      "train_flow_matching_loss": 0.1884765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 11650,
      "train_flow_matching_loss": 0.189453125,
      "train_physics_loss": 0.0
    },
    {
      "step": 11650,
      "train_flow_matching_loss": 0.1845703125,
      "train_physics_loss": 0.0
    },
    {
      "step": 11650,
      "train_flow_matching_loss": 0.1884765625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08463867008686066,
      "learning_rate": 5e-05,
      "loss": 0.1481,
      "step": 11660
    },
    {
      "step": 11660,
      "train_flow_matching_loss": 0.1318359375,
      "train_physics_loss": 0.0
    },
    {
      "step": 11660,
      "train_flow_matching_loss": 0.1796875,
      "train_physics_loss": 0.0
    },
    {
      "step": 11660,
      "train_flow_matching_loss": 0.181640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 11660,
      "train_flow_matching_loss": 0.154296875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08578003942966461,
      "learning_rate": 5e-05,
      "loss": 0.1567,
      "step": 11670
    },
    {
      "step": 11670,
      "train_flow_matching_loss": 0.11279296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 11670,
      "train_flow_matching_loss": 0.1796875,
      "train_physics_loss": 0.0
    },
    {
      "step": 11670,
      "train_flow_matching_loss": 0.11279296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 11670,
      "train_flow_matching_loss": 0.09326171875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.07996276766061783,
      "learning_rate": 5e-05,
      "loss": 0.1485,
      "step": 11680
    },
    {
      "step": 11680,
      "train_flow_matching_loss": 0.154296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 11680,
      "train_flow_matching_loss": 0.1396484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 11680,
      "train_flow_matching_loss": 0.1083984375,
      "train_physics_loss": 0.0
    },
    {
      "step": 11680,
      "train_flow_matching_loss": 0.2236328125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.12199077755212784,
      "learning_rate": 5e-05,
      "loss": 0.154,
      "step": 11690
    },
    {
      "step": 11690,
      "train_flow_matching_loss": 0.142578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 11690,
      "train_flow_matching_loss": 0.1923828125,
      "train_physics_loss": 0.0
    },
    {
      "step": 11690,
      "train_flow_matching_loss": 0.154296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 11690,
      "train_flow_matching_loss": 0.1728515625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08514820039272308,
      "learning_rate": 5e-05,
      "loss": 0.1556,
      "step": 11700
    },
    {
      "step": 11700,
      "train_flow_matching_loss": 0.203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 11700,
      "train_flow_matching_loss": 0.1611328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 11700,
      "train_flow_matching_loss": 0.1748046875,
      "train_physics_loss": 0.0
    },
    {
      "step": 11700,
      "train_flow_matching_loss": 0.1376953125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.18855150043964386,
      "learning_rate": 5e-05,
      "loss": 0.1559,
      "step": 11710
    },
    {
      "step": 11710,
      "train_flow_matching_loss": 0.1572265625,
      "train_physics_loss": 0.0
    },
    {
      "step": 11710,
      "train_flow_matching_loss": 0.259765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 11710,
      "train_flow_matching_loss": 0.1943359375,
      "train_physics_loss": 0.0
    },
    {
      "step": 11710,
      "train_flow_matching_loss": 0.10791015625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08643609285354614,
      "learning_rate": 5e-05,
      "loss": 0.1483,
      "step": 11720
    },
    {
      "step": 11720,
      "train_flow_matching_loss": 0.1484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 11720,
      "train_flow_matching_loss": 0.1591796875,
      "train_physics_loss": 0.0
    },
    {
      "step": 11720,
      "train_flow_matching_loss": 0.2119140625,
      "train_physics_loss": 0.0
    },
    {
      "step": 11720,
      "train_flow_matching_loss": 0.1513671875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.19388318061828613,
      "learning_rate": 5e-05,
      "loss": 0.1536,
      "step": 11730
    },
    {
      "step": 11730,
      "train_flow_matching_loss": 0.1875,
      "train_physics_loss": 0.0
    },
    {
      "step": 11730,
      "train_flow_matching_loss": 0.1064453125,
      "train_physics_loss": 0.0
    },
    {
      "step": 11730,
      "train_flow_matching_loss": 0.193359375,
      "train_physics_loss": 0.0
    },
    {
      "step": 11730,
      "train_flow_matching_loss": 0.07421875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11764205992221832,
      "learning_rate": 5e-05,
      "loss": 0.1464,
      "step": 11740
    },
    {
      "step": 11740,
      "train_flow_matching_loss": 0.138671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 11740,
      "train_flow_matching_loss": 0.1318359375,
      "train_physics_loss": 0.0
    },
    {
      "step": 11740,
      "train_flow_matching_loss": 0.205078125,
      "train_physics_loss": 0.0
    },
    {
      "step": 11740,
      "train_flow_matching_loss": 0.201171875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11100908368825912,
      "learning_rate": 5e-05,
      "loss": 0.1503,
      "step": 11750
    },
    {
      "step": 11750,
      "train_flow_matching_loss": 0.0966796875,
      "train_physics_loss": 0.0
    },
    {
      "step": 11750,
      "train_flow_matching_loss": 0.166015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 11750,
      "train_flow_matching_loss": 0.08447265625,
      "train_physics_loss": 0.0
    },
    {
      "step": 11750,
      "train_flow_matching_loss": 0.1533203125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11461525410413742,
      "learning_rate": 5e-05,
      "loss": 0.1491,
      "step": 11760
    },
    {
      "step": 11760,
      "train_flow_matching_loss": 0.12890625,
      "train_physics_loss": 0.0
    },
    {
      "step": 11760,
      "train_flow_matching_loss": 0.140625,
      "train_physics_loss": 0.0
    },
    {
      "step": 11760,
      "train_flow_matching_loss": 0.1435546875,
      "train_physics_loss": 0.0
    },
    {
      "step": 11760,
      "train_flow_matching_loss": 0.1875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09347642213106155,
      "learning_rate": 5e-05,
      "loss": 0.1455,
      "step": 11770
    },
    {
      "step": 11770,
      "train_flow_matching_loss": 0.1513671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 11770,
      "train_flow_matching_loss": 0.1572265625,
      "train_physics_loss": 0.0
    },
    {
      "step": 11770,
      "train_flow_matching_loss": 0.11279296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 11770,
      "train_flow_matching_loss": 0.11962890625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09758294373750687,
      "learning_rate": 5e-05,
      "loss": 0.1473,
      "step": 11780
    },
    {
      "step": 11780,
      "train_flow_matching_loss": 0.1298828125,
      "train_physics_loss": 0.0
    },
    {
      "step": 11780,
      "train_flow_matching_loss": 0.11962890625,
      "train_physics_loss": 0.0
    },
    {
      "step": 11780,
      "train_flow_matching_loss": 0.1650390625,
      "train_physics_loss": 0.0
    },
    {
      "step": 11780,
      "train_flow_matching_loss": 0.12890625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.07781567424535751,
      "learning_rate": 5e-05,
      "loss": 0.1495,
      "step": 11790
    },
    {
      "step": 11790,
      "train_flow_matching_loss": 0.1572265625,
      "train_physics_loss": 0.0
    },
    {
      "step": 11790,
      "train_flow_matching_loss": 0.146484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 11790,
      "train_flow_matching_loss": 0.158203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 11790,
      "train_flow_matching_loss": 0.150390625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09691512584686279,
      "learning_rate": 5e-05,
      "loss": 0.1469,
      "step": 11800
    },
    {
      "step": 11800,
      "train_flow_matching_loss": 0.09521484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 11800,
      "train_flow_matching_loss": 0.1884765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 11800,
      "train_flow_matching_loss": 0.11279296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 11800,
      "train_flow_matching_loss": 0.1630859375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09757324308156967,
      "learning_rate": 5e-05,
      "loss": 0.1471,
      "step": 11810
    },
    {
      "step": 11810,
      "train_flow_matching_loss": 0.275390625,
      "train_physics_loss": 0.0
    },
    {
      "step": 11810,
      "train_flow_matching_loss": 0.0830078125,
      "train_physics_loss": 0.0
    },
    {
      "step": 11810,
      "train_flow_matching_loss": 0.078125,
      "train_physics_loss": 0.0
    },
    {
      "step": 11810,
      "train_flow_matching_loss": 0.1328125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.1460341364145279,
      "learning_rate": 5e-05,
      "loss": 0.1531,
      "step": 11820
    },
    {
      "step": 11820,
      "train_flow_matching_loss": 0.1962890625,
      "train_physics_loss": 0.0
    },
    {
      "step": 11820,
      "train_flow_matching_loss": 0.1826171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 11820,
      "train_flow_matching_loss": 0.2041015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 11820,
      "train_flow_matching_loss": 0.177734375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10072782635688782,
      "learning_rate": 5e-05,
      "loss": 0.1485,
      "step": 11830
    },
    {
      "step": 11830,
      "train_flow_matching_loss": 0.193359375,
      "train_physics_loss": 0.0
    },
    {
      "step": 11830,
      "train_flow_matching_loss": 0.12255859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 11830,
      "train_flow_matching_loss": 0.09716796875,
      "train_physics_loss": 0.0
    },
    {
      "step": 11830,
      "train_flow_matching_loss": 0.12255859375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08502661436796188,
      "learning_rate": 5e-05,
      "loss": 0.1455,
      "step": 11840
    },
    {
      "step": 11840,
      "train_flow_matching_loss": 0.12109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 11840,
      "train_flow_matching_loss": 0.193359375,
      "train_physics_loss": 0.0
    },
    {
      "step": 11840,
      "train_flow_matching_loss": 0.09326171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 11840,
      "train_flow_matching_loss": 0.09765625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08086476475000381,
      "learning_rate": 5e-05,
      "loss": 0.147,
      "step": 11850
    },
    {
      "step": 11850,
      "train_flow_matching_loss": 0.150390625,
      "train_physics_loss": 0.0
    },
    {
      "step": 11850,
      "train_flow_matching_loss": 0.2138671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 11850,
      "train_flow_matching_loss": 0.1865234375,
      "train_physics_loss": 0.0
    },
    {
      "step": 11850,
      "train_flow_matching_loss": 0.1708984375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.07662493735551834,
      "learning_rate": 5e-05,
      "loss": 0.1532,
      "step": 11860
    },
    {
      "step": 11860,
      "train_flow_matching_loss": 0.1611328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 11860,
      "train_flow_matching_loss": 0.11376953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 11860,
      "train_flow_matching_loss": 0.11572265625,
      "train_physics_loss": 0.0
    },
    {
      "step": 11860,
      "train_flow_matching_loss": 0.146484375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.18577931821346283,
      "learning_rate": 5e-05,
      "loss": 0.1524,
      "step": 11870
    },
    {
      "step": 11870,
      "train_flow_matching_loss": 0.12060546875,
      "train_physics_loss": 0.0
    },
    {
      "step": 11870,
      "train_flow_matching_loss": 0.1533203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 11870,
      "train_flow_matching_loss": 0.15625,
      "train_physics_loss": 0.0
    },
    {
      "step": 11870,
      "train_flow_matching_loss": 0.10302734375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.12506578862667084,
      "learning_rate": 5e-05,
      "loss": 0.1465,
      "step": 11880
    },
    {
      "step": 11880,
      "train_flow_matching_loss": 0.1884765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 11880,
      "train_flow_matching_loss": 0.2197265625,
      "train_physics_loss": 0.0
    },
    {
      "step": 11880,
      "train_flow_matching_loss": 0.1259765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 11880,
      "train_flow_matching_loss": 0.232421875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.36334171891212463,
      "learning_rate": 5e-05,
      "loss": 0.1481,
      "step": 11890
    },
    {
      "step": 11890,
      "train_flow_matching_loss": 0.1923828125,
      "train_physics_loss": 0.0
    },
    {
      "step": 11890,
      "train_flow_matching_loss": 0.197265625,
      "train_physics_loss": 0.0
    },
    {
      "step": 11890,
      "train_flow_matching_loss": 0.1455078125,
      "train_physics_loss": 0.0
    },
    {
      "step": 11890,
      "train_flow_matching_loss": 0.1630859375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11283085495233536,
      "learning_rate": 5e-05,
      "loss": 0.1522,
      "step": 11900
    },
    {
      "step": 11900,
      "train_flow_matching_loss": 0.1376953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 11900,
      "train_flow_matching_loss": 0.25390625,
      "train_physics_loss": 0.0
    },
    {
      "step": 11900,
      "train_flow_matching_loss": 0.1591796875,
      "train_physics_loss": 0.0
    },
    {
      "step": 11900,
      "train_flow_matching_loss": 0.1533203125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.12074869871139526,
      "learning_rate": 5e-05,
      "loss": 0.1494,
      "step": 11910
    },
    {
      "step": 11910,
      "train_flow_matching_loss": 0.1669921875,
      "train_physics_loss": 0.0
    },
    {
      "step": 11910,
      "train_flow_matching_loss": 0.12158203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 11910,
      "train_flow_matching_loss": 0.0966796875,
      "train_physics_loss": 0.0
    },
    {
      "step": 11910,
      "train_flow_matching_loss": 0.1611328125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.0726502314209938,
      "learning_rate": 5e-05,
      "loss": 0.1486,
      "step": 11920
    },
    {
      "step": 11920,
      "train_flow_matching_loss": 0.08251953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 11920,
      "train_flow_matching_loss": 0.197265625,
      "train_physics_loss": 0.0
    },
    {
      "step": 11920,
      "train_flow_matching_loss": 0.1201171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 11920,
      "train_flow_matching_loss": 0.12158203125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.07854725420475006,
      "learning_rate": 5e-05,
      "loss": 0.1456,
      "step": 11930
    },
    {
      "step": 11930,
      "train_flow_matching_loss": 0.1748046875,
      "train_physics_loss": 0.0
    },
    {
      "step": 11930,
      "train_flow_matching_loss": 0.13671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 11930,
      "train_flow_matching_loss": 0.2109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 11930,
      "train_flow_matching_loss": 0.1640625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10293098539113998,
      "learning_rate": 5e-05,
      "loss": 0.1486,
      "step": 11940
    },
    {
      "step": 11940,
      "train_flow_matching_loss": 0.1630859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 11940,
      "train_flow_matching_loss": 0.2451171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 11940,
      "train_flow_matching_loss": 0.1591796875,
      "train_physics_loss": 0.0
    },
    {
      "step": 11940,
      "train_flow_matching_loss": 0.134765625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10494773089885712,
      "learning_rate": 5e-05,
      "loss": 0.1432,
      "step": 11950
    },
    {
      "step": 11950,
      "train_flow_matching_loss": 0.126953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 11950,
      "train_flow_matching_loss": 0.1923828125,
      "train_physics_loss": 0.0
    },
    {
      "step": 11950,
      "train_flow_matching_loss": 0.228515625,
      "train_physics_loss": 0.0
    },
    {
      "step": 11950,
      "train_flow_matching_loss": 0.11767578125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.069607675075531,
      "learning_rate": 5e-05,
      "loss": 0.1458,
      "step": 11960
    },
    {
      "step": 11960,
      "train_flow_matching_loss": 0.2490234375,
      "train_physics_loss": 0.0
    },
    {
      "step": 11960,
      "train_flow_matching_loss": 0.1318359375,
      "train_physics_loss": 0.0
    },
    {
      "step": 11960,
      "train_flow_matching_loss": 0.07763671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 11960,
      "train_flow_matching_loss": 0.09814453125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10807735472917557,
      "learning_rate": 5e-05,
      "loss": 0.1521,
      "step": 11970
    },
    {
      "step": 11970,
      "train_flow_matching_loss": 0.1923828125,
      "train_physics_loss": 0.0
    },
    {
      "step": 11970,
      "train_flow_matching_loss": 0.205078125,
      "train_physics_loss": 0.0
    },
    {
      "step": 11970,
      "train_flow_matching_loss": 0.1044921875,
      "train_physics_loss": 0.0
    },
    {
      "step": 11970,
      "train_flow_matching_loss": 0.140625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09337348490953445,
      "learning_rate": 5e-05,
      "loss": 0.1446,
      "step": 11980
    },
    {
      "step": 11980,
      "train_flow_matching_loss": 0.1123046875,
      "train_physics_loss": 0.0
    },
    {
      "step": 11980,
      "train_flow_matching_loss": 0.10986328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 11980,
      "train_flow_matching_loss": 0.08056640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 11980,
      "train_flow_matching_loss": 0.1748046875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11035298556089401,
      "learning_rate": 5e-05,
      "loss": 0.1483,
      "step": 11990
    },
    {
      "step": 11990,
      "train_flow_matching_loss": 0.140625,
      "train_physics_loss": 0.0
    },
    {
      "step": 11990,
      "train_flow_matching_loss": 0.162109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 11990,
      "train_flow_matching_loss": 0.09423828125,
      "train_physics_loss": 0.0
    },
    {
      "step": 11990,
      "train_flow_matching_loss": 0.1416015625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.1412060558795929,
      "learning_rate": 5e-05,
      "loss": 0.1446,
      "step": 12000
    },
    {
      "step": 12000,
      "train_flow_matching_loss": 0.150390625,
      "train_physics_loss": 0.0
    },
    {
      "step": 12000,
      "train_flow_matching_loss": 0.2216796875,
      "train_physics_loss": 0.0
    },
    {
      "step": 12000,
      "train_flow_matching_loss": 0.109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 12000,
      "train_flow_matching_loss": 0.1748046875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.1230836734175682,
      "learning_rate": 5e-05,
      "loss": 0.1495,
      "step": 12010
    },
    {
      "step": 12010,
      "train_flow_matching_loss": 0.1328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 12010,
      "train_flow_matching_loss": 0.271484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 12010,
      "train_flow_matching_loss": 0.1123046875,
      "train_physics_loss": 0.0
    },
    {
      "step": 12010,
      "train_flow_matching_loss": 0.171875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11697296798229218,
      "learning_rate": 5e-05,
      "loss": 0.1506,
      "step": 12020
    },
    {
      "step": 12020,
      "train_flow_matching_loss": 0.2236328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 12020,
      "train_flow_matching_loss": 0.2158203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 12020,
      "train_flow_matching_loss": 0.2421875,
      "train_physics_loss": 0.0
    },
    {
      "step": 12020,
      "train_flow_matching_loss": 0.1005859375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.13042712211608887,
      "learning_rate": 5e-05,
      "loss": 0.1502,
      "step": 12030
    },
    {
      "step": 12030,
      "train_flow_matching_loss": 0.109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 12030,
      "train_flow_matching_loss": 0.248046875,
      "train_physics_loss": 0.0
    },
    {
      "step": 12030,
      "train_flow_matching_loss": 0.12255859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 12030,
      "train_flow_matching_loss": 0.07177734375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11777787655591965,
      "learning_rate": 5e-05,
      "loss": 0.1439,
      "step": 12040
    },
    {
      "step": 12040,
      "train_flow_matching_loss": 0.05126953125,
      "train_physics_loss": 0.6905488967895508
    },
    {
      "step": 12040,
      "train_flow_matching_loss": 0.2001953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 12040,
      "train_flow_matching_loss": 0.1728515625,
      "train_physics_loss": 0.0
    },
    {
      "step": 12040,
      "train_flow_matching_loss": 0.134765625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.12658198177814484,
      "learning_rate": 5e-05,
      "loss": 0.1525,
      "step": 12050
    },
    {
      "step": 12050,
      "train_flow_matching_loss": 0.08984375,
      "train_physics_loss": 0.6684324741363525
    },
    {
      "step": 12050,
      "train_flow_matching_loss": 0.1552734375,
      "train_physics_loss": 0.0
    },
    {
      "step": 12050,
      "train_flow_matching_loss": 0.2080078125,
      "train_physics_loss": 0.0
    },
    {
      "step": 12050,
      "train_flow_matching_loss": 0.2041015625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09068953990936279,
      "learning_rate": 5e-05,
      "loss": 0.1502,
      "step": 12060
    },
    {
      "step": 12060,
      "train_flow_matching_loss": 0.09716796875,
      "train_physics_loss": 0.669750988483429
    },
    {
      "step": 12060,
      "train_flow_matching_loss": 0.12060546875,
      "train_physics_loss": 0.0
    },
    {
      "step": 12060,
      "train_flow_matching_loss": 0.169921875,
      "train_physics_loss": 0.0
    },
    {
      "step": 12060,
      "train_flow_matching_loss": 0.1162109375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08315644413232803,
      "learning_rate": 5e-05,
      "loss": 0.1478,
      "step": 12070
    },
    {
      "step": 12070,
      "train_flow_matching_loss": 0.1259765625,
      "train_physics_loss": 0.6835718750953674
    },
    {
      "step": 12070,
      "train_flow_matching_loss": 0.25,
      "train_physics_loss": 0.0
    },
    {
      "step": 12070,
      "train_flow_matching_loss": 0.11962890625,
      "train_physics_loss": 0.0
    },
    {
      "step": 12070,
      "train_flow_matching_loss": 0.1494140625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.12542207539081573,
      "learning_rate": 5e-05,
      "loss": 0.1437,
      "step": 12080
    },
    {
      "step": 12080,
      "train_flow_matching_loss": 0.045654296875,
      "train_physics_loss": 0.7028636336326599
    },
    {
      "step": 12080,
      "train_flow_matching_loss": 0.240234375,
      "train_physics_loss": 0.0
    },
    {
      "step": 12080,
      "train_flow_matching_loss": 0.2080078125,
      "train_physics_loss": 0.0
    },
    {
      "step": 12080,
      "train_flow_matching_loss": 0.1103515625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08865167945623398,
      "learning_rate": 5e-05,
      "loss": 0.1421,
      "step": 12090
    },
    {
      "step": 12090,
      "train_flow_matching_loss": 0.06884765625,
      "train_physics_loss": 0.6108679175376892
    },
    {
      "step": 12090,
      "train_flow_matching_loss": 0.2041015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 12090,
      "train_flow_matching_loss": 0.1083984375,
      "train_physics_loss": 0.0
    },
    {
      "step": 12090,
      "train_flow_matching_loss": 0.134765625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.16979201138019562,
      "learning_rate": 5e-05,
      "loss": 0.1489,
      "step": 12100
    },
    {
      "step": 12100,
      "train_flow_matching_loss": 0.0927734375,
      "train_physics_loss": 0.6082064509391785
    },
    {
      "step": 12100,
      "train_flow_matching_loss": 0.1552734375,
      "train_physics_loss": 0.0
    },
    {
      "step": 12100,
      "train_flow_matching_loss": 0.1494140625,
      "train_physics_loss": 0.0
    },
    {
      "step": 12100,
      "train_flow_matching_loss": 0.0986328125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.086146280169487,
      "learning_rate": 5e-05,
      "loss": 0.147,
      "step": 12110
    },
    {
      "step": 12110,
      "train_flow_matching_loss": 0.06884765625,
      "train_physics_loss": 0.6502268314361572
    },
    {
      "step": 12110,
      "train_flow_matching_loss": 0.1513671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 12110,
      "train_flow_matching_loss": 0.1318359375,
      "train_physics_loss": 0.0
    },
    {
      "step": 12110,
      "train_flow_matching_loss": 0.09375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.07792559266090393,
      "learning_rate": 5e-05,
      "loss": 0.149,
      "step": 12120
    },
    {
      "step": 12120,
      "train_flow_matching_loss": 0.060791015625,
      "train_physics_loss": 0.6252107620239258
    },
    {
      "step": 12120,
      "train_flow_matching_loss": 0.166015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 12120,
      "train_flow_matching_loss": 0.1025390625,
      "train_physics_loss": 0.0
    },
    {
      "step": 12120,
      "train_flow_matching_loss": 0.126953125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.14844833314418793,
      "learning_rate": 5e-05,
      "loss": 0.1452,
      "step": 12130
    },
    {
      "step": 12130,
      "train_flow_matching_loss": 0.045166015625,
      "train_physics_loss": 0.6267894506454468
    },
    {
      "step": 12130,
      "train_flow_matching_loss": 0.1748046875,
      "train_physics_loss": 0.0
    },
    {
      "step": 12130,
      "train_flow_matching_loss": 0.1650390625,
      "train_physics_loss": 0.0
    },
    {
      "step": 12130,
      "train_flow_matching_loss": 0.1533203125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10029509663581848,
      "learning_rate": 5e-05,
      "loss": 0.1482,
      "step": 12140
    },
    {
      "step": 12140,
      "train_flow_matching_loss": 0.09423828125,
      "train_physics_loss": 0.6416319608688354
    },
    {
      "step": 12140,
      "train_flow_matching_loss": 0.1513671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 12140,
      "train_flow_matching_loss": 0.11669921875,
      "train_physics_loss": 0.0
    },
    {
      "step": 12140,
      "train_flow_matching_loss": 0.10546875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08605927973985672,
      "learning_rate": 5e-05,
      "loss": 0.1441,
      "step": 12150
    },
    {
      "step": 12150,
      "train_flow_matching_loss": 0.06640625,
      "train_physics_loss": 0.6114657521247864
    },
    {
      "step": 12150,
      "train_flow_matching_loss": 0.04638671875,
      "train_physics_loss": 0.5778919458389282
    },
    {
      "step": 12150,
      "train_flow_matching_loss": 0.2353515625,
      "train_physics_loss": 0.0
    },
    {
      "step": 12150,
      "train_flow_matching_loss": 0.1513671875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08039865642786026,
      "learning_rate": 5e-05,
      "loss": 0.1523,
      "step": 12160
    },
    {
      "step": 12160,
      "train_flow_matching_loss": 0.068359375,
      "train_physics_loss": 0.6228850483894348
    },
    {
      "step": 12160,
      "train_flow_matching_loss": 0.1318359375,
      "train_physics_loss": 0.6809952855110168
    },
    {
      "step": 12160,
      "train_flow_matching_loss": 0.1669921875,
      "train_physics_loss": 0.0
    },
    {
      "step": 12160,
      "train_flow_matching_loss": 0.1494140625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11496985703706741,
      "learning_rate": 5e-05,
      "loss": 0.1468,
      "step": 12170
    },
    {
      "step": 12170,
      "train_flow_matching_loss": 0.10107421875,
      "train_physics_loss": 0.6242273449897766
    },
    {
      "step": 12170,
      "train_flow_matching_loss": 0.0791015625,
      "train_physics_loss": 0.6827747225761414
    },
    {
      "step": 12170,
      "train_flow_matching_loss": 0.1279296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 12170,
      "train_flow_matching_loss": 0.1474609375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08426567167043686,
      "learning_rate": 5e-05,
      "loss": 0.1482,
      "step": 12180
    },
    {
      "step": 12180,
      "train_flow_matching_loss": 0.0986328125,
      "train_physics_loss": 0.6676936745643616
    },
    {
      "step": 12180,
      "train_flow_matching_loss": 0.046875,
      "train_physics_loss": 0.6885567903518677
    },
    {
      "step": 12180,
      "train_flow_matching_loss": 0.1787109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 12180,
      "train_flow_matching_loss": 0.201171875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.130489781498909,
      "learning_rate": 5e-05,
      "loss": 0.154,
      "step": 12190
    },
    {
      "step": 12190,
      "train_flow_matching_loss": 0.0673828125,
      "train_physics_loss": 0.6400671601295471
    },
    {
      "step": 12190,
      "train_flow_matching_loss": 0.044189453125,
      "train_physics_loss": 0.6353613138198853
    },
    {
      "step": 12190,
      "train_flow_matching_loss": 0.0693359375,
      "train_physics_loss": 0.0
    },
    {
      "step": 12190,
      "train_flow_matching_loss": 0.038330078125,
      "train_physics_loss": 0.6307622790336609
    },
    {
      "grad_norm": 0.09621015936136246,
      "learning_rate": 5e-05,
      "loss": 0.1471,
      "step": 12200
    },
    {
      "step": 12200,
      "train_flow_matching_loss": 0.10302734375,
      "train_physics_loss": 0.6578107476234436
    },
    {
      "step": 12200,
      "train_flow_matching_loss": 0.036376953125,
      "train_physics_loss": 0.6613573431968689
    },
    {
      "step": 12200,
      "train_flow_matching_loss": 0.1337890625,
      "train_physics_loss": 0.0
    },
    {
      "step": 12200,
      "train_flow_matching_loss": 0.042724609375,
      "train_physics_loss": 0.7270750999450684
    },
    {
      "grad_norm": 0.07948026806116104,
      "learning_rate": 5e-05,
      "loss": 0.1489,
      "step": 12210
    },
    {
      "step": 12210,
      "train_flow_matching_loss": 0.060546875,
      "train_physics_loss": 0.6180981993675232
    },
    {
      "step": 12210,
      "train_flow_matching_loss": 0.0693359375,
      "train_physics_loss": 0.651862621307373
    },
    {
      "step": 12210,
      "train_flow_matching_loss": 0.1982421875,
      "train_physics_loss": 0.0
    },
    {
      "step": 12210,
      "train_flow_matching_loss": 0.041748046875,
      "train_physics_loss": 0.5956721901893616
    },
    {
      "grad_norm": 0.18224895000457764,
      "learning_rate": 5e-05,
      "loss": 0.1504,
      "step": 12220
    },
    {
      "step": 12220,
      "train_flow_matching_loss": 0.11474609375,
      "train_physics_loss": 0.7071207165718079
    },
    {
      "step": 12220,
      "train_flow_matching_loss": 0.04833984375,
      "train_physics_loss": 0.6249430775642395
    },
    {
      "step": 12220,
      "train_flow_matching_loss": 0.1240234375,
      "train_physics_loss": 0.0
    },
    {
      "step": 12220,
      "train_flow_matching_loss": 0.09619140625,
      "train_physics_loss": 0.6614636778831482
    },
    {
      "grad_norm": 0.16314737498760223,
      "learning_rate": 5e-05,
      "loss": 0.1542,
      "step": 12230
    },
    {
      "step": 12230,
      "train_flow_matching_loss": 0.1259765625,
      "train_physics_loss": 0.6399523019790649
    },
    {
      "step": 12230,
      "train_flow_matching_loss": 0.03857421875,
      "train_physics_loss": 0.6142847537994385
    },
    {
      "step": 12230,
      "train_flow_matching_loss": 0.1796875,
      "train_physics_loss": 0.0
    },
    {
      "step": 12230,
      "train_flow_matching_loss": 0.09375,
      "train_physics_loss": 0.6638307571411133
    },
    {
      "grad_norm": 0.08856307715177536,
      "learning_rate": 5e-05,
      "loss": 0.1469,
      "step": 12240
    },
    {
      "step": 12240,
      "train_flow_matching_loss": 0.09619140625,
      "train_physics_loss": 0.6721986532211304
    },
    {
      "step": 12240,
      "train_flow_matching_loss": 0.07763671875,
      "train_physics_loss": 0.6687635183334351
    },
    {
      "step": 12240,
      "train_flow_matching_loss": 0.1298828125,
      "train_physics_loss": 0.0
    },
    {
      "step": 12240,
      "train_flow_matching_loss": 0.0390625,
      "train_physics_loss": 0.6804574728012085
    },
    {
      "grad_norm": 0.12551698088645935,
      "learning_rate": 5e-05,
      "loss": 0.1454,
      "step": 12250
    },
    {
      "step": 12250,
      "train_flow_matching_loss": 0.05419921875,
      "train_physics_loss": 0.5903021097183228
    },
    {
      "step": 12250,
      "train_flow_matching_loss": 0.07666015625,
      "train_physics_loss": 0.6674636006355286
    },
    {
      "step": 12250,
      "train_flow_matching_loss": 0.2099609375,
      "train_physics_loss": 0.0
    },
    {
      "step": 12250,
      "train_flow_matching_loss": 0.07080078125,
      "train_physics_loss": 0.6459048986434937
    },
    {
      "grad_norm": 0.12087113410234451,
      "learning_rate": 5e-05,
      "loss": 0.15,
      "step": 12260
    },
    {
      "step": 12260,
      "train_flow_matching_loss": 0.06494140625,
      "train_physics_loss": 0.617138683795929
    },
    {
      "step": 12260,
      "train_flow_matching_loss": 0.058349609375,
      "train_physics_loss": 0.6651809811592102
    },
    {
      "step": 12260,
      "train_flow_matching_loss": 0.201171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 12260,
      "train_flow_matching_loss": 0.060791015625,
      "train_physics_loss": 0.6811864376068115
    },
    {
      "grad_norm": 0.11331881582736969,
      "learning_rate": 5e-05,
      "loss": 0.1497,
      "step": 12270
    },
    {
      "step": 12270,
      "train_flow_matching_loss": 0.06494140625,
      "train_physics_loss": 0.6836932301521301
    },
    {
      "step": 12270,
      "train_flow_matching_loss": 0.07421875,
      "train_physics_loss": 0.5892125964164734
    },
    {
      "step": 12270,
      "train_flow_matching_loss": 0.1357421875,
      "train_physics_loss": 0.0
    },
    {
      "step": 12270,
      "train_flow_matching_loss": 0.07177734375,
      "train_physics_loss": 0.5971441268920898
    },
    {
      "grad_norm": 0.11390642076730728,
      "learning_rate": 5e-05,
      "loss": 0.1484,
      "step": 12280
    },
    {
      "step": 12280,
      "train_flow_matching_loss": 0.0341796875,
      "train_physics_loss": 0.6458973288536072
    },
    {
      "step": 12280,
      "train_flow_matching_loss": 0.05712890625,
      "train_physics_loss": 0.6143069863319397
    },
    {
      "step": 12280,
      "train_flow_matching_loss": 0.173828125,
      "train_physics_loss": 0.0
    },
    {
      "step": 12280,
      "train_flow_matching_loss": 0.07666015625,
      "train_physics_loss": 0.5929605960845947
    },
    {
      "grad_norm": 0.10451456904411316,
      "learning_rate": 5e-05,
      "loss": 0.1454,
      "step": 12290
    },
    {
      "step": 12290,
      "train_flow_matching_loss": 0.140625,
      "train_physics_loss": 0.0
    },
    {
      "step": 12290,
      "train_flow_matching_loss": 0.0869140625,
      "train_physics_loss": 0.6574084758758545
    },
    {
      "step": 12290,
      "train_flow_matching_loss": 0.11572265625,
      "train_physics_loss": 0.0
    },
    {
      "step": 12290,
      "train_flow_matching_loss": 0.09814453125,
      "train_physics_loss": 0.7003625631332397
    },
    {
      "grad_norm": 0.13109388947486877,
      "learning_rate": 5e-05,
      "loss": 0.1467,
      "step": 12300
    },
    {
      "step": 12300,
      "train_flow_matching_loss": 0.12451171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 12300,
      "train_flow_matching_loss": 0.08447265625,
      "train_physics_loss": 0.624492883682251
    },
    {
      "step": 12300,
      "train_flow_matching_loss": 0.1376953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 12300,
      "train_flow_matching_loss": 0.024658203125,
      "train_physics_loss": 0.6131467223167419
    },
    {
      "grad_norm": 0.09920691698789597,
      "learning_rate": 5e-05,
      "loss": 0.1453,
      "step": 12310
    },
    {
      "step": 12310,
      "train_flow_matching_loss": 0.1630859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 12310,
      "train_flow_matching_loss": 0.0927734375,
      "train_physics_loss": 0.5970215797424316
    },
    {
      "step": 12310,
      "train_flow_matching_loss": 0.1337890625,
      "train_physics_loss": 0.0
    },
    {
      "step": 12310,
      "train_flow_matching_loss": 0.03955078125,
      "train_physics_loss": 0.610499382019043
    },
    {
      "grad_norm": 0.41055721044540405,
      "learning_rate": 5e-05,
      "loss": 0.1454,
      "step": 12320
    },
    {
      "step": 12320,
      "train_flow_matching_loss": 0.1767578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 12320,
      "train_flow_matching_loss": 0.06689453125,
      "train_physics_loss": 0.6078653335571289
    },
    {
      "step": 12320,
      "train_flow_matching_loss": 0.1328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 12320,
      "train_flow_matching_loss": 0.09326171875,
      "train_physics_loss": 0.6366735696792603
    },
    {
      "grad_norm": 0.09666409343481064,
      "learning_rate": 5e-05,
      "loss": 0.1444,
      "step": 12330
    },
    {
      "step": 12330,
      "train_flow_matching_loss": 0.169921875,
      "train_physics_loss": 0.0
    },
    {
      "step": 12330,
      "train_flow_matching_loss": 0.0908203125,
      "train_physics_loss": 0.6116131544113159
    },
    {
      "step": 12330,
      "train_flow_matching_loss": 0.1044921875,
      "train_physics_loss": 0.0
    },
    {
      "step": 12330,
      "train_flow_matching_loss": 0.04638671875,
      "train_physics_loss": 0.6037408709526062
    },
    {
      "grad_norm": 0.15271200239658356,
      "learning_rate": 5e-05,
      "loss": 0.1479,
      "step": 12340
    },
    {
      "step": 12340,
      "train_flow_matching_loss": 0.1201171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 12340,
      "train_flow_matching_loss": 0.1298828125,
      "train_physics_loss": 0.7242223024368286
    },
    {
      "step": 12340,
      "train_flow_matching_loss": 0.1591796875,
      "train_physics_loss": 0.0
    },
    {
      "step": 12340,
      "train_flow_matching_loss": 0.025390625,
      "train_physics_loss": 0.60097736120224
    },
    {
      "grad_norm": 0.26332443952560425,
      "learning_rate": 5e-05,
      "loss": 0.1499,
      "step": 12350
    },
    {
      "step": 12350,
      "train_flow_matching_loss": 0.08349609375,
      "train_physics_loss": 0.0
    },
    {
      "step": 12350,
      "train_flow_matching_loss": 0.07421875,
      "train_physics_loss": 0.6662667989730835
    },
    {
      "step": 12350,
      "train_flow_matching_loss": 0.1533203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 12350,
      "train_flow_matching_loss": 0.0277099609375,
      "train_physics_loss": 0.6174055337905884
    },
    {
      "grad_norm": 0.09867967665195465,
      "learning_rate": 5e-05,
      "loss": 0.1465,
      "step": 12360
    },
    {
      "step": 12360,
      "train_flow_matching_loss": 0.09912109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 12360,
      "train_flow_matching_loss": 0.03515625,
      "train_physics_loss": 0.590154230594635
    },
    {
      "step": 12360,
      "train_flow_matching_loss": 0.12109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 12360,
      "train_flow_matching_loss": 0.052001953125,
      "train_physics_loss": 0.6088687777519226
    },
    {
      "grad_norm": 0.10549134761095047,
      "learning_rate": 5e-05,
      "loss": 0.144,
      "step": 12370
    },
    {
      "step": 12370,
      "train_flow_matching_loss": 0.1826171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 12370,
      "train_flow_matching_loss": 0.064453125,
      "train_physics_loss": 0.6492936015129089
    },
    {
      "step": 12370,
      "train_flow_matching_loss": 0.1357421875,
      "train_physics_loss": 0.0
    },
    {
      "step": 12370,
      "train_flow_matching_loss": 0.09619140625,
      "train_physics_loss": 0.6256867051124573
    },
    {
      "grad_norm": 0.09295989573001862,
      "learning_rate": 5e-05,
      "loss": 0.1524,
      "step": 12380
    },
    {
      "step": 12380,
      "train_flow_matching_loss": 0.1748046875,
      "train_physics_loss": 0.0
    },
    {
      "step": 12380,
      "train_flow_matching_loss": 0.04638671875,
      "train_physics_loss": 0.6229615211486816
    },
    {
      "step": 12380,
      "train_flow_matching_loss": 0.09521484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 12380,
      "train_flow_matching_loss": 0.037109375,
      "train_physics_loss": 0.6053617000579834
    },
    {
      "grad_norm": 0.11685453355312347,
      "learning_rate": 5e-05,
      "loss": 0.145,
      "step": 12390
    },
    {
      "step": 12390,
      "train_flow_matching_loss": 0.1591796875,
      "train_physics_loss": 0.0
    },
    {
      "step": 12390,
      "train_flow_matching_loss": 0.05126953125,
      "train_physics_loss": 0.6004153490066528
    },
    {
      "step": 12390,
      "train_flow_matching_loss": 0.142578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 12390,
      "train_flow_matching_loss": 0.0712890625,
      "train_physics_loss": 0.6081439852714539
    },
    {
      "grad_norm": 0.10550831258296967,
      "learning_rate": 5e-05,
      "loss": 0.1511,
      "step": 12400
    },
    {
      "step": 12400,
      "train_flow_matching_loss": 0.1640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 12400,
      "train_flow_matching_loss": 0.053955078125,
      "train_physics_loss": 0.615753710269928
    },
    {
      "step": 12400,
      "train_flow_matching_loss": 0.1572265625,
      "train_physics_loss": 0.0
    },
    {
      "step": 12400,
      "train_flow_matching_loss": 0.064453125,
      "train_physics_loss": 0.5979684591293335
    },
    {
      "grad_norm": 0.07310279458761215,
      "learning_rate": 5e-05,
      "loss": 0.1495,
      "step": 12410
    },
    {
      "step": 12410,
      "train_flow_matching_loss": 0.2177734375,
      "train_physics_loss": 0.0
    },
    {
      "step": 12410,
      "train_flow_matching_loss": 0.0703125,
      "train_physics_loss": 0.6518213748931885
    },
    {
      "step": 12410,
      "train_flow_matching_loss": 0.1591796875,
      "train_physics_loss": 0.0
    },
    {
      "step": 12410,
      "train_flow_matching_loss": 0.07080078125,
      "train_physics_loss": 0.5640745162963867
    },
    {
      "grad_norm": 0.09577307105064392,
      "learning_rate": 5e-05,
      "loss": 0.1485,
      "step": 12420
    },
    {
      "step": 12420,
      "train_flow_matching_loss": 0.2431640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 12420,
      "train_flow_matching_loss": 0.07373046875,
      "train_physics_loss": 0.6979124546051025
    },
    {
      "step": 12420,
      "train_flow_matching_loss": 0.109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 12420,
      "train_flow_matching_loss": 0.057861328125,
      "train_physics_loss": 0.6868740320205688
    },
    {
      "grad_norm": 0.10723213106393814,
      "learning_rate": 5e-05,
      "loss": 0.1494,
      "step": 12430
    },
    {
      "step": 12430,
      "train_flow_matching_loss": 0.1689453125,
      "train_physics_loss": 0.0
    },
    {
      "step": 12430,
      "train_flow_matching_loss": 0.1201171875,
      "train_physics_loss": 0.5353472232818604
    },
    {
      "step": 12430,
      "train_flow_matching_loss": 0.2001953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 12430,
      "train_flow_matching_loss": 0.042724609375,
      "train_physics_loss": 0.58029705286026
    },
    {
      "grad_norm": 0.08102940022945404,
      "learning_rate": 5e-05,
      "loss": 0.1475,
      "step": 12440
    },
    {
      "step": 12440,
      "train_flow_matching_loss": 0.10498046875,
      "train_physics_loss": 0.0
    },
    {
      "step": 12440,
      "train_flow_matching_loss": 0.050537109375,
      "train_physics_loss": 0.6109086871147156
    },
    {
      "step": 12440,
      "train_flow_matching_loss": 0.1728515625,
      "train_physics_loss": 0.0
    },
    {
      "step": 12440,
      "train_flow_matching_loss": 0.05029296875,
      "train_physics_loss": 0.6245026588439941
    },
    {
      "grad_norm": 0.10534479469060898,
      "learning_rate": 5e-05,
      "loss": 0.1451,
      "step": 12450
    },
    {
      "step": 12450,
      "train_flow_matching_loss": 0.2294921875,
      "train_physics_loss": 0.0
    },
    {
      "step": 12450,
      "train_flow_matching_loss": 0.1025390625,
      "train_physics_loss": 0.6186361908912659
    },
    {
      "step": 12450,
      "train_flow_matching_loss": 0.1328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 12450,
      "train_flow_matching_loss": 0.228515625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.07792925089597702,
      "learning_rate": 5e-05,
      "loss": 0.1512,
      "step": 12460
    },
    {
      "step": 12460,
      "train_flow_matching_loss": 0.1943359375,
      "train_physics_loss": 0.0
    },
    {
      "step": 12460,
      "train_flow_matching_loss": 0.07763671875,
      "train_physics_loss": 0.6611670255661011
    },
    {
      "step": 12460,
      "train_flow_matching_loss": 0.255859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 12460,
      "train_flow_matching_loss": 0.1875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10049429535865784,
      "learning_rate": 5e-05,
      "loss": 0.1438,
      "step": 12470
    },
    {
      "step": 12470,
      "train_flow_matching_loss": 0.10400390625,
      "train_physics_loss": 0.0
    },
    {
      "step": 12470,
      "train_flow_matching_loss": 0.15625,
      "train_physics_loss": 0.5202507972717285
    },
    {
      "step": 12470,
      "train_flow_matching_loss": 0.1748046875,
      "train_physics_loss": 0.0
    },
    {
      "step": 12470,
      "train_flow_matching_loss": 0.138671875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09558115154504776,
      "learning_rate": 5e-05,
      "loss": 0.1507,
      "step": 12480
    },
    {
      "step": 12480,
      "train_flow_matching_loss": 0.173828125,
      "train_physics_loss": 0.0
    },
    {
      "step": 12480,
      "train_flow_matching_loss": 0.08935546875,
      "train_physics_loss": 0.6333060264587402
    },
    {
      "step": 12480,
      "train_flow_matching_loss": 0.1748046875,
      "train_physics_loss": 0.0
    },
    {
      "step": 12480,
      "train_flow_matching_loss": 0.1337890625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08217790722846985,
      "learning_rate": 5e-05,
      "loss": 0.1456,
      "step": 12490
    },
    {
      "step": 12490,
      "train_flow_matching_loss": 0.15625,
      "train_physics_loss": 0.0
    },
    {
      "step": 12490,
      "train_flow_matching_loss": 0.0712890625,
      "train_physics_loss": 0.5996473431587219
    },
    {
      "step": 12490,
      "train_flow_matching_loss": 0.1982421875,
      "train_physics_loss": 0.0
    },
    {
      "step": 12490,
      "train_flow_matching_loss": 0.1064453125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.07404857128858566,
      "learning_rate": 5e-05,
      "loss": 0.1449,
      "step": 12500
    },
    {
      "step": 12500,
      "train_flow_matching_loss": 0.1728515625,
      "train_physics_loss": 0.0
    },
    {
      "step": 12500,
      "train_flow_matching_loss": 0.048828125,
      "train_physics_loss": 0.5646234750747681
    },
    {
      "step": 12500,
      "train_flow_matching_loss": 0.1015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 12500,
      "train_flow_matching_loss": 0.09765625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11368373036384583,
      "learning_rate": 5e-05,
      "loss": 0.1481,
      "step": 12510
    },
    {
      "step": 12510,
      "train_flow_matching_loss": 0.2236328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 12510,
      "train_flow_matching_loss": 0.07373046875,
      "train_physics_loss": 0.6076894402503967
    },
    {
      "step": 12510,
      "train_flow_matching_loss": 0.1015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 12510,
      "train_flow_matching_loss": 0.171875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.07949022948741913,
      "learning_rate": 5e-05,
      "loss": 0.1442,
      "step": 12520
    },
    {
      "step": 12520,
      "train_flow_matching_loss": 0.1337890625,
      "train_physics_loss": 0.0
    },
    {
      "step": 12520,
      "train_flow_matching_loss": 0.060302734375,
      "train_physics_loss": 0.572978138923645
    },
    {
      "step": 12520,
      "train_flow_matching_loss": 0.134765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 12520,
      "train_flow_matching_loss": 0.05419921875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11712365597486496,
      "learning_rate": 5e-05,
      "loss": 0.1435,
      "step": 12530
    },
    {
      "step": 12530,
      "train_flow_matching_loss": 0.185546875,
      "train_physics_loss": 0.0
    },
    {
      "step": 12530,
      "train_flow_matching_loss": 0.09814453125,
      "train_physics_loss": 0.6024534106254578
    },
    {
      "step": 12530,
      "train_flow_matching_loss": 0.1865234375,
      "train_physics_loss": 0.0
    },
    {
      "step": 12530,
      "train_flow_matching_loss": 0.2333984375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08454035967588425,
      "learning_rate": 5e-05,
      "loss": 0.1407,
      "step": 12540
    },
    {
      "step": 12540,
      "train_flow_matching_loss": 0.126953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 12540,
      "train_flow_matching_loss": 0.059326171875,
      "train_physics_loss": 0.6210848093032837
    },
    {
      "step": 12540,
      "train_flow_matching_loss": 0.15625,
      "train_physics_loss": 0.0
    },
    {
      "step": 12540,
      "train_flow_matching_loss": 0.185546875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.1474543958902359,
      "learning_rate": 5e-05,
      "loss": 0.144,
      "step": 12550
    },
    {
      "step": 12550,
      "train_flow_matching_loss": 0.1513671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 12550,
      "train_flow_matching_loss": 0.1142578125,
      "train_physics_loss": 0.5254189968109131
    },
    {
      "step": 12550,
      "train_flow_matching_loss": 0.1240234375,
      "train_physics_loss": 0.0
    },
    {
      "step": 12550,
      "train_flow_matching_loss": 0.10400390625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.07917473465204239,
      "learning_rate": 5e-05,
      "loss": 0.1406,
      "step": 12560
    },
    {
      "step": 12560,
      "train_flow_matching_loss": 0.1572265625,
      "train_physics_loss": 0.0
    },
    {
      "step": 12560,
      "train_flow_matching_loss": 0.07275390625,
      "train_physics_loss": 0.0
    },
    {
      "step": 12560,
      "train_flow_matching_loss": 0.10595703125,
      "train_physics_loss": 0.0
    },
    {
      "step": 12560,
      "train_flow_matching_loss": 0.10205078125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10213124006986618,
      "learning_rate": 5e-05,
      "loss": 0.1468,
      "step": 12570
    },
    {
      "step": 12570,
      "train_flow_matching_loss": 0.1728515625,
      "train_physics_loss": 0.0
    },
    {
      "step": 12570,
      "train_flow_matching_loss": 0.08056640625,
      "train_physics_loss": 0.6155820488929749
    },
    {
      "step": 12570,
      "train_flow_matching_loss": 0.267578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 12570,
      "train_flow_matching_loss": 0.1572265625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08637513220310211,
      "learning_rate": 5e-05,
      "loss": 0.1432,
      "step": 12580
    },
    {
      "step": 12580,
      "train_flow_matching_loss": 0.1875,
      "train_physics_loss": 0.0
    },
    {
      "step": 12580,
      "train_flow_matching_loss": 0.031982421875,
      "train_physics_loss": 0.5717250108718872
    },
    {
      "step": 12580,
      "train_flow_matching_loss": 0.1875,
      "train_physics_loss": 0.0
    },
    {
      "step": 12580,
      "train_flow_matching_loss": 0.1669921875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.0974104180932045,
      "learning_rate": 5e-05,
      "loss": 0.1416,
      "step": 12590
    },
    {
      "step": 12590,
      "train_flow_matching_loss": 0.1220703125,
      "train_physics_loss": 0.0
    },
    {
      "step": 12590,
      "train_flow_matching_loss": 0.06298828125,
      "train_physics_loss": 0.6078248620033264
    },
    {
      "step": 12590,
      "train_flow_matching_loss": 0.11474609375,
      "train_physics_loss": 0.0
    },
    {
      "step": 12590,
      "train_flow_matching_loss": 0.203125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.20501376688480377,
      "learning_rate": 5e-05,
      "loss": 0.1462,
      "step": 12600
    },
    {
      "step": 12600,
      "train_flow_matching_loss": 0.162109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 12600,
      "train_flow_matching_loss": 0.06201171875,
      "train_physics_loss": 0.6038104891777039
    },
    {
      "step": 12600,
      "train_flow_matching_loss": 0.158203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 12600,
      "train_flow_matching_loss": 0.162109375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10317407548427582,
      "learning_rate": 5e-05,
      "loss": 0.1398,
      "step": 12610
    },
    {
      "step": 12610,
      "train_flow_matching_loss": 0.09375,
      "train_physics_loss": 0.0
    },
    {
      "step": 12610,
      "train_flow_matching_loss": 0.057373046875,
      "train_physics_loss": 0.5639416575431824
    },
    {
      "step": 12610,
      "train_flow_matching_loss": 0.1796875,
      "train_physics_loss": 0.0
    },
    {
      "step": 12610,
      "train_flow_matching_loss": 0.203125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08795735239982605,
      "learning_rate": 5e-05,
      "loss": 0.1437,
      "step": 12620
    },
    {
      "step": 12620,
      "train_flow_matching_loss": 0.193359375,
      "train_physics_loss": 0.0
    },
    {
      "step": 12620,
      "train_flow_matching_loss": 0.04345703125,
      "train_physics_loss": 0.5689607858657837
    },
    {
      "step": 12620,
      "train_flow_matching_loss": 0.2275390625,
      "train_physics_loss": 0.0
    },
    {
      "step": 12620,
      "train_flow_matching_loss": 0.11474609375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11100281774997711,
      "learning_rate": 5e-05,
      "loss": 0.1448,
      "step": 12630
    },
    {
      "step": 12630,
      "train_flow_matching_loss": 0.169921875,
      "train_physics_loss": 0.0
    },
    {
      "step": 12630,
      "train_flow_matching_loss": 0.045166015625,
      "train_physics_loss": 0.5339780449867249
    },
    {
      "step": 12630,
      "train_flow_matching_loss": 0.2216796875,
      "train_physics_loss": 0.0
    },
    {
      "step": 12630,
      "train_flow_matching_loss": 0.1201171875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08846411108970642,
      "learning_rate": 5e-05,
      "loss": 0.1445,
      "step": 12640
    },
    {
      "step": 12640,
      "train_flow_matching_loss": 0.11474609375,
      "train_physics_loss": 0.0
    },
    {
      "step": 12640,
      "train_flow_matching_loss": 0.043701171875,
      "train_physics_loss": 0.6336011290550232
    },
    {
      "step": 12640,
      "train_flow_matching_loss": 0.138671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 12640,
      "train_flow_matching_loss": 0.138671875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10147139430046082,
      "learning_rate": 5e-05,
      "loss": 0.1479,
      "step": 12650
    },
    {
      "step": 12650,
      "train_flow_matching_loss": 0.10498046875,
      "train_physics_loss": 0.0
    },
    {
      "step": 12650,
      "train_flow_matching_loss": 0.04345703125,
      "train_physics_loss": 0.6002920866012573
    },
    {
      "step": 12650,
      "train_flow_matching_loss": 0.2060546875,
      "train_physics_loss": 0.0
    },
    {
      "step": 12650,
      "train_flow_matching_loss": 0.1025390625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08507557213306427,
      "learning_rate": 5e-05,
      "loss": 0.1456,
      "step": 12660
    },
    {
      "step": 12660,
      "train_flow_matching_loss": 0.2001953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 12660,
      "train_flow_matching_loss": 0.1318359375,
      "train_physics_loss": 0.0
    },
    {
      "step": 12660,
      "train_flow_matching_loss": 0.1611328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 12660,
      "train_flow_matching_loss": 0.10107421875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10334846377372742,
      "learning_rate": 5e-05,
      "loss": 0.1514,
      "step": 12670
    },
    {
      "step": 12670,
      "train_flow_matching_loss": 0.2060546875,
      "train_physics_loss": 0.0
    },
    {
      "step": 12670,
      "train_flow_matching_loss": 0.1416015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 12670,
      "train_flow_matching_loss": 0.1748046875,
      "train_physics_loss": 0.0
    },
    {
      "step": 12670,
      "train_flow_matching_loss": 0.11669921875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08658421784639359,
      "learning_rate": 5e-05,
      "loss": 0.1426,
      "step": 12680
    },
    {
      "step": 12680,
      "train_flow_matching_loss": 0.138671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 12680,
      "train_flow_matching_loss": 0.10888671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 12680,
      "train_flow_matching_loss": 0.1337890625,
      "train_physics_loss": 0.0
    },
    {
      "step": 12680,
      "train_flow_matching_loss": 0.1435546875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09990105032920837,
      "learning_rate": 5e-05,
      "loss": 0.1494,
      "step": 12690
    },
    {
      "step": 12690,
      "train_flow_matching_loss": 0.255859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 12690,
      "train_flow_matching_loss": 0.07861328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 12690,
      "train_flow_matching_loss": 0.189453125,
      "train_physics_loss": 0.0
    },
    {
      "step": 12690,
      "train_flow_matching_loss": 0.1552734375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10790000110864639,
      "learning_rate": 5e-05,
      "loss": 0.1488,
      "step": 12700
    },
    {
      "step": 12700,
      "train_flow_matching_loss": 0.08544921875,
      "train_physics_loss": 0.0
    },
    {
      "step": 12700,
      "train_flow_matching_loss": 0.185546875,
      "train_physics_loss": 0.0
    },
    {
      "step": 12700,
      "train_flow_matching_loss": 0.1650390625,
      "train_physics_loss": 0.0
    },
    {
      "step": 12700,
      "train_flow_matching_loss": 0.1279296875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10119875520467758,
      "learning_rate": 5e-05,
      "loss": 0.14,
      "step": 12710
    },
    {
      "step": 12710,
      "train_flow_matching_loss": 0.12255859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 12710,
      "train_flow_matching_loss": 0.1650390625,
      "train_physics_loss": 0.0
    },
    {
      "step": 12710,
      "train_flow_matching_loss": 0.197265625,
      "train_physics_loss": 0.0
    },
    {
      "step": 12710,
      "train_flow_matching_loss": 0.1845703125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11741263419389725,
      "learning_rate": 5e-05,
      "loss": 0.1462,
      "step": 12720
    },
    {
      "step": 12720,
      "train_flow_matching_loss": 0.1328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 12720,
      "train_flow_matching_loss": 0.1396484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 12720,
      "train_flow_matching_loss": 0.09326171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 12720,
      "train_flow_matching_loss": 0.1591796875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.16594192385673523,
      "learning_rate": 5e-05,
      "loss": 0.1473,
      "step": 12730
    },
    {
      "step": 12730,
      "train_flow_matching_loss": 0.2060546875,
      "train_physics_loss": 0.0
    },
    {
      "step": 12730,
      "train_flow_matching_loss": 0.244140625,
      "train_physics_loss": 0.0
    },
    {
      "step": 12730,
      "train_flow_matching_loss": 0.138671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 12730,
      "train_flow_matching_loss": 0.1279296875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08927638828754425,
      "learning_rate": 5e-05,
      "loss": 0.1447,
      "step": 12740
    },
    {
      "step": 12740,
      "train_flow_matching_loss": 0.11865234375,
      "train_physics_loss": 0.0
    },
    {
      "step": 12740,
      "train_flow_matching_loss": 0.1328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 12740,
      "train_flow_matching_loss": 0.1162109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 12740,
      "train_flow_matching_loss": 0.1630859375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11029656231403351,
      "learning_rate": 5e-05,
      "loss": 0.1434,
      "step": 12750
    },
    {
      "step": 12750,
      "train_flow_matching_loss": 0.134765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 12750,
      "train_flow_matching_loss": 0.1806640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 12750,
      "train_flow_matching_loss": 0.1845703125,
      "train_physics_loss": 0.0
    },
    {
      "step": 12750,
      "train_flow_matching_loss": 0.11767578125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09206056594848633,
      "learning_rate": 5e-05,
      "loss": 0.1432,
      "step": 12760
    },
    {
      "step": 12760,
      "train_flow_matching_loss": 0.1767578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 12760,
      "train_flow_matching_loss": 0.1484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 12760,
      "train_flow_matching_loss": 0.1572265625,
      "train_physics_loss": 0.0
    },
    {
      "step": 12760,
      "train_flow_matching_loss": 0.12255859375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.07491324841976166,
      "learning_rate": 5e-05,
      "loss": 0.1412,
      "step": 12770
    },
    {
      "step": 12770,
      "train_flow_matching_loss": 0.140625,
      "train_physics_loss": 0.0
    },
    {
      "step": 12770,
      "train_flow_matching_loss": 0.1484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 12770,
      "train_flow_matching_loss": 0.1416015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 12770,
      "train_flow_matching_loss": 0.2294921875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11710335314273834,
      "learning_rate": 5e-05,
      "loss": 0.1495,
      "step": 12780
    },
    {
      "step": 12780,
      "train_flow_matching_loss": 0.09521484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 12780,
      "train_flow_matching_loss": 0.142578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 12780,
      "train_flow_matching_loss": 0.08203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 12780,
      "train_flow_matching_loss": 0.25390625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08665399253368378,
      "learning_rate": 5e-05,
      "loss": 0.15,
      "step": 12790
    },
    {
      "step": 12790,
      "train_flow_matching_loss": 0.173828125,
      "train_physics_loss": 0.0
    },
    {
      "step": 12790,
      "train_flow_matching_loss": 0.1962890625,
      "train_physics_loss": 0.0
    },
    {
      "step": 12790,
      "train_flow_matching_loss": 0.1494140625,
      "train_physics_loss": 0.0
    },
    {
      "step": 12790,
      "train_flow_matching_loss": 0.1171875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.07391123473644257,
      "learning_rate": 5e-05,
      "loss": 0.1469,
      "step": 12800
    },
    {
      "step": 12800,
      "train_flow_matching_loss": 0.1689453125,
      "train_physics_loss": 0.0
    },
    {
      "step": 12800,
      "train_flow_matching_loss": 0.1357421875,
      "train_physics_loss": 0.0
    },
    {
      "step": 12800,
      "train_flow_matching_loss": 0.1845703125,
      "train_physics_loss": 0.0
    },
    {
      "step": 12800,
      "train_flow_matching_loss": 0.1689453125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.19463947415351868,
      "learning_rate": 5e-05,
      "loss": 0.1499,
      "step": 12810
    },
    {
      "step": 12810,
      "train_flow_matching_loss": 0.255859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 12810,
      "train_flow_matching_loss": 0.142578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 12810,
      "train_flow_matching_loss": 0.2265625,
      "train_physics_loss": 0.0
    },
    {
      "step": 12810,
      "train_flow_matching_loss": 0.09765625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.0937717854976654,
      "learning_rate": 5e-05,
      "loss": 0.1466,
      "step": 12820
    },
    {
      "step": 12820,
      "train_flow_matching_loss": 0.0634765625,
      "train_physics_loss": 0.6275708079338074
    },
    {
      "step": 12820,
      "train_flow_matching_loss": 0.1357421875,
      "train_physics_loss": 0.0
    },
    {
      "step": 12820,
      "train_flow_matching_loss": 0.11181640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 12820,
      "train_flow_matching_loss": 0.0849609375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08207585662603378,
      "learning_rate": 5e-05,
      "loss": 0.1425,
      "step": 12830
    },
    {
      "step": 12830,
      "train_flow_matching_loss": 0.059814453125,
      "train_physics_loss": 0.5710270404815674
    },
    {
      "step": 12830,
      "train_flow_matching_loss": 0.1337890625,
      "train_physics_loss": 0.0
    },
    {
      "step": 12830,
      "train_flow_matching_loss": 0.08642578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 12830,
      "train_flow_matching_loss": 0.197265625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.07958545535802841,
      "learning_rate": 5e-05,
      "loss": 0.1471,
      "step": 12840
    },
    {
      "step": 12840,
      "train_flow_matching_loss": 0.0859375,
      "train_physics_loss": 0.547249436378479
    },
    {
      "step": 12840,
      "train_flow_matching_loss": 0.1396484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 12840,
      "train_flow_matching_loss": 0.1728515625,
      "train_physics_loss": 0.0
    },
    {
      "step": 12840,
      "train_flow_matching_loss": 0.05908203125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.14308711886405945,
      "learning_rate": 5e-05,
      "loss": 0.1412,
      "step": 12850
    },
    {
      "step": 12850,
      "train_flow_matching_loss": 0.06494140625,
      "train_physics_loss": 0.6048845052719116
    },
    {
      "step": 12850,
      "train_flow_matching_loss": 0.1796875,
      "train_physics_loss": 0.0
    },
    {
      "step": 12850,
      "train_flow_matching_loss": 0.158203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 12850,
      "train_flow_matching_loss": 0.197265625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.12575043737888336,
      "learning_rate": 5e-05,
      "loss": 0.1452,
      "step": 12860
    },
    {
      "step": 12860,
      "train_flow_matching_loss": 0.057861328125,
      "train_physics_loss": 0.5655050873756409
    },
    {
      "step": 12860,
      "train_flow_matching_loss": 0.11279296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 12860,
      "train_flow_matching_loss": 0.1083984375,
      "train_physics_loss": 0.0
    },
    {
      "step": 12860,
      "train_flow_matching_loss": 0.11279296875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.0873633325099945,
      "learning_rate": 5e-05,
      "loss": 0.1439,
      "step": 12870
    },
    {
      "step": 12870,
      "train_flow_matching_loss": 0.0791015625,
      "train_physics_loss": 0.6191795468330383
    },
    {
      "step": 12870,
      "train_flow_matching_loss": 0.10107421875,
      "train_physics_loss": 0.0
    },
    {
      "step": 12870,
      "train_flow_matching_loss": 0.11865234375,
      "train_physics_loss": 0.0
    },
    {
      "step": 12870,
      "train_flow_matching_loss": 0.1513671875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09515372663736343,
      "learning_rate": 5e-05,
      "loss": 0.1427,
      "step": 12880
    },
    {
      "step": 12880,
      "train_flow_matching_loss": 0.08642578125,
      "train_physics_loss": 0.5772161483764648
    },
    {
      "step": 12880,
      "train_flow_matching_loss": 0.154296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 12880,
      "train_flow_matching_loss": 0.169921875,
      "train_physics_loss": 0.0
    },
    {
      "step": 12880,
      "train_flow_matching_loss": 0.12060546875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.25231191515922546,
      "learning_rate": 5e-05,
      "loss": 0.1436,
      "step": 12890
    },
    {
      "step": 12890,
      "train_flow_matching_loss": 0.0673828125,
      "train_physics_loss": 0.5575106143951416
    },
    {
      "step": 12890,
      "train_flow_matching_loss": 0.1416015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 12890,
      "train_flow_matching_loss": 0.1767578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 12890,
      "train_flow_matching_loss": 0.1728515625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.0851062685251236,
      "learning_rate": 5e-05,
      "loss": 0.1443,
      "step": 12900
    },
    {
      "step": 12900,
      "train_flow_matching_loss": 0.050048828125,
      "train_physics_loss": 0.5683948397636414
    },
    {
      "step": 12900,
      "train_flow_matching_loss": 0.11572265625,
      "train_physics_loss": 0.0
    },
    {
      "step": 12900,
      "train_flow_matching_loss": 0.1708984375,
      "train_physics_loss": 0.0
    },
    {
      "step": 12900,
      "train_flow_matching_loss": 0.173828125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10332545638084412,
      "learning_rate": 5e-05,
      "loss": 0.1452,
      "step": 12910
    },
    {
      "step": 12910,
      "train_flow_matching_loss": 0.050048828125,
      "train_physics_loss": 0.5157502889633179
    },
    {
      "step": 12910,
      "train_flow_matching_loss": 0.1923828125,
      "train_physics_loss": 0.0
    },
    {
      "step": 12910,
      "train_flow_matching_loss": 0.10400390625,
      "train_physics_loss": 0.0
    },
    {
      "step": 12910,
      "train_flow_matching_loss": 0.1015625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.07594352215528488,
      "learning_rate": 5e-05,
      "loss": 0.146,
      "step": 12920
    },
    {
      "step": 12920,
      "train_flow_matching_loss": 0.1298828125,
      "train_physics_loss": 0.6350038051605225
    },
    {
      "step": 12920,
      "train_flow_matching_loss": 0.181640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 12920,
      "train_flow_matching_loss": 0.1572265625,
      "train_physics_loss": 0.0
    },
    {
      "step": 12920,
      "train_flow_matching_loss": 0.1396484375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.17387782037258148,
      "learning_rate": 5e-05,
      "loss": 0.1443,
      "step": 12930
    },
    {
      "step": 12930,
      "train_flow_matching_loss": 0.078125,
      "train_physics_loss": 0.6415193676948547
    },
    {
      "step": 12930,
      "train_flow_matching_loss": 0.185546875,
      "train_physics_loss": 0.0
    },
    {
      "step": 12930,
      "train_flow_matching_loss": 0.11865234375,
      "train_physics_loss": 0.0
    },
    {
      "step": 12930,
      "train_flow_matching_loss": 0.10693359375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.05753621459007263,
      "learning_rate": 5e-05,
      "loss": 0.1419,
      "step": 12940
    },
    {
      "step": 12940,
      "train_flow_matching_loss": 0.0888671875,
      "train_physics_loss": 0.6048800349235535
    },
    {
      "step": 12940,
      "train_flow_matching_loss": 0.2060546875,
      "train_physics_loss": 0.0
    },
    {
      "step": 12940,
      "train_flow_matching_loss": 0.1416015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 12940,
      "train_flow_matching_loss": 0.162109375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.1043257787823677,
      "learning_rate": 5e-05,
      "loss": 0.1441,
      "step": 12950
    },
    {
      "step": 12950,
      "train_flow_matching_loss": 0.053955078125,
      "train_physics_loss": 0.5980367660522461
    },
    {
      "step": 12950,
      "train_flow_matching_loss": 0.2333984375,
      "train_physics_loss": 0.0
    },
    {
      "step": 12950,
      "train_flow_matching_loss": 0.1767578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 12950,
      "train_flow_matching_loss": 0.08642578125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08036669343709946,
      "learning_rate": 5e-05,
      "loss": 0.1414,
      "step": 12960
    },
    {
      "step": 12960,
      "train_flow_matching_loss": 0.11181640625,
      "train_physics_loss": 0.6939288973808289
    },
    {
      "step": 12960,
      "train_flow_matching_loss": 0.07421875,
      "train_physics_loss": 0.0
    },
    {
      "step": 12960,
      "train_flow_matching_loss": 0.1787109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 12960,
      "train_flow_matching_loss": 0.1181640625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.0982985720038414,
      "learning_rate": 5e-05,
      "loss": 0.1444,
      "step": 12970
    },
    {
      "step": 12970,
      "train_flow_matching_loss": 0.04443359375,
      "train_physics_loss": 0.5199130177497864
    },
    {
      "step": 12970,
      "train_flow_matching_loss": 0.0966796875,
      "train_physics_loss": 0.0
    },
    {
      "step": 12970,
      "train_flow_matching_loss": 0.208984375,
      "train_physics_loss": 0.0
    },
    {
      "step": 12970,
      "train_flow_matching_loss": 0.134765625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.18924197554588318,
      "learning_rate": 5e-05,
      "loss": 0.1448,
      "step": 12980
    },
    {
      "step": 12980,
      "train_flow_matching_loss": 0.1015625,
      "train_physics_loss": 0.5338549017906189
    },
    {
      "step": 12980,
      "train_flow_matching_loss": 0.0810546875,
      "train_physics_loss": 0.0
    },
    {
      "step": 12980,
      "train_flow_matching_loss": 0.177734375,
      "train_physics_loss": 0.0
    },
    {
      "step": 12980,
      "train_flow_matching_loss": 0.1240234375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08823633193969727,
      "learning_rate": 5e-05,
      "loss": 0.1458,
      "step": 12990
    },
    {
      "step": 12990,
      "train_flow_matching_loss": 0.057373046875,
      "train_physics_loss": 0.5236258506774902
    },
    {
      "step": 12990,
      "train_flow_matching_loss": 0.166015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 12990,
      "train_flow_matching_loss": 0.1201171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 12990,
      "train_flow_matching_loss": 0.1162109375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.1024790108203888,
      "learning_rate": 5e-05,
      "loss": 0.1454,
      "step": 13000
    },
    {
      "step": 13000,
      "train_flow_matching_loss": 0.059326171875,
      "train_physics_loss": 0.5610686540603638
    },
    {
      "step": 13000,
      "train_flow_matching_loss": 0.0927734375,
      "train_physics_loss": 0.0
    },
    {
      "step": 13000,
      "train_flow_matching_loss": 0.1376953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 13000,
      "train_flow_matching_loss": 0.1787109375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10440017282962799,
      "learning_rate": 5e-05,
      "loss": 0.1481,
      "step": 13010
    },
    {
      "step": 13010,
      "train_flow_matching_loss": 0.1123046875,
      "train_physics_loss": 0.5901439785957336
    },
    {
      "step": 13010,
      "train_flow_matching_loss": 0.1796875,
      "train_physics_loss": 0.0
    },
    {
      "step": 13010,
      "train_flow_matching_loss": 0.154296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 13010,
      "train_flow_matching_loss": 0.142578125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10378670692443848,
      "learning_rate": 5e-05,
      "loss": 0.1423,
      "step": 13020
    },
    {
      "step": 13020,
      "train_flow_matching_loss": 0.06298828125,
      "train_physics_loss": 0.572398841381073
    },
    {
      "step": 13020,
      "train_flow_matching_loss": 0.1787109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 13020,
      "train_flow_matching_loss": 0.1640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 13020,
      "train_flow_matching_loss": 0.1435546875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.1728595346212387,
      "learning_rate": 5e-05,
      "loss": 0.1442,
      "step": 13030
    },
    {
      "step": 13030,
      "train_flow_matching_loss": 0.0634765625,
      "train_physics_loss": 0.6389660239219666
    },
    {
      "step": 13030,
      "train_flow_matching_loss": 0.13671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 13030,
      "train_flow_matching_loss": 0.1396484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 13030,
      "train_flow_matching_loss": 0.177734375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.0768958106637001,
      "learning_rate": 5e-05,
      "loss": 0.1424,
      "step": 13040
    },
    {
      "step": 13040,
      "train_flow_matching_loss": 0.0751953125,
      "train_physics_loss": 0.5680235028266907
    },
    {
      "step": 13040,
      "train_flow_matching_loss": 0.1611328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 13040,
      "train_flow_matching_loss": 0.07861328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 13040,
      "train_flow_matching_loss": 0.1591796875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09987699240446091,
      "learning_rate": 5e-05,
      "loss": 0.1447,
      "step": 13050
    },
    {
      "step": 13050,
      "train_flow_matching_loss": 0.03369140625,
      "train_physics_loss": 0.5664889216423035
    },
    {
      "step": 13050,
      "train_flow_matching_loss": 0.1015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 13050,
      "train_flow_matching_loss": 0.1796875,
      "train_physics_loss": 0.0
    },
    {
      "step": 13050,
      "train_flow_matching_loss": 0.1748046875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11300074309110641,
      "learning_rate": 5e-05,
      "loss": 0.1465,
      "step": 13060
    },
    {
      "step": 13060,
      "train_flow_matching_loss": 0.035888671875,
      "train_physics_loss": 0.5248883366584778
    },
    {
      "step": 13060,
      "train_flow_matching_loss": 0.10888671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 13060,
      "train_flow_matching_loss": 0.11962890625,
      "train_physics_loss": 0.0
    },
    {
      "step": 13060,
      "train_flow_matching_loss": 0.035888671875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.124582439661026,
      "learning_rate": 5e-05,
      "loss": 0.138,
      "step": 13070
    },
    {
      "step": 13070,
      "train_flow_matching_loss": 0.08447265625,
      "train_physics_loss": 0.5465453267097473
    },
    {
      "step": 13070,
      "train_flow_matching_loss": 0.1650390625,
      "train_physics_loss": 0.0
    },
    {
      "step": 13070,
      "train_flow_matching_loss": 0.169921875,
      "train_physics_loss": 0.0
    },
    {
      "step": 13070,
      "train_flow_matching_loss": 0.130859375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11215104162693024,
      "learning_rate": 5e-05,
      "loss": 0.1433,
      "step": 13080
    },
    {
      "step": 13080,
      "train_flow_matching_loss": 0.1865234375,
      "train_physics_loss": 0.0
    },
    {
      "step": 13080,
      "train_flow_matching_loss": 0.1015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 13080,
      "train_flow_matching_loss": 0.1318359375,
      "train_physics_loss": 0.0
    },
    {
      "step": 13080,
      "train_flow_matching_loss": 0.134765625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.0795990452170372,
      "learning_rate": 5e-05,
      "loss": 0.1443,
      "step": 13090
    },
    {
      "step": 13090,
      "train_flow_matching_loss": 0.1015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 13090,
      "train_flow_matching_loss": 0.1416015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 13090,
      "train_flow_matching_loss": 0.1279296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 13090,
      "train_flow_matching_loss": 0.1044921875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09313444048166275,
      "learning_rate": 5e-05,
      "loss": 0.1431,
      "step": 13100
    },
    {
      "step": 13100,
      "train_flow_matching_loss": 0.1767578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 13100,
      "train_flow_matching_loss": 0.30078125,
      "train_physics_loss": 0.0
    },
    {
      "step": 13100,
      "train_flow_matching_loss": 0.11328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 13100,
      "train_flow_matching_loss": 0.1474609375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.07196172326803207,
      "learning_rate": 5e-05,
      "loss": 0.1419,
      "step": 13110
    },
    {
      "step": 13110,
      "train_flow_matching_loss": 0.142578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 13110,
      "train_flow_matching_loss": 0.1435546875,
      "train_physics_loss": 0.0
    },
    {
      "step": 13110,
      "train_flow_matching_loss": 0.2001953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 13110,
      "train_flow_matching_loss": 0.08251953125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.1348866969347,
      "learning_rate": 5e-05,
      "loss": 0.1468,
      "step": 13120
    },
    {
      "step": 13120,
      "train_flow_matching_loss": 0.11572265625,
      "train_physics_loss": 0.0
    },
    {
      "step": 13120,
      "train_flow_matching_loss": 0.263671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 13120,
      "train_flow_matching_loss": 0.12890625,
      "train_physics_loss": 0.0
    },
    {
      "step": 13120,
      "train_flow_matching_loss": 0.08203125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.1429612785577774,
      "learning_rate": 5e-05,
      "loss": 0.148,
      "step": 13130
    },
    {
      "step": 13130,
      "train_flow_matching_loss": 0.177734375,
      "train_physics_loss": 0.0
    },
    {
      "step": 13130,
      "train_flow_matching_loss": 0.150390625,
      "train_physics_loss": 0.0
    },
    {
      "step": 13130,
      "train_flow_matching_loss": 0.1396484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 13130,
      "train_flow_matching_loss": 0.0791015625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10372084379196167,
      "learning_rate": 5e-05,
      "loss": 0.1419,
      "step": 13140
    },
    {
      "step": 13140,
      "train_flow_matching_loss": 0.10888671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 13140,
      "train_flow_matching_loss": 0.169921875,
      "train_physics_loss": 0.0
    },
    {
      "step": 13140,
      "train_flow_matching_loss": 0.1474609375,
      "train_physics_loss": 0.0
    },
    {
      "step": 13140,
      "train_flow_matching_loss": 0.197265625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10993526875972748,
      "learning_rate": 5e-05,
      "loss": 0.1439,
      "step": 13150
    },
    {
      "step": 13150,
      "train_flow_matching_loss": 0.1845703125,
      "train_physics_loss": 0.0
    },
    {
      "step": 13150,
      "train_flow_matching_loss": 0.146484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 13150,
      "train_flow_matching_loss": 0.1201171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 13150,
      "train_flow_matching_loss": 0.0791015625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.07780591398477554,
      "learning_rate": 5e-05,
      "loss": 0.1388,
      "step": 13160
    },
    {
      "step": 13160,
      "train_flow_matching_loss": 0.1806640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 13160,
      "train_flow_matching_loss": 0.2490234375,
      "train_physics_loss": 0.0
    },
    {
      "step": 13160,
      "train_flow_matching_loss": 0.12451171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 13160,
      "train_flow_matching_loss": 0.142578125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.1350005716085434,
      "learning_rate": 5e-05,
      "loss": 0.1425,
      "step": 13170
    },
    {
      "step": 13170,
      "train_flow_matching_loss": 0.1845703125,
      "train_physics_loss": 0.0
    },
    {
      "step": 13170,
      "train_flow_matching_loss": 0.06689453125,
      "train_physics_loss": 0.5943811535835266
    },
    {
      "step": 13170,
      "train_flow_matching_loss": 0.1640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 13170,
      "train_flow_matching_loss": 0.1103515625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.1275300681591034,
      "learning_rate": 5e-05,
      "loss": 0.1441,
      "step": 13180
    },
    {
      "step": 13180,
      "train_flow_matching_loss": 0.10791015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 13180,
      "train_flow_matching_loss": 0.10302734375,
      "train_physics_loss": 0.6198968291282654
    },
    {
      "step": 13180,
      "train_flow_matching_loss": 0.1455078125,
      "train_physics_loss": 0.0
    },
    {
      "step": 13180,
      "train_flow_matching_loss": 0.0927734375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.06759755313396454,
      "learning_rate": 5e-05,
      "loss": 0.1384,
      "step": 13190
    },
    {
      "step": 13190,
      "train_flow_matching_loss": 0.189453125,
      "train_physics_loss": 0.0
    },
    {
      "step": 13190,
      "train_flow_matching_loss": 0.05615234375,
      "train_physics_loss": 0.47006770968437195
    },
    {
      "step": 13190,
      "train_flow_matching_loss": 0.1875,
      "train_physics_loss": 0.0
    },
    {
      "step": 13190,
      "train_flow_matching_loss": 0.08544921875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10649221390485764,
      "learning_rate": 5e-05,
      "loss": 0.1498,
      "step": 13200
    },
    {
      "step": 13200,
      "train_flow_matching_loss": 0.1513671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 13200,
      "train_flow_matching_loss": 0.035888671875,
      "train_physics_loss": 0.5057836771011353
    },
    {
      "step": 13200,
      "train_flow_matching_loss": 0.11328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 13200,
      "train_flow_matching_loss": 0.11572265625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.07229616492986679,
      "learning_rate": 5e-05,
      "loss": 0.1431,
      "step": 13210
    },
    {
      "step": 13210,
      "train_flow_matching_loss": 0.15625,
      "train_physics_loss": 0.0
    },
    {
      "step": 13210,
      "train_flow_matching_loss": 0.04052734375,
      "train_physics_loss": 0.5183820128440857
    },
    {
      "step": 13210,
      "train_flow_matching_loss": 0.1279296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 13210,
      "train_flow_matching_loss": 0.1171875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09276603162288666,
      "learning_rate": 5e-05,
      "loss": 0.1448,
      "step": 13220
    },
    {
      "step": 13220,
      "train_flow_matching_loss": 0.166015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 13220,
      "train_flow_matching_loss": 0.040771484375,
      "train_physics_loss": 0.5598906874656677
    },
    {
      "step": 13220,
      "train_flow_matching_loss": 0.2138671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 13220,
      "train_flow_matching_loss": 0.1474609375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.14794422686100006,
      "learning_rate": 5e-05,
      "loss": 0.1466,
      "step": 13230
    },
    {
      "step": 13230,
      "train_flow_matching_loss": 0.1357421875,
      "train_physics_loss": 0.0
    },
    {
      "step": 13230,
      "train_flow_matching_loss": 0.059326171875,
      "train_physics_loss": 0.5475823879241943
    },
    {
      "step": 13230,
      "train_flow_matching_loss": 0.1513671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 13230,
      "train_flow_matching_loss": 0.166015625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10524102300405502,
      "learning_rate": 5e-05,
      "loss": 0.1428,
      "step": 13240
    },
    {
      "step": 13240,
      "train_flow_matching_loss": 0.146484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 13240,
      "train_flow_matching_loss": 0.058837890625,
      "train_physics_loss": 0.5842326283454895
    },
    {
      "step": 13240,
      "train_flow_matching_loss": 0.263671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 13240,
      "train_flow_matching_loss": 0.11083984375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.0970182865858078,
      "learning_rate": 5e-05,
      "loss": 0.1469,
      "step": 13250
    },
    {
      "step": 13250,
      "train_flow_matching_loss": 0.12158203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 13250,
      "train_flow_matching_loss": 0.0751953125,
      "train_physics_loss": 0.6331334710121155
    },
    {
      "step": 13250,
      "train_flow_matching_loss": 0.1494140625,
      "train_physics_loss": 0.0
    },
    {
      "step": 13250,
      "train_flow_matching_loss": 0.158203125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08502712100744247,
      "learning_rate": 5e-05,
      "loss": 0.1471,
      "step": 13260
    },
    {
      "step": 13260,
      "train_flow_matching_loss": 0.1630859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 13260,
      "train_flow_matching_loss": 0.04443359375,
      "train_physics_loss": 0.5475220084190369
    },
    {
      "step": 13260,
      "train_flow_matching_loss": 0.1484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 13260,
      "train_flow_matching_loss": 0.1552734375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10596468299627304,
      "learning_rate": 5e-05,
      "loss": 0.1475,
      "step": 13270
    },
    {
      "step": 13270,
      "train_flow_matching_loss": 0.1748046875,
      "train_physics_loss": 0.0
    },
    {
      "step": 13270,
      "train_flow_matching_loss": 0.0966796875,
      "train_physics_loss": 0.5749532580375671
    },
    {
      "step": 13270,
      "train_flow_matching_loss": 0.2333984375,
      "train_physics_loss": 0.0
    },
    {
      "step": 13270,
      "train_flow_matching_loss": 0.10888671875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.06968491524457932,
      "learning_rate": 5e-05,
      "loss": 0.1414,
      "step": 13280
    },
    {
      "step": 13280,
      "train_flow_matching_loss": 0.1669921875,
      "train_physics_loss": 0.0
    },
    {
      "step": 13280,
      "train_flow_matching_loss": 0.0751953125,
      "train_physics_loss": 0.4972655773162842
    },
    {
      "step": 13280,
      "train_flow_matching_loss": 0.1923828125,
      "train_physics_loss": 0.0
    },
    {
      "step": 13280,
      "train_flow_matching_loss": 0.1767578125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11574403941631317,
      "learning_rate": 5e-05,
      "loss": 0.1476,
      "step": 13290
    },
    {
      "step": 13290,
      "train_flow_matching_loss": 0.1435546875,
      "train_physics_loss": 0.0
    },
    {
      "step": 13290,
      "train_flow_matching_loss": 0.062255859375,
      "train_physics_loss": 0.5734683275222778
    },
    {
      "step": 13290,
      "train_flow_matching_loss": 0.1982421875,
      "train_physics_loss": 0.0
    },
    {
      "step": 13290,
      "train_flow_matching_loss": 0.1015625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09332116693258286,
      "learning_rate": 5e-05,
      "loss": 0.147,
      "step": 13300
    },
    {
      "step": 13300,
      "train_flow_matching_loss": 0.2060546875,
      "train_physics_loss": 0.0
    },
    {
      "step": 13300,
      "train_flow_matching_loss": 0.045654296875,
      "train_physics_loss": 0.5669358968734741
    },
    {
      "step": 13300,
      "train_flow_matching_loss": 0.2197265625,
      "train_physics_loss": 0.0
    },
    {
      "step": 13300,
      "train_flow_matching_loss": 0.2158203125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08053693920373917,
      "learning_rate": 5e-05,
      "loss": 0.1371,
      "step": 13310
    },
    {
      "step": 13310,
      "train_flow_matching_loss": 0.1337890625,
      "train_physics_loss": 0.0
    },
    {
      "step": 13310,
      "train_flow_matching_loss": 0.12353515625,
      "train_physics_loss": 0.5457478761672974
    },
    {
      "step": 13310,
      "train_flow_matching_loss": 0.1533203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 13310,
      "train_flow_matching_loss": 0.08837890625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.1427958905696869,
      "learning_rate": 5e-05,
      "loss": 0.1447,
      "step": 13320
    },
    {
      "step": 13320,
      "train_flow_matching_loss": 0.1435546875,
      "train_physics_loss": 0.0
    },
    {
      "step": 13320,
      "train_flow_matching_loss": 0.0546875,
      "train_physics_loss": 0.55079185962677
    },
    {
      "step": 13320,
      "train_flow_matching_loss": 0.328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 13320,
      "train_flow_matching_loss": 0.1416015625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.07641550898551941,
      "learning_rate": 5e-05,
      "loss": 0.1436,
      "step": 13330
    },
    {
      "step": 13330,
      "train_flow_matching_loss": 0.09375,
      "train_physics_loss": 0.5929238796234131
    },
    {
      "step": 13330,
      "train_flow_matching_loss": 0.06005859375,
      "train_physics_loss": 0.5521538257598877
    },
    {
      "step": 13330,
      "train_flow_matching_loss": 0.1640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 13330,
      "train_flow_matching_loss": 0.1923828125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08653002977371216,
      "learning_rate": 5e-05,
      "loss": 0.1423,
      "step": 13340
    },
    {
      "step": 13340,
      "train_flow_matching_loss": 0.044189453125,
      "train_physics_loss": 0.5917474627494812
    },
    {
      "step": 13340,
      "train_flow_matching_loss": 0.09326171875,
      "train_physics_loss": 0.4891117513179779
    },
    {
      "step": 13340,
      "train_flow_matching_loss": 0.171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 13340,
      "train_flow_matching_loss": 0.1630859375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.06414114683866501,
      "learning_rate": 5e-05,
      "loss": 0.1362,
      "step": 13350
    },
    {
      "step": 13350,
      "train_flow_matching_loss": 0.043212890625,
      "train_physics_loss": 0.5658658146858215
    },
    {
      "step": 13350,
      "train_flow_matching_loss": 0.07177734375,
      "train_physics_loss": 0.5403662919998169
    },
    {
      "step": 13350,
      "train_flow_matching_loss": 0.1591796875,
      "train_physics_loss": 0.0
    },
    {
      "step": 13350,
      "train_flow_matching_loss": 0.1240234375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.07664921134710312,
      "learning_rate": 5e-05,
      "loss": 0.1412,
      "step": 13360
    },
    {
      "step": 13360,
      "train_flow_matching_loss": 0.049072265625,
      "train_physics_loss": 0.5616970658302307
    },
    {
      "step": 13360,
      "train_flow_matching_loss": 0.1044921875,
      "train_physics_loss": 0.6843598484992981
    },
    {
      "step": 13360,
      "train_flow_matching_loss": 0.162109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 13360,
      "train_flow_matching_loss": 0.1416015625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.1111908107995987,
      "learning_rate": 5e-05,
      "loss": 0.1383,
      "step": 13370
    },
    {
      "step": 13370,
      "train_flow_matching_loss": 0.07861328125,
      "train_physics_loss": 0.6433392763137817
    },
    {
      "step": 13370,
      "train_flow_matching_loss": 0.044921875,
      "train_physics_loss": 0.5761138200759888
    },
    {
      "step": 13370,
      "train_flow_matching_loss": 0.1259765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 13370,
      "train_flow_matching_loss": 0.1435546875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.0945654883980751,
      "learning_rate": 5e-05,
      "loss": 0.1387,
      "step": 13380
    },
    {
      "step": 13380,
      "train_flow_matching_loss": 0.058837890625,
      "train_physics_loss": 0.5543280243873596
    },
    {
      "step": 13380,
      "train_flow_matching_loss": 0.046142578125,
      "train_physics_loss": 0.5043354034423828
    },
    {
      "step": 13380,
      "train_flow_matching_loss": 0.1357421875,
      "train_physics_loss": 0.0
    },
    {
      "step": 13380,
      "train_flow_matching_loss": 0.138671875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.0896901860833168,
      "learning_rate": 5e-05,
      "loss": 0.1442,
      "step": 13390
    },
    {
      "step": 13390,
      "train_flow_matching_loss": 0.057373046875,
      "train_physics_loss": 0.5164749026298523
    },
    {
      "step": 13390,
      "train_flow_matching_loss": 0.054443359375,
      "train_physics_loss": 0.5097777843475342
    },
    {
      "step": 13390,
      "train_flow_matching_loss": 0.146484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 13390,
      "train_flow_matching_loss": 0.07275390625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10711207985877991,
      "learning_rate": 5e-05,
      "loss": 0.1367,
      "step": 13400
    },
    {
      "step": 13400,
      "train_flow_matching_loss": 0.0869140625,
      "train_physics_loss": 0.6281992793083191
    },
    {
      "step": 13400,
      "train_flow_matching_loss": 0.05712890625,
      "train_physics_loss": 0.5154262781143188
    },
    {
      "step": 13400,
      "train_flow_matching_loss": 0.1875,
      "train_physics_loss": 0.0
    },
    {
      "step": 13400,
      "train_flow_matching_loss": 0.130859375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08358393609523773,
      "learning_rate": 5e-05,
      "loss": 0.1402,
      "step": 13410
    },
    {
      "step": 13410,
      "train_flow_matching_loss": 0.1015625,
      "train_physics_loss": 0.606894850730896
    },
    {
      "step": 13410,
      "train_flow_matching_loss": 0.052001953125,
      "train_physics_loss": 0.5444029569625854
    },
    {
      "step": 13410,
      "train_flow_matching_loss": 0.1435546875,
      "train_physics_loss": 0.0
    },
    {
      "step": 13410,
      "train_flow_matching_loss": 0.173828125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09456732124090195,
      "learning_rate": 5e-05,
      "loss": 0.1434,
      "step": 13420
    },
    {
      "step": 13420,
      "train_flow_matching_loss": 0.08447265625,
      "train_physics_loss": 0.5585141777992249
    },
    {
      "step": 13420,
      "train_flow_matching_loss": 0.09326171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 13420,
      "train_flow_matching_loss": 0.2431640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 13420,
      "train_flow_matching_loss": 0.12109375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.17740464210510254,
      "learning_rate": 5e-05,
      "loss": 0.1465,
      "step": 13430
    },
    {
      "step": 13430,
      "train_flow_matching_loss": 0.06591796875,
      "train_physics_loss": 0.5452973246574402
    },
    {
      "step": 13430,
      "train_flow_matching_loss": 0.126953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 13430,
      "train_flow_matching_loss": 0.265625,
      "train_physics_loss": 0.0
    },
    {
      "step": 13430,
      "train_flow_matching_loss": 0.1337890625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10477649420499802,
      "learning_rate": 5e-05,
      "loss": 0.1453,
      "step": 13440
    },
    {
      "step": 13440,
      "train_flow_matching_loss": 0.08056640625,
      "train_physics_loss": 0.5680809020996094
    },
    {
      "step": 13440,
      "train_flow_matching_loss": 0.1318359375,
      "train_physics_loss": 0.0
    },
    {
      "step": 13440,
      "train_flow_matching_loss": 0.1767578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 13440,
      "train_flow_matching_loss": 0.09033203125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.13871440291404724,
      "learning_rate": 5e-05,
      "loss": 0.147,
      "step": 13450
    },
    {
      "step": 13450,
      "train_flow_matching_loss": 0.060302734375,
      "train_physics_loss": 0.5955352783203125
    },
    {
      "step": 13450,
      "train_flow_matching_loss": 0.1455078125,
      "train_physics_loss": 0.0
    },
    {
      "step": 13450,
      "train_flow_matching_loss": 0.162109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 13450,
      "train_flow_matching_loss": 0.09326171875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11069118976593018,
      "learning_rate": 5e-05,
      "loss": 0.1423,
      "step": 13460
    },
    {
      "step": 13460,
      "train_flow_matching_loss": 0.07568359375,
      "train_physics_loss": 0.5047733783721924
    },
    {
      "step": 13460,
      "train_flow_matching_loss": 0.189453125,
      "train_physics_loss": 0.0
    },
    {
      "step": 13460,
      "train_flow_matching_loss": 0.1533203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 13460,
      "train_flow_matching_loss": 0.08642578125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08037852495908737,
      "learning_rate": 5e-05,
      "loss": 0.1436,
      "step": 13470
    },
    {
      "step": 13470,
      "train_flow_matching_loss": 0.041259765625,
      "train_physics_loss": 0.46311506628990173
    },
    {
      "step": 13470,
      "train_flow_matching_loss": 0.142578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 13470,
      "train_flow_matching_loss": 0.1923828125,
      "train_physics_loss": 0.0
    },
    {
      "step": 13470,
      "train_flow_matching_loss": 0.09326171875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08222074806690216,
      "learning_rate": 5e-05,
      "loss": 0.1402,
      "step": 13480
    },
    {
      "step": 13480,
      "train_flow_matching_loss": 0.07275390625,
      "train_physics_loss": 0.5798841118812561
    },
    {
      "step": 13480,
      "train_flow_matching_loss": 0.10107421875,
      "train_physics_loss": 0.0
    },
    {
      "step": 13480,
      "train_flow_matching_loss": 0.169921875,
      "train_physics_loss": 0.5486432909965515
    },
    {
      "step": 13480,
      "train_flow_matching_loss": 0.142578125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10381629317998886,
      "learning_rate": 5e-05,
      "loss": 0.1518,
      "step": 13490
    },
    {
      "step": 13490,
      "train_flow_matching_loss": 0.125,
      "train_physics_loss": 0.5412933826446533
    },
    {
      "step": 13490,
      "train_flow_matching_loss": 0.2275390625,
      "train_physics_loss": 0.0
    },
    {
      "step": 13490,
      "train_flow_matching_loss": 0.053955078125,
      "train_physics_loss": 0.5412256121635437
    },
    {
      "step": 13490,
      "train_flow_matching_loss": 0.158203125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08653941750526428,
      "learning_rate": 5e-05,
      "loss": 0.1434,
      "step": 13500
    },
    {
      "step": 13500,
      "train_flow_matching_loss": 0.0625,
      "train_physics_loss": 0.5312855839729309
    },
    {
      "step": 13500,
      "train_flow_matching_loss": 0.2333984375,
      "train_physics_loss": 0.0
    },
    {
      "step": 13500,
      "train_flow_matching_loss": 0.1640625,
      "train_physics_loss": 0.49828803539276123
    },
    {
      "step": 13500,
      "train_flow_matching_loss": 0.0751953125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11344141513109207,
      "learning_rate": 5e-05,
      "loss": 0.1455,
      "step": 13510
    },
    {
      "step": 13510,
      "train_flow_matching_loss": 0.08935546875,
      "train_physics_loss": 0.5084046125411987
    },
    {
      "step": 13510,
      "train_flow_matching_loss": 0.1826171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 13510,
      "train_flow_matching_loss": 0.04931640625,
      "train_physics_loss": 0.5096040964126587
    },
    {
      "step": 13510,
      "train_flow_matching_loss": 0.10498046875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.15403230488300323,
      "learning_rate": 5e-05,
      "loss": 0.1415,
      "step": 13520
    },
    {
      "step": 13520,
      "train_flow_matching_loss": 0.03369140625,
      "train_physics_loss": 0.5374127626419067
    },
    {
      "step": 13520,
      "train_flow_matching_loss": 0.1806640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 13520,
      "train_flow_matching_loss": 0.0712890625,
      "train_physics_loss": 0.5284395217895508
    },
    {
      "step": 13520,
      "train_flow_matching_loss": 0.11669921875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.1712426245212555,
      "learning_rate": 5e-05,
      "loss": 0.1414,
      "step": 13530
    },
    {
      "step": 13530,
      "train_flow_matching_loss": 0.06201171875,
      "train_physics_loss": 0.53216952085495
    },
    {
      "step": 13530,
      "train_flow_matching_loss": 0.181640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 13530,
      "train_flow_matching_loss": 0.06298828125,
      "train_physics_loss": 0.5379720330238342
    },
    {
      "step": 13530,
      "train_flow_matching_loss": 0.07568359375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11771569401025772,
      "learning_rate": 5e-05,
      "loss": 0.1461,
      "step": 13540
    },
    {
      "step": 13540,
      "train_flow_matching_loss": 0.0859375,
      "train_physics_loss": 0.5384583473205566
    },
    {
      "step": 13540,
      "train_flow_matching_loss": 0.1474609375,
      "train_physics_loss": 0.0
    },
    {
      "step": 13540,
      "train_flow_matching_loss": 0.07763671875,
      "train_physics_loss": 0.5621698498725891
    },
    {
      "step": 13540,
      "train_flow_matching_loss": 0.2099609375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09482621401548386,
      "learning_rate": 5e-05,
      "loss": 0.1463,
      "step": 13550
    },
    {
      "step": 13550,
      "train_flow_matching_loss": 0.09130859375,
      "train_physics_loss": 0.5402094125747681
    },
    {
      "step": 13550,
      "train_flow_matching_loss": 0.1689453125,
      "train_physics_loss": 0.0
    },
    {
      "step": 13550,
      "train_flow_matching_loss": 0.059326171875,
      "train_physics_loss": 0.5183961391448975
    },
    {
      "step": 13550,
      "train_flow_matching_loss": 0.146484375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.12733420729637146,
      "learning_rate": 5e-05,
      "loss": 0.1477,
      "step": 13560
    },
    {
      "step": 13560,
      "train_flow_matching_loss": 0.05419921875,
      "train_physics_loss": 0.595061719417572
    },
    {
      "step": 13560,
      "train_flow_matching_loss": 0.158203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 13560,
      "train_flow_matching_loss": 0.06103515625,
      "train_physics_loss": 0.5141715407371521
    },
    {
      "step": 13560,
      "train_flow_matching_loss": 0.1376953125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08744199573993683,
      "learning_rate": 5e-05,
      "loss": 0.137,
      "step": 13570
    },
    {
      "step": 13570,
      "train_flow_matching_loss": 0.12255859375,
      "train_physics_loss": 0.6506703495979309
    },
    {
      "step": 13570,
      "train_flow_matching_loss": 0.27734375,
      "train_physics_loss": 0.0
    },
    {
      "step": 13570,
      "train_flow_matching_loss": 0.07470703125,
      "train_physics_loss": 0.4866844415664673
    },
    {
      "step": 13570,
      "train_flow_matching_loss": 0.158203125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.07761654257774353,
      "learning_rate": 5e-05,
      "loss": 0.1472,
      "step": 13580
    },
    {
      "step": 13580,
      "train_flow_matching_loss": 0.07177734375,
      "train_physics_loss": 0.5326005816459656
    },
    {
      "step": 13580,
      "train_flow_matching_loss": 0.162109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 13580,
      "train_flow_matching_loss": 0.12353515625,
      "train_physics_loss": 0.5396987199783325
    },
    {
      "step": 13580,
      "train_flow_matching_loss": 0.1669921875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09162817150354385,
      "learning_rate": 5e-05,
      "loss": 0.1423,
      "step": 13590
    },
    {
      "step": 13590,
      "train_flow_matching_loss": 0.0693359375,
      "train_physics_loss": 0.5134056210517883
    },
    {
      "step": 13590,
      "train_flow_matching_loss": 0.0791015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 13590,
      "train_flow_matching_loss": 0.045166015625,
      "train_physics_loss": 0.5428693890571594
    },
    {
      "step": 13590,
      "train_flow_matching_loss": 0.138671875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09748910367488861,
      "learning_rate": 5e-05,
      "loss": 0.1429,
      "step": 13600
    },
    {
      "step": 13600,
      "train_flow_matching_loss": 0.0927734375,
      "train_physics_loss": 0.9206795692443848
    },
    {
      "step": 13600,
      "train_flow_matching_loss": 0.1201171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 13600,
      "train_flow_matching_loss": 0.0986328125,
      "train_physics_loss": 0.5210053324699402
    },
    {
      "step": 13600,
      "train_flow_matching_loss": 0.08837890625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.12788568437099457,
      "learning_rate": 5e-05,
      "loss": 0.1427,
      "step": 13610
    },
    {
      "step": 13610,
      "train_flow_matching_loss": 0.08251953125,
      "train_physics_loss": 0.5274748802185059
    },
    {
      "step": 13610,
      "train_flow_matching_loss": 0.1357421875,
      "train_physics_loss": 0.0
    },
    {
      "step": 13610,
      "train_flow_matching_loss": 0.06103515625,
      "train_physics_loss": 0.5685727596282959
    },
    {
      "step": 13610,
      "train_flow_matching_loss": 0.1845703125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.13927672803401947,
      "learning_rate": 5e-05,
      "loss": 0.144,
      "step": 13620
    },
    {
      "step": 13620,
      "train_flow_matching_loss": 0.109375,
      "train_physics_loss": 0.6587162613868713
    },
    {
      "step": 13620,
      "train_flow_matching_loss": 0.146484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 13620,
      "train_flow_matching_loss": 0.10595703125,
      "train_physics_loss": 0.4478304088115692
    },
    {
      "step": 13620,
      "train_flow_matching_loss": 0.177734375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.14453567564487457,
      "learning_rate": 5e-05,
      "loss": 0.1476,
      "step": 13630
    },
    {
      "step": 13630,
      "train_flow_matching_loss": 0.03955078125,
      "train_physics_loss": 0.6005984544754028
    },
    {
      "step": 13630,
      "train_flow_matching_loss": 0.181640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 13630,
      "train_flow_matching_loss": 0.036376953125,
      "train_physics_loss": 0.48659834265708923
    },
    {
      "step": 13630,
      "train_flow_matching_loss": 0.126953125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08493853360414505,
      "learning_rate": 5e-05,
      "loss": 0.1449,
      "step": 13640
    },
    {
      "step": 13640,
      "train_flow_matching_loss": 0.078125,
      "train_physics_loss": 0.6858162879943848
    },
    {
      "step": 13640,
      "train_flow_matching_loss": 0.1650390625,
      "train_physics_loss": 0.0
    },
    {
      "step": 13640,
      "train_flow_matching_loss": 0.059814453125,
      "train_physics_loss": 0.48872965574264526
    },
    {
      "step": 13640,
      "train_flow_matching_loss": 0.0830078125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.13161446154117584,
      "learning_rate": 5e-05,
      "loss": 0.143,
      "step": 13650
    },
    {
      "step": 13650,
      "train_flow_matching_loss": 0.173828125,
      "train_physics_loss": 0.5731845498085022
    },
    {
      "step": 13650,
      "train_flow_matching_loss": 0.205078125,
      "train_physics_loss": 0.0
    },
    {
      "step": 13650,
      "train_flow_matching_loss": 0.08056640625,
      "train_physics_loss": 0.48490384221076965
    },
    {
      "step": 13650,
      "train_flow_matching_loss": 0.1591796875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.14839106798171997,
      "learning_rate": 5e-05,
      "loss": 0.1474,
      "step": 13660
    },
    {
      "step": 13660,
      "train_flow_matching_loss": 0.0791015625,
      "train_physics_loss": 0.5455568432807922
    },
    {
      "step": 13660,
      "train_flow_matching_loss": 0.12353515625,
      "train_physics_loss": 0.0
    },
    {
      "step": 13660,
      "train_flow_matching_loss": 0.04931640625,
      "train_physics_loss": 0.5988472700119019
    },
    {
      "step": 13660,
      "train_flow_matching_loss": 0.1552734375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09027969092130661,
      "learning_rate": 5e-05,
      "loss": 0.1446,
      "step": 13670
    },
    {
      "step": 13670,
      "train_flow_matching_loss": 0.08642578125,
      "train_physics_loss": 0.4867352545261383
    },
    {
      "step": 13670,
      "train_flow_matching_loss": 0.2197265625,
      "train_physics_loss": 0.0
    },
    {
      "step": 13670,
      "train_flow_matching_loss": 0.1015625,
      "train_physics_loss": 0.6665010452270508
    },
    {
      "step": 13670,
      "train_flow_matching_loss": 0.1005859375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.07628229260444641,
      "learning_rate": 5e-05,
      "loss": 0.1441,
      "step": 13680
    },
    {
      "step": 13680,
      "train_flow_matching_loss": 0.08740234375,
      "train_physics_loss": 0.5672062039375305
    },
    {
      "step": 13680,
      "train_flow_matching_loss": 0.1455078125,
      "train_physics_loss": 0.0
    },
    {
      "step": 13680,
      "train_flow_matching_loss": 0.039306640625,
      "train_physics_loss": 0.45912623405456543
    },
    {
      "step": 13680,
      "train_flow_matching_loss": 0.09521484375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.1328280121088028,
      "learning_rate": 5e-05,
      "loss": 0.1385,
      "step": 13690
    },
    {
      "step": 13690,
      "train_flow_matching_loss": 0.048583984375,
      "train_physics_loss": 0.5034458637237549
    },
    {
      "step": 13690,
      "train_flow_matching_loss": 0.1142578125,
      "train_physics_loss": 0.6219753623008728
    },
    {
      "step": 13690,
      "train_flow_matching_loss": 0.08251953125,
      "train_physics_loss": 0.5326421856880188
    },
    {
      "step": 13690,
      "train_flow_matching_loss": 0.158203125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.07584349811077118,
      "learning_rate": 5e-05,
      "loss": 0.1428,
      "step": 13700
    },
    {
      "step": 13700,
      "train_flow_matching_loss": 0.058349609375,
      "train_physics_loss": 0.5161749124526978
    },
    {
      "step": 13700,
      "train_flow_matching_loss": 0.055419921875,
      "train_physics_loss": 0.5337357521057129
    },
    {
      "step": 13700,
      "train_flow_matching_loss": 0.09765625,
      "train_physics_loss": 0.606281578540802
    },
    {
      "step": 13700,
      "train_flow_matching_loss": 0.1650390625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09865792095661163,
      "learning_rate": 5e-05,
      "loss": 0.1437,
      "step": 13710
    },
    {
      "step": 13710,
      "train_flow_matching_loss": 0.0634765625,
      "train_physics_loss": 0.5518600344657898
    },
    {
      "step": 13710,
      "train_flow_matching_loss": 0.1298828125,
      "train_physics_loss": 0.5592809915542603
    },
    {
      "step": 13710,
      "train_flow_matching_loss": 0.0849609375,
      "train_physics_loss": 0.5926632285118103
    },
    {
      "step": 13710,
      "train_flow_matching_loss": 0.06494140625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08850464224815369,
      "learning_rate": 5e-05,
      "loss": 0.1402,
      "step": 13720
    },
    {
      "step": 13720,
      "train_flow_matching_loss": 0.064453125,
      "train_physics_loss": 0.5566110014915466
    },
    {
      "step": 13720,
      "train_flow_matching_loss": 0.0712890625,
      "train_physics_loss": 0.4708033502101898
    },
    {
      "step": 13720,
      "train_flow_matching_loss": 0.040771484375,
      "train_physics_loss": 0.5166180729866028
    },
    {
      "step": 13720,
      "train_flow_matching_loss": 0.142578125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.1279284805059433,
      "learning_rate": 5e-05,
      "loss": 0.1406,
      "step": 13730
    },
    {
      "step": 13730,
      "train_flow_matching_loss": 0.048583984375,
      "train_physics_loss": 0.5370196104049683
    },
    {
      "step": 13730,
      "train_flow_matching_loss": 0.056396484375,
      "train_physics_loss": 0.5408833622932434
    },
    {
      "step": 13730,
      "train_flow_matching_loss": 0.053955078125,
      "train_physics_loss": 0.5815435647964478
    },
    {
      "step": 13730,
      "train_flow_matching_loss": 0.1396484375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09597991406917572,
      "learning_rate": 5e-05,
      "loss": 0.1417,
      "step": 13740
    },
    {
      "step": 13740,
      "train_flow_matching_loss": 0.1689453125,
      "train_physics_loss": 0.6519774198532104
    },
    {
      "step": 13740,
      "train_flow_matching_loss": 0.06884765625,
      "train_physics_loss": 0.5591563582420349
    },
    {
      "step": 13740,
      "train_flow_matching_loss": 0.1484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 13740,
      "train_flow_matching_loss": 0.11865234375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.12600432336330414,
      "learning_rate": 5e-05,
      "loss": 0.1476,
      "step": 13750
    },
    {
      "step": 13750,
      "train_flow_matching_loss": 0.049560546875,
      "train_physics_loss": 0.48111164569854736
    },
    {
      "step": 13750,
      "train_flow_matching_loss": 0.0712890625,
      "train_physics_loss": 0.49000585079193115
    },
    {
      "step": 13750,
      "train_flow_matching_loss": 0.171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 13750,
      "train_flow_matching_loss": 0.2021484375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09685985743999481,
      "learning_rate": 5e-05,
      "loss": 0.1406,
      "step": 13760
    },
    {
      "step": 13760,
      "train_flow_matching_loss": 0.060302734375,
      "train_physics_loss": 0.5358156561851501
    },
    {
      "step": 13760,
      "train_flow_matching_loss": 0.050537109375,
      "train_physics_loss": 0.5788434743881226
    },
    {
      "step": 13760,
      "train_flow_matching_loss": 0.1875,
      "train_physics_loss": 0.0
    },
    {
      "step": 13760,
      "train_flow_matching_loss": 0.1083984375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09212185442447662,
      "learning_rate": 5e-05,
      "loss": 0.1412,
      "step": 13770
    },
    {
      "step": 13770,
      "train_flow_matching_loss": 0.11767578125,
      "train_physics_loss": 0.6359449625015259
    },
    {
      "step": 13770,
      "train_flow_matching_loss": 0.09765625,
      "train_physics_loss": 0.5326741337776184
    },
    {
      "step": 13770,
      "train_flow_matching_loss": 0.0859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 13770,
      "train_flow_matching_loss": 0.1689453125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.07178157567977905,
      "learning_rate": 5e-05,
      "loss": 0.1396,
      "step": 13780
    },
    {
      "step": 13780,
      "train_flow_matching_loss": 0.058837890625,
      "train_physics_loss": 0.5263181924819946
    },
    {
      "step": 13780,
      "train_flow_matching_loss": 0.06982421875,
      "train_physics_loss": 0.5151779055595398
    },
    {
      "step": 13780,
      "train_flow_matching_loss": 0.1484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 13780,
      "train_flow_matching_loss": 0.1689453125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10930275917053223,
      "learning_rate": 5e-05,
      "loss": 0.1463,
      "step": 13790
    },
    {
      "step": 13790,
      "train_flow_matching_loss": 0.042236328125,
      "train_physics_loss": 0.5124998092651367
    },
    {
      "step": 13790,
      "train_flow_matching_loss": 0.10986328125,
      "train_physics_loss": 0.5296639800071716
    },
    {
      "step": 13790,
      "train_flow_matching_loss": 0.1591796875,
      "train_physics_loss": 0.0
    },
    {
      "step": 13790,
      "train_flow_matching_loss": 0.138671875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.07032228261232376,
      "learning_rate": 5e-05,
      "loss": 0.1443,
      "step": 13800
    },
    {
      "step": 13800,
      "train_flow_matching_loss": 0.10791015625,
      "train_physics_loss": 0.6656738519668579
    },
    {
      "step": 13800,
      "train_flow_matching_loss": 0.044921875,
      "train_physics_loss": 0.52130126953125
    },
    {
      "step": 13800,
      "train_flow_matching_loss": 0.1923828125,
      "train_physics_loss": 0.0
    },
    {
      "step": 13800,
      "train_flow_matching_loss": 0.0693359375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09890166670084,
      "learning_rate": 5e-05,
      "loss": 0.1471,
      "step": 13810
    },
    {
      "step": 13810,
      "train_flow_matching_loss": 0.07080078125,
      "train_physics_loss": 0.5671100616455078
    },
    {
      "step": 13810,
      "train_flow_matching_loss": 0.040771484375,
      "train_physics_loss": 0.5133415460586548
    },
    {
      "step": 13810,
      "train_flow_matching_loss": 0.146484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 13810,
      "train_flow_matching_loss": 0.140625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08122668415307999,
      "learning_rate": 5e-05,
      "loss": 0.1424,
      "step": 13820
    },
    {
      "step": 13820,
      "train_flow_matching_loss": 0.0927734375,
      "train_physics_loss": 0.5071226358413696
    },
    {
      "step": 13820,
      "train_flow_matching_loss": 0.11181640625,
      "train_physics_loss": 0.5263448357582092
    },
    {
      "step": 13820,
      "train_flow_matching_loss": 0.1455078125,
      "train_physics_loss": 0.0
    },
    {
      "step": 13820,
      "train_flow_matching_loss": 0.1826171875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.0945652425289154,
      "learning_rate": 5e-05,
      "loss": 0.1408,
      "step": 13830
    },
    {
      "step": 13830,
      "train_flow_matching_loss": 0.09375,
      "train_physics_loss": 0.5228934288024902
    },
    {
      "step": 13830,
      "train_flow_matching_loss": 0.051025390625,
      "train_physics_loss": 0.4887750446796417
    },
    {
      "step": 13830,
      "train_flow_matching_loss": 0.12158203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 13830,
      "train_flow_matching_loss": 0.11865234375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10683096200227737,
      "learning_rate": 5e-05,
      "loss": 0.1417,
      "step": 13840
    },
    {
      "step": 13840,
      "train_flow_matching_loss": 0.07958984375,
      "train_physics_loss": 0.0
    },
    {
      "step": 13840,
      "train_flow_matching_loss": 0.05517578125,
      "train_physics_loss": 0.4507363736629486
    },
    {
      "step": 13840,
      "train_flow_matching_loss": 0.1533203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 13840,
      "train_flow_matching_loss": 0.1328125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.06924284249544144,
      "learning_rate": 5e-05,
      "loss": 0.1367,
      "step": 13850
    },
    {
      "step": 13850,
      "train_flow_matching_loss": 0.173828125,
      "train_physics_loss": 0.0
    },
    {
      "step": 13850,
      "train_flow_matching_loss": 0.1552734375,
      "train_physics_loss": 0.6210078597068787
    },
    {
      "step": 13850,
      "train_flow_matching_loss": 0.1416015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 13850,
      "train_flow_matching_loss": 0.1083984375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.13140243291854858,
      "learning_rate": 5e-05,
      "loss": 0.142,
      "step": 13860
    },
    {
      "step": 13860,
      "train_flow_matching_loss": 0.09814453125,
      "train_physics_loss": 0.0
    },
    {
      "step": 13860,
      "train_flow_matching_loss": 0.052001953125,
      "train_physics_loss": 0.5686313509941101
    },
    {
      "step": 13860,
      "train_flow_matching_loss": 0.193359375,
      "train_physics_loss": 0.0
    },
    {
      "step": 13860,
      "train_flow_matching_loss": 0.1767578125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08978927880525589,
      "learning_rate": 5e-05,
      "loss": 0.139,
      "step": 13870
    },
    {
      "step": 13870,
      "train_flow_matching_loss": 0.1904296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 13870,
      "train_flow_matching_loss": 0.06982421875,
      "train_physics_loss": 0.4920171797275543
    },
    {
      "step": 13870,
      "train_flow_matching_loss": 0.1689453125,
      "train_physics_loss": 0.0
    },
    {
      "step": 13870,
      "train_flow_matching_loss": 0.0966796875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.13586300611495972,
      "learning_rate": 5e-05,
      "loss": 0.1384,
      "step": 13880
    },
    {
      "step": 13880,
      "train_flow_matching_loss": 0.1494140625,
      "train_physics_loss": 0.0
    },
    {
      "step": 13880,
      "train_flow_matching_loss": 0.1083984375,
      "train_physics_loss": 0.547295093536377
    },
    {
      "step": 13880,
      "train_flow_matching_loss": 0.1953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 13880,
      "train_flow_matching_loss": 0.2470703125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09598669409751892,
      "learning_rate": 5e-05,
      "loss": 0.1438,
      "step": 13890
    },
    {
      "step": 13890,
      "train_flow_matching_loss": 0.125,
      "train_physics_loss": 0.0
    },
    {
      "step": 13890,
      "train_flow_matching_loss": 0.08544921875,
      "train_physics_loss": 0.4405263066291809
    },
    {
      "step": 13890,
      "train_flow_matching_loss": 0.203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 13890,
      "train_flow_matching_loss": 0.126953125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09825576096773148,
      "learning_rate": 5e-05,
      "loss": 0.1417,
      "step": 13900
    },
    {
      "step": 13900,
      "train_flow_matching_loss": 0.09228515625,
      "train_physics_loss": 0.0
    },
    {
      "step": 13900,
      "train_flow_matching_loss": 0.07275390625,
      "train_physics_loss": 0.594031810760498
    },
    {
      "step": 13900,
      "train_flow_matching_loss": 0.1015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 13900,
      "train_flow_matching_loss": 0.126953125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.06469212472438812,
      "learning_rate": 5e-05,
      "loss": 0.1389,
      "step": 13910
    },
    {
      "step": 13910,
      "train_flow_matching_loss": 0.10009765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 13910,
      "train_flow_matching_loss": 0.034912109375,
      "train_physics_loss": 0.4868779480457306
    },
    {
      "step": 13910,
      "train_flow_matching_loss": 0.10302734375,
      "train_physics_loss": 0.0
    },
    {
      "step": 13910,
      "train_flow_matching_loss": 0.154296875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09487304091453552,
      "learning_rate": 5e-05,
      "loss": 0.1436,
      "step": 13920
    },
    {
      "step": 13920,
      "train_flow_matching_loss": 0.12255859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 13920,
      "train_flow_matching_loss": 0.0654296875,
      "train_physics_loss": 0.5382136702537537
    },
    {
      "step": 13920,
      "train_flow_matching_loss": 0.1435546875,
      "train_physics_loss": 0.0
    },
    {
      "step": 13920,
      "train_flow_matching_loss": 0.1455078125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.1534988135099411,
      "learning_rate": 5e-05,
      "loss": 0.1406,
      "step": 13930
    },
    {
      "step": 13930,
      "train_flow_matching_loss": 0.1298828125,
      "train_physics_loss": 0.0
    },
    {
      "step": 13930,
      "train_flow_matching_loss": 0.07080078125,
      "train_physics_loss": 0.5502580404281616
    },
    {
      "step": 13930,
      "train_flow_matching_loss": 0.146484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 13930,
      "train_flow_matching_loss": 0.2001953125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09344195574522018,
      "learning_rate": 5e-05,
      "loss": 0.1383,
      "step": 13940
    },
    {
      "step": 13940,
      "train_flow_matching_loss": 0.166015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 13940,
      "train_flow_matching_loss": 0.158203125,
      "train_physics_loss": 0.476464182138443
    },
    {
      "step": 13940,
      "train_flow_matching_loss": 0.1337890625,
      "train_physics_loss": 0.0
    },
    {
      "step": 13940,
      "train_flow_matching_loss": 0.201171875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11842228472232819,
      "learning_rate": 5e-05,
      "loss": 0.1411,
      "step": 13950
    },
    {
      "step": 13950,
      "train_flow_matching_loss": 0.1630859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 13950,
      "train_flow_matching_loss": 0.06884765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 13950,
      "train_flow_matching_loss": 0.109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 13950,
      "train_flow_matching_loss": 0.10205078125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.1491789072751999,
      "learning_rate": 5e-05,
      "loss": 0.1372,
      "step": 13960
    },
    {
      "step": 13960,
      "train_flow_matching_loss": 0.09814453125,
      "train_physics_loss": 0.0
    },
    {
      "step": 13960,
      "train_flow_matching_loss": 0.10498046875,
      "train_physics_loss": 0.0
    },
    {
      "step": 13960,
      "train_flow_matching_loss": 0.203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 13960,
      "train_flow_matching_loss": 0.10302734375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09922787547111511,
      "learning_rate": 5e-05,
      "loss": 0.1423,
      "step": 13970
    },
    {
      "step": 13970,
      "train_flow_matching_loss": 0.1171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 13970,
      "train_flow_matching_loss": 0.2255859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 13970,
      "train_flow_matching_loss": 0.1181640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 13970,
      "train_flow_matching_loss": 0.146484375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.152213916182518,
      "learning_rate": 5e-05,
      "loss": 0.1417,
      "step": 13980
    },
    {
      "step": 13980,
      "train_flow_matching_loss": 0.1103515625,
      "train_physics_loss": 0.0
    },
    {
      "step": 13980,
      "train_flow_matching_loss": 0.203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 13980,
      "train_flow_matching_loss": 0.169921875,
      "train_physics_loss": 0.0
    },
    {
      "step": 13980,
      "train_flow_matching_loss": 0.2109375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.12185034900903702,
      "learning_rate": 5e-05,
      "loss": 0.1377,
      "step": 13990
    },
    {
      "step": 13990,
      "train_flow_matching_loss": 0.1015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 13990,
      "train_flow_matching_loss": 0.1611328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 13990,
      "train_flow_matching_loss": 0.2158203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 13990,
      "train_flow_matching_loss": 0.1181640625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.07578324526548386,
      "learning_rate": 5e-05,
      "loss": 0.136,
      "step": 14000
    },
    {
      "step": 14000,
      "train_flow_matching_loss": 0.140625,
      "train_physics_loss": 0.0
    },
    {
      "step": 14000,
      "train_flow_matching_loss": 0.1064453125,
      "train_physics_loss": 0.0
    },
    {
      "step": 14000,
      "train_flow_matching_loss": 0.1796875,
      "train_physics_loss": 0.0
    },
    {
      "step": 14000,
      "train_flow_matching_loss": 0.1083984375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.07540524750947952,
      "learning_rate": 5e-05,
      "loss": 0.1414,
      "step": 14010
    },
    {
      "step": 14010,
      "train_flow_matching_loss": 0.12353515625,
      "train_physics_loss": 0.0
    },
    {
      "step": 14010,
      "train_flow_matching_loss": 0.1455078125,
      "train_physics_loss": 0.0
    },
    {
      "step": 14010,
      "train_flow_matching_loss": 0.10986328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 14010,
      "train_flow_matching_loss": 0.1787109375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.13058805465698242,
      "learning_rate": 5e-05,
      "loss": 0.1419,
      "step": 14020
    },
    {
      "step": 14020,
      "train_flow_matching_loss": 0.0849609375,
      "train_physics_loss": 0.0
    },
    {
      "step": 14020,
      "train_flow_matching_loss": 0.1357421875,
      "train_physics_loss": 0.0
    },
    {
      "step": 14020,
      "train_flow_matching_loss": 0.1259765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 14020,
      "train_flow_matching_loss": 0.10986328125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.1052531972527504,
      "learning_rate": 5e-05,
      "loss": 0.14,
      "step": 14030
    },
    {
      "step": 14030,
      "train_flow_matching_loss": 0.09765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 14030,
      "train_flow_matching_loss": 0.171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 14030,
      "train_flow_matching_loss": 0.2216796875,
      "train_physics_loss": 0.0
    },
    {
      "step": 14030,
      "train_flow_matching_loss": 0.1494140625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.1258251816034317,
      "learning_rate": 5e-05,
      "loss": 0.1411,
      "step": 14040
    },
    {
      "step": 14040,
      "train_flow_matching_loss": 0.11328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 14040,
      "train_flow_matching_loss": 0.140625,
      "train_physics_loss": 0.0
    },
    {
      "step": 14040,
      "train_flow_matching_loss": 0.1767578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 14040,
      "train_flow_matching_loss": 0.1484375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10561259835958481,
      "learning_rate": 5e-05,
      "loss": 0.1353,
      "step": 14050
    },
    {
      "step": 14050,
      "train_flow_matching_loss": 0.169921875,
      "train_physics_loss": 0.0
    },
    {
      "step": 14050,
      "train_flow_matching_loss": 0.134765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 14050,
      "train_flow_matching_loss": 0.189453125,
      "train_physics_loss": 0.0
    },
    {
      "step": 14050,
      "train_flow_matching_loss": 0.12353515625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.069070965051651,
      "learning_rate": 5e-05,
      "loss": 0.1386,
      "step": 14060
    },
    {
      "step": 14060,
      "train_flow_matching_loss": 0.08447265625,
      "train_physics_loss": 0.0
    },
    {
      "step": 14060,
      "train_flow_matching_loss": 0.193359375,
      "train_physics_loss": 0.0
    },
    {
      "step": 14060,
      "train_flow_matching_loss": 0.173828125,
      "train_physics_loss": 0.0
    },
    {
      "step": 14060,
      "train_flow_matching_loss": 0.1455078125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08711210638284683,
      "learning_rate": 5e-05,
      "loss": 0.1419,
      "step": 14070
    },
    {
      "step": 14070,
      "train_flow_matching_loss": 0.2109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 14070,
      "train_flow_matching_loss": 0.1845703125,
      "train_physics_loss": 0.0
    },
    {
      "step": 14070,
      "train_flow_matching_loss": 0.220703125,
      "train_physics_loss": 0.0
    },
    {
      "step": 14070,
      "train_flow_matching_loss": 0.2060546875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.07431162148714066,
      "learning_rate": 5e-05,
      "loss": 0.141,
      "step": 14080
    },
    {
      "step": 14080,
      "train_flow_matching_loss": 0.130859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 14080,
      "train_flow_matching_loss": 0.1318359375,
      "train_physics_loss": 0.0
    },
    {
      "step": 14080,
      "train_flow_matching_loss": 0.1298828125,
      "train_physics_loss": 0.0
    },
    {
      "step": 14080,
      "train_flow_matching_loss": 0.0849609375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.142583429813385,
      "learning_rate": 5e-05,
      "loss": 0.1411,
      "step": 14090
    },
    {
      "step": 14090,
      "train_flow_matching_loss": 0.044677734375,
      "train_physics_loss": 0.5445124506950378
    },
    {
      "step": 14090,
      "train_flow_matching_loss": 0.1953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 14090,
      "train_flow_matching_loss": 0.11279296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 14090,
      "train_flow_matching_loss": 0.1494140625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11690323054790497,
      "learning_rate": 5e-05,
      "loss": 0.1389,
      "step": 14100
    },
    {
      "step": 14100,
      "train_flow_matching_loss": 0.08056640625,
      "train_physics_loss": 0.5005038976669312
    },
    {
      "step": 14100,
      "train_flow_matching_loss": 0.15625,
      "train_physics_loss": 0.0
    },
    {
      "step": 14100,
      "train_flow_matching_loss": 0.1455078125,
      "train_physics_loss": 0.0
    },
    {
      "step": 14100,
      "train_flow_matching_loss": 0.138671875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.07929664850234985,
      "learning_rate": 5e-05,
      "loss": 0.1447,
      "step": 14110
    },
    {
      "step": 14110,
      "train_flow_matching_loss": 0.06201171875,
      "train_physics_loss": 0.5618734359741211
    },
    {
      "step": 14110,
      "train_flow_matching_loss": 0.1669921875,
      "train_physics_loss": 0.0
    },
    {
      "step": 14110,
      "train_flow_matching_loss": 0.1396484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 14110,
      "train_flow_matching_loss": 0.1572265625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08853788673877716,
      "learning_rate": 5e-05,
      "loss": 0.1399,
      "step": 14120
    },
    {
      "step": 14120,
      "train_flow_matching_loss": 0.05029296875,
      "train_physics_loss": 0.5258551239967346
    },
    {
      "step": 14120,
      "train_flow_matching_loss": 0.138671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 14120,
      "train_flow_matching_loss": 0.1103515625,
      "train_physics_loss": 0.0
    },
    {
      "step": 14120,
      "train_flow_matching_loss": 0.146484375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09839729219675064,
      "learning_rate": 5e-05,
      "loss": 0.1444,
      "step": 14130
    },
    {
      "step": 14130,
      "train_flow_matching_loss": 0.08935546875,
      "train_physics_loss": 0.5138475298881531
    },
    {
      "step": 14130,
      "train_flow_matching_loss": 0.09716796875,
      "train_physics_loss": 0.0
    },
    {
      "step": 14130,
      "train_flow_matching_loss": 0.11962890625,
      "train_physics_loss": 0.0
    },
    {
      "step": 14130,
      "train_flow_matching_loss": 0.1396484375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08814162760972977,
      "learning_rate": 5e-05,
      "loss": 0.1435,
      "step": 14140
    },
    {
      "step": 14140,
      "train_flow_matching_loss": 0.08740234375,
      "train_physics_loss": 0.4566287696361542
    },
    {
      "step": 14140,
      "train_flow_matching_loss": 0.1279296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 14140,
      "train_flow_matching_loss": 0.0966796875,
      "train_physics_loss": 0.0
    },
    {
      "step": 14140,
      "train_flow_matching_loss": 0.0810546875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10998815298080444,
      "learning_rate": 5e-05,
      "loss": 0.1377,
      "step": 14150
    },
    {
      "step": 14150,
      "train_flow_matching_loss": 0.048095703125,
      "train_physics_loss": 0.5007959604263306
    },
    {
      "step": 14150,
      "train_flow_matching_loss": 0.126953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 14150,
      "train_flow_matching_loss": 0.1513671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 14150,
      "train_flow_matching_loss": 0.208984375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09048923850059509,
      "learning_rate": 5e-05,
      "loss": 0.1391,
      "step": 14160
    },
    {
      "step": 14160,
      "train_flow_matching_loss": 0.1279296875,
      "train_physics_loss": 0.5327332019805908
    },
    {
      "step": 14160,
      "train_flow_matching_loss": 0.1640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 14160,
      "train_flow_matching_loss": 0.1474609375,
      "train_physics_loss": 0.0
    },
    {
      "step": 14160,
      "train_flow_matching_loss": 0.1298828125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08964093029499054,
      "learning_rate": 5e-05,
      "loss": 0.1428,
      "step": 14170
    },
    {
      "step": 14170,
      "train_flow_matching_loss": 0.056884765625,
      "train_physics_loss": 0.5230916142463684
    },
    {
      "step": 14170,
      "train_flow_matching_loss": 0.13671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 14170,
      "train_flow_matching_loss": 0.2119140625,
      "train_physics_loss": 0.0
    },
    {
      "step": 14170,
      "train_flow_matching_loss": 0.169921875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.0870717316865921,
      "learning_rate": 5e-05,
      "loss": 0.1417,
      "step": 14180
    },
    {
      "step": 14180,
      "train_flow_matching_loss": 0.037109375,
      "train_physics_loss": 0.4667988717556
    },
    {
      "step": 14180,
      "train_flow_matching_loss": 0.1064453125,
      "train_physics_loss": 0.0
    },
    {
      "step": 14180,
      "train_flow_matching_loss": 0.10400390625,
      "train_physics_loss": 0.0
    },
    {
      "step": 14180,
      "train_flow_matching_loss": 0.1416015625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.13643163442611694,
      "learning_rate": 5e-05,
      "loss": 0.1418,
      "step": 14190
    },
    {
      "step": 14190,
      "train_flow_matching_loss": 0.1494140625,
      "train_physics_loss": 0.5477119088172913
    },
    {
      "step": 14190,
      "train_flow_matching_loss": 0.2275390625,
      "train_physics_loss": 0.0
    },
    {
      "step": 14190,
      "train_flow_matching_loss": 0.1591796875,
      "train_physics_loss": 0.0
    },
    {
      "step": 14190,
      "train_flow_matching_loss": 0.1435546875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11298070847988129,
      "learning_rate": 5e-05,
      "loss": 0.1406,
      "step": 14200
    },
    {
      "step": 14200,
      "train_flow_matching_loss": 0.05224609375,
      "train_physics_loss": 0.489852637052536
    },
    {
      "step": 14200,
      "train_flow_matching_loss": 0.154296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 14200,
      "train_flow_matching_loss": 0.13671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 14200,
      "train_flow_matching_loss": 0.12451171875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.12046285718679428,
      "learning_rate": 5e-05,
      "loss": 0.1396,
      "step": 14210
    },
    {
      "step": 14210,
      "train_flow_matching_loss": 0.054931640625,
      "train_physics_loss": 0.5129054188728333
    },
    {
      "step": 14210,
      "train_flow_matching_loss": 0.146484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 14210,
      "train_flow_matching_loss": 0.1806640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 14210,
      "train_flow_matching_loss": 0.126953125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08669010549783707,
      "learning_rate": 5e-05,
      "loss": 0.1454,
      "step": 14220
    },
    {
      "step": 14220,
      "train_flow_matching_loss": 0.0751953125,
      "train_physics_loss": 0.5149661302566528
    },
    {
      "step": 14220,
      "train_flow_matching_loss": 0.138671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 14220,
      "train_flow_matching_loss": 0.126953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 14220,
      "train_flow_matching_loss": 0.169921875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09430156648159027,
      "learning_rate": 5e-05,
      "loss": 0.1429,
      "step": 14230
    },
    {
      "step": 14230,
      "train_flow_matching_loss": 0.08544921875,
      "train_physics_loss": 0.5536693930625916
    },
    {
      "step": 14230,
      "train_flow_matching_loss": 0.1103515625,
      "train_physics_loss": 0.0
    },
    {
      "step": 14230,
      "train_flow_matching_loss": 0.2177734375,
      "train_physics_loss": 0.0
    },
    {
      "step": 14230,
      "train_flow_matching_loss": 0.1240234375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.1344670057296753,
      "learning_rate": 5e-05,
      "loss": 0.1363,
      "step": 14240
    },
    {
      "step": 14240,
      "train_flow_matching_loss": 0.0634765625,
      "train_physics_loss": 0.4550507962703705
    },
    {
      "step": 14240,
      "train_flow_matching_loss": 0.189453125,
      "train_physics_loss": 0.0
    },
    {
      "step": 14240,
      "train_flow_matching_loss": 0.162109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 14240,
      "train_flow_matching_loss": 0.1279296875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09075253456830978,
      "learning_rate": 5e-05,
      "loss": 0.1449,
      "step": 14250
    },
    {
      "step": 14250,
      "train_flow_matching_loss": 0.05419921875,
      "train_physics_loss": 0.49928218126296997
    },
    {
      "step": 14250,
      "train_flow_matching_loss": 0.189453125,
      "train_physics_loss": 0.0
    },
    {
      "step": 14250,
      "train_flow_matching_loss": 0.341796875,
      "train_physics_loss": 0.0
    },
    {
      "step": 14250,
      "train_flow_matching_loss": 0.150390625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.12163114547729492,
      "learning_rate": 5e-05,
      "loss": 0.1454,
      "step": 14260
    },
    {
      "step": 14260,
      "train_flow_matching_loss": 0.08056640625,
      "train_physics_loss": 0.521673321723938
    },
    {
      "step": 14260,
      "train_flow_matching_loss": 0.1708984375,
      "train_physics_loss": 0.0
    },
    {
      "step": 14260,
      "train_flow_matching_loss": 0.2060546875,
      "train_physics_loss": 0.0
    },
    {
      "step": 14260,
      "train_flow_matching_loss": 0.1123046875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10269806534051895,
      "learning_rate": 5e-05,
      "loss": 0.1399,
      "step": 14270
    },
    {
      "step": 14270,
      "train_flow_matching_loss": 0.04345703125,
      "train_physics_loss": 0.49896010756492615
    },
    {
      "step": 14270,
      "train_flow_matching_loss": 0.11083984375,
      "train_physics_loss": 0.0
    },
    {
      "step": 14270,
      "train_flow_matching_loss": 0.142578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 14270,
      "train_flow_matching_loss": 0.138671875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.07712706923484802,
      "learning_rate": 5e-05,
      "loss": 0.1427,
      "step": 14280
    },
    {
      "step": 14280,
      "train_flow_matching_loss": 0.06689453125,
      "train_physics_loss": 0.42835181951522827
    },
    {
      "step": 14280,
      "train_flow_matching_loss": 0.1669921875,
      "train_physics_loss": 0.0
    },
    {
      "step": 14280,
      "train_flow_matching_loss": 0.0703125,
      "train_physics_loss": 0.0
    },
    {
      "step": 14280,
      "train_flow_matching_loss": 0.1337890625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09921456128358841,
      "learning_rate": 5e-05,
      "loss": 0.1361,
      "step": 14290
    },
    {
      "step": 14290,
      "train_flow_matching_loss": 0.08544921875,
      "train_physics_loss": 0.5986888408660889
    },
    {
      "step": 14290,
      "train_flow_matching_loss": 0.0947265625,
      "train_physics_loss": 0.0
    },
    {
      "step": 14290,
      "train_flow_matching_loss": 0.11083984375,
      "train_physics_loss": 0.0
    },
    {
      "step": 14290,
      "train_flow_matching_loss": 0.234375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11278045177459717,
      "learning_rate": 5e-05,
      "loss": 0.1427,
      "step": 14300
    },
    {
      "step": 14300,
      "train_flow_matching_loss": 0.06298828125,
      "train_physics_loss": 0.5145019292831421
    },
    {
      "step": 14300,
      "train_flow_matching_loss": 0.1845703125,
      "train_physics_loss": 0.0
    },
    {
      "step": 14300,
      "train_flow_matching_loss": 0.11962890625,
      "train_physics_loss": 0.0
    },
    {
      "step": 14300,
      "train_flow_matching_loss": 0.1142578125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09597263485193253,
      "learning_rate": 5e-05,
      "loss": 0.141,
      "step": 14310
    },
    {
      "step": 14310,
      "train_flow_matching_loss": 0.04052734375,
      "train_physics_loss": 0.4875347316265106
    },
    {
      "step": 14310,
      "train_flow_matching_loss": 0.177734375,
      "train_physics_loss": 0.0
    },
    {
      "step": 14310,
      "train_flow_matching_loss": 0.2080078125,
      "train_physics_loss": 0.0
    },
    {
      "step": 14310,
      "train_flow_matching_loss": 0.25390625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10644907504320145,
      "learning_rate": 5e-05,
      "loss": 0.1447,
      "step": 14320
    },
    {
      "step": 14320,
      "train_flow_matching_loss": 0.043701171875,
      "train_physics_loss": 0.5016383528709412
    },
    {
      "step": 14320,
      "train_flow_matching_loss": 0.2314453125,
      "train_physics_loss": 0.0
    },
    {
      "step": 14320,
      "train_flow_matching_loss": 0.220703125,
      "train_physics_loss": 0.0
    },
    {
      "step": 14320,
      "train_flow_matching_loss": 0.13671875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10891515761613846,
      "learning_rate": 5e-05,
      "loss": 0.1434,
      "step": 14330
    },
    {
      "step": 14330,
      "train_flow_matching_loss": 0.053955078125,
      "train_physics_loss": 0.46617671847343445
    },
    {
      "step": 14330,
      "train_flow_matching_loss": 0.154296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 14330,
      "train_flow_matching_loss": 0.134765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 14330,
      "train_flow_matching_loss": 0.1640625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09578026831150055,
      "learning_rate": 5e-05,
      "loss": 0.1435,
      "step": 14340
    },
    {
      "step": 14340,
      "train_flow_matching_loss": 0.0625,
      "train_physics_loss": 0.49087080359458923
    },
    {
      "step": 14340,
      "train_flow_matching_loss": 0.1171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 14340,
      "train_flow_matching_loss": 0.166015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 14340,
      "train_flow_matching_loss": 0.1201171875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10918024927377701,
      "learning_rate": 5e-05,
      "loss": 0.1427,
      "step": 14350
    },
    {
      "step": 14350,
      "train_flow_matching_loss": 0.1572265625,
      "train_physics_loss": 0.0
    },
    {
      "step": 14350,
      "train_flow_matching_loss": 0.1083984375,
      "train_physics_loss": 0.0
    },
    {
      "step": 14350,
      "train_flow_matching_loss": 0.1396484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 14350,
      "train_flow_matching_loss": 0.11376953125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.0843542292714119,
      "learning_rate": 5e-05,
      "loss": 0.1431,
      "step": 14360
    },
    {
      "step": 14360,
      "train_flow_matching_loss": 0.1640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 14360,
      "train_flow_matching_loss": 0.1572265625,
      "train_physics_loss": 0.0
    },
    {
      "step": 14360,
      "train_flow_matching_loss": 0.193359375,
      "train_physics_loss": 0.0
    },
    {
      "step": 14360,
      "train_flow_matching_loss": 0.1708984375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.15948280692100525,
      "learning_rate": 5e-05,
      "loss": 0.1437,
      "step": 14370
    },
    {
      "step": 14370,
      "train_flow_matching_loss": 0.177734375,
      "train_physics_loss": 0.0
    },
    {
      "step": 14370,
      "train_flow_matching_loss": 0.1865234375,
      "train_physics_loss": 0.0
    },
    {
      "step": 14370,
      "train_flow_matching_loss": 0.162109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 14370,
      "train_flow_matching_loss": 0.158203125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.12213480472564697,
      "learning_rate": 5e-05,
      "loss": 0.1456,
      "step": 14380
    },
    {
      "step": 14380,
      "train_flow_matching_loss": 0.1591796875,
      "train_physics_loss": 0.0
    },
    {
      "step": 14380,
      "train_flow_matching_loss": 0.08056640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 14380,
      "train_flow_matching_loss": 0.2001953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 14380,
      "train_flow_matching_loss": 0.173828125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11494160443544388,
      "learning_rate": 5e-05,
      "loss": 0.1403,
      "step": 14390
    },
    {
      "step": 14390,
      "train_flow_matching_loss": 0.162109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 14390,
      "train_flow_matching_loss": 0.142578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 14390,
      "train_flow_matching_loss": 0.0908203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 14390,
      "train_flow_matching_loss": 0.1552734375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10777406394481659,
      "learning_rate": 5e-05,
      "loss": 0.1452,
      "step": 14400
    },
    {
      "step": 14400,
      "train_flow_matching_loss": 0.1572265625,
      "train_physics_loss": 0.0
    },
    {
      "step": 14400,
      "train_flow_matching_loss": 0.1513671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 14400,
      "train_flow_matching_loss": 0.1337890625,
      "train_physics_loss": 0.0
    },
    {
      "step": 14400,
      "train_flow_matching_loss": 0.150390625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11075058579444885,
      "learning_rate": 5e-05,
      "loss": 0.1431,
      "step": 14410
    },
    {
      "step": 14410,
      "train_flow_matching_loss": 0.205078125,
      "train_physics_loss": 0.0
    },
    {
      "step": 14410,
      "train_flow_matching_loss": 0.1083984375,
      "train_physics_loss": 0.0
    },
    {
      "step": 14410,
      "train_flow_matching_loss": 0.1259765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 14410,
      "train_flow_matching_loss": 0.1767578125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.13380005955696106,
      "learning_rate": 5e-05,
      "loss": 0.1395,
      "step": 14420
    },
    {
      "step": 14420,
      "train_flow_matching_loss": 0.12158203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 14420,
      "train_flow_matching_loss": 0.07861328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 14420,
      "train_flow_matching_loss": 0.11474609375,
      "train_physics_loss": 0.0
    },
    {
      "step": 14420,
      "train_flow_matching_loss": 0.189453125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.1103835180401802,
      "learning_rate": 5e-05,
      "loss": 0.1357,
      "step": 14430
    },
    {
      "step": 14430,
      "train_flow_matching_loss": 0.1689453125,
      "train_physics_loss": 0.0
    },
    {
      "step": 14430,
      "train_flow_matching_loss": 0.09521484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 14430,
      "train_flow_matching_loss": 0.1484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 14430,
      "train_flow_matching_loss": 0.10791015625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.0962274968624115,
      "learning_rate": 5e-05,
      "loss": 0.1345,
      "step": 14440
    },
    {
      "step": 14440,
      "train_flow_matching_loss": 0.1630859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 14440,
      "train_flow_matching_loss": 0.1328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 14440,
      "train_flow_matching_loss": 0.1728515625,
      "train_physics_loss": 0.0
    },
    {
      "step": 14440,
      "train_flow_matching_loss": 0.1103515625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09127029776573181,
      "learning_rate": 5e-05,
      "loss": 0.149,
      "step": 14450
    },
    {
      "step": 14450,
      "train_flow_matching_loss": 0.1396484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 14450,
      "train_flow_matching_loss": 0.1591796875,
      "train_physics_loss": 0.0
    },
    {
      "step": 14450,
      "train_flow_matching_loss": 0.1357421875,
      "train_physics_loss": 0.0
    },
    {
      "step": 14450,
      "train_flow_matching_loss": 0.2294921875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11928507685661316,
      "learning_rate": 5e-05,
      "loss": 0.1397,
      "step": 14460
    },
    {
      "step": 14460,
      "train_flow_matching_loss": 0.1484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 14460,
      "train_flow_matching_loss": 0.1787109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 14460,
      "train_flow_matching_loss": 0.1552734375,
      "train_physics_loss": 0.0
    },
    {
      "step": 14460,
      "train_flow_matching_loss": 0.06640625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09360893815755844,
      "learning_rate": 5e-05,
      "loss": 0.1425,
      "step": 14470
    },
    {
      "step": 14470,
      "train_flow_matching_loss": 0.1748046875,
      "train_physics_loss": 0.0
    },
    {
      "step": 14470,
      "train_flow_matching_loss": 0.1357421875,
      "train_physics_loss": 0.0
    },
    {
      "step": 14470,
      "train_flow_matching_loss": 0.1396484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 14470,
      "train_flow_matching_loss": 0.1513671875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08364734053611755,
      "learning_rate": 5e-05,
      "loss": 0.1441,
      "step": 14480
    },
    {
      "step": 14480,
      "train_flow_matching_loss": 0.21875,
      "train_physics_loss": 0.0
    },
    {
      "step": 14480,
      "train_flow_matching_loss": 0.1142578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 14480,
      "train_flow_matching_loss": 0.1416015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 14480,
      "train_flow_matching_loss": 0.1435546875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.14139176905155182,
      "learning_rate": 5e-05,
      "loss": 0.1417,
      "step": 14490
    },
    {
      "step": 14490,
      "train_flow_matching_loss": 0.220703125,
      "train_physics_loss": 0.0
    },
    {
      "step": 14490,
      "train_flow_matching_loss": 0.06689453125,
      "train_physics_loss": 0.0
    },
    {
      "step": 14490,
      "train_flow_matching_loss": 0.10693359375,
      "train_physics_loss": 0.0
    },
    {
      "step": 14490,
      "train_flow_matching_loss": 0.08740234375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.19646768271923065,
      "learning_rate": 5e-05,
      "loss": 0.133,
      "step": 14500
    },
    {
      "step": 14500,
      "train_flow_matching_loss": 0.125,
      "train_physics_loss": 0.0
    },
    {
      "step": 14500,
      "train_flow_matching_loss": 0.1396484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 14500,
      "train_flow_matching_loss": 0.1767578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 14500,
      "train_flow_matching_loss": 0.09765625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08411276340484619,
      "learning_rate": 5e-05,
      "loss": 0.1392,
      "step": 14510
    },
    {
      "step": 14510,
      "train_flow_matching_loss": 0.1689453125,
      "train_physics_loss": 0.0
    },
    {
      "step": 14510,
      "train_flow_matching_loss": 0.10986328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 14510,
      "train_flow_matching_loss": 0.173828125,
      "train_physics_loss": 0.0
    },
    {
      "step": 14510,
      "train_flow_matching_loss": 0.0546875,
      "train_physics_loss": 0.4859444797039032
    },
    {
      "grad_norm": 0.23875200748443604,
      "learning_rate": 5e-05,
      "loss": 0.1376,
      "step": 14520
    },
    {
      "step": 14520,
      "train_flow_matching_loss": 0.173828125,
      "train_physics_loss": 0.0
    },
    {
      "step": 14520,
      "train_flow_matching_loss": 0.1865234375,
      "train_physics_loss": 0.0
    },
    {
      "step": 14520,
      "train_flow_matching_loss": 0.166015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 14520,
      "train_flow_matching_loss": 0.04150390625,
      "train_physics_loss": 0.49148431420326233
    },
    {
      "grad_norm": 0.09613458812236786,
      "learning_rate": 5e-05,
      "loss": 0.1369,
      "step": 14530
    },
    {
      "step": 14530,
      "train_flow_matching_loss": 0.2138671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 14530,
      "train_flow_matching_loss": 0.185546875,
      "train_physics_loss": 0.0
    },
    {
      "step": 14530,
      "train_flow_matching_loss": 0.154296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 14530,
      "train_flow_matching_loss": 0.07666015625,
      "train_physics_loss": 0.4653259515762329
    },
    {
      "grad_norm": 0.13281576335430145,
      "learning_rate": 5e-05,
      "loss": 0.1465,
      "step": 14540
    },
    {
      "step": 14540,
      "train_flow_matching_loss": 0.166015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 14540,
      "train_flow_matching_loss": 0.134765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 14540,
      "train_flow_matching_loss": 0.171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 14540,
      "train_flow_matching_loss": 0.06982421875,
      "train_physics_loss": 0.47035402059555054
    },
    {
      "grad_norm": 0.10913684964179993,
      "learning_rate": 5e-05,
      "loss": 0.1418,
      "step": 14550
    },
    {
      "step": 14550,
      "train_flow_matching_loss": 0.267578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 14550,
      "train_flow_matching_loss": 0.109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 14550,
      "train_flow_matching_loss": 0.1640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 14550,
      "train_flow_matching_loss": 0.047607421875,
      "train_physics_loss": 0.47499001026153564
    },
    {
      "grad_norm": 0.08093580603599548,
      "learning_rate": 5e-05,
      "loss": 0.1385,
      "step": 14560
    },
    {
      "step": 14560,
      "train_flow_matching_loss": 0.201171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 14560,
      "train_flow_matching_loss": 0.1708984375,
      "train_physics_loss": 0.0
    },
    {
      "step": 14560,
      "train_flow_matching_loss": 0.2294921875,
      "train_physics_loss": 0.0
    },
    {
      "step": 14560,
      "train_flow_matching_loss": 0.11669921875,
      "train_physics_loss": 0.5366604924201965
    },
    {
      "grad_norm": 0.08661065995693207,
      "learning_rate": 5e-05,
      "loss": 0.1392,
      "step": 14570
    },
    {
      "step": 14570,
      "train_flow_matching_loss": 0.1962890625,
      "train_physics_loss": 0.0
    },
    {
      "step": 14570,
      "train_flow_matching_loss": 0.1396484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 14570,
      "train_flow_matching_loss": 0.224609375,
      "train_physics_loss": 0.0
    },
    {
      "step": 14570,
      "train_flow_matching_loss": 0.08154296875,
      "train_physics_loss": 0.48463305830955505
    },
    {
      "grad_norm": 0.061241745948791504,
      "learning_rate": 5e-05,
      "loss": 0.142,
      "step": 14580
    },
    {
      "step": 14580,
      "train_flow_matching_loss": 0.197265625,
      "train_physics_loss": 0.0
    },
    {
      "step": 14580,
      "train_flow_matching_loss": 0.0771484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 14580,
      "train_flow_matching_loss": 0.1796875,
      "train_physics_loss": 0.0
    },
    {
      "step": 14580,
      "train_flow_matching_loss": 0.025634765625,
      "train_physics_loss": 0.4919135570526123
    },
    {
      "grad_norm": 0.07873758673667908,
      "learning_rate": 5e-05,
      "loss": 0.136,
      "step": 14590
    },
    {
      "step": 14590,
      "train_flow_matching_loss": 0.1484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 14590,
      "train_flow_matching_loss": 0.1455078125,
      "train_physics_loss": 0.0
    },
    {
      "step": 14590,
      "train_flow_matching_loss": 0.1484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 14590,
      "train_flow_matching_loss": 0.0888671875,
      "train_physics_loss": 0.5895126461982727
    },
    {
      "grad_norm": 0.08782992511987686,
      "learning_rate": 5e-05,
      "loss": 0.1392,
      "step": 14600
    },
    {
      "step": 14600,
      "train_flow_matching_loss": 0.12158203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 14600,
      "train_flow_matching_loss": 0.2236328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 14600,
      "train_flow_matching_loss": 0.169921875,
      "train_physics_loss": 0.0
    },
    {
      "step": 14600,
      "train_flow_matching_loss": 0.060546875,
      "train_physics_loss": 0.4701158106327057
    },
    {
      "grad_norm": 0.11073684692382812,
      "learning_rate": 5e-05,
      "loss": 0.1401,
      "step": 14610
    },
    {
      "step": 14610,
      "train_flow_matching_loss": 0.1484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 14610,
      "train_flow_matching_loss": 0.1123046875,
      "train_physics_loss": 0.0
    },
    {
      "step": 14610,
      "train_flow_matching_loss": 0.1728515625,
      "train_physics_loss": 0.0
    },
    {
      "step": 14610,
      "train_flow_matching_loss": 0.08935546875,
      "train_physics_loss": 0.4842158555984497
    },
    {
      "grad_norm": 0.18881145119667053,
      "learning_rate": 5e-05,
      "loss": 0.1399,
      "step": 14620
    },
    {
      "step": 14620,
      "train_flow_matching_loss": 0.1484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 14620,
      "train_flow_matching_loss": 0.154296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 14620,
      "train_flow_matching_loss": 0.1201171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 14620,
      "train_flow_matching_loss": 0.0654296875,
      "train_physics_loss": 0.45666030049324036
    },
    {
      "grad_norm": 0.09899760037660599,
      "learning_rate": 5e-05,
      "loss": 0.1467,
      "step": 14630
    },
    {
      "step": 14630,
      "train_flow_matching_loss": 0.1337890625,
      "train_physics_loss": 0.0
    },
    {
      "step": 14630,
      "train_flow_matching_loss": 0.11572265625,
      "train_physics_loss": 0.0
    },
    {
      "step": 14630,
      "train_flow_matching_loss": 0.1279296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 14630,
      "train_flow_matching_loss": 0.08642578125,
      "train_physics_loss": 0.5127437710762024
    },
    {
      "grad_norm": 0.17033864557743073,
      "learning_rate": 5e-05,
      "loss": 0.1408,
      "step": 14640
    },
    {
      "step": 14640,
      "train_flow_matching_loss": 0.1787109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 14640,
      "train_flow_matching_loss": 0.1552734375,
      "train_physics_loss": 0.0
    },
    {
      "step": 14640,
      "train_flow_matching_loss": 0.1259765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 14640,
      "train_flow_matching_loss": 0.08154296875,
      "train_physics_loss": 0.49497026205062866
    },
    {
      "grad_norm": 0.11161508411169052,
      "learning_rate": 5e-05,
      "loss": 0.14,
      "step": 14650
    },
    {
      "step": 14650,
      "train_flow_matching_loss": 0.1005859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 14650,
      "train_flow_matching_loss": 0.1689453125,
      "train_physics_loss": 0.0
    },
    {
      "step": 14650,
      "train_flow_matching_loss": 0.2255859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 14650,
      "train_flow_matching_loss": 0.1015625,
      "train_physics_loss": 0.5293346047401428
    },
    {
      "grad_norm": 0.09140318632125854,
      "learning_rate": 5e-05,
      "loss": 0.1393,
      "step": 14660
    },
    {
      "step": 14660,
      "train_flow_matching_loss": 0.1123046875,
      "train_physics_loss": 0.0
    },
    {
      "step": 14660,
      "train_flow_matching_loss": 0.1279296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 14660,
      "train_flow_matching_loss": 0.1298828125,
      "train_physics_loss": 0.0
    },
    {
      "step": 14660,
      "train_flow_matching_loss": 0.044189453125,
      "train_physics_loss": 0.4732816219329834
    },
    {
      "grad_norm": 0.11689877510070801,
      "learning_rate": 5e-05,
      "loss": 0.137,
      "step": 14670
    },
    {
      "step": 14670,
      "train_flow_matching_loss": 0.1630859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 14670,
      "train_flow_matching_loss": 0.12353515625,
      "train_physics_loss": 0.0
    },
    {
      "step": 14670,
      "train_flow_matching_loss": 0.1806640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 14670,
      "train_flow_matching_loss": 0.039794921875,
      "train_physics_loss": 0.5328750014305115
    },
    {
      "grad_norm": 0.09841272979974747,
      "learning_rate": 5e-05,
      "loss": 0.1411,
      "step": 14680
    },
    {
      "step": 14680,
      "train_flow_matching_loss": 0.10546875,
      "train_physics_loss": 0.0
    },
    {
      "step": 14680,
      "train_flow_matching_loss": 0.150390625,
      "train_physics_loss": 0.0
    },
    {
      "step": 14680,
      "train_flow_matching_loss": 0.1884765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 14680,
      "train_flow_matching_loss": 0.048095703125,
      "train_physics_loss": 0.4920625388622284
    },
    {
      "grad_norm": 0.2942334711551666,
      "learning_rate": 5e-05,
      "loss": 0.1364,
      "step": 14690
    },
    {
      "step": 14690,
      "train_flow_matching_loss": 0.1494140625,
      "train_physics_loss": 0.0
    },
    {
      "step": 14690,
      "train_flow_matching_loss": 0.2236328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 14690,
      "train_flow_matching_loss": 0.09033203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 14690,
      "train_flow_matching_loss": 0.09912109375,
      "train_physics_loss": 0.5076122879981995
    },
    {
      "grad_norm": 0.11239825189113617,
      "learning_rate": 5e-05,
      "loss": 0.1421,
      "step": 14700
    },
    {
      "step": 14700,
      "train_flow_matching_loss": 0.177734375,
      "train_physics_loss": 0.0
    },
    {
      "step": 14700,
      "train_flow_matching_loss": 0.1435546875,
      "train_physics_loss": 0.0
    },
    {
      "step": 14700,
      "train_flow_matching_loss": 0.10986328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 14700,
      "train_flow_matching_loss": 0.10302734375,
      "train_physics_loss": 0.4696432650089264
    },
    {
      "grad_norm": 0.0954582616686821,
      "learning_rate": 5e-05,
      "loss": 0.1441,
      "step": 14710
    },
    {
      "step": 14710,
      "train_flow_matching_loss": 0.1533203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 14710,
      "train_flow_matching_loss": 0.12060546875,
      "train_physics_loss": 0.0
    },
    {
      "step": 14710,
      "train_flow_matching_loss": 0.169921875,
      "train_physics_loss": 0.0
    },
    {
      "step": 14710,
      "train_flow_matching_loss": 0.06591796875,
      "train_physics_loss": 0.4963972866535187
    },
    {
      "grad_norm": 0.10206077247858047,
      "learning_rate": 5e-05,
      "loss": 0.1416,
      "step": 14720
    },
    {
      "step": 14720,
      "train_flow_matching_loss": 0.1455078125,
      "train_physics_loss": 0.0
    },
    {
      "step": 14720,
      "train_flow_matching_loss": 0.1044921875,
      "train_physics_loss": 0.0
    },
    {
      "step": 14720,
      "train_flow_matching_loss": 0.125,
      "train_physics_loss": 0.0
    },
    {
      "step": 14720,
      "train_flow_matching_loss": 0.09228515625,
      "train_physics_loss": 0.4883729815483093
    },
    {
      "grad_norm": 0.10573887079954147,
      "learning_rate": 5e-05,
      "loss": 0.1367,
      "step": 14730
    },
    {
      "step": 14730,
      "train_flow_matching_loss": 0.162109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 14730,
      "train_flow_matching_loss": 0.0966796875,
      "train_physics_loss": 0.0
    },
    {
      "step": 14730,
      "train_flow_matching_loss": 0.11572265625,
      "train_physics_loss": 0.0
    },
    {
      "step": 14730,
      "train_flow_matching_loss": 0.11572265625,
      "train_physics_loss": 0.6129489541053772
    },
    {
      "grad_norm": 0.1149250715970993,
      "learning_rate": 5e-05,
      "loss": 0.141,
      "step": 14740
    },
    {
      "step": 14740,
      "train_flow_matching_loss": 0.12255859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 14740,
      "train_flow_matching_loss": 0.134765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 14740,
      "train_flow_matching_loss": 0.11328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 14740,
      "train_flow_matching_loss": 0.10302734375,
      "train_physics_loss": 0.535083532333374
    },
    {
      "grad_norm": 0.13155001401901245,
      "learning_rate": 5e-05,
      "loss": 0.1438,
      "step": 14750
    },
    {
      "step": 14750,
      "train_flow_matching_loss": 0.2001953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 14750,
      "train_flow_matching_loss": 0.1259765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 14750,
      "train_flow_matching_loss": 0.12060546875,
      "train_physics_loss": 0.0
    },
    {
      "step": 14750,
      "train_flow_matching_loss": 0.036376953125,
      "train_physics_loss": 0.42604151368141174
    },
    {
      "grad_norm": 0.11349663883447647,
      "learning_rate": 5e-05,
      "loss": 0.1379,
      "step": 14760
    },
    {
      "step": 14760,
      "train_flow_matching_loss": 0.173828125,
      "train_physics_loss": 0.0
    },
    {
      "step": 14760,
      "train_flow_matching_loss": 0.1904296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 14760,
      "train_flow_matching_loss": 0.1513671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 14760,
      "train_flow_matching_loss": 0.0771484375,
      "train_physics_loss": 0.4227786660194397
    },
    {
      "grad_norm": 0.07602278143167496,
      "learning_rate": 5e-05,
      "loss": 0.146,
      "step": 14770
    },
    {
      "step": 14770,
      "train_flow_matching_loss": 0.1298828125,
      "train_physics_loss": 0.0
    },
    {
      "step": 14770,
      "train_flow_matching_loss": 0.0888671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 14770,
      "train_flow_matching_loss": 0.177734375,
      "train_physics_loss": 0.0
    },
    {
      "step": 14770,
      "train_flow_matching_loss": 0.1943359375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10362166911363602,
      "learning_rate": 5e-05,
      "loss": 0.1398,
      "step": 14780
    },
    {
      "step": 14780,
      "train_flow_matching_loss": 0.1396484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 14780,
      "train_flow_matching_loss": 0.126953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 14780,
      "train_flow_matching_loss": 0.234375,
      "train_physics_loss": 0.0
    },
    {
      "step": 14780,
      "train_flow_matching_loss": 0.1943359375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09779636561870575,
      "learning_rate": 5e-05,
      "loss": 0.1397,
      "step": 14790
    },
    {
      "step": 14790,
      "train_flow_matching_loss": 0.1806640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 14790,
      "train_flow_matching_loss": 0.0947265625,
      "train_physics_loss": 0.0
    },
    {
      "step": 14790,
      "train_flow_matching_loss": 0.1611328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 14790,
      "train_flow_matching_loss": 0.2216796875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.06249051168560982,
      "learning_rate": 5e-05,
      "loss": 0.1354,
      "step": 14800
    },
    {
      "step": 14800,
      "train_flow_matching_loss": 0.1279296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 14800,
      "train_flow_matching_loss": 0.1806640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 14800,
      "train_flow_matching_loss": 0.1591796875,
      "train_physics_loss": 0.0
    },
    {
      "step": 14800,
      "train_flow_matching_loss": 0.1591796875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08815553039312363,
      "learning_rate": 5e-05,
      "loss": 0.1394,
      "step": 14810
    },
    {
      "step": 14810,
      "train_flow_matching_loss": 0.1357421875,
      "train_physics_loss": 0.0
    },
    {
      "step": 14810,
      "train_flow_matching_loss": 0.06884765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 14810,
      "train_flow_matching_loss": 0.1904296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 14810,
      "train_flow_matching_loss": 0.11376953125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10808554291725159,
      "learning_rate": 5e-05,
      "loss": 0.1388,
      "step": 14820
    },
    {
      "step": 14820,
      "train_flow_matching_loss": 0.166015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 14820,
      "train_flow_matching_loss": 0.205078125,
      "train_physics_loss": 0.0
    },
    {
      "step": 14820,
      "train_flow_matching_loss": 0.1630859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 14820,
      "train_flow_matching_loss": 0.1826171875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.14096303284168243,
      "learning_rate": 5e-05,
      "loss": 0.1385,
      "step": 14830
    },
    {
      "step": 14830,
      "train_flow_matching_loss": 0.09326171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 14830,
      "train_flow_matching_loss": 0.150390625,
      "train_physics_loss": 0.0
    },
    {
      "step": 14830,
      "train_flow_matching_loss": 0.1806640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 14830,
      "train_flow_matching_loss": 0.12060546875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11359623819589615,
      "learning_rate": 5e-05,
      "loss": 0.138,
      "step": 14840
    },
    {
      "step": 14840,
      "train_flow_matching_loss": 0.1455078125,
      "train_physics_loss": 0.0
    },
    {
      "step": 14840,
      "train_flow_matching_loss": 0.15625,
      "train_physics_loss": 0.0
    },
    {
      "step": 14840,
      "train_flow_matching_loss": 0.1611328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 14840,
      "train_flow_matching_loss": 0.0869140625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08576595038175583,
      "learning_rate": 5e-05,
      "loss": 0.1358,
      "step": 14850
    },
    {
      "step": 14850,
      "train_flow_matching_loss": 0.1630859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 14850,
      "train_flow_matching_loss": 0.197265625,
      "train_physics_loss": 0.0
    },
    {
      "step": 14850,
      "train_flow_matching_loss": 0.1240234375,
      "train_physics_loss": 0.0
    },
    {
      "step": 14850,
      "train_flow_matching_loss": 0.1142578125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.12327726185321808,
      "learning_rate": 5e-05,
      "loss": 0.1389,
      "step": 14860
    },
    {
      "step": 14860,
      "train_flow_matching_loss": 0.1796875,
      "train_physics_loss": 0.0
    },
    {
      "step": 14860,
      "train_flow_matching_loss": 0.1591796875,
      "train_physics_loss": 0.0
    },
    {
      "step": 14860,
      "train_flow_matching_loss": 0.201171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 14860,
      "train_flow_matching_loss": 0.1611328125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09244239330291748,
      "learning_rate": 5e-05,
      "loss": 0.1425,
      "step": 14870
    },
    {
      "step": 14870,
      "train_flow_matching_loss": 0.1708984375,
      "train_physics_loss": 0.0
    },
    {
      "step": 14870,
      "train_flow_matching_loss": 0.201171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 14870,
      "train_flow_matching_loss": 0.150390625,
      "train_physics_loss": 0.0
    },
    {
      "step": 14870,
      "train_flow_matching_loss": 0.15625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.0934281200170517,
      "learning_rate": 5e-05,
      "loss": 0.1397,
      "step": 14880
    },
    {
      "step": 14880,
      "train_flow_matching_loss": 0.1181640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 14880,
      "train_flow_matching_loss": 0.08056640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 14880,
      "train_flow_matching_loss": 0.177734375,
      "train_physics_loss": 0.0
    },
    {
      "step": 14880,
      "train_flow_matching_loss": 0.171875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.1266254037618637,
      "learning_rate": 5e-05,
      "loss": 0.1376,
      "step": 14890
    },
    {
      "step": 14890,
      "train_flow_matching_loss": 0.162109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 14890,
      "train_flow_matching_loss": 0.10693359375,
      "train_physics_loss": 0.0
    },
    {
      "step": 14890,
      "train_flow_matching_loss": 0.154296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 14890,
      "train_flow_matching_loss": 0.1162109375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.1324175000190735,
      "learning_rate": 5e-05,
      "loss": 0.1352,
      "step": 14900
    },
    {
      "step": 14900,
      "train_flow_matching_loss": 0.1279296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 14900,
      "train_flow_matching_loss": 0.1904296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 14900,
      "train_flow_matching_loss": 0.134765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 14900,
      "train_flow_matching_loss": 0.1044921875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.14737403392791748,
      "learning_rate": 5e-05,
      "loss": 0.1358,
      "step": 14910
    },
    {
      "step": 14910,
      "train_flow_matching_loss": 0.150390625,
      "train_physics_loss": 0.0
    },
    {
      "step": 14910,
      "train_flow_matching_loss": 0.09912109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 14910,
      "train_flow_matching_loss": 0.201171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 14910,
      "train_flow_matching_loss": 0.1494140625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10873666405677795,
      "learning_rate": 5e-05,
      "loss": 0.1469,
      "step": 14920
    },
    {
      "step": 14920,
      "train_flow_matching_loss": 0.1298828125,
      "train_physics_loss": 0.0
    },
    {
      "step": 14920,
      "train_flow_matching_loss": 0.1474609375,
      "train_physics_loss": 0.0
    },
    {
      "step": 14920,
      "train_flow_matching_loss": 0.043701171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 14920,
      "train_flow_matching_loss": 0.11083984375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08544749766588211,
      "learning_rate": 5e-05,
      "loss": 0.1368,
      "step": 14930
    },
    {
      "step": 14930,
      "train_flow_matching_loss": 0.1552734375,
      "train_physics_loss": 0.0
    },
    {
      "step": 14930,
      "train_flow_matching_loss": 0.1630859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 14930,
      "train_flow_matching_loss": 0.232421875,
      "train_physics_loss": 0.0
    },
    {
      "step": 14930,
      "train_flow_matching_loss": 0.2041015625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.13457471132278442,
      "learning_rate": 5e-05,
      "loss": 0.1415,
      "step": 14940
    },
    {
      "step": 14940,
      "train_flow_matching_loss": 0.0732421875,
      "train_physics_loss": 0.0
    },
    {
      "step": 14940,
      "train_flow_matching_loss": 0.26171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 14940,
      "train_flow_matching_loss": 0.126953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 14940,
      "train_flow_matching_loss": 0.1337890625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.1146920919418335,
      "learning_rate": 5e-05,
      "loss": 0.1396,
      "step": 14950
    },
    {
      "step": 14950,
      "train_flow_matching_loss": 0.1455078125,
      "train_physics_loss": 0.0
    },
    {
      "step": 14950,
      "train_flow_matching_loss": 0.11083984375,
      "train_physics_loss": 0.0
    },
    {
      "step": 14950,
      "train_flow_matching_loss": 0.125,
      "train_physics_loss": 0.0
    },
    {
      "step": 14950,
      "train_flow_matching_loss": 0.1767578125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.07271361351013184,
      "learning_rate": 5e-05,
      "loss": 0.1352,
      "step": 14960
    },
    {
      "step": 14960,
      "train_flow_matching_loss": 0.1533203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 14960,
      "train_flow_matching_loss": 0.134765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 14960,
      "train_flow_matching_loss": 0.1767578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 14960,
      "train_flow_matching_loss": 0.2119140625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.19265717267990112,
      "learning_rate": 5e-05,
      "loss": 0.1362,
      "step": 14970
    },
    {
      "step": 14970,
      "train_flow_matching_loss": 0.158203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 14970,
      "train_flow_matching_loss": 0.2177734375,
      "train_physics_loss": 0.0
    },
    {
      "step": 14970,
      "train_flow_matching_loss": 0.1572265625,
      "train_physics_loss": 0.0
    },
    {
      "step": 14970,
      "train_flow_matching_loss": 0.1337890625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.0754137709736824,
      "learning_rate": 5e-05,
      "loss": 0.1369,
      "step": 14980
    },
    {
      "step": 14980,
      "train_flow_matching_loss": 0.11376953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 14980,
      "train_flow_matching_loss": 0.09912109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 14980,
      "train_flow_matching_loss": 0.166015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 14980,
      "train_flow_matching_loss": 0.0908203125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11526960879564285,
      "learning_rate": 5e-05,
      "loss": 0.1464,
      "step": 14990
    },
    {
      "step": 14990,
      "train_flow_matching_loss": 0.11328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 14990,
      "train_flow_matching_loss": 0.1396484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 14990,
      "train_flow_matching_loss": 0.12451171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 14990,
      "train_flow_matching_loss": 0.1318359375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09407442063093185,
      "learning_rate": 5e-05,
      "loss": 0.1346,
      "step": 15000
    },
    {
      "step": 15000,
      "train_flow_matching_loss": 0.0966796875,
      "train_physics_loss": 0.0
    },
    {
      "step": 15000,
      "train_flow_matching_loss": 0.1357421875,
      "train_physics_loss": 0.0
    },
    {
      "step": 15000,
      "train_flow_matching_loss": 0.1298828125,
      "train_physics_loss": 0.0
    },
    {
      "step": 15000,
      "train_flow_matching_loss": 0.1533203125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08500252664089203,
      "learning_rate": 5e-05,
      "loss": 0.1404,
      "step": 15010
    },
    {
      "step": 15010,
      "train_flow_matching_loss": 0.1474609375,
      "train_physics_loss": 0.0
    },
    {
      "step": 15010,
      "train_flow_matching_loss": 0.11474609375,
      "train_physics_loss": 0.0
    },
    {
      "step": 15010,
      "train_flow_matching_loss": 0.2353515625,
      "train_physics_loss": 0.0
    },
    {
      "step": 15010,
      "train_flow_matching_loss": 0.0966796875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10911952704191208,
      "learning_rate": 5e-05,
      "loss": 0.1339,
      "step": 15020
    },
    {
      "step": 15020,
      "train_flow_matching_loss": 0.12353515625,
      "train_physics_loss": 0.0
    },
    {
      "step": 15020,
      "train_flow_matching_loss": 0.197265625,
      "train_physics_loss": 0.0
    },
    {
      "step": 15020,
      "train_flow_matching_loss": 0.134765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 15020,
      "train_flow_matching_loss": 0.1474609375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.12117145955562592,
      "learning_rate": 5e-05,
      "loss": 0.141,
      "step": 15030
    },
    {
      "step": 15030,
      "train_flow_matching_loss": 0.138671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 15030,
      "train_flow_matching_loss": 0.1689453125,
      "train_physics_loss": 0.0
    },
    {
      "step": 15030,
      "train_flow_matching_loss": 0.2001953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 15030,
      "train_flow_matching_loss": 0.11572265625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09375333040952682,
      "learning_rate": 5e-05,
      "loss": 0.1379,
      "step": 15040
    },
    {
      "step": 15040,
      "train_flow_matching_loss": 0.1259765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 15040,
      "train_flow_matching_loss": 0.12255859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 15040,
      "train_flow_matching_loss": 0.142578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 15040,
      "train_flow_matching_loss": 0.1591796875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.1294279843568802,
      "learning_rate": 5e-05,
      "loss": 0.1385,
      "step": 15050
    },
    {
      "step": 15050,
      "train_flow_matching_loss": 0.11376953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 15050,
      "train_flow_matching_loss": 0.15625,
      "train_physics_loss": 0.0
    },
    {
      "step": 15050,
      "train_flow_matching_loss": 0.1728515625,
      "train_physics_loss": 0.0
    },
    {
      "step": 15050,
      "train_flow_matching_loss": 0.12255859375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.15659867227077484,
      "learning_rate": 5e-05,
      "loss": 0.1384,
      "step": 15060
    },
    {
      "step": 15060,
      "train_flow_matching_loss": 0.1083984375,
      "train_physics_loss": 0.0
    },
    {
      "step": 15060,
      "train_flow_matching_loss": 0.177734375,
      "train_physics_loss": 0.0
    },
    {
      "step": 15060,
      "train_flow_matching_loss": 0.10693359375,
      "train_physics_loss": 0.0
    },
    {
      "step": 15060,
      "train_flow_matching_loss": 0.1533203125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08140257745981216,
      "learning_rate": 5e-05,
      "loss": 0.1407,
      "step": 15070
    },
    {
      "step": 15070,
      "train_flow_matching_loss": 0.1162109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 15070,
      "train_flow_matching_loss": 0.1552734375,
      "train_physics_loss": 0.0
    },
    {
      "step": 15070,
      "train_flow_matching_loss": 0.06884765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 15070,
      "train_flow_matching_loss": 0.09033203125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08914367109537125,
      "learning_rate": 5e-05,
      "loss": 0.1361,
      "step": 15080
    },
    {
      "step": 15080,
      "train_flow_matching_loss": 0.0751953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 15080,
      "train_flow_matching_loss": 0.10791015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 15080,
      "train_flow_matching_loss": 0.1865234375,
      "train_physics_loss": 0.0
    },
    {
      "step": 15080,
      "train_flow_matching_loss": 0.1513671875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.0819920003414154,
      "learning_rate": 5e-05,
      "loss": 0.1402,
      "step": 15090
    },
    {
      "step": 15090,
      "train_flow_matching_loss": 0.125,
      "train_physics_loss": 0.0
    },
    {
      "step": 15090,
      "train_flow_matching_loss": 0.1845703125,
      "train_physics_loss": 0.0
    },
    {
      "step": 15090,
      "train_flow_matching_loss": 0.142578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 15090,
      "train_flow_matching_loss": 0.0869140625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08168147504329681,
      "learning_rate": 5e-05,
      "loss": 0.1375,
      "step": 15100
    },
    {
      "step": 15100,
      "train_flow_matching_loss": 0.15625,
      "train_physics_loss": 0.0
    },
    {
      "step": 15100,
      "train_flow_matching_loss": 0.0849609375,
      "train_physics_loss": 0.0
    },
    {
      "step": 15100,
      "train_flow_matching_loss": 0.1669921875,
      "train_physics_loss": 0.0
    },
    {
      "step": 15100,
      "train_flow_matching_loss": 0.11474609375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.12298641353845596,
      "learning_rate": 5e-05,
      "loss": 0.1455,
      "step": 15110
    },
    {
      "step": 15110,
      "train_flow_matching_loss": 0.1796875,
      "train_physics_loss": 0.0
    },
    {
      "step": 15110,
      "train_flow_matching_loss": 0.12158203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 15110,
      "train_flow_matching_loss": 0.1279296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 15110,
      "train_flow_matching_loss": 0.1416015625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10169751197099686,
      "learning_rate": 5e-05,
      "loss": 0.1425,
      "step": 15120
    },
    {
      "step": 15120,
      "train_flow_matching_loss": 0.1611328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 15120,
      "train_flow_matching_loss": 0.09423828125,
      "train_physics_loss": 0.0
    },
    {
      "step": 15120,
      "train_flow_matching_loss": 0.1484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 15120,
      "train_flow_matching_loss": 0.1533203125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.07573933899402618,
      "learning_rate": 5e-05,
      "loss": 0.1398,
      "step": 15130
    },
    {
      "step": 15130,
      "train_flow_matching_loss": 0.1865234375,
      "train_physics_loss": 0.0
    },
    {
      "step": 15130,
      "train_flow_matching_loss": 0.1826171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 15130,
      "train_flow_matching_loss": 0.11279296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 15130,
      "train_flow_matching_loss": 0.08447265625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08751090615987778,
      "learning_rate": 5e-05,
      "loss": 0.138,
      "step": 15140
    },
    {
      "step": 15140,
      "train_flow_matching_loss": 0.11767578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 15140,
      "train_flow_matching_loss": 0.08154296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 15140,
      "train_flow_matching_loss": 0.162109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 15140,
      "train_flow_matching_loss": 0.1298828125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.17508564889431,
      "learning_rate": 5e-05,
      "loss": 0.1375,
      "step": 15150
    },
    {
      "step": 15150,
      "train_flow_matching_loss": 0.1689453125,
      "train_physics_loss": 0.0
    },
    {
      "step": 15150,
      "train_flow_matching_loss": 0.224609375,
      "train_physics_loss": 0.0
    },
    {
      "step": 15150,
      "train_flow_matching_loss": 0.09228515625,
      "train_physics_loss": 0.0
    },
    {
      "step": 15150,
      "train_flow_matching_loss": 0.1328125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.07786083221435547,
      "learning_rate": 5e-05,
      "loss": 0.1367,
      "step": 15160
    },
    {
      "step": 15160,
      "train_flow_matching_loss": 0.1513671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 15160,
      "train_flow_matching_loss": 0.2080078125,
      "train_physics_loss": 0.0
    },
    {
      "step": 15160,
      "train_flow_matching_loss": 0.11669921875,
      "train_physics_loss": 0.0
    },
    {
      "step": 15160,
      "train_flow_matching_loss": 0.08203125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.06622079759836197,
      "learning_rate": 5e-05,
      "loss": 0.1389,
      "step": 15170
    },
    {
      "step": 15170,
      "train_flow_matching_loss": 0.12109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 15170,
      "train_flow_matching_loss": 0.1640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 15170,
      "train_flow_matching_loss": 0.10791015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 15170,
      "train_flow_matching_loss": 0.06494140625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11816391348838806,
      "learning_rate": 5e-05,
      "loss": 0.1376,
      "step": 15180
    },
    {
      "step": 15180,
      "train_flow_matching_loss": 0.1943359375,
      "train_physics_loss": 0.0
    },
    {
      "step": 15180,
      "train_flow_matching_loss": 0.12158203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 15180,
      "train_flow_matching_loss": 0.1708984375,
      "train_physics_loss": 0.0
    },
    {
      "step": 15180,
      "train_flow_matching_loss": 0.1279296875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08370979875326157,
      "learning_rate": 5e-05,
      "loss": 0.138,
      "step": 15190
    },
    {
      "step": 15190,
      "train_flow_matching_loss": 0.1337890625,
      "train_physics_loss": 0.0
    },
    {
      "step": 15190,
      "train_flow_matching_loss": 0.11962890625,
      "train_physics_loss": 0.0
    },
    {
      "step": 15190,
      "train_flow_matching_loss": 0.142578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 15190,
      "train_flow_matching_loss": 0.138671875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10768818110227585,
      "learning_rate": 5e-05,
      "loss": 0.1418,
      "step": 15200
    },
    {
      "step": 15200,
      "train_flow_matching_loss": 0.150390625,
      "train_physics_loss": 0.0
    },
    {
      "step": 15200,
      "train_flow_matching_loss": 0.1650390625,
      "train_physics_loss": 0.0
    },
    {
      "step": 15200,
      "train_flow_matching_loss": 0.07861328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 15200,
      "train_flow_matching_loss": 0.1982421875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11743833124637604,
      "learning_rate": 5e-05,
      "loss": 0.1405,
      "step": 15210
    },
    {
      "step": 15210,
      "train_flow_matching_loss": 0.1494140625,
      "train_physics_loss": 0.0
    },
    {
      "step": 15210,
      "train_flow_matching_loss": 0.10693359375,
      "train_physics_loss": 0.0
    },
    {
      "step": 15210,
      "train_flow_matching_loss": 0.0791015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 15210,
      "train_flow_matching_loss": 0.1259765625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.076980821788311,
      "learning_rate": 5e-05,
      "loss": 0.1342,
      "step": 15220
    },
    {
      "step": 15220,
      "train_flow_matching_loss": 0.1484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 15220,
      "train_flow_matching_loss": 0.1201171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 15220,
      "train_flow_matching_loss": 0.10498046875,
      "train_physics_loss": 0.0
    },
    {
      "step": 15220,
      "train_flow_matching_loss": 0.134765625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08164877444505692,
      "learning_rate": 5e-05,
      "loss": 0.1374,
      "step": 15230
    },
    {
      "step": 15230,
      "train_flow_matching_loss": 0.197265625,
      "train_physics_loss": 0.0
    },
    {
      "step": 15230,
      "train_flow_matching_loss": 0.10791015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 15230,
      "train_flow_matching_loss": 0.142578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 15230,
      "train_flow_matching_loss": 0.2216796875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09302470088005066,
      "learning_rate": 5e-05,
      "loss": 0.1374,
      "step": 15240
    },
    {
      "step": 15240,
      "train_flow_matching_loss": 0.142578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 15240,
      "train_flow_matching_loss": 0.08447265625,
      "train_physics_loss": 0.0
    },
    {
      "step": 15240,
      "train_flow_matching_loss": 0.1005859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 15240,
      "train_flow_matching_loss": 0.1748046875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09506456553936005,
      "learning_rate": 5e-05,
      "loss": 0.1402,
      "step": 15250
    },
    {
      "step": 15250,
      "train_flow_matching_loss": 0.142578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 15250,
      "train_flow_matching_loss": 0.07958984375,
      "train_physics_loss": 0.0
    },
    {
      "step": 15250,
      "train_flow_matching_loss": 0.134765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 15250,
      "train_flow_matching_loss": 0.0908203125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10600747913122177,
      "learning_rate": 5e-05,
      "loss": 0.1364,
      "step": 15260
    },
    {
      "step": 15260,
      "train_flow_matching_loss": 0.177734375,
      "train_physics_loss": 0.0
    },
    {
      "step": 15260,
      "train_flow_matching_loss": 0.11767578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 15260,
      "train_flow_matching_loss": 0.1337890625,
      "train_physics_loss": 0.0
    },
    {
      "step": 15260,
      "train_flow_matching_loss": 0.1826171875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.12210611253976822,
      "learning_rate": 5e-05,
      "loss": 0.1384,
      "step": 15270
    },
    {
      "step": 15270,
      "train_flow_matching_loss": 0.09619140625,
      "train_physics_loss": 0.0
    },
    {
      "step": 15270,
      "train_flow_matching_loss": 0.181640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 15270,
      "train_flow_matching_loss": 0.15625,
      "train_physics_loss": 0.0
    },
    {
      "step": 15270,
      "train_flow_matching_loss": 0.12109375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10508526116609573,
      "learning_rate": 5e-05,
      "loss": 0.1413,
      "step": 15280
    },
    {
      "step": 15280,
      "train_flow_matching_loss": 0.142578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 15280,
      "train_flow_matching_loss": 0.10107421875,
      "train_physics_loss": 0.0
    },
    {
      "step": 15280,
      "train_flow_matching_loss": 0.08154296875,
      "train_physics_loss": 0.4877016544342041
    },
    {
      "step": 15280,
      "train_flow_matching_loss": 0.130859375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.0723804235458374,
      "learning_rate": 5e-05,
      "loss": 0.1412,
      "step": 15290
    },
    {
      "step": 15290,
      "train_flow_matching_loss": 0.09228515625,
      "train_physics_loss": 0.0
    },
    {
      "step": 15290,
      "train_flow_matching_loss": 0.1806640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 15290,
      "train_flow_matching_loss": 0.08935546875,
      "train_physics_loss": 0.48611029982566833
    },
    {
      "step": 15290,
      "train_flow_matching_loss": 0.08984375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09438132494688034,
      "learning_rate": 5e-05,
      "loss": 0.1358,
      "step": 15300
    },
    {
      "step": 15300,
      "train_flow_matching_loss": 0.1669921875,
      "train_physics_loss": 0.0
    },
    {
      "step": 15300,
      "train_flow_matching_loss": 0.1494140625,
      "train_physics_loss": 0.0
    },
    {
      "step": 15300,
      "train_flow_matching_loss": 0.09130859375,
      "train_physics_loss": 0.42936205863952637
    },
    {
      "step": 15300,
      "train_flow_matching_loss": 0.09814453125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.33956921100616455,
      "learning_rate": 5e-05,
      "loss": 0.1408,
      "step": 15310
    },
    {
      "step": 15310,
      "train_flow_matching_loss": 0.1923828125,
      "train_physics_loss": 0.0
    },
    {
      "step": 15310,
      "train_flow_matching_loss": 0.134765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 15310,
      "train_flow_matching_loss": 0.11181640625,
      "train_physics_loss": 0.48742491006851196
    },
    {
      "step": 15310,
      "train_flow_matching_loss": 0.10107421875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.1251327246427536,
      "learning_rate": 5e-05,
      "loss": 0.1473,
      "step": 15320
    },
    {
      "step": 15320,
      "train_flow_matching_loss": 0.150390625,
      "train_physics_loss": 0.0
    },
    {
      "step": 15320,
      "train_flow_matching_loss": 0.2099609375,
      "train_physics_loss": 0.0
    },
    {
      "step": 15320,
      "train_flow_matching_loss": 0.046630859375,
      "train_physics_loss": 0.47661900520324707
    },
    {
      "step": 15320,
      "train_flow_matching_loss": 0.228515625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.0874103456735611,
      "learning_rate": 5e-05,
      "loss": 0.1442,
      "step": 15330
    },
    {
      "step": 15330,
      "train_flow_matching_loss": 0.1279296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 15330,
      "train_flow_matching_loss": 0.09814453125,
      "train_physics_loss": 0.0
    },
    {
      "step": 15330,
      "train_flow_matching_loss": 0.09521484375,
      "train_physics_loss": 0.5544835925102234
    },
    {
      "step": 15330,
      "train_flow_matching_loss": 0.1484375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08365847170352936,
      "learning_rate": 5e-05,
      "loss": 0.1402,
      "step": 15340
    },
    {
      "step": 15340,
      "train_flow_matching_loss": 0.208984375,
      "train_physics_loss": 0.0
    },
    {
      "step": 15340,
      "train_flow_matching_loss": 0.1396484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 15340,
      "train_flow_matching_loss": 0.06396484375,
      "train_physics_loss": 0.5198897123336792
    },
    {
      "step": 15340,
      "train_flow_matching_loss": 0.1630859375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.07770109921693802,
      "learning_rate": 5e-05,
      "loss": 0.1375,
      "step": 15350
    },
    {
      "step": 15350,
      "train_flow_matching_loss": 0.09228515625,
      "train_physics_loss": 0.0
    },
    {
      "step": 15350,
      "train_flow_matching_loss": 0.10693359375,
      "train_physics_loss": 0.0
    },
    {
      "step": 15350,
      "train_flow_matching_loss": 0.04736328125,
      "train_physics_loss": 0.4238298237323761
    },
    {
      "step": 15350,
      "train_flow_matching_loss": 0.10302734375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.1062915027141571,
      "learning_rate": 5e-05,
      "loss": 0.1366,
      "step": 15360
    },
    {
      "step": 15360,
      "train_flow_matching_loss": 0.08251953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 15360,
      "train_flow_matching_loss": 0.1689453125,
      "train_physics_loss": 0.0
    },
    {
      "step": 15360,
      "train_flow_matching_loss": 0.07861328125,
      "train_physics_loss": 0.4368528723716736
    },
    {
      "step": 15360,
      "train_flow_matching_loss": 0.142578125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.05648857355117798,
      "learning_rate": 5e-05,
      "loss": 0.135,
      "step": 15370
    },
    {
      "step": 15370,
      "train_flow_matching_loss": 0.1962890625,
      "train_physics_loss": 0.0
    },
    {
      "step": 15370,
      "train_flow_matching_loss": 0.1640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 15370,
      "train_flow_matching_loss": 0.042236328125,
      "train_physics_loss": 0.5283318758010864
    },
    {
      "step": 15370,
      "train_flow_matching_loss": 0.2060546875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.07227209210395813,
      "learning_rate": 5e-05,
      "loss": 0.1383,
      "step": 15380
    },
    {
      "step": 15380,
      "train_flow_matching_loss": 0.1513671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 15380,
      "train_flow_matching_loss": 0.1748046875,
      "train_physics_loss": 0.0
    },
    {
      "step": 15380,
      "train_flow_matching_loss": 0.05517578125,
      "train_physics_loss": 0.45429477095603943
    },
    {
      "step": 15380,
      "train_flow_matching_loss": 0.158203125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09757515043020248,
      "learning_rate": 5e-05,
      "loss": 0.1419,
      "step": 15390
    },
    {
      "step": 15390,
      "train_flow_matching_loss": 0.1640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 15390,
      "train_flow_matching_loss": 0.1748046875,
      "train_physics_loss": 0.0
    },
    {
      "step": 15390,
      "train_flow_matching_loss": 0.048095703125,
      "train_physics_loss": 0.4555404782295227
    },
    {
      "step": 15390,
      "train_flow_matching_loss": 0.1337890625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09091021865606308,
      "learning_rate": 5e-05,
      "loss": 0.1419,
      "step": 15400
    },
    {
      "step": 15400,
      "train_flow_matching_loss": 0.1552734375,
      "train_physics_loss": 0.0
    },
    {
      "step": 15400,
      "train_flow_matching_loss": 0.2119140625,
      "train_physics_loss": 0.0
    },
    {
      "step": 15400,
      "train_flow_matching_loss": 0.154296875,
      "train_physics_loss": 0.48938679695129395
    },
    {
      "step": 15400,
      "train_flow_matching_loss": 0.2451171875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.8649597764015198,
      "learning_rate": 5e-05,
      "loss": 0.1375,
      "step": 15410
    },
    {
      "step": 15410,
      "train_flow_matching_loss": 0.10009765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 15410,
      "train_flow_matching_loss": 0.171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 15410,
      "train_flow_matching_loss": 0.0810546875,
      "train_physics_loss": 0.5388220548629761
    },
    {
      "step": 15410,
      "train_flow_matching_loss": 0.1650390625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10256778448820114,
      "learning_rate": 5e-05,
      "loss": 0.1431,
      "step": 15420
    },
    {
      "step": 15420,
      "train_flow_matching_loss": 0.1279296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 15420,
      "train_flow_matching_loss": 0.1484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 15420,
      "train_flow_matching_loss": 0.05615234375,
      "train_physics_loss": 0.43889179825782776
    },
    {
      "step": 15420,
      "train_flow_matching_loss": 0.1396484375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.21653343737125397,
      "learning_rate": 5e-05,
      "loss": 0.1379,
      "step": 15430
    },
    {
      "step": 15430,
      "train_flow_matching_loss": 0.1796875,
      "train_physics_loss": 0.0
    },
    {
      "step": 15430,
      "train_flow_matching_loss": 0.11279296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 15430,
      "train_flow_matching_loss": 0.068359375,
      "train_physics_loss": 0.4830441176891327
    },
    {
      "step": 15430,
      "train_flow_matching_loss": 0.1298828125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.0765182301402092,
      "learning_rate": 5e-05,
      "loss": 0.137,
      "step": 15440
    },
    {
      "step": 15440,
      "train_flow_matching_loss": 0.1875,
      "train_physics_loss": 0.0
    },
    {
      "step": 15440,
      "train_flow_matching_loss": 0.150390625,
      "train_physics_loss": 0.0
    },
    {
      "step": 15440,
      "train_flow_matching_loss": 0.048095703125,
      "train_physics_loss": 0.5288510918617249
    },
    {
      "step": 15440,
      "train_flow_matching_loss": 0.146484375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.12262129038572311,
      "learning_rate": 5e-05,
      "loss": 0.1397,
      "step": 15450
    },
    {
      "step": 15450,
      "train_flow_matching_loss": 0.1748046875,
      "train_physics_loss": 0.0
    },
    {
      "step": 15450,
      "train_flow_matching_loss": 0.162109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 15450,
      "train_flow_matching_loss": 0.06787109375,
      "train_physics_loss": 0.560839056968689
    },
    {
      "step": 15450,
      "train_flow_matching_loss": 0.09814453125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09411384165287018,
      "learning_rate": 5e-05,
      "loss": 0.1409,
      "step": 15460
    },
    {
      "step": 15460,
      "train_flow_matching_loss": 0.1396484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 15460,
      "train_flow_matching_loss": 0.1904296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 15460,
      "train_flow_matching_loss": 0.1064453125,
      "train_physics_loss": 0.5403193831443787
    },
    {
      "step": 15460,
      "train_flow_matching_loss": 0.12890625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09635304659605026,
      "learning_rate": 5e-05,
      "loss": 0.1367,
      "step": 15470
    },
    {
      "step": 15470,
      "train_flow_matching_loss": 0.1279296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 15470,
      "train_flow_matching_loss": 0.146484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 15470,
      "train_flow_matching_loss": 0.042724609375,
      "train_physics_loss": 0.4826945662498474
    },
    {
      "step": 15470,
      "train_flow_matching_loss": 0.109375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11356497555971146,
      "learning_rate": 5e-05,
      "loss": 0.134,
      "step": 15480
    },
    {
      "step": 15480,
      "train_flow_matching_loss": 0.158203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 15480,
      "train_flow_matching_loss": 0.142578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 15480,
      "train_flow_matching_loss": 0.0712890625,
      "train_physics_loss": 0.42900383472442627
    },
    {
      "step": 15480,
      "train_flow_matching_loss": 0.1357421875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.14432989060878754,
      "learning_rate": 5e-05,
      "loss": 0.1412,
      "step": 15490
    },
    {
      "step": 15490,
      "train_flow_matching_loss": 0.1728515625,
      "train_physics_loss": 0.0
    },
    {
      "step": 15490,
      "train_flow_matching_loss": 0.11865234375,
      "train_physics_loss": 0.0
    },
    {
      "step": 15490,
      "train_flow_matching_loss": 0.034912109375,
      "train_physics_loss": 0.48918256163597107
    },
    {
      "step": 15490,
      "train_flow_matching_loss": 0.1845703125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.0848502665758133,
      "learning_rate": 5e-05,
      "loss": 0.1411,
      "step": 15500
    },
    {
      "step": 15500,
      "train_flow_matching_loss": 0.1474609375,
      "train_physics_loss": 0.0
    },
    {
      "step": 15500,
      "train_flow_matching_loss": 0.181640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 15500,
      "train_flow_matching_loss": 0.09814453125,
      "train_physics_loss": 0.4823071360588074
    },
    {
      "step": 15500,
      "train_flow_matching_loss": 0.11279296875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.07792361825704575,
      "learning_rate": 5e-05,
      "loss": 0.1374,
      "step": 15510
    },
    {
      "step": 15510,
      "train_flow_matching_loss": 0.193359375,
      "train_physics_loss": 0.0
    },
    {
      "step": 15510,
      "train_flow_matching_loss": 0.166015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 15510,
      "train_flow_matching_loss": 0.0625,
      "train_physics_loss": 0.5483680963516235
    },
    {
      "step": 15510,
      "train_flow_matching_loss": 0.1103515625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.07994668185710907,
      "learning_rate": 5e-05,
      "loss": 0.1363,
      "step": 15520
    },
    {
      "step": 15520,
      "train_flow_matching_loss": 0.10302734375,
      "train_physics_loss": 0.0
    },
    {
      "step": 15520,
      "train_flow_matching_loss": 0.1201171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 15520,
      "train_flow_matching_loss": 0.049072265625,
      "train_physics_loss": 0.4115137755870819
    },
    {
      "step": 15520,
      "train_flow_matching_loss": 0.1689453125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10164375603199005,
      "learning_rate": 5e-05,
      "loss": 0.1459,
      "step": 15530
    },
    {
      "step": 15530,
      "train_flow_matching_loss": 0.09375,
      "train_physics_loss": 0.0
    },
    {
      "step": 15530,
      "train_flow_matching_loss": 0.09912109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 15530,
      "train_flow_matching_loss": 0.11572265625,
      "train_physics_loss": 0.5515661239624023
    },
    {
      "step": 15530,
      "train_flow_matching_loss": 0.169921875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10540987551212311,
      "learning_rate": 5e-05,
      "loss": 0.1417,
      "step": 15540
    },
    {
      "step": 15540,
      "train_flow_matching_loss": 0.11279296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 15540,
      "train_flow_matching_loss": 0.1904296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 15540,
      "train_flow_matching_loss": 0.1787109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 15540,
      "train_flow_matching_loss": 0.10986328125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.29119619727134705,
      "learning_rate": 5e-05,
      "loss": 0.1375,
      "step": 15550
    },
    {
      "step": 15550,
      "train_flow_matching_loss": 0.134765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 15550,
      "train_flow_matching_loss": 0.10205078125,
      "train_physics_loss": 0.0
    },
    {
      "step": 15550,
      "train_flow_matching_loss": 0.1572265625,
      "train_physics_loss": 0.0
    },
    {
      "step": 15550,
      "train_flow_matching_loss": 0.1279296875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09686028212308884,
      "learning_rate": 5e-05,
      "loss": 0.1391,
      "step": 15560
    },
    {
      "step": 15560,
      "train_flow_matching_loss": 0.171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 15560,
      "train_flow_matching_loss": 0.1669921875,
      "train_physics_loss": 0.0
    },
    {
      "step": 15560,
      "train_flow_matching_loss": 0.06494140625,
      "train_physics_loss": 0.0
    },
    {
      "step": 15560,
      "train_flow_matching_loss": 0.1533203125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09609020501375198,
      "learning_rate": 5e-05,
      "loss": 0.1376,
      "step": 15570
    },
    {
      "step": 15570,
      "train_flow_matching_loss": 0.142578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 15570,
      "train_flow_matching_loss": 0.2080078125,
      "train_physics_loss": 0.0
    },
    {
      "step": 15570,
      "train_flow_matching_loss": 0.1572265625,
      "train_physics_loss": 0.0
    },
    {
      "step": 15570,
      "train_flow_matching_loss": 0.09912109375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10144457221031189,
      "learning_rate": 5e-05,
      "loss": 0.1367,
      "step": 15580
    },
    {
      "step": 15580,
      "train_flow_matching_loss": 0.1904296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 15580,
      "train_flow_matching_loss": 0.1298828125,
      "train_physics_loss": 0.0
    },
    {
      "step": 15580,
      "train_flow_matching_loss": 0.12255859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 15580,
      "train_flow_matching_loss": 0.1728515625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10506860911846161,
      "learning_rate": 5e-05,
      "loss": 0.1391,
      "step": 15590
    },
    {
      "step": 15590,
      "train_flow_matching_loss": 0.0966796875,
      "train_physics_loss": 0.0
    },
    {
      "step": 15590,
      "train_flow_matching_loss": 0.1865234375,
      "train_physics_loss": 0.0
    },
    {
      "step": 15590,
      "train_flow_matching_loss": 0.10888671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 15590,
      "train_flow_matching_loss": 0.1513671875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10467854142189026,
      "learning_rate": 5e-05,
      "loss": 0.1354,
      "step": 15600
    },
    {
      "step": 15600,
      "train_flow_matching_loss": 0.1337890625,
      "train_physics_loss": 0.0
    },
    {
      "step": 15600,
      "train_flow_matching_loss": 0.1640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 15600,
      "train_flow_matching_loss": 0.138671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 15600,
      "train_flow_matching_loss": 0.1669921875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.1235220804810524,
      "learning_rate": 5e-05,
      "loss": 0.1386,
      "step": 15610
    },
    {
      "step": 15610,
      "train_flow_matching_loss": 0.11279296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 15610,
      "train_flow_matching_loss": 0.162109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 15610,
      "train_flow_matching_loss": 0.1279296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 15610,
      "train_flow_matching_loss": 0.146484375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10033700615167618,
      "learning_rate": 5e-05,
      "loss": 0.1391,
      "step": 15620
    },
    {
      "step": 15620,
      "train_flow_matching_loss": 0.1943359375,
      "train_physics_loss": 0.0
    },
    {
      "step": 15620,
      "train_flow_matching_loss": 0.08349609375,
      "train_physics_loss": 0.0
    },
    {
      "step": 15620,
      "train_flow_matching_loss": 0.09375,
      "train_physics_loss": 0.0
    },
    {
      "step": 15620,
      "train_flow_matching_loss": 0.2236328125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.9821517467498779,
      "learning_rate": 5e-05,
      "loss": 0.1307,
      "step": 15630
    },
    {
      "step": 15630,
      "train_flow_matching_loss": 0.126953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 15630,
      "train_flow_matching_loss": 0.1298828125,
      "train_physics_loss": 0.0
    },
    {
      "step": 15630,
      "train_flow_matching_loss": 0.1259765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 15630,
      "train_flow_matching_loss": 0.1337890625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10532821714878082,
      "learning_rate": 5e-05,
      "loss": 0.1349,
      "step": 15640
    },
    {
      "step": 15640,
      "train_flow_matching_loss": 0.10791015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 15640,
      "train_flow_matching_loss": 0.1474609375,
      "train_physics_loss": 0.0
    },
    {
      "step": 15640,
      "train_flow_matching_loss": 0.15625,
      "train_physics_loss": 0.0
    },
    {
      "step": 15640,
      "train_flow_matching_loss": 0.1201171875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10772021114826202,
      "learning_rate": 5e-05,
      "loss": 0.1387,
      "step": 15650
    },
    {
      "step": 15650,
      "train_flow_matching_loss": 0.1826171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 15650,
      "train_flow_matching_loss": 0.1259765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 15650,
      "train_flow_matching_loss": 0.1201171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 15650,
      "train_flow_matching_loss": 0.1474609375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08424073457717896,
      "learning_rate": 5e-05,
      "loss": 0.1397,
      "step": 15660
    },
    {
      "step": 15660,
      "train_flow_matching_loss": 0.11328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 15660,
      "train_flow_matching_loss": 0.10888671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 15660,
      "train_flow_matching_loss": 0.1279296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 15660,
      "train_flow_matching_loss": 0.2109375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.07797741889953613,
      "learning_rate": 5e-05,
      "loss": 0.1358,
      "step": 15670
    },
    {
      "step": 15670,
      "train_flow_matching_loss": 0.130859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 15670,
      "train_flow_matching_loss": 0.12451171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 15670,
      "train_flow_matching_loss": 0.0908203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 15670,
      "train_flow_matching_loss": 0.1767578125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10851754993200302,
      "learning_rate": 5e-05,
      "loss": 0.1356,
      "step": 15680
    },
    {
      "step": 15680,
      "train_flow_matching_loss": 0.142578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 15680,
      "train_flow_matching_loss": 0.11962890625,
      "train_physics_loss": 0.0
    },
    {
      "step": 15680,
      "train_flow_matching_loss": 0.07861328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 15680,
      "train_flow_matching_loss": 0.146484375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10397902131080627,
      "learning_rate": 5e-05,
      "loss": 0.1356,
      "step": 15690
    },
    {
      "step": 15690,
      "train_flow_matching_loss": 0.12890625,
      "train_physics_loss": 0.0
    },
    {
      "step": 15690,
      "train_flow_matching_loss": 0.150390625,
      "train_physics_loss": 0.0
    },
    {
      "step": 15690,
      "train_flow_matching_loss": 0.138671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 15690,
      "train_flow_matching_loss": 0.11328125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10578665882349014,
      "learning_rate": 5e-05,
      "loss": 0.1341,
      "step": 15700
    },
    {
      "step": 15700,
      "train_flow_matching_loss": 0.08984375,
      "train_physics_loss": 0.0
    },
    {
      "step": 15700,
      "train_flow_matching_loss": 0.177734375,
      "train_physics_loss": 0.0
    },
    {
      "step": 15700,
      "train_flow_matching_loss": 0.1669921875,
      "train_physics_loss": 0.0
    },
    {
      "step": 15700,
      "train_flow_matching_loss": 0.154296875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.07667627185583115,
      "learning_rate": 5e-05,
      "loss": 0.1381,
      "step": 15710
    },
    {
      "step": 15710,
      "train_flow_matching_loss": 0.2431640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 15710,
      "train_flow_matching_loss": 0.1533203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 15710,
      "train_flow_matching_loss": 0.11083984375,
      "train_physics_loss": 0.0
    },
    {
      "step": 15710,
      "train_flow_matching_loss": 0.08837890625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.05508473515510559,
      "learning_rate": 5e-05,
      "loss": 0.136,
      "step": 15720
    },
    {
      "step": 15720,
      "train_flow_matching_loss": 0.1142578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 15720,
      "train_flow_matching_loss": 0.2177734375,
      "train_physics_loss": 0.0
    },
    {
      "step": 15720,
      "train_flow_matching_loss": 0.1005859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 15720,
      "train_flow_matching_loss": 0.134765625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09372356534004211,
      "learning_rate": 5e-05,
      "loss": 0.1355,
      "step": 15730
    },
    {
      "step": 15730,
      "train_flow_matching_loss": 0.1884765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 15730,
      "train_flow_matching_loss": 0.1171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 15730,
      "train_flow_matching_loss": 0.11767578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 15730,
      "train_flow_matching_loss": 0.1259765625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.1253298819065094,
      "learning_rate": 5e-05,
      "loss": 0.1376,
      "step": 15740
    },
    {
      "step": 15740,
      "train_flow_matching_loss": 0.1396484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 15740,
      "train_flow_matching_loss": 0.1455078125,
      "train_physics_loss": 0.0
    },
    {
      "step": 15740,
      "train_flow_matching_loss": 0.1103515625,
      "train_physics_loss": 0.0
    },
    {
      "step": 15740,
      "train_flow_matching_loss": 0.2197265625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08512961119413376,
      "learning_rate": 5e-05,
      "loss": 0.1368,
      "step": 15750
    },
    {
      "step": 15750,
      "train_flow_matching_loss": 0.1591796875,
      "train_physics_loss": 0.0
    },
    {
      "step": 15750,
      "train_flow_matching_loss": 0.2216796875,
      "train_physics_loss": 0.0
    },
    {
      "step": 15750,
      "train_flow_matching_loss": 0.0986328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 15750,
      "train_flow_matching_loss": 0.1728515625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08728566020727158,
      "learning_rate": 5e-05,
      "loss": 0.1391,
      "step": 15760
    },
    {
      "step": 15760,
      "train_flow_matching_loss": 0.1591796875,
      "train_physics_loss": 0.0
    },
    {
      "step": 15760,
      "train_flow_matching_loss": 0.1640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 15760,
      "train_flow_matching_loss": 0.08544921875,
      "train_physics_loss": 0.0
    },
    {
      "step": 15760,
      "train_flow_matching_loss": 0.09130859375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11963008344173431,
      "learning_rate": 5e-05,
      "loss": 0.1388,
      "step": 15770
    },
    {
      "step": 15770,
      "train_flow_matching_loss": 0.1591796875,
      "train_physics_loss": 0.0
    },
    {
      "step": 15770,
      "train_flow_matching_loss": 0.10498046875,
      "train_physics_loss": 0.0
    },
    {
      "step": 15770,
      "train_flow_matching_loss": 0.04931640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 15770,
      "train_flow_matching_loss": 0.154296875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11787401884794235,
      "learning_rate": 5e-05,
      "loss": 0.143,
      "step": 15780
    },
    {
      "step": 15780,
      "train_flow_matching_loss": 0.1083984375,
      "train_physics_loss": 0.0
    },
    {
      "step": 15780,
      "train_flow_matching_loss": 0.1787109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 15780,
      "train_flow_matching_loss": 0.1474609375,
      "train_physics_loss": 0.0
    },
    {
      "step": 15780,
      "train_flow_matching_loss": 0.1474609375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08328364044427872,
      "learning_rate": 5e-05,
      "loss": 0.1331,
      "step": 15790
    },
    {
      "step": 15790,
      "train_flow_matching_loss": 0.10693359375,
      "train_physics_loss": 0.0
    },
    {
      "step": 15790,
      "train_flow_matching_loss": 0.2236328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 15790,
      "train_flow_matching_loss": 0.1826171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 15790,
      "train_flow_matching_loss": 0.0859375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09948001056909561,
      "learning_rate": 5e-05,
      "loss": 0.1362,
      "step": 15800
    },
    {
      "step": 15800,
      "train_flow_matching_loss": 0.11083984375,
      "train_physics_loss": 0.0
    },
    {
      "step": 15800,
      "train_flow_matching_loss": 0.1669921875,
      "train_physics_loss": 0.0
    },
    {
      "step": 15800,
      "train_flow_matching_loss": 0.138671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 15800,
      "train_flow_matching_loss": 0.1845703125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.1618686467409134,
      "learning_rate": 5e-05,
      "loss": 0.136,
      "step": 15810
    },
    {
      "step": 15810,
      "train_flow_matching_loss": 0.09521484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 15810,
      "train_flow_matching_loss": 0.1015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 15810,
      "train_flow_matching_loss": 0.12255859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 15810,
      "train_flow_matching_loss": 0.12158203125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.16540972888469696,
      "learning_rate": 5e-05,
      "loss": 0.1368,
      "step": 15820
    },
    {
      "step": 15820,
      "train_flow_matching_loss": 0.1513671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 15820,
      "train_flow_matching_loss": 0.1748046875,
      "train_physics_loss": 0.0
    },
    {
      "step": 15820,
      "train_flow_matching_loss": 0.126953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 15820,
      "train_flow_matching_loss": 0.1591796875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08939839154481888,
      "learning_rate": 5e-05,
      "loss": 0.1406,
      "step": 15830
    },
    {
      "step": 15830,
      "train_flow_matching_loss": 0.09912109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 15830,
      "train_flow_matching_loss": 0.1337890625,
      "train_physics_loss": 0.0
    },
    {
      "step": 15830,
      "train_flow_matching_loss": 0.126953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 15830,
      "train_flow_matching_loss": 0.1494140625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.06401089578866959,
      "learning_rate": 5e-05,
      "loss": 0.1341,
      "step": 15840
    },
    {
      "step": 15840,
      "train_flow_matching_loss": 0.142578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 15840,
      "train_flow_matching_loss": 0.1298828125,
      "train_physics_loss": 0.0
    },
    {
      "step": 15840,
      "train_flow_matching_loss": 0.173828125,
      "train_physics_loss": 0.0
    },
    {
      "step": 15840,
      "train_flow_matching_loss": 0.28515625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.30656754970550537,
      "learning_rate": 5e-05,
      "loss": 0.1399,
      "step": 15850
    },
    {
      "step": 15850,
      "train_flow_matching_loss": 0.10791015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 15850,
      "train_flow_matching_loss": 0.130859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 15850,
      "train_flow_matching_loss": 0.1220703125,
      "train_physics_loss": 0.0
    },
    {
      "step": 15850,
      "train_flow_matching_loss": 0.10791015625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.13189367949962616,
      "learning_rate": 5e-05,
      "loss": 0.1348,
      "step": 15860
    },
    {
      "step": 15860,
      "train_flow_matching_loss": 0.142578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 15860,
      "train_flow_matching_loss": 0.1611328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 15860,
      "train_flow_matching_loss": 0.10107421875,
      "train_physics_loss": 0.0
    },
    {
      "step": 15860,
      "train_flow_matching_loss": 0.1259765625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.0642257034778595,
      "learning_rate": 5e-05,
      "loss": 0.1348,
      "step": 15870
    },
    {
      "step": 15870,
      "train_flow_matching_loss": 0.205078125,
      "train_physics_loss": 0.0
    },
    {
      "step": 15870,
      "train_flow_matching_loss": 0.130859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 15870,
      "train_flow_matching_loss": 0.11474609375,
      "train_physics_loss": 0.0
    },
    {
      "step": 15870,
      "train_flow_matching_loss": 0.10888671875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.1109623834490776,
      "learning_rate": 5e-05,
      "loss": 0.1409,
      "step": 15880
    },
    {
      "step": 15880,
      "train_flow_matching_loss": 0.1630859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 15880,
      "train_flow_matching_loss": 0.11181640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 15880,
      "train_flow_matching_loss": 0.13671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 15880,
      "train_flow_matching_loss": 0.197265625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11286380887031555,
      "learning_rate": 5e-05,
      "loss": 0.1372,
      "step": 15890
    },
    {
      "step": 15890,
      "train_flow_matching_loss": 0.11669921875,
      "train_physics_loss": 0.0
    },
    {
      "step": 15890,
      "train_flow_matching_loss": 0.1240234375,
      "train_physics_loss": 0.0
    },
    {
      "step": 15890,
      "train_flow_matching_loss": 0.1455078125,
      "train_physics_loss": 0.0
    },
    {
      "step": 15890,
      "train_flow_matching_loss": 0.2197265625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10130240768194199,
      "learning_rate": 5e-05,
      "loss": 0.1403,
      "step": 15900
    },
    {
      "step": 15900,
      "train_flow_matching_loss": 0.126953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 15900,
      "train_flow_matching_loss": 0.126953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 15900,
      "train_flow_matching_loss": 0.12353515625,
      "train_physics_loss": 0.0
    },
    {
      "step": 15900,
      "train_flow_matching_loss": 0.197265625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08775720745325089,
      "learning_rate": 5e-05,
      "loss": 0.141,
      "step": 15910
    },
    {
      "step": 15910,
      "train_flow_matching_loss": 0.09326171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 15910,
      "train_flow_matching_loss": 0.10791015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 15910,
      "train_flow_matching_loss": 0.142578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 15910,
      "train_flow_matching_loss": 0.154296875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.14230455458164215,
      "learning_rate": 5e-05,
      "loss": 0.1336,
      "step": 15920
    },
    {
      "step": 15920,
      "train_flow_matching_loss": 0.197265625,
      "train_physics_loss": 0.0
    },
    {
      "step": 15920,
      "train_flow_matching_loss": 0.177734375,
      "train_physics_loss": 0.0
    },
    {
      "step": 15920,
      "train_flow_matching_loss": 0.2265625,
      "train_physics_loss": 0.0
    },
    {
      "step": 15920,
      "train_flow_matching_loss": 0.1669921875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11157055199146271,
      "learning_rate": 5e-05,
      "loss": 0.1433,
      "step": 15930
    },
    {
      "step": 15930,
      "train_flow_matching_loss": 0.27734375,
      "train_physics_loss": 0.0
    },
    {
      "step": 15930,
      "train_flow_matching_loss": 0.1748046875,
      "train_physics_loss": 0.0
    },
    {
      "step": 15930,
      "train_flow_matching_loss": 0.08642578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 15930,
      "train_flow_matching_loss": 0.11181640625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.112178273499012,
      "learning_rate": 5e-05,
      "loss": 0.134,
      "step": 15940
    },
    {
      "step": 15940,
      "train_flow_matching_loss": 0.185546875,
      "train_physics_loss": 0.0
    },
    {
      "step": 15940,
      "train_flow_matching_loss": 0.1484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 15940,
      "train_flow_matching_loss": 0.0859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 15940,
      "train_flow_matching_loss": 0.142578125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10137400031089783,
      "learning_rate": 5e-05,
      "loss": 0.139,
      "step": 15950
    },
    {
      "step": 15950,
      "train_flow_matching_loss": 0.2294921875,
      "train_physics_loss": 0.0
    },
    {
      "step": 15950,
      "train_flow_matching_loss": 0.1787109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 15950,
      "train_flow_matching_loss": 0.1416015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 15950,
      "train_flow_matching_loss": 0.1708984375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10347674787044525,
      "learning_rate": 5e-05,
      "loss": 0.1418,
      "step": 15960
    },
    {
      "step": 15960,
      "train_flow_matching_loss": 0.1357421875,
      "train_physics_loss": 0.0
    },
    {
      "step": 15960,
      "train_flow_matching_loss": 0.06396484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 15960,
      "train_flow_matching_loss": 0.1533203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 15960,
      "train_flow_matching_loss": 0.1826171875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10186557471752167,
      "learning_rate": 5e-05,
      "loss": 0.1391,
      "step": 15970
    },
    {
      "step": 15970,
      "train_flow_matching_loss": 0.1845703125,
      "train_physics_loss": 0.0
    },
    {
      "step": 15970,
      "train_flow_matching_loss": 0.1875,
      "train_physics_loss": 0.0
    },
    {
      "step": 15970,
      "train_flow_matching_loss": 0.197265625,
      "train_physics_loss": 0.0
    },
    {
      "step": 15970,
      "train_flow_matching_loss": 0.1318359375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.06890220940113068,
      "learning_rate": 5e-05,
      "loss": 0.1359,
      "step": 15980
    },
    {
      "step": 15980,
      "train_flow_matching_loss": 0.1669921875,
      "train_physics_loss": 0.0
    },
    {
      "step": 15980,
      "train_flow_matching_loss": 0.0260009765625,
      "train_physics_loss": 0.43278902769088745
    },
    {
      "step": 15980,
      "train_flow_matching_loss": 0.1923828125,
      "train_physics_loss": 0.0
    },
    {
      "step": 15980,
      "train_flow_matching_loss": 0.1943359375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08345348387956619,
      "learning_rate": 5e-05,
      "loss": 0.1343,
      "step": 15990
    },
    {
      "step": 15990,
      "train_flow_matching_loss": 0.150390625,
      "train_physics_loss": 0.0
    },
    {
      "step": 15990,
      "train_flow_matching_loss": 0.048583984375,
      "train_physics_loss": 0.44496822357177734
    },
    {
      "step": 15990,
      "train_flow_matching_loss": 0.08837890625,
      "train_physics_loss": 0.0
    },
    {
      "step": 15990,
      "train_flow_matching_loss": 0.1044921875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.13404062390327454,
      "learning_rate": 5e-05,
      "loss": 0.1375,
      "step": 16000
    },
    {
      "step": 16000,
      "train_flow_matching_loss": 0.162109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 16000,
      "train_flow_matching_loss": 0.043701171875,
      "train_physics_loss": 0.44353944063186646
    },
    {
      "step": 16000,
      "train_flow_matching_loss": 0.1591796875,
      "train_physics_loss": 0.0
    },
    {
      "step": 16000,
      "train_flow_matching_loss": 0.10498046875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.1238219290971756,
      "learning_rate": 5e-05,
      "loss": 0.1364,
      "step": 16010
    },
    {
      "step": 16010,
      "train_flow_matching_loss": 0.08740234375,
      "train_physics_loss": 0.0
    },
    {
      "step": 16010,
      "train_flow_matching_loss": 0.046630859375,
      "train_physics_loss": 0.45190274715423584
    },
    {
      "step": 16010,
      "train_flow_matching_loss": 0.140625,
      "train_physics_loss": 0.0
    },
    {
      "step": 16010,
      "train_flow_matching_loss": 0.1875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08907845616340637,
      "learning_rate": 5e-05,
      "loss": 0.1326,
      "step": 16020
    },
    {
      "step": 16020,
      "train_flow_matching_loss": 0.1728515625,
      "train_physics_loss": 0.0
    },
    {
      "step": 16020,
      "train_flow_matching_loss": 0.052001953125,
      "train_physics_loss": 0.41535043716430664
    },
    {
      "step": 16020,
      "train_flow_matching_loss": 0.1611328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 16020,
      "train_flow_matching_loss": 0.2080078125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.07022511959075928,
      "learning_rate": 5e-05,
      "loss": 0.1349,
      "step": 16030
    },
    {
      "step": 16030,
      "train_flow_matching_loss": 0.12353515625,
      "train_physics_loss": 0.0
    },
    {
      "step": 16030,
      "train_flow_matching_loss": 0.08349609375,
      "train_physics_loss": 0.41542789340019226
    },
    {
      "step": 16030,
      "train_flow_matching_loss": 0.171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 16030,
      "train_flow_matching_loss": 0.1103515625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.17975878715515137,
      "learning_rate": 5e-05,
      "loss": 0.1362,
      "step": 16040
    },
    {
      "step": 16040,
      "train_flow_matching_loss": 0.125,
      "train_physics_loss": 0.0
    },
    {
      "step": 16040,
      "train_flow_matching_loss": 0.0654296875,
      "train_physics_loss": 0.46694663166999817
    },
    {
      "step": 16040,
      "train_flow_matching_loss": 0.158203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 16040,
      "train_flow_matching_loss": 0.1708984375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.07511463761329651,
      "learning_rate": 5e-05,
      "loss": 0.1362,
      "step": 16050
    },
    {
      "step": 16050,
      "train_flow_matching_loss": 0.11962890625,
      "train_physics_loss": 0.0
    },
    {
      "step": 16050,
      "train_flow_matching_loss": 0.06591796875,
      "train_physics_loss": 0.6369867324829102
    },
    {
      "step": 16050,
      "train_flow_matching_loss": 0.1787109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 16050,
      "train_flow_matching_loss": 0.0986328125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11165697127580643,
      "learning_rate": 5e-05,
      "loss": 0.1346,
      "step": 16060
    },
    {
      "step": 16060,
      "train_flow_matching_loss": 0.1259765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 16060,
      "train_flow_matching_loss": 0.07470703125,
      "train_physics_loss": 0.44688910245895386
    },
    {
      "step": 16060,
      "train_flow_matching_loss": 0.0927734375,
      "train_physics_loss": 0.0
    },
    {
      "step": 16060,
      "train_flow_matching_loss": 0.07421875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.07149883359670639,
      "learning_rate": 5e-05,
      "loss": 0.1393,
      "step": 16070
    },
    {
      "step": 16070,
      "train_flow_matching_loss": 0.138671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 16070,
      "train_flow_matching_loss": 0.0625,
      "train_physics_loss": 0.49913355708122253
    },
    {
      "step": 16070,
      "train_flow_matching_loss": 0.1318359375,
      "train_physics_loss": 0.0
    },
    {
      "step": 16070,
      "train_flow_matching_loss": 0.0947265625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11939989775419235,
      "learning_rate": 5e-05,
      "loss": 0.1423,
      "step": 16080
    },
    {
      "step": 16080,
      "train_flow_matching_loss": 0.1708984375,
      "train_physics_loss": 0.0
    },
    {
      "step": 16080,
      "train_flow_matching_loss": 0.126953125,
      "train_physics_loss": 0.43045124411582947
    },
    {
      "step": 16080,
      "train_flow_matching_loss": 0.1806640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 16080,
      "train_flow_matching_loss": 0.169921875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08592528849840164,
      "learning_rate": 5e-05,
      "loss": 0.1397,
      "step": 16090
    },
    {
      "step": 16090,
      "train_flow_matching_loss": 0.1650390625,
      "train_physics_loss": 0.0
    },
    {
      "step": 16090,
      "train_flow_matching_loss": 0.03515625,
      "train_physics_loss": 0.4792134761810303
    },
    {
      "step": 16090,
      "train_flow_matching_loss": 0.146484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 16090,
      "train_flow_matching_loss": 0.1474609375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08740340918302536,
      "learning_rate": 5e-05,
      "loss": 0.1366,
      "step": 16100
    },
    {
      "step": 16100,
      "train_flow_matching_loss": 0.1708984375,
      "train_physics_loss": 0.0
    },
    {
      "step": 16100,
      "train_flow_matching_loss": 0.051513671875,
      "train_physics_loss": 0.40064436197280884
    },
    {
      "step": 16100,
      "train_flow_matching_loss": 0.1513671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 16100,
      "train_flow_matching_loss": 0.189453125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.0721786618232727,
      "learning_rate": 5e-05,
      "loss": 0.1409,
      "step": 16110
    },
    {
      "step": 16110,
      "train_flow_matching_loss": 0.185546875,
      "train_physics_loss": 0.0
    },
    {
      "step": 16110,
      "train_flow_matching_loss": 0.06640625,
      "train_physics_loss": 0.4887370765209198
    },
    {
      "step": 16110,
      "train_flow_matching_loss": 0.1640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 16110,
      "train_flow_matching_loss": 0.1455078125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.06897297501564026,
      "learning_rate": 5e-05,
      "loss": 0.1381,
      "step": 16120
    },
    {
      "step": 16120,
      "train_flow_matching_loss": 0.12158203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 16120,
      "train_flow_matching_loss": 0.1103515625,
      "train_physics_loss": 0.442701518535614
    },
    {
      "step": 16120,
      "train_flow_matching_loss": 0.1005859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 16120,
      "train_flow_matching_loss": 0.1259765625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.12044677138328552,
      "learning_rate": 5e-05,
      "loss": 0.1339,
      "step": 16130
    },
    {
      "step": 16130,
      "train_flow_matching_loss": 0.12451171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 16130,
      "train_flow_matching_loss": 0.0159912109375,
      "train_physics_loss": 0.4137090742588043
    },
    {
      "step": 16130,
      "train_flow_matching_loss": 0.154296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 16130,
      "train_flow_matching_loss": 0.2158203125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.12308747321367264,
      "learning_rate": 5e-05,
      "loss": 0.1405,
      "step": 16140
    },
    {
      "step": 16140,
      "train_flow_matching_loss": 0.1298828125,
      "train_physics_loss": 0.0
    },
    {
      "step": 16140,
      "train_flow_matching_loss": 0.03369140625,
      "train_physics_loss": 0.45102450251579285
    },
    {
      "step": 16140,
      "train_flow_matching_loss": 0.1357421875,
      "train_physics_loss": 0.0
    },
    {
      "step": 16140,
      "train_flow_matching_loss": 0.177734375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.2401352822780609,
      "learning_rate": 5e-05,
      "loss": 0.1334,
      "step": 16150
    },
    {
      "step": 16150,
      "train_flow_matching_loss": 0.171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 16150,
      "train_flow_matching_loss": 0.07861328125,
      "train_physics_loss": 0.44686779379844666
    },
    {
      "step": 16150,
      "train_flow_matching_loss": 0.1083984375,
      "train_physics_loss": 0.0
    },
    {
      "step": 16150,
      "train_flow_matching_loss": 0.263671875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09934668987989426,
      "learning_rate": 5e-05,
      "loss": 0.1402,
      "step": 16160
    },
    {
      "step": 16160,
      "train_flow_matching_loss": 0.171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 16160,
      "train_flow_matching_loss": 0.08544921875,
      "train_physics_loss": 0.42187485098838806
    },
    {
      "step": 16160,
      "train_flow_matching_loss": 0.2001953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 16160,
      "train_flow_matching_loss": 0.1787109375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10250754654407501,
      "learning_rate": 5e-05,
      "loss": 0.1334,
      "step": 16170
    },
    {
      "step": 16170,
      "train_flow_matching_loss": 0.1015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 16170,
      "train_flow_matching_loss": 0.036376953125,
      "train_physics_loss": 0.4420827627182007
    },
    {
      "step": 16170,
      "train_flow_matching_loss": 0.1357421875,
      "train_physics_loss": 0.0
    },
    {
      "step": 16170,
      "train_flow_matching_loss": 0.12158203125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.29546231031417847,
      "learning_rate": 5e-05,
      "loss": 0.1405,
      "step": 16180
    },
    {
      "step": 16180,
      "train_flow_matching_loss": 0.1123046875,
      "train_physics_loss": 0.0
    },
    {
      "step": 16180,
      "train_flow_matching_loss": 0.09716796875,
      "train_physics_loss": 0.4437144994735718
    },
    {
      "step": 16180,
      "train_flow_matching_loss": 0.08056640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 16180,
      "train_flow_matching_loss": 0.2119140625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.1057034507393837,
      "learning_rate": 5e-05,
      "loss": 0.1403,
      "step": 16190
    },
    {
      "step": 16190,
      "train_flow_matching_loss": 0.2080078125,
      "train_physics_loss": 0.0
    },
    {
      "step": 16190,
      "train_flow_matching_loss": 0.048828125,
      "train_physics_loss": 0.4393121898174286
    },
    {
      "step": 16190,
      "train_flow_matching_loss": 0.1064453125,
      "train_physics_loss": 0.0
    },
    {
      "step": 16190,
      "train_flow_matching_loss": 0.1748046875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.3790835738182068,
      "learning_rate": 5e-05,
      "loss": 0.1402,
      "step": 16200
    },
    {
      "step": 16200,
      "train_flow_matching_loss": 0.181640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 16200,
      "train_flow_matching_loss": 0.078125,
      "train_physics_loss": 0.44644561409950256
    },
    {
      "step": 16200,
      "train_flow_matching_loss": 0.11865234375,
      "train_physics_loss": 0.0
    },
    {
      "step": 16200,
      "train_flow_matching_loss": 0.1318359375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09365010261535645,
      "learning_rate": 5e-05,
      "loss": 0.1375,
      "step": 16210
    },
    {
      "step": 16210,
      "train_flow_matching_loss": 0.158203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 16210,
      "train_flow_matching_loss": 0.022705078125,
      "train_physics_loss": 0.4387159049510956
    },
    {
      "step": 16210,
      "train_flow_matching_loss": 0.0888671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 16210,
      "train_flow_matching_loss": 0.1259765625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.07261950522661209,
      "learning_rate": 5e-05,
      "loss": 0.1369,
      "step": 16220
    },
    {
      "step": 16220,
      "train_flow_matching_loss": 0.1962890625,
      "train_physics_loss": 0.0
    },
    {
      "step": 16220,
      "train_flow_matching_loss": 0.162109375,
      "train_physics_loss": 0.4619278013706207
    },
    {
      "step": 16220,
      "train_flow_matching_loss": 0.1953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 16220,
      "train_flow_matching_loss": 0.09033203125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10823575407266617,
      "learning_rate": 5e-05,
      "loss": 0.1394,
      "step": 16230
    },
    {
      "step": 16230,
      "train_flow_matching_loss": 0.1005859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 16230,
      "train_flow_matching_loss": 0.08447265625,
      "train_physics_loss": 0.41419708728790283
    },
    {
      "step": 16230,
      "train_flow_matching_loss": 0.1962890625,
      "train_physics_loss": 0.0
    },
    {
      "step": 16230,
      "train_flow_matching_loss": 0.1650390625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.0759219229221344,
      "learning_rate": 5e-05,
      "loss": 0.139,
      "step": 16240
    },
    {
      "step": 16240,
      "train_flow_matching_loss": 0.09716796875,
      "train_physics_loss": 0.0
    },
    {
      "step": 16240,
      "train_flow_matching_loss": 0.08740234375,
      "train_physics_loss": 0.4172540605068207
    },
    {
      "step": 16240,
      "train_flow_matching_loss": 0.166015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 16240,
      "train_flow_matching_loss": 0.162109375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08398652076721191,
      "learning_rate": 5e-05,
      "loss": 0.1374,
      "step": 16250
    },
    {
      "step": 16250,
      "train_flow_matching_loss": 0.25390625,
      "train_physics_loss": 0.0
    },
    {
      "step": 16250,
      "train_flow_matching_loss": 0.0712890625,
      "train_physics_loss": 0.4646586775779724
    },
    {
      "step": 16250,
      "train_flow_matching_loss": 0.12060546875,
      "train_physics_loss": 0.0
    },
    {
      "step": 16250,
      "train_flow_matching_loss": 0.1953125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08186598122119904,
      "learning_rate": 5e-05,
      "loss": 0.1385,
      "step": 16260
    },
    {
      "step": 16260,
      "train_flow_matching_loss": 0.1376953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 16260,
      "train_flow_matching_loss": 0.10888671875,
      "train_physics_loss": 0.5743127465248108
    },
    {
      "step": 16260,
      "train_flow_matching_loss": 0.1123046875,
      "train_physics_loss": 0.0
    },
    {
      "step": 16260,
      "train_flow_matching_loss": 0.1611328125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.28392302989959717,
      "learning_rate": 5e-05,
      "loss": 0.1381,
      "step": 16270
    },
    {
      "step": 16270,
      "train_flow_matching_loss": 0.12109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 16270,
      "train_flow_matching_loss": 0.0869140625,
      "train_physics_loss": 0.3987542390823364
    },
    {
      "step": 16270,
      "train_flow_matching_loss": 0.1298828125,
      "train_physics_loss": 0.0
    },
    {
      "step": 16270,
      "train_flow_matching_loss": 0.1259765625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.20893187820911407,
      "learning_rate": 5e-05,
      "loss": 0.1337,
      "step": 16280
    },
    {
      "step": 16280,
      "train_flow_matching_loss": 0.1337890625,
      "train_physics_loss": 0.0
    },
    {
      "step": 16280,
      "train_flow_matching_loss": 0.06005859375,
      "train_physics_loss": 0.5073346495628357
    },
    {
      "step": 16280,
      "train_flow_matching_loss": 0.1416015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 16280,
      "train_flow_matching_loss": 0.197265625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09416130185127258,
      "learning_rate": 5e-05,
      "loss": 0.1373,
      "step": 16290
    },
    {
      "step": 16290,
      "train_flow_matching_loss": 0.224609375,
      "train_physics_loss": 0.0
    },
    {
      "step": 16290,
      "train_flow_matching_loss": 0.09912109375,
      "train_physics_loss": 0.3972088694572449
    },
    {
      "step": 16290,
      "train_flow_matching_loss": 0.060791015625,
      "train_physics_loss": 0.40410950779914856
    },
    {
      "step": 16290,
      "train_flow_matching_loss": 0.10546875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.13970531523227692,
      "learning_rate": 5e-05,
      "loss": 0.1365,
      "step": 16300
    },
    {
      "step": 16300,
      "train_flow_matching_loss": 0.21875,
      "train_physics_loss": 0.0
    },
    {
      "step": 16300,
      "train_flow_matching_loss": 0.07568359375,
      "train_physics_loss": 0.42813846468925476
    },
    {
      "step": 16300,
      "train_flow_matching_loss": 0.044921875,
      "train_physics_loss": 0.46666082739830017
    },
    {
      "step": 16300,
      "train_flow_matching_loss": 0.138671875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.07058744877576828,
      "learning_rate": 5e-05,
      "loss": 0.1362,
      "step": 16310
    },
    {
      "step": 16310,
      "train_flow_matching_loss": 0.158203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 16310,
      "train_flow_matching_loss": 0.10107421875,
      "train_physics_loss": 0.45272985100746155
    },
    {
      "step": 16310,
      "train_flow_matching_loss": 0.06689453125,
      "train_physics_loss": 0.42808473110198975
    },
    {
      "step": 16310,
      "train_flow_matching_loss": 0.08447265625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.0755370631814003,
      "learning_rate": 5e-05,
      "loss": 0.1338,
      "step": 16320
    },
    {
      "step": 16320,
      "train_flow_matching_loss": 0.1396484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 16320,
      "train_flow_matching_loss": 0.06396484375,
      "train_physics_loss": 0.47953253984451294
    },
    {
      "step": 16320,
      "train_flow_matching_loss": 0.0263671875,
      "train_physics_loss": 0.4326748251914978
    },
    {
      "step": 16320,
      "train_flow_matching_loss": 0.2177734375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09074758738279343,
      "learning_rate": 5e-05,
      "loss": 0.1375,
      "step": 16330
    },
    {
      "step": 16330,
      "train_flow_matching_loss": 0.1630859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 16330,
      "train_flow_matching_loss": 0.07421875,
      "train_physics_loss": 0.4837624430656433
    },
    {
      "step": 16330,
      "train_flow_matching_loss": 0.056884765625,
      "train_physics_loss": 0.4631340503692627
    },
    {
      "step": 16330,
      "train_flow_matching_loss": 0.1552734375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.12720409035682678,
      "learning_rate": 5e-05,
      "loss": 0.1327,
      "step": 16340
    },
    {
      "step": 16340,
      "train_flow_matching_loss": 0.1357421875,
      "train_physics_loss": 0.0
    },
    {
      "step": 16340,
      "train_flow_matching_loss": 0.062255859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 16340,
      "train_flow_matching_loss": 0.04931640625,
      "train_physics_loss": 0.42125919461250305
    },
    {
      "step": 16340,
      "train_flow_matching_loss": 0.10009765625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.06138471141457558,
      "learning_rate": 5e-05,
      "loss": 0.1306,
      "step": 16350
    },
    {
      "step": 16350,
      "train_flow_matching_loss": 0.150390625,
      "train_physics_loss": 0.0
    },
    {
      "step": 16350,
      "train_flow_matching_loss": 0.12158203125,
      "train_physics_loss": 0.4024377763271332
    },
    {
      "step": 16350,
      "train_flow_matching_loss": 0.044677734375,
      "train_physics_loss": 0.5578294992446899
    },
    {
      "step": 16350,
      "train_flow_matching_loss": 0.1435546875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10905923694372177,
      "learning_rate": 5e-05,
      "loss": 0.1367,
      "step": 16360
    },
    {
      "step": 16360,
      "train_flow_matching_loss": 0.1884765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 16360,
      "train_flow_matching_loss": 0.09423828125,
      "train_physics_loss": 0.4098307192325592
    },
    {
      "step": 16360,
      "train_flow_matching_loss": 0.0888671875,
      "train_physics_loss": 0.4520365595817566
    },
    {
      "step": 16360,
      "train_flow_matching_loss": 0.10791015625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.07483577728271484,
      "learning_rate": 5e-05,
      "loss": 0.1369,
      "step": 16370
    },
    {
      "step": 16370,
      "train_flow_matching_loss": 0.0869140625,
      "train_physics_loss": 0.0
    },
    {
      "step": 16370,
      "train_flow_matching_loss": 0.07373046875,
      "train_physics_loss": 0.41936182975769043
    },
    {
      "step": 16370,
      "train_flow_matching_loss": 0.078125,
      "train_physics_loss": 0.4694300591945648
    },
    {
      "step": 16370,
      "train_flow_matching_loss": 0.1279296875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.1220056489109993,
      "learning_rate": 5e-05,
      "loss": 0.1398,
      "step": 16380
    },
    {
      "step": 16380,
      "train_flow_matching_loss": 0.0751953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 16380,
      "train_flow_matching_loss": 0.056640625,
      "train_physics_loss": 0.508339524269104
    },
    {
      "step": 16380,
      "train_flow_matching_loss": 0.059814453125,
      "train_physics_loss": 0.4060460925102234
    },
    {
      "step": 16380,
      "train_flow_matching_loss": 0.1376953125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09648419171571732,
      "learning_rate": 5e-05,
      "loss": 0.136,
      "step": 16390
    },
    {
      "step": 16390,
      "train_flow_matching_loss": 0.1689453125,
      "train_physics_loss": 0.0
    },
    {
      "step": 16390,
      "train_flow_matching_loss": 0.03271484375,
      "train_physics_loss": 0.39935678243637085
    },
    {
      "step": 16390,
      "train_flow_matching_loss": 0.060791015625,
      "train_physics_loss": 0.4395064115524292
    },
    {
      "step": 16390,
      "train_flow_matching_loss": 0.1875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08826830238103867,
      "learning_rate": 5e-05,
      "loss": 0.1351,
      "step": 16400
    },
    {
      "step": 16400,
      "train_flow_matching_loss": 0.09423828125,
      "train_physics_loss": 0.0
    },
    {
      "step": 16400,
      "train_flow_matching_loss": 0.0654296875,
      "train_physics_loss": 0.42183446884155273
    },
    {
      "step": 16400,
      "train_flow_matching_loss": 0.078125,
      "train_physics_loss": 0.4538635313510895
    },
    {
      "step": 16400,
      "train_flow_matching_loss": 0.09912109375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10486242920160294,
      "learning_rate": 5e-05,
      "loss": 0.1361,
      "step": 16410
    },
    {
      "step": 16410,
      "train_flow_matching_loss": 0.236328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 16410,
      "train_flow_matching_loss": 0.0299072265625,
      "train_physics_loss": 0.4148198664188385
    },
    {
      "step": 16410,
      "train_flow_matching_loss": 0.043701171875,
      "train_physics_loss": 0.4640271067619324
    },
    {
      "step": 16410,
      "train_flow_matching_loss": 0.11865234375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10310588032007217,
      "learning_rate": 5e-05,
      "loss": 0.1318,
      "step": 16420
    },
    {
      "step": 16420,
      "train_flow_matching_loss": 0.1640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 16420,
      "train_flow_matching_loss": 0.052001953125,
      "train_physics_loss": 0.4510272145271301
    },
    {
      "step": 16420,
      "train_flow_matching_loss": 0.0311279296875,
      "train_physics_loss": 0.4151761531829834
    },
    {
      "step": 16420,
      "train_flow_matching_loss": 0.1708984375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08422515541315079,
      "learning_rate": 5e-05,
      "loss": 0.1305,
      "step": 16430
    },
    {
      "step": 16430,
      "train_flow_matching_loss": 0.1728515625,
      "train_physics_loss": 0.0
    },
    {
      "step": 16430,
      "train_flow_matching_loss": 0.060302734375,
      "train_physics_loss": 0.47522810101509094
    },
    {
      "step": 16430,
      "train_flow_matching_loss": 0.038818359375,
      "train_physics_loss": 0.4168815314769745
    },
    {
      "step": 16430,
      "train_flow_matching_loss": 0.1611328125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10160360485315323,
      "learning_rate": 5e-05,
      "loss": 0.1355,
      "step": 16440
    },
    {
      "step": 16440,
      "train_flow_matching_loss": 0.134765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 16440,
      "train_flow_matching_loss": 0.04931640625,
      "train_physics_loss": 0.41556546092033386
    },
    {
      "step": 16440,
      "train_flow_matching_loss": 0.04150390625,
      "train_physics_loss": 0.3953983187675476
    },
    {
      "step": 16440,
      "train_flow_matching_loss": 0.1279296875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.0892341211438179,
      "learning_rate": 5e-05,
      "loss": 0.1317,
      "step": 16450
    },
    {
      "step": 16450,
      "train_flow_matching_loss": 0.1748046875,
      "train_physics_loss": 0.0
    },
    {
      "step": 16450,
      "train_flow_matching_loss": 0.08349609375,
      "train_physics_loss": 0.4137800931930542
    },
    {
      "step": 16450,
      "train_flow_matching_loss": 0.0712890625,
      "train_physics_loss": 0.3576895296573639
    },
    {
      "step": 16450,
      "train_flow_matching_loss": 0.11474609375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.1053849309682846,
      "learning_rate": 5e-05,
      "loss": 0.1336,
      "step": 16460
    },
    {
      "step": 16460,
      "train_flow_matching_loss": 0.162109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 16460,
      "train_flow_matching_loss": 0.07666015625,
      "train_physics_loss": 0.41827520728111267
    },
    {
      "step": 16460,
      "train_flow_matching_loss": 0.0341796875,
      "train_physics_loss": 0.39304688572883606
    },
    {
      "step": 16460,
      "train_flow_matching_loss": 0.06982421875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09757518768310547,
      "learning_rate": 5e-05,
      "loss": 0.1294,
      "step": 16470
    },
    {
      "step": 16470,
      "train_flow_matching_loss": 0.1767578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 16470,
      "train_flow_matching_loss": 0.059814453125,
      "train_physics_loss": 0.4515064060688019
    },
    {
      "step": 16470,
      "train_flow_matching_loss": 0.1259765625,
      "train_physics_loss": 0.5552309155464172
    },
    {
      "step": 16470,
      "train_flow_matching_loss": 0.1083984375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10915850102901459,
      "learning_rate": 5e-05,
      "loss": 0.1332,
      "step": 16480
    },
    {
      "step": 16480,
      "train_flow_matching_loss": 0.1162109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 16480,
      "train_flow_matching_loss": 0.06689453125,
      "train_physics_loss": 0.4107041656970978
    },
    {
      "step": 16480,
      "train_flow_matching_loss": 0.10693359375,
      "train_physics_loss": 0.3878869116306305
    },
    {
      "step": 16480,
      "train_flow_matching_loss": 0.162109375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08840745687484741,
      "learning_rate": 5e-05,
      "loss": 0.1398,
      "step": 16490
    },
    {
      "step": 16490,
      "train_flow_matching_loss": 0.1533203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 16490,
      "train_flow_matching_loss": 0.0849609375,
      "train_physics_loss": 0.0
    },
    {
      "step": 16490,
      "train_flow_matching_loss": 0.10205078125,
      "train_physics_loss": 0.34512901306152344
    },
    {
      "step": 16490,
      "train_flow_matching_loss": 0.140625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10103483498096466,
      "learning_rate": 5e-05,
      "loss": 0.1308,
      "step": 16500
    },
    {
      "step": 16500,
      "train_flow_matching_loss": 0.10400390625,
      "train_physics_loss": 0.0
    },
    {
      "step": 16500,
      "train_flow_matching_loss": 0.1455078125,
      "train_physics_loss": 0.0
    },
    {
      "step": 16500,
      "train_flow_matching_loss": 0.05126953125,
      "train_physics_loss": 0.4454532265663147
    },
    {
      "step": 16500,
      "train_flow_matching_loss": 0.08447265625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09329575300216675,
      "learning_rate": 5e-05,
      "loss": 0.1296,
      "step": 16510
    },
    {
      "step": 16510,
      "train_flow_matching_loss": 0.12451171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 16510,
      "train_flow_matching_loss": 0.1591796875,
      "train_physics_loss": 0.0
    },
    {
      "step": 16510,
      "train_flow_matching_loss": 0.044921875,
      "train_physics_loss": 0.401406854391098
    },
    {
      "step": 16510,
      "train_flow_matching_loss": 0.162109375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08181340992450714,
      "learning_rate": 5e-05,
      "loss": 0.1339,
      "step": 16520
    },
    {
      "step": 16520,
      "train_flow_matching_loss": 0.2041015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 16520,
      "train_flow_matching_loss": 0.25,
      "train_physics_loss": 0.0
    },
    {
      "step": 16520,
      "train_flow_matching_loss": 0.059814453125,
      "train_physics_loss": 0.4109176695346832
    },
    {
      "step": 16520,
      "train_flow_matching_loss": 0.177734375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.1046636551618576,
      "learning_rate": 5e-05,
      "loss": 0.1344,
      "step": 16530
    },
    {
      "step": 16530,
      "train_flow_matching_loss": 0.1015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 16530,
      "train_flow_matching_loss": 0.169921875,
      "train_physics_loss": 0.0
    },
    {
      "step": 16530,
      "train_flow_matching_loss": 0.0267333984375,
      "train_physics_loss": 0.429096519947052
    },
    {
      "step": 16530,
      "train_flow_matching_loss": 0.08203125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.3235691487789154,
      "learning_rate": 5e-05,
      "loss": 0.1338,
      "step": 16540
    },
    {
      "step": 16540,
      "train_flow_matching_loss": 0.1298828125,
      "train_physics_loss": 0.0
    },
    {
      "step": 16540,
      "train_flow_matching_loss": 0.1318359375,
      "train_physics_loss": 0.0
    },
    {
      "step": 16540,
      "train_flow_matching_loss": 0.046875,
      "train_physics_loss": 0.47538119554519653
    },
    {
      "step": 16540,
      "train_flow_matching_loss": 0.11083984375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.1024763211607933,
      "learning_rate": 5e-05,
      "loss": 0.14,
      "step": 16550
    },
    {
      "step": 16550,
      "train_flow_matching_loss": 0.1435546875,
      "train_physics_loss": 0.0
    },
    {
      "step": 16550,
      "train_flow_matching_loss": 0.1484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 16550,
      "train_flow_matching_loss": 0.10400390625,
      "train_physics_loss": 0.0
    },
    {
      "step": 16550,
      "train_flow_matching_loss": 0.1259765625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11263430118560791,
      "learning_rate": 5e-05,
      "loss": 0.1322,
      "step": 16560
    },
    {
      "step": 16560,
      "train_flow_matching_loss": 0.1376953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 16560,
      "train_flow_matching_loss": 0.1298828125,
      "train_physics_loss": 0.0
    },
    {
      "step": 16560,
      "train_flow_matching_loss": 0.1494140625,
      "train_physics_loss": 0.0
    },
    {
      "step": 16560,
      "train_flow_matching_loss": 0.0947265625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.083173468708992,
      "learning_rate": 5e-05,
      "loss": 0.1362,
      "step": 16570
    },
    {
      "step": 16570,
      "train_flow_matching_loss": 0.11767578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 16570,
      "train_flow_matching_loss": 0.11572265625,
      "train_physics_loss": 0.0
    },
    {
      "step": 16570,
      "train_flow_matching_loss": 0.171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 16570,
      "train_flow_matching_loss": 0.240234375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.1267733871936798,
      "learning_rate": 5e-05,
      "loss": 0.1404,
      "step": 16580
    },
    {
      "step": 16580,
      "train_flow_matching_loss": 0.1298828125,
      "train_physics_loss": 0.0
    },
    {
      "step": 16580,
      "train_flow_matching_loss": 0.1787109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 16580,
      "train_flow_matching_loss": 0.126953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 16580,
      "train_flow_matching_loss": 0.1630859375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11699742078781128,
      "learning_rate": 5e-05,
      "loss": 0.1344,
      "step": 16590
    },
    {
      "step": 16590,
      "train_flow_matching_loss": 0.1103515625,
      "train_physics_loss": 0.0
    },
    {
      "step": 16590,
      "train_flow_matching_loss": 0.1181640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 16590,
      "train_flow_matching_loss": 0.134765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 16590,
      "train_flow_matching_loss": 0.11376953125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.12886637449264526,
      "learning_rate": 5e-05,
      "loss": 0.1379,
      "step": 16600
    },
    {
      "step": 16600,
      "train_flow_matching_loss": 0.126953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 16600,
      "train_flow_matching_loss": 0.15625,
      "train_physics_loss": 0.0
    },
    {
      "step": 16600,
      "train_flow_matching_loss": 0.130859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 16600,
      "train_flow_matching_loss": 0.1201171875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11690055578947067,
      "learning_rate": 5e-05,
      "loss": 0.1358,
      "step": 16610
    },
    {
      "step": 16610,
      "train_flow_matching_loss": 0.1318359375,
      "train_physics_loss": 0.0
    },
    {
      "step": 16610,
      "train_flow_matching_loss": 0.130859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 16610,
      "train_flow_matching_loss": 0.1396484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 16610,
      "train_flow_matching_loss": 0.1181640625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10369376093149185,
      "learning_rate": 5e-05,
      "loss": 0.1383,
      "step": 16620
    },
    {
      "step": 16620,
      "train_flow_matching_loss": 0.126953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 16620,
      "train_flow_matching_loss": 0.138671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 16620,
      "train_flow_matching_loss": 0.1181640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 16620,
      "train_flow_matching_loss": 0.08642578125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08927435427904129,
      "learning_rate": 5e-05,
      "loss": 0.1324,
      "step": 16630
    },
    {
      "step": 16630,
      "train_flow_matching_loss": 0.2216796875,
      "train_physics_loss": 0.0
    },
    {
      "step": 16630,
      "train_flow_matching_loss": 0.09716796875,
      "train_physics_loss": 0.0
    },
    {
      "step": 16630,
      "train_flow_matching_loss": 0.10595703125,
      "train_physics_loss": 0.0
    },
    {
      "step": 16630,
      "train_flow_matching_loss": 0.181640625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09100035578012466,
      "learning_rate": 5e-05,
      "loss": 0.1369,
      "step": 16640
    },
    {
      "step": 16640,
      "train_flow_matching_loss": 0.1259765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 16640,
      "train_flow_matching_loss": 0.1787109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 16640,
      "train_flow_matching_loss": 0.1416015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 16640,
      "train_flow_matching_loss": 0.10205078125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10827035456895828,
      "learning_rate": 5e-05,
      "loss": 0.1393,
      "step": 16650
    },
    {
      "step": 16650,
      "train_flow_matching_loss": 0.21875,
      "train_physics_loss": 0.0
    },
    {
      "step": 16650,
      "train_flow_matching_loss": 0.1259765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 16650,
      "train_flow_matching_loss": 0.1220703125,
      "train_physics_loss": 0.0
    },
    {
      "step": 16650,
      "train_flow_matching_loss": 0.1748046875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.07204524427652359,
      "learning_rate": 5e-05,
      "loss": 0.13,
      "step": 16660
    },
    {
      "step": 16660,
      "train_flow_matching_loss": 0.2138671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 16660,
      "train_flow_matching_loss": 0.1376953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 16660,
      "train_flow_matching_loss": 0.1416015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 16660,
      "train_flow_matching_loss": 0.12353515625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11934295296669006,
      "learning_rate": 5e-05,
      "loss": 0.1355,
      "step": 16670
    },
    {
      "step": 16670,
      "train_flow_matching_loss": 0.181640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 16670,
      "train_flow_matching_loss": 0.220703125,
      "train_physics_loss": 0.0
    },
    {
      "step": 16670,
      "train_flow_matching_loss": 0.140625,
      "train_physics_loss": 0.0
    },
    {
      "step": 16670,
      "train_flow_matching_loss": 0.1953125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.3939994275569916,
      "learning_rate": 5e-05,
      "loss": 0.1338,
      "step": 16680
    },
    {
      "step": 16680,
      "train_flow_matching_loss": 0.0830078125,
      "train_physics_loss": 0.0
    },
    {
      "step": 16680,
      "train_flow_matching_loss": 0.2294921875,
      "train_physics_loss": 0.0
    },
    {
      "step": 16680,
      "train_flow_matching_loss": 0.1357421875,
      "train_physics_loss": 0.0
    },
    {
      "step": 16680,
      "train_flow_matching_loss": 0.11279296875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10067585855722427,
      "learning_rate": 5e-05,
      "loss": 0.1325,
      "step": 16690
    },
    {
      "step": 16690,
      "train_flow_matching_loss": 0.1337890625,
      "train_physics_loss": 0.0
    },
    {
      "step": 16690,
      "train_flow_matching_loss": 0.208984375,
      "train_physics_loss": 0.0
    },
    {
      "step": 16690,
      "train_flow_matching_loss": 0.10595703125,
      "train_physics_loss": 0.0
    },
    {
      "step": 16690,
      "train_flow_matching_loss": 0.208984375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.1846601516008377,
      "learning_rate": 5e-05,
      "loss": 0.1383,
      "step": 16700
    },
    {
      "step": 16700,
      "train_flow_matching_loss": 0.12451171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 16700,
      "train_flow_matching_loss": 0.244140625,
      "train_physics_loss": 0.0
    },
    {
      "step": 16700,
      "train_flow_matching_loss": 0.158203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 16700,
      "train_flow_matching_loss": 0.130859375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.12297917157411575,
      "learning_rate": 5e-05,
      "loss": 0.131,
      "step": 16710
    },
    {
      "step": 16710,
      "train_flow_matching_loss": 0.1201171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 16710,
      "train_flow_matching_loss": 0.1396484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 16710,
      "train_flow_matching_loss": 0.10107421875,
      "train_physics_loss": 0.0
    },
    {
      "step": 16710,
      "train_flow_matching_loss": 0.189453125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09929110109806061,
      "learning_rate": 5e-05,
      "loss": 0.1373,
      "step": 16720
    },
    {
      "step": 16720,
      "train_flow_matching_loss": 0.10595703125,
      "train_physics_loss": 0.0
    },
    {
      "step": 16720,
      "train_flow_matching_loss": 0.1025390625,
      "train_physics_loss": 0.0
    },
    {
      "step": 16720,
      "train_flow_matching_loss": 0.177734375,
      "train_physics_loss": 0.0
    },
    {
      "step": 16720,
      "train_flow_matching_loss": 0.1318359375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08913774788379669,
      "learning_rate": 5e-05,
      "loss": 0.131,
      "step": 16730
    },
    {
      "step": 16730,
      "train_flow_matching_loss": 0.1455078125,
      "train_physics_loss": 0.0
    },
    {
      "step": 16730,
      "train_flow_matching_loss": 0.154296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 16730,
      "train_flow_matching_loss": 0.08349609375,
      "train_physics_loss": 0.0
    },
    {
      "step": 16730,
      "train_flow_matching_loss": 0.1171875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.07676814496517181,
      "learning_rate": 5e-05,
      "loss": 0.1268,
      "step": 16740
    },
    {
      "step": 16740,
      "train_flow_matching_loss": 0.126953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 16740,
      "train_flow_matching_loss": 0.0888671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 16740,
      "train_flow_matching_loss": 0.126953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 16740,
      "train_flow_matching_loss": 0.146484375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.07027620822191238,
      "learning_rate": 5e-05,
      "loss": 0.1304,
      "step": 16750
    },
    {
      "step": 16750,
      "train_flow_matching_loss": 0.11865234375,
      "train_physics_loss": 0.0
    },
    {
      "step": 16750,
      "train_flow_matching_loss": 0.1708984375,
      "train_physics_loss": 0.0
    },
    {
      "step": 16750,
      "train_flow_matching_loss": 0.1220703125,
      "train_physics_loss": 0.0
    },
    {
      "step": 16750,
      "train_flow_matching_loss": 0.2021484375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.30536434054374695,
      "learning_rate": 5e-05,
      "loss": 0.137,
      "step": 16760
    },
    {
      "step": 16760,
      "train_flow_matching_loss": 0.201171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 16760,
      "train_flow_matching_loss": 0.1591796875,
      "train_physics_loss": 0.0
    },
    {
      "step": 16760,
      "train_flow_matching_loss": 0.1875,
      "train_physics_loss": 0.0
    },
    {
      "step": 16760,
      "train_flow_matching_loss": 0.162109375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.07751469314098358,
      "learning_rate": 5e-05,
      "loss": 0.1407,
      "step": 16770
    },
    {
      "step": 16770,
      "train_flow_matching_loss": 0.1337890625,
      "train_physics_loss": 0.0
    },
    {
      "step": 16770,
      "train_flow_matching_loss": 0.0869140625,
      "train_physics_loss": 0.0
    },
    {
      "step": 16770,
      "train_flow_matching_loss": 0.11572265625,
      "train_physics_loss": 0.0
    },
    {
      "step": 16770,
      "train_flow_matching_loss": 0.130859375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10414283722639084,
      "learning_rate": 5e-05,
      "loss": 0.132,
      "step": 16780
    },
    {
      "step": 16780,
      "train_flow_matching_loss": 0.291015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 16780,
      "train_flow_matching_loss": 0.08837890625,
      "train_physics_loss": 0.0
    },
    {
      "step": 16780,
      "train_flow_matching_loss": 0.1806640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 16780,
      "train_flow_matching_loss": 0.2099609375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.1116599440574646,
      "learning_rate": 5e-05,
      "loss": 0.1354,
      "step": 16790
    },
    {
      "step": 16790,
      "train_flow_matching_loss": 0.11767578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 16790,
      "train_flow_matching_loss": 0.201171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 16790,
      "train_flow_matching_loss": 0.15625,
      "train_physics_loss": 0.0
    },
    {
      "step": 16790,
      "train_flow_matching_loss": 0.234375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08436319977045059,
      "learning_rate": 5e-05,
      "loss": 0.1359,
      "step": 16800
    },
    {
      "step": 16800,
      "train_flow_matching_loss": 0.1640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 16800,
      "train_flow_matching_loss": 0.1826171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 16800,
      "train_flow_matching_loss": 0.1376953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 16800,
      "train_flow_matching_loss": 0.11376953125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10555562376976013,
      "learning_rate": 5e-05,
      "loss": 0.1393,
      "step": 16810
    },
    {
      "step": 16810,
      "train_flow_matching_loss": 0.09375,
      "train_physics_loss": 0.0
    },
    {
      "step": 16810,
      "train_flow_matching_loss": 0.1455078125,
      "train_physics_loss": 0.0
    },
    {
      "step": 16810,
      "train_flow_matching_loss": 0.12451171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 16810,
      "train_flow_matching_loss": 0.140625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.077812060713768,
      "learning_rate": 5e-05,
      "loss": 0.1348,
      "step": 16820
    },
    {
      "step": 16820,
      "train_flow_matching_loss": 0.1904296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 16820,
      "train_flow_matching_loss": 0.11865234375,
      "train_physics_loss": 0.0
    },
    {
      "step": 16820,
      "train_flow_matching_loss": 0.1015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 16820,
      "train_flow_matching_loss": 0.1591796875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08408434689044952,
      "learning_rate": 5e-05,
      "loss": 0.1317,
      "step": 16830
    },
    {
      "step": 16830,
      "train_flow_matching_loss": 0.134765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 16830,
      "train_flow_matching_loss": 0.0908203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 16830,
      "train_flow_matching_loss": 0.09765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 16830,
      "train_flow_matching_loss": 0.1162109375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08735095709562302,
      "learning_rate": 5e-05,
      "loss": 0.1362,
      "step": 16840
    },
    {
      "step": 16840,
      "train_flow_matching_loss": 0.236328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 16840,
      "train_flow_matching_loss": 0.09228515625,
      "train_physics_loss": 0.0
    },
    {
      "step": 16840,
      "train_flow_matching_loss": 0.12255859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 16840,
      "train_flow_matching_loss": 0.0966796875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10877827554941177,
      "learning_rate": 5e-05,
      "loss": 0.1348,
      "step": 16850
    },
    {
      "step": 16850,
      "train_flow_matching_loss": 0.15625,
      "train_physics_loss": 0.0
    },
    {
      "step": 16850,
      "train_flow_matching_loss": 0.1611328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 16850,
      "train_flow_matching_loss": 0.171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 16850,
      "train_flow_matching_loss": 0.1708984375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.1188538670539856,
      "learning_rate": 5e-05,
      "loss": 0.1306,
      "step": 16860
    },
    {
      "step": 16860,
      "train_flow_matching_loss": 0.1435546875,
      "train_physics_loss": 0.0
    },
    {
      "step": 16860,
      "train_flow_matching_loss": 0.0869140625,
      "train_physics_loss": 0.0
    },
    {
      "step": 16860,
      "train_flow_matching_loss": 0.08203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 16860,
      "train_flow_matching_loss": 0.1337890625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.12387775629758835,
      "learning_rate": 5e-05,
      "loss": 0.1322,
      "step": 16870
    },
    {
      "step": 16870,
      "train_flow_matching_loss": 0.173828125,
      "train_physics_loss": 0.0
    },
    {
      "step": 16870,
      "train_flow_matching_loss": 0.1015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 16870,
      "train_flow_matching_loss": 0.11669921875,
      "train_physics_loss": 0.0
    },
    {
      "step": 16870,
      "train_flow_matching_loss": 0.2197265625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10947274416685104,
      "learning_rate": 5e-05,
      "loss": 0.137,
      "step": 16880
    },
    {
      "step": 16880,
      "train_flow_matching_loss": 0.1904296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 16880,
      "train_flow_matching_loss": 0.126953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 16880,
      "train_flow_matching_loss": 0.1162109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 16880,
      "train_flow_matching_loss": 0.08056640625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09667959064245224,
      "learning_rate": 5e-05,
      "loss": 0.1405,
      "step": 16890
    },
    {
      "step": 16890,
      "train_flow_matching_loss": 0.10693359375,
      "train_physics_loss": 0.37243422865867615
    },
    {
      "step": 16890,
      "train_flow_matching_loss": 0.134765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 16890,
      "train_flow_matching_loss": 0.0810546875,
      "train_physics_loss": 0.0
    },
    {
      "step": 16890,
      "train_flow_matching_loss": 0.10546875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10017538070678711,
      "learning_rate": 5e-05,
      "loss": 0.1428,
      "step": 16900
    },
    {
      "step": 16900,
      "train_flow_matching_loss": 0.0869140625,
      "train_physics_loss": 0.43485429883003235
    },
    {
      "step": 16900,
      "train_flow_matching_loss": 0.11962890625,
      "train_physics_loss": 0.0
    },
    {
      "step": 16900,
      "train_flow_matching_loss": 0.07958984375,
      "train_physics_loss": 0.0
    },
    {
      "step": 16900,
      "train_flow_matching_loss": 0.10791015625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.0728551521897316,
      "learning_rate": 5e-05,
      "loss": 0.133,
      "step": 16910
    },
    {
      "step": 16910,
      "train_flow_matching_loss": 0.05908203125,
      "train_physics_loss": 0.4069073796272278
    },
    {
      "step": 16910,
      "train_flow_matching_loss": 0.1298828125,
      "train_physics_loss": 0.0
    },
    {
      "step": 16910,
      "train_flow_matching_loss": 0.07958984375,
      "train_physics_loss": 0.0
    },
    {
      "step": 16910,
      "train_flow_matching_loss": 0.0986328125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.1197754442691803,
      "learning_rate": 5e-05,
      "loss": 0.1326,
      "step": 16920
    },
    {
      "step": 16920,
      "train_flow_matching_loss": 0.039794921875,
      "train_physics_loss": 0.4022892117500305
    },
    {
      "step": 16920,
      "train_flow_matching_loss": 0.11474609375,
      "train_physics_loss": 0.0
    },
    {
      "step": 16920,
      "train_flow_matching_loss": 0.1083984375,
      "train_physics_loss": 0.0
    },
    {
      "step": 16920,
      "train_flow_matching_loss": 0.1328125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.0795893445611,
      "learning_rate": 5e-05,
      "loss": 0.1322,
      "step": 16930
    },
    {
      "step": 16930,
      "train_flow_matching_loss": 0.08837890625,
      "train_physics_loss": 0.4333896040916443
    },
    {
      "step": 16930,
      "train_flow_matching_loss": 0.1376953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 16930,
      "train_flow_matching_loss": 0.05419921875,
      "train_physics_loss": 0.0
    },
    {
      "step": 16930,
      "train_flow_matching_loss": 0.11181640625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08548784255981445,
      "learning_rate": 5e-05,
      "loss": 0.1332,
      "step": 16940
    },
    {
      "step": 16940,
      "train_flow_matching_loss": 0.078125,
      "train_physics_loss": 0.48811954259872437
    },
    {
      "step": 16940,
      "train_flow_matching_loss": 0.1240234375,
      "train_physics_loss": 0.0
    },
    {
      "step": 16940,
      "train_flow_matching_loss": 0.140625,
      "train_physics_loss": 0.0
    },
    {
      "step": 16940,
      "train_flow_matching_loss": 0.12451171875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.05616561323404312,
      "learning_rate": 5e-05,
      "loss": 0.1322,
      "step": 16950
    },
    {
      "step": 16950,
      "train_flow_matching_loss": 0.059814453125,
      "train_physics_loss": 0.4110567271709442
    },
    {
      "step": 16950,
      "train_flow_matching_loss": 0.09765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 16950,
      "train_flow_matching_loss": 0.10546875,
      "train_physics_loss": 0.0
    },
    {
      "step": 16950,
      "train_flow_matching_loss": 0.1435546875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.13204212486743927,
      "learning_rate": 5e-05,
      "loss": 0.1338,
      "step": 16960
    },
    {
      "step": 16960,
      "train_flow_matching_loss": 0.026123046875,
      "train_physics_loss": 0.3486100733280182
    },
    {
      "step": 16960,
      "train_flow_matching_loss": 0.10205078125,
      "train_physics_loss": 0.0
    },
    {
      "step": 16960,
      "train_flow_matching_loss": 0.1533203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 16960,
      "train_flow_matching_loss": 0.1494140625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.0785045474767685,
      "learning_rate": 5e-05,
      "loss": 0.1387,
      "step": 16970
    },
    {
      "step": 16970,
      "train_flow_matching_loss": 0.0712890625,
      "train_physics_loss": 0.0
    },
    {
      "step": 16970,
      "train_flow_matching_loss": 0.08544921875,
      "train_physics_loss": 0.0
    },
    {
      "step": 16970,
      "train_flow_matching_loss": 0.07568359375,
      "train_physics_loss": 0.0
    },
    {
      "step": 16970,
      "train_flow_matching_loss": 0.1630859375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09287883341312408,
      "learning_rate": 5e-05,
      "loss": 0.1326,
      "step": 16980
    },
    {
      "step": 16980,
      "train_flow_matching_loss": 0.0625,
      "train_physics_loss": 0.4691436290740967
    },
    {
      "step": 16980,
      "train_flow_matching_loss": 0.169921875,
      "train_physics_loss": 0.0
    },
    {
      "step": 16980,
      "train_flow_matching_loss": 0.12158203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 16980,
      "train_flow_matching_loss": 0.0869140625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09385160356760025,
      "learning_rate": 5e-05,
      "loss": 0.1364,
      "step": 16990
    },
    {
      "step": 16990,
      "train_flow_matching_loss": 0.09033203125,
      "train_physics_loss": 0.47657889127731323
    },
    {
      "step": 16990,
      "train_flow_matching_loss": 0.1533203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 16990,
      "train_flow_matching_loss": 0.08935546875,
      "train_physics_loss": 0.0
    },
    {
      "step": 16990,
      "train_flow_matching_loss": 0.1025390625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11162716150283813,
      "learning_rate": 5e-05,
      "loss": 0.1424,
      "step": 17000
    },
    {
      "step": 17000,
      "train_flow_matching_loss": 0.052734375,
      "train_physics_loss": 0.35624629259109497
    },
    {
      "step": 17000,
      "train_flow_matching_loss": 0.1474609375,
      "train_physics_loss": 0.0
    },
    {
      "step": 17000,
      "train_flow_matching_loss": 0.1513671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 17000,
      "train_flow_matching_loss": 0.150390625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.14672045409679413,
      "learning_rate": 5e-05,
      "loss": 0.136,
      "step": 17010
    },
    {
      "step": 17010,
      "train_flow_matching_loss": 0.09912109375,
      "train_physics_loss": 0.4534081220626831
    },
    {
      "step": 17010,
      "train_flow_matching_loss": 0.09619140625,
      "train_physics_loss": 0.0
    },
    {
      "step": 17010,
      "train_flow_matching_loss": 0.166015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 17010,
      "train_flow_matching_loss": 0.126953125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.05636212229728699,
      "learning_rate": 5e-05,
      "loss": 0.1361,
      "step": 17020
    },
    {
      "step": 17020,
      "train_flow_matching_loss": 0.0751953125,
      "train_physics_loss": 0.41710907220840454
    },
    {
      "step": 17020,
      "train_flow_matching_loss": 0.1396484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 17020,
      "train_flow_matching_loss": 0.142578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 17020,
      "train_flow_matching_loss": 0.1025390625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09572219848632812,
      "learning_rate": 5e-05,
      "loss": 0.1382,
      "step": 17030
    },
    {
      "step": 17030,
      "train_flow_matching_loss": 0.0625,
      "train_physics_loss": 0.3679787814617157
    },
    {
      "step": 17030,
      "train_flow_matching_loss": 0.1796875,
      "train_physics_loss": 0.0
    },
    {
      "step": 17030,
      "train_flow_matching_loss": 0.1572265625,
      "train_physics_loss": 0.0
    },
    {
      "step": 17030,
      "train_flow_matching_loss": 0.1513671875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08238841593265533,
      "learning_rate": 5e-05,
      "loss": 0.1318,
      "step": 17040
    },
    {
      "step": 17040,
      "train_flow_matching_loss": 0.07666015625,
      "train_physics_loss": 0.40635764598846436
    },
    {
      "step": 17040,
      "train_flow_matching_loss": 0.1552734375,
      "train_physics_loss": 0.0
    },
    {
      "step": 17040,
      "train_flow_matching_loss": 0.1494140625,
      "train_physics_loss": 0.0
    },
    {
      "step": 17040,
      "train_flow_matching_loss": 0.1533203125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.07711054384708405,
      "learning_rate": 5e-05,
      "loss": 0.1326,
      "step": 17050
    },
    {
      "step": 17050,
      "train_flow_matching_loss": 0.1416015625,
      "train_physics_loss": 0.5071556568145752
    },
    {
      "step": 17050,
      "train_flow_matching_loss": 0.171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 17050,
      "train_flow_matching_loss": 0.1376953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 17050,
      "train_flow_matching_loss": 0.126953125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08892817795276642,
      "learning_rate": 5e-05,
      "loss": 0.1316,
      "step": 17060
    },
    {
      "step": 17060,
      "train_flow_matching_loss": 0.049560546875,
      "train_physics_loss": 0.40319380164146423
    },
    {
      "step": 17060,
      "train_flow_matching_loss": 0.09228515625,
      "train_physics_loss": 0.0
    },
    {
      "step": 17060,
      "train_flow_matching_loss": 0.10888671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 17060,
      "train_flow_matching_loss": 0.0262451171875,
      "train_physics_loss": 0.36331844329833984
    },
    {
      "grad_norm": 0.06854937225580215,
      "learning_rate": 5e-05,
      "loss": 0.1345,
      "step": 17070
    },
    {
      "step": 17070,
      "train_flow_matching_loss": 0.134765625,
      "train_physics_loss": 0.5109431147575378
    },
    {
      "step": 17070,
      "train_flow_matching_loss": 0.244140625,
      "train_physics_loss": 0.0
    },
    {
      "step": 17070,
      "train_flow_matching_loss": 0.11376953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 17070,
      "train_flow_matching_loss": 0.06396484375,
      "train_physics_loss": 0.4098433554172516
    },
    {
      "grad_norm": 0.08621799200773239,
      "learning_rate": 5e-05,
      "loss": 0.1436,
      "step": 17080
    },
    {
      "step": 17080,
      "train_flow_matching_loss": 0.06787109375,
      "train_physics_loss": 0.3673316240310669
    },
    {
      "step": 17080,
      "train_flow_matching_loss": 0.10791015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 17080,
      "train_flow_matching_loss": 0.10009765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 17080,
      "train_flow_matching_loss": 0.051513671875,
      "train_physics_loss": 0.47102028131484985
    },
    {
      "grad_norm": 0.09617643803358078,
      "learning_rate": 5e-05,
      "loss": 0.132,
      "step": 17090
    },
    {
      "step": 17090,
      "train_flow_matching_loss": 0.052978515625,
      "train_physics_loss": 0.34816086292266846
    },
    {
      "step": 17090,
      "train_flow_matching_loss": 0.11572265625,
      "train_physics_loss": 0.0
    },
    {
      "step": 17090,
      "train_flow_matching_loss": 0.189453125,
      "train_physics_loss": 0.0
    },
    {
      "step": 17090,
      "train_flow_matching_loss": 0.05712890625,
      "train_physics_loss": 0.38781148195266724
    },
    {
      "grad_norm": 0.1524619162082672,
      "learning_rate": 5e-05,
      "loss": 0.1335,
      "step": 17100
    },
    {
      "step": 17100,
      "train_flow_matching_loss": 0.056396484375,
      "train_physics_loss": 0.42837703227996826
    },
    {
      "step": 17100,
      "train_flow_matching_loss": 0.130859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 17100,
      "train_flow_matching_loss": 0.1357421875,
      "train_physics_loss": 0.0
    },
    {
      "step": 17100,
      "train_flow_matching_loss": 0.054443359375,
      "train_physics_loss": 0.3748905062675476
    },
    {
      "grad_norm": 0.13997529447078705,
      "learning_rate": 5e-05,
      "loss": 0.1343,
      "step": 17110
    },
    {
      "step": 17110,
      "train_flow_matching_loss": 0.061767578125,
      "train_physics_loss": 0.37553635239601135
    },
    {
      "step": 17110,
      "train_flow_matching_loss": 0.08935546875,
      "train_physics_loss": 0.0
    },
    {
      "step": 17110,
      "train_flow_matching_loss": 0.09716796875,
      "train_physics_loss": 0.0
    },
    {
      "step": 17110,
      "train_flow_matching_loss": 0.055419921875,
      "train_physics_loss": 0.4032548666000366
    },
    {
      "grad_norm": 0.13064061105251312,
      "learning_rate": 5e-05,
      "loss": 0.1379,
      "step": 17120
    },
    {
      "step": 17120,
      "train_flow_matching_loss": 0.046875,
      "train_physics_loss": 0.38779979944229126
    },
    {
      "step": 17120,
      "train_flow_matching_loss": 0.1455078125,
      "train_physics_loss": 0.0
    },
    {
      "step": 17120,
      "train_flow_matching_loss": 0.1005859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 17120,
      "train_flow_matching_loss": 0.0308837890625,
      "train_physics_loss": 0.3760910630226135
    },
    {
      "grad_norm": 0.08619234710931778,
      "learning_rate": 5e-05,
      "loss": 0.144,
      "step": 17130
    },
    {
      "step": 17130,
      "train_flow_matching_loss": 0.049560546875,
      "train_physics_loss": 0.3796572983264923
    },
    {
      "step": 17130,
      "train_flow_matching_loss": 0.244140625,
      "train_physics_loss": 0.0
    },
    {
      "step": 17130,
      "train_flow_matching_loss": 0.0927734375,
      "train_physics_loss": 0.0
    },
    {
      "step": 17130,
      "train_flow_matching_loss": 0.0654296875,
      "train_physics_loss": 0.5090402364730835
    },
    {
      "grad_norm": 0.09600158035755157,
      "learning_rate": 5e-05,
      "loss": 0.1401,
      "step": 17140
    },
    {
      "step": 17140,
      "train_flow_matching_loss": 0.07568359375,
      "train_physics_loss": 0.3976304233074188
    },
    {
      "step": 17140,
      "train_flow_matching_loss": 0.1650390625,
      "train_physics_loss": 0.0
    },
    {
      "step": 17140,
      "train_flow_matching_loss": 0.1279296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 17140,
      "train_flow_matching_loss": 0.036376953125,
      "train_physics_loss": 0.37212592363357544
    },
    {
      "grad_norm": 0.06963006407022476,
      "learning_rate": 5e-05,
      "loss": 0.1369,
      "step": 17150
    },
    {
      "step": 17150,
      "train_flow_matching_loss": 0.048583984375,
      "train_physics_loss": 0.3739180564880371
    },
    {
      "step": 17150,
      "train_flow_matching_loss": 0.1279296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 17150,
      "train_flow_matching_loss": 0.1904296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 17150,
      "train_flow_matching_loss": 0.1005859375,
      "train_physics_loss": 0.44226518273353577
    },
    {
      "grad_norm": 0.09096731245517731,
      "learning_rate": 5e-05,
      "loss": 0.1377,
      "step": 17160
    },
    {
      "step": 17160,
      "train_flow_matching_loss": 0.07275390625,
      "train_physics_loss": 0.39275383949279785
    },
    {
      "step": 17160,
      "train_flow_matching_loss": 0.10791015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 17160,
      "train_flow_matching_loss": 0.162109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 17160,
      "train_flow_matching_loss": 0.059326171875,
      "train_physics_loss": 0.4002891182899475
    },
    {
      "grad_norm": 0.08952459692955017,
      "learning_rate": 5e-05,
      "loss": 0.1306,
      "step": 17170
    },
    {
      "step": 17170,
      "train_flow_matching_loss": 0.0859375,
      "train_physics_loss": 0.43061473965644836
    },
    {
      "step": 17170,
      "train_flow_matching_loss": 0.1767578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 17170,
      "train_flow_matching_loss": 0.1396484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 17170,
      "train_flow_matching_loss": 0.068359375,
      "train_physics_loss": 0.3554278314113617
    },
    {
      "grad_norm": 0.0880519449710846,
      "learning_rate": 5e-05,
      "loss": 0.1336,
      "step": 17180
    },
    {
      "step": 17180,
      "train_flow_matching_loss": 0.038818359375,
      "train_physics_loss": 0.4023350179195404
    },
    {
      "step": 17180,
      "train_flow_matching_loss": 0.1474609375,
      "train_physics_loss": 0.0
    },
    {
      "step": 17180,
      "train_flow_matching_loss": 0.1455078125,
      "train_physics_loss": 0.0
    },
    {
      "step": 17180,
      "train_flow_matching_loss": 0.10498046875,
      "train_physics_loss": 0.48499763011932373
    },
    {
      "grad_norm": 0.07156910747289658,
      "learning_rate": 5e-05,
      "loss": 0.131,
      "step": 17190
    },
    {
      "step": 17190,
      "train_flow_matching_loss": 0.05517578125,
      "train_physics_loss": 0.4277819097042084
    },
    {
      "step": 17190,
      "train_flow_matching_loss": 0.1484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 17190,
      "train_flow_matching_loss": 0.1689453125,
      "train_physics_loss": 0.0
    },
    {
      "step": 17190,
      "train_flow_matching_loss": 0.05029296875,
      "train_physics_loss": 0.40393078327178955
    },
    {
      "grad_norm": 0.09824364632368088,
      "learning_rate": 5e-05,
      "loss": 0.1351,
      "step": 17200
    },
    {
      "step": 17200,
      "train_flow_matching_loss": 0.050537109375,
      "train_physics_loss": 0.44588491320610046
    },
    {
      "step": 17200,
      "train_flow_matching_loss": 0.2431640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 17200,
      "train_flow_matching_loss": 0.1728515625,
      "train_physics_loss": 0.0
    },
    {
      "step": 17200,
      "train_flow_matching_loss": 0.08251953125,
      "train_physics_loss": 0.4133254587650299
    },
    {
      "grad_norm": 0.08057958632707596,
      "learning_rate": 5e-05,
      "loss": 0.1394,
      "step": 17210
    },
    {
      "step": 17210,
      "train_flow_matching_loss": 0.0255126953125,
      "train_physics_loss": 0.3857806324958801
    },
    {
      "step": 17210,
      "train_flow_matching_loss": 0.10498046875,
      "train_physics_loss": 0.0
    },
    {
      "step": 17210,
      "train_flow_matching_loss": 0.12890625,
      "train_physics_loss": 0.0
    },
    {
      "step": 17210,
      "train_flow_matching_loss": 0.08349609375,
      "train_physics_loss": 0.3983805179595947
    },
    {
      "grad_norm": 0.10925596952438354,
      "learning_rate": 5e-05,
      "loss": 0.1337,
      "step": 17220
    },
    {
      "step": 17220,
      "train_flow_matching_loss": 0.03662109375,
      "train_physics_loss": 0.324028879404068
    },
    {
      "step": 17220,
      "train_flow_matching_loss": 0.162109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 17220,
      "train_flow_matching_loss": 0.10546875,
      "train_physics_loss": 0.0
    },
    {
      "step": 17220,
      "train_flow_matching_loss": 0.08349609375,
      "train_physics_loss": 0.4152125418186188
    },
    {
      "grad_norm": 2.3195481300354004,
      "learning_rate": 5e-05,
      "loss": 0.1318,
      "step": 17230
    },
    {
      "step": 17230,
      "train_flow_matching_loss": 0.036376953125,
      "train_physics_loss": 0.404584139585495
    },
    {
      "step": 17230,
      "train_flow_matching_loss": 0.07421875,
      "train_physics_loss": 0.0
    },
    {
      "step": 17230,
      "train_flow_matching_loss": 0.162109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 17230,
      "train_flow_matching_loss": 0.10107421875,
      "train_physics_loss": 0.4047988951206207
    },
    {
      "grad_norm": 0.10221529006958008,
      "learning_rate": 5e-05,
      "loss": 0.1357,
      "step": 17240
    },
    {
      "step": 17240,
      "train_flow_matching_loss": 0.035400390625,
      "train_physics_loss": 0.366818904876709
    },
    {
      "step": 17240,
      "train_flow_matching_loss": 0.1787109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 17240,
      "train_flow_matching_loss": 0.1318359375,
      "train_physics_loss": 0.0
    },
    {
      "step": 17240,
      "train_flow_matching_loss": 0.0283203125,
      "train_physics_loss": 0.3944336473941803
    },
    {
      "grad_norm": 0.0836704820394516,
      "learning_rate": 5e-05,
      "loss": 0.1384,
      "step": 17250
    },
    {
      "step": 17250,
      "train_flow_matching_loss": 0.060791015625,
      "train_physics_loss": 0.40067335963249207
    },
    {
      "step": 17250,
      "train_flow_matching_loss": 0.1455078125,
      "train_physics_loss": 0.0
    },
    {
      "step": 17250,
      "train_flow_matching_loss": 0.1279296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 17250,
      "train_flow_matching_loss": 0.08740234375,
      "train_physics_loss": 0.3537503778934479
    },
    {
      "grad_norm": 0.09022406488656998,
      "learning_rate": 5e-05,
      "loss": 0.1323,
      "step": 17260
    },
    {
      "step": 17260,
      "train_flow_matching_loss": 0.07421875,
      "train_physics_loss": 0.36258018016815186
    },
    {
      "step": 17260,
      "train_flow_matching_loss": 0.1474609375,
      "train_physics_loss": 0.0
    },
    {
      "step": 17260,
      "train_flow_matching_loss": 0.220703125,
      "train_physics_loss": 0.0
    },
    {
      "step": 17260,
      "train_flow_matching_loss": 0.0732421875,
      "train_physics_loss": 0.3431922197341919
    },
    {
      "grad_norm": 0.0932457372546196,
      "learning_rate": 5e-05,
      "loss": 0.1358,
      "step": 17270
    },
    {
      "step": 17270,
      "train_flow_matching_loss": 0.1044921875,
      "train_physics_loss": 0.379167377948761
    },
    {
      "step": 17270,
      "train_flow_matching_loss": 0.1435546875,
      "train_physics_loss": 0.0
    },
    {
      "step": 17270,
      "train_flow_matching_loss": 0.2294921875,
      "train_physics_loss": 0.0
    },
    {
      "step": 17270,
      "train_flow_matching_loss": 0.051513671875,
      "train_physics_loss": 0.37184906005859375
    },
    {
      "grad_norm": 0.07265932857990265,
      "learning_rate": 5e-05,
      "loss": 0.1379,
      "step": 17280
    },
    {
      "step": 17280,
      "train_flow_matching_loss": 0.146484375,
      "train_physics_loss": 0.42109784483909607
    },
    {
      "step": 17280,
      "train_flow_matching_loss": 0.08251953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 17280,
      "train_flow_matching_loss": 0.1630859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 17280,
      "train_flow_matching_loss": 0.048095703125,
      "train_physics_loss": 0.38951268792152405
    },
    {
      "grad_norm": 0.1041565090417862,
      "learning_rate": 5e-05,
      "loss": 0.1366,
      "step": 17290
    },
    {
      "step": 17290,
      "train_flow_matching_loss": 0.03369140625,
      "train_physics_loss": 0.3433452248573303
    },
    {
      "step": 17290,
      "train_flow_matching_loss": 0.1474609375,
      "train_physics_loss": 0.0
    },
    {
      "step": 17290,
      "train_flow_matching_loss": 0.1767578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 17290,
      "train_flow_matching_loss": 0.0849609375,
      "train_physics_loss": 0.46610867977142334
    },
    {
      "grad_norm": 0.12387540936470032,
      "learning_rate": 5e-05,
      "loss": 0.1398,
      "step": 17300
    },
    {
      "step": 17300,
      "train_flow_matching_loss": 0.05615234375,
      "train_physics_loss": 0.3125597834587097
    },
    {
      "step": 17300,
      "train_flow_matching_loss": 0.1298828125,
      "train_physics_loss": 0.0
    },
    {
      "step": 17300,
      "train_flow_matching_loss": 0.1591796875,
      "train_physics_loss": 0.0
    },
    {
      "step": 17300,
      "train_flow_matching_loss": 0.07080078125,
      "train_physics_loss": 0.35218513011932373
    },
    {
      "grad_norm": 0.14077617228031158,
      "learning_rate": 5e-05,
      "loss": 0.1355,
      "step": 17310
    },
    {
      "step": 17310,
      "train_flow_matching_loss": 0.07568359375,
      "train_physics_loss": 0.45755261182785034
    },
    {
      "step": 17310,
      "train_flow_matching_loss": 0.2021484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 17310,
      "train_flow_matching_loss": 0.08056640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 17310,
      "train_flow_matching_loss": 0.1552734375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09259439259767532,
      "learning_rate": 5e-05,
      "loss": 0.131,
      "step": 17320
    },
    {
      "step": 17320,
      "train_flow_matching_loss": 0.050048828125,
      "train_physics_loss": 0.33700239658355713
    },
    {
      "step": 17320,
      "train_flow_matching_loss": 0.11767578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 17320,
      "train_flow_matching_loss": 0.2060546875,
      "train_physics_loss": 0.0
    },
    {
      "step": 17320,
      "train_flow_matching_loss": 0.11962890625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08196473866701126,
      "learning_rate": 5e-05,
      "loss": 0.1358,
      "step": 17330
    },
    {
      "step": 17330,
      "train_flow_matching_loss": 0.054931640625,
      "train_physics_loss": 0.33278408646583557
    },
    {
      "step": 17330,
      "train_flow_matching_loss": 0.10205078125,
      "train_physics_loss": 0.0
    },
    {
      "step": 17330,
      "train_flow_matching_loss": 0.146484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 17330,
      "train_flow_matching_loss": 0.146484375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.0987023189663887,
      "learning_rate": 5e-05,
      "loss": 0.1374,
      "step": 17340
    },
    {
      "step": 17340,
      "train_flow_matching_loss": 0.05810546875,
      "train_physics_loss": 0.36972033977508545
    },
    {
      "step": 17340,
      "train_flow_matching_loss": 0.10693359375,
      "train_physics_loss": 0.0
    },
    {
      "step": 17340,
      "train_flow_matching_loss": 0.169921875,
      "train_physics_loss": 0.0
    },
    {
      "step": 17340,
      "train_flow_matching_loss": 0.2216796875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09227844327688217,
      "learning_rate": 5e-05,
      "loss": 0.1342,
      "step": 17350
    },
    {
      "step": 17350,
      "train_flow_matching_loss": 0.06591796875,
      "train_physics_loss": 0.367430716753006
    },
    {
      "step": 17350,
      "train_flow_matching_loss": 0.2099609375,
      "train_physics_loss": 0.0
    },
    {
      "step": 17350,
      "train_flow_matching_loss": 0.1513671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 17350,
      "train_flow_matching_loss": 0.150390625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10652261227369308,
      "learning_rate": 5e-05,
      "loss": 0.1368,
      "step": 17360
    },
    {
      "step": 17360,
      "train_flow_matching_loss": 0.0634765625,
      "train_physics_loss": 0.40317192673683167
    },
    {
      "step": 17360,
      "train_flow_matching_loss": 0.126953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 17360,
      "train_flow_matching_loss": 0.154296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 17360,
      "train_flow_matching_loss": 0.185546875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10601402819156647,
      "learning_rate": 5e-05,
      "loss": 0.1331,
      "step": 17370
    },
    {
      "step": 17370,
      "train_flow_matching_loss": 0.09765625,
      "train_physics_loss": 0.4766532778739929
    },
    {
      "step": 17370,
      "train_flow_matching_loss": 0.1171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 17370,
      "train_flow_matching_loss": 0.138671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 17370,
      "train_flow_matching_loss": 0.11328125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.0805194154381752,
      "learning_rate": 5e-05,
      "loss": 0.1311,
      "step": 17380
    },
    {
      "step": 17380,
      "train_flow_matching_loss": 0.06591796875,
      "train_physics_loss": 0.4109370708465576
    },
    {
      "step": 17380,
      "train_flow_matching_loss": 0.1474609375,
      "train_physics_loss": 0.0
    },
    {
      "step": 17380,
      "train_flow_matching_loss": 0.181640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 17380,
      "train_flow_matching_loss": 0.1767578125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.1322629749774933,
      "learning_rate": 5e-05,
      "loss": 0.1345,
      "step": 17390
    },
    {
      "step": 17390,
      "train_flow_matching_loss": 0.033203125,
      "train_physics_loss": 0.37599483132362366
    },
    {
      "step": 17390,
      "train_flow_matching_loss": 0.1064453125,
      "train_physics_loss": 0.0
    },
    {
      "step": 17390,
      "train_flow_matching_loss": 0.1474609375,
      "train_physics_loss": 0.0
    },
    {
      "step": 17390,
      "train_flow_matching_loss": 0.11083984375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10140521079301834,
      "learning_rate": 5e-05,
      "loss": 0.1327,
      "step": 17400
    },
    {
      "step": 17400,
      "train_flow_matching_loss": 0.1748046875,
      "train_physics_loss": 0.0
    },
    {
      "step": 17400,
      "train_flow_matching_loss": 0.1474609375,
      "train_physics_loss": 0.0
    },
    {
      "step": 17400,
      "train_flow_matching_loss": 0.11865234375,
      "train_physics_loss": 0.0
    },
    {
      "step": 17400,
      "train_flow_matching_loss": 0.166015625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11768219619989395,
      "learning_rate": 5e-05,
      "loss": 0.1372,
      "step": 17410
    },
    {
      "step": 17410,
      "train_flow_matching_loss": 0.11572265625,
      "train_physics_loss": 0.0
    },
    {
      "step": 17410,
      "train_flow_matching_loss": 0.11328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 17410,
      "train_flow_matching_loss": 0.11669921875,
      "train_physics_loss": 0.0
    },
    {
      "step": 17410,
      "train_flow_matching_loss": 0.138671875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10348916053771973,
      "learning_rate": 5e-05,
      "loss": 0.1369,
      "step": 17420
    },
    {
      "step": 17420,
      "train_flow_matching_loss": 0.1767578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 17420,
      "train_flow_matching_loss": 0.1181640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 17420,
      "train_flow_matching_loss": 0.0810546875,
      "train_physics_loss": 0.0
    },
    {
      "step": 17420,
      "train_flow_matching_loss": 0.12353515625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.0912446677684784,
      "learning_rate": 5e-05,
      "loss": 0.1397,
      "step": 17430
    },
    {
      "step": 17430,
      "train_flow_matching_loss": 0.1572265625,
      "train_physics_loss": 0.0
    },
    {
      "step": 17430,
      "train_flow_matching_loss": 0.08349609375,
      "train_physics_loss": 0.0
    },
    {
      "step": 17430,
      "train_flow_matching_loss": 0.1533203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 17430,
      "train_flow_matching_loss": 0.1552734375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10126598179340363,
      "learning_rate": 5e-05,
      "loss": 0.1307,
      "step": 17440
    },
    {
      "step": 17440,
      "train_flow_matching_loss": 0.171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 17440,
      "train_flow_matching_loss": 0.171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 17440,
      "train_flow_matching_loss": 0.1279296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 17440,
      "train_flow_matching_loss": 0.09228515625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08111345767974854,
      "learning_rate": 5e-05,
      "loss": 0.1342,
      "step": 17450
    },
    {
      "step": 17450,
      "train_flow_matching_loss": 0.11376953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 17450,
      "train_flow_matching_loss": 0.1279296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 17450,
      "train_flow_matching_loss": 0.1513671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 17450,
      "train_flow_matching_loss": 0.09619140625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08119836449623108,
      "learning_rate": 5e-05,
      "loss": 0.1331,
      "step": 17460
    },
    {
      "step": 17460,
      "train_flow_matching_loss": 0.1142578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 17460,
      "train_flow_matching_loss": 0.1044921875,
      "train_physics_loss": 0.0
    },
    {
      "step": 17460,
      "train_flow_matching_loss": 0.10546875,
      "train_physics_loss": 0.0
    },
    {
      "step": 17460,
      "train_flow_matching_loss": 0.1455078125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.3632412850856781,
      "learning_rate": 5e-05,
      "loss": 0.1327,
      "step": 17470
    },
    {
      "step": 17470,
      "train_flow_matching_loss": 0.1162109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 17470,
      "train_flow_matching_loss": 0.09912109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 17470,
      "train_flow_matching_loss": 0.1787109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 17470,
      "train_flow_matching_loss": 0.09228515625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09437524527311325,
      "learning_rate": 5e-05,
      "loss": 0.1335,
      "step": 17480
    },
    {
      "step": 17480,
      "train_flow_matching_loss": 0.1279296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 17480,
      "train_flow_matching_loss": 0.1640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 17480,
      "train_flow_matching_loss": 0.1279296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 17480,
      "train_flow_matching_loss": 0.12353515625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.2200397104024887,
      "learning_rate": 5e-05,
      "loss": 0.1353,
      "step": 17490
    },
    {
      "step": 17490,
      "train_flow_matching_loss": 0.12109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 17490,
      "train_flow_matching_loss": 0.12255859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 17490,
      "train_flow_matching_loss": 0.1240234375,
      "train_physics_loss": 0.0
    },
    {
      "step": 17490,
      "train_flow_matching_loss": 0.09765625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.3026432991027832,
      "learning_rate": 5e-05,
      "loss": 0.128,
      "step": 17500
    },
    {
      "step": 17500,
      "train_flow_matching_loss": 0.10595703125,
      "train_physics_loss": 0.0
    },
    {
      "step": 17500,
      "train_flow_matching_loss": 0.1259765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 17500,
      "train_flow_matching_loss": 0.1318359375,
      "train_physics_loss": 0.0
    },
    {
      "step": 17500,
      "train_flow_matching_loss": 0.1376953125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08854041248559952,
      "learning_rate": 5e-05,
      "loss": 0.1292,
      "step": 17510
    },
    {
      "step": 17510,
      "train_flow_matching_loss": 0.10546875,
      "train_physics_loss": 0.0
    },
    {
      "step": 17510,
      "train_flow_matching_loss": 0.1611328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 17510,
      "train_flow_matching_loss": 0.13671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 17510,
      "train_flow_matching_loss": 0.173828125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.07702381163835526,
      "learning_rate": 5e-05,
      "loss": 0.1337,
      "step": 17520
    },
    {
      "step": 17520,
      "train_flow_matching_loss": 0.2314453125,
      "train_physics_loss": 0.0
    },
    {
      "step": 17520,
      "train_flow_matching_loss": 0.130859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 17520,
      "train_flow_matching_loss": 0.1259765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 17520,
      "train_flow_matching_loss": 0.1240234375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.1139383465051651,
      "learning_rate": 5e-05,
      "loss": 0.1359,
      "step": 17530
    },
    {
      "step": 17530,
      "train_flow_matching_loss": 0.1513671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 17530,
      "train_flow_matching_loss": 0.1181640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 17530,
      "train_flow_matching_loss": 0.1015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 17530,
      "train_flow_matching_loss": 0.11328125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09368199855089188,
      "learning_rate": 5e-05,
      "loss": 0.1388,
      "step": 17540
    },
    {
      "step": 17540,
      "train_flow_matching_loss": 0.1396484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 17540,
      "train_flow_matching_loss": 0.134765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 17540,
      "train_flow_matching_loss": 0.1240234375,
      "train_physics_loss": 0.0
    },
    {
      "step": 17540,
      "train_flow_matching_loss": 0.10205078125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08331505954265594,
      "learning_rate": 5e-05,
      "loss": 0.1373,
      "step": 17550
    },
    {
      "step": 17550,
      "train_flow_matching_loss": 0.11376953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 17550,
      "train_flow_matching_loss": 0.1728515625,
      "train_physics_loss": 0.0
    },
    {
      "step": 17550,
      "train_flow_matching_loss": 0.15625,
      "train_physics_loss": 0.0
    },
    {
      "step": 17550,
      "train_flow_matching_loss": 0.1591796875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08635850250720978,
      "learning_rate": 5e-05,
      "loss": 0.1346,
      "step": 17560
    },
    {
      "step": 17560,
      "train_flow_matching_loss": 0.1162109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 17560,
      "train_flow_matching_loss": 0.11376953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 17560,
      "train_flow_matching_loss": 0.08642578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 17560,
      "train_flow_matching_loss": 0.08837890625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09064111858606339,
      "learning_rate": 5e-05,
      "loss": 0.1345,
      "step": 17570
    },
    {
      "step": 17570,
      "train_flow_matching_loss": 0.1591796875,
      "train_physics_loss": 0.0
    },
    {
      "step": 17570,
      "train_flow_matching_loss": 0.08642578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 17570,
      "train_flow_matching_loss": 0.1982421875,
      "train_physics_loss": 0.0
    },
    {
      "step": 17570,
      "train_flow_matching_loss": 0.171875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 1.2193028926849365,
      "learning_rate": 5e-05,
      "loss": 0.1351,
      "step": 17580
    },
    {
      "step": 17580,
      "train_flow_matching_loss": 0.12060546875,
      "train_physics_loss": 0.0
    },
    {
      "step": 17580,
      "train_flow_matching_loss": 0.1396484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 17580,
      "train_flow_matching_loss": 0.1279296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 17580,
      "train_flow_matching_loss": 0.1611328125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.07286284118890762,
      "learning_rate": 5e-05,
      "loss": 0.1271,
      "step": 17590
    },
    {
      "step": 17590,
      "train_flow_matching_loss": 0.1513671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 17590,
      "train_flow_matching_loss": 0.08837890625,
      "train_physics_loss": 0.0
    },
    {
      "step": 17590,
      "train_flow_matching_loss": 0.11328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 17590,
      "train_flow_matching_loss": 0.1748046875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10708209127187729,
      "learning_rate": 5e-05,
      "loss": 0.1302,
      "step": 17600
    },
    {
      "step": 17600,
      "train_flow_matching_loss": 0.11962890625,
      "train_physics_loss": 0.0
    },
    {
      "step": 17600,
      "train_flow_matching_loss": 0.1005859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 17600,
      "train_flow_matching_loss": 0.11328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 17600,
      "train_flow_matching_loss": 0.12060546875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08443376421928406,
      "learning_rate": 5e-05,
      "loss": 0.1278,
      "step": 17610
    },
    {
      "step": 17610,
      "train_flow_matching_loss": 0.1533203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 17610,
      "train_flow_matching_loss": 0.10107421875,
      "train_physics_loss": 0.0
    },
    {
      "step": 17610,
      "train_flow_matching_loss": 0.146484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 17610,
      "train_flow_matching_loss": 0.10009765625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11405123770236969,
      "learning_rate": 5e-05,
      "loss": 0.1307,
      "step": 17620
    },
    {
      "step": 17620,
      "train_flow_matching_loss": 0.1640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 17620,
      "train_flow_matching_loss": 0.1357421875,
      "train_physics_loss": 0.0
    },
    {
      "step": 17620,
      "train_flow_matching_loss": 0.1650390625,
      "train_physics_loss": 0.0
    },
    {
      "step": 17620,
      "train_flow_matching_loss": 0.26171875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.1281951665878296,
      "learning_rate": 5e-05,
      "loss": 0.1318,
      "step": 17630
    },
    {
      "step": 17630,
      "train_flow_matching_loss": 0.150390625,
      "train_physics_loss": 0.0
    },
    {
      "step": 17630,
      "train_flow_matching_loss": 0.06787109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 17630,
      "train_flow_matching_loss": 0.0849609375,
      "train_physics_loss": 0.0
    },
    {
      "step": 17630,
      "train_flow_matching_loss": 0.1435546875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.0823139101266861,
      "learning_rate": 5e-05,
      "loss": 0.1376,
      "step": 17640
    },
    {
      "step": 17640,
      "train_flow_matching_loss": 0.1533203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 17640,
      "train_flow_matching_loss": 0.224609375,
      "train_physics_loss": 0.0
    },
    {
      "step": 17640,
      "train_flow_matching_loss": 0.0869140625,
      "train_physics_loss": 0.0
    },
    {
      "step": 17640,
      "train_flow_matching_loss": 0.11865234375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10619883239269257,
      "learning_rate": 5e-05,
      "loss": 0.1337,
      "step": 17650
    },
    {
      "step": 17650,
      "train_flow_matching_loss": 0.1259765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 17650,
      "train_flow_matching_loss": 0.12060546875,
      "train_physics_loss": 0.0
    },
    {
      "step": 17650,
      "train_flow_matching_loss": 0.1181640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 17650,
      "train_flow_matching_loss": 0.09521484375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.07707928866147995,
      "learning_rate": 5e-05,
      "loss": 0.1335,
      "step": 17660
    },
    {
      "step": 17660,
      "train_flow_matching_loss": 0.1923828125,
      "train_physics_loss": 0.0
    },
    {
      "step": 17660,
      "train_flow_matching_loss": 0.0703125,
      "train_physics_loss": 0.0
    },
    {
      "step": 17660,
      "train_flow_matching_loss": 0.10546875,
      "train_physics_loss": 0.0
    },
    {
      "step": 17660,
      "train_flow_matching_loss": 0.1767578125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08232368528842926,
      "learning_rate": 5e-05,
      "loss": 0.1356,
      "step": 17670
    },
    {
      "step": 17670,
      "train_flow_matching_loss": 0.1259765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 17670,
      "train_flow_matching_loss": 0.1171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 17670,
      "train_flow_matching_loss": 0.08544921875,
      "train_physics_loss": 0.0
    },
    {
      "step": 17670,
      "train_flow_matching_loss": 0.1044921875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09391064196825027,
      "learning_rate": 5e-05,
      "loss": 0.1288,
      "step": 17680
    },
    {
      "step": 17680,
      "train_flow_matching_loss": 0.1982421875,
      "train_physics_loss": 0.0
    },
    {
      "step": 17680,
      "train_flow_matching_loss": 0.1669921875,
      "train_physics_loss": 0.0
    },
    {
      "step": 17680,
      "train_flow_matching_loss": 0.171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 17680,
      "train_flow_matching_loss": 0.10107421875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10017833113670349,
      "learning_rate": 5e-05,
      "loss": 0.1318,
      "step": 17690
    },
    {
      "step": 17690,
      "train_flow_matching_loss": 0.2373046875,
      "train_physics_loss": 0.0
    },
    {
      "step": 17690,
      "train_flow_matching_loss": 0.181640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 17690,
      "train_flow_matching_loss": 0.06982421875,
      "train_physics_loss": 0.0
    },
    {
      "step": 17690,
      "train_flow_matching_loss": 0.1630859375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08403000235557556,
      "learning_rate": 5e-05,
      "loss": 0.1449,
      "step": 17700
    },
    {
      "step": 17700,
      "train_flow_matching_loss": 0.1650390625,
      "train_physics_loss": 0.0
    },
    {
      "step": 17700,
      "train_flow_matching_loss": 0.10107421875,
      "train_physics_loss": 0.0
    },
    {
      "step": 17700,
      "train_flow_matching_loss": 0.12255859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 17700,
      "train_flow_matching_loss": 0.1318359375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.06991875171661377,
      "learning_rate": 5e-05,
      "loss": 0.1315,
      "step": 17710
    },
    {
      "step": 17710,
      "train_flow_matching_loss": 0.1962890625,
      "train_physics_loss": 0.0
    },
    {
      "step": 17710,
      "train_flow_matching_loss": 0.1416015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 17710,
      "train_flow_matching_loss": 0.1572265625,
      "train_physics_loss": 0.0
    },
    {
      "step": 17710,
      "train_flow_matching_loss": 0.1220703125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.1650877445936203,
      "learning_rate": 5e-05,
      "loss": 0.1324,
      "step": 17720
    },
    {
      "step": 17720,
      "train_flow_matching_loss": 0.1708984375,
      "train_physics_loss": 0.0
    },
    {
      "step": 17720,
      "train_flow_matching_loss": 0.12890625,
      "train_physics_loss": 0.0
    },
    {
      "step": 17720,
      "train_flow_matching_loss": 0.1005859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 17720,
      "train_flow_matching_loss": 0.142578125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11051962524652481,
      "learning_rate": 5e-05,
      "loss": 0.1323,
      "step": 17730
    },
    {
      "step": 17730,
      "train_flow_matching_loss": 0.126953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 17730,
      "train_flow_matching_loss": 0.158203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 17730,
      "train_flow_matching_loss": 0.1240234375,
      "train_physics_loss": 0.0
    },
    {
      "step": 17730,
      "train_flow_matching_loss": 0.10546875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11468426138162613,
      "learning_rate": 5e-05,
      "loss": 0.1312,
      "step": 17740
    },
    {
      "step": 17740,
      "train_flow_matching_loss": 0.146484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 17740,
      "train_flow_matching_loss": 0.1484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 17740,
      "train_flow_matching_loss": 0.13671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 17740,
      "train_flow_matching_loss": 0.1416015625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08868303894996643,
      "learning_rate": 5e-05,
      "loss": 0.129,
      "step": 17750
    },
    {
      "step": 17750,
      "train_flow_matching_loss": 0.109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 17750,
      "train_flow_matching_loss": 0.08349609375,
      "train_physics_loss": 0.0
    },
    {
      "step": 17750,
      "train_flow_matching_loss": 0.10595703125,
      "train_physics_loss": 0.0
    },
    {
      "step": 17750,
      "train_flow_matching_loss": 0.1279296875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.06433029472827911,
      "learning_rate": 5e-05,
      "loss": 0.132,
      "step": 17760
    },
    {
      "step": 17760,
      "train_flow_matching_loss": 0.150390625,
      "train_physics_loss": 0.0
    },
    {
      "step": 17760,
      "train_flow_matching_loss": 0.1611328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 17760,
      "train_flow_matching_loss": 0.11572265625,
      "train_physics_loss": 0.0
    },
    {
      "step": 17760,
      "train_flow_matching_loss": 0.12890625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09009543061256409,
      "learning_rate": 5e-05,
      "loss": 0.1331,
      "step": 17770
    },
    {
      "step": 17770,
      "train_flow_matching_loss": 0.1826171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 17770,
      "train_flow_matching_loss": 0.1552734375,
      "train_physics_loss": 0.0
    },
    {
      "step": 17770,
      "train_flow_matching_loss": 0.0888671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 17770,
      "train_flow_matching_loss": 0.1484375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09998331218957901,
      "learning_rate": 5e-05,
      "loss": 0.1316,
      "step": 17780
    },
    {
      "step": 17780,
      "train_flow_matching_loss": 0.205078125,
      "train_physics_loss": 0.0
    },
    {
      "step": 17780,
      "train_flow_matching_loss": 0.265625,
      "train_physics_loss": 0.0
    },
    {
      "step": 17780,
      "train_flow_matching_loss": 0.138671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 17780,
      "train_flow_matching_loss": 0.11669921875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.07818524539470673,
      "learning_rate": 5e-05,
      "loss": 0.1289,
      "step": 17790
    },
    {
      "step": 17790,
      "train_flow_matching_loss": 0.1640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 17790,
      "train_flow_matching_loss": 0.1865234375,
      "train_physics_loss": 0.0
    },
    {
      "step": 17790,
      "train_flow_matching_loss": 0.11474609375,
      "train_physics_loss": 0.0
    },
    {
      "step": 17790,
      "train_flow_matching_loss": 0.10791015625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10647507756948471,
      "learning_rate": 5e-05,
      "loss": 0.1308,
      "step": 17800
    },
    {
      "step": 17800,
      "train_flow_matching_loss": 0.1884765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 17800,
      "train_flow_matching_loss": 0.1416015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 17800,
      "train_flow_matching_loss": 0.11083984375,
      "train_physics_loss": 0.0
    },
    {
      "step": 17800,
      "train_flow_matching_loss": 0.10205078125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10143224149942398,
      "learning_rate": 5e-05,
      "loss": 0.1339,
      "step": 17810
    },
    {
      "step": 17810,
      "train_flow_matching_loss": 0.2001953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 17810,
      "train_flow_matching_loss": 0.1416015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 17810,
      "train_flow_matching_loss": 0.158203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 17810,
      "train_flow_matching_loss": 0.1376953125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10460410267114639,
      "learning_rate": 5e-05,
      "loss": 0.1321,
      "step": 17820
    },
    {
      "step": 17820,
      "train_flow_matching_loss": 0.1845703125,
      "train_physics_loss": 0.0
    },
    {
      "step": 17820,
      "train_flow_matching_loss": 0.08837890625,
      "train_physics_loss": 0.0
    },
    {
      "step": 17820,
      "train_flow_matching_loss": 0.10302734375,
      "train_physics_loss": 0.0
    },
    {
      "step": 17820,
      "train_flow_matching_loss": 0.1025390625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08973146975040436,
      "learning_rate": 5e-05,
      "loss": 0.1306,
      "step": 17830
    },
    {
      "step": 17830,
      "train_flow_matching_loss": 0.1494140625,
      "train_physics_loss": 0.0
    },
    {
      "step": 17830,
      "train_flow_matching_loss": 0.1533203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 17830,
      "train_flow_matching_loss": 0.1669921875,
      "train_physics_loss": 0.0
    },
    {
      "step": 17830,
      "train_flow_matching_loss": 0.1416015625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11419594287872314,
      "learning_rate": 5e-05,
      "loss": 0.1363,
      "step": 17840
    },
    {
      "step": 17840,
      "train_flow_matching_loss": 0.1591796875,
      "train_physics_loss": 0.0
    },
    {
      "step": 17840,
      "train_flow_matching_loss": 0.1484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 17840,
      "train_flow_matching_loss": 0.08056640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 17840,
      "train_flow_matching_loss": 0.12890625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09302981197834015,
      "learning_rate": 5e-05,
      "loss": 0.1418,
      "step": 17850
    },
    {
      "step": 17850,
      "train_flow_matching_loss": 0.1748046875,
      "train_physics_loss": 0.0
    },
    {
      "step": 17850,
      "train_flow_matching_loss": 0.1025390625,
      "train_physics_loss": 0.0
    },
    {
      "step": 17850,
      "train_flow_matching_loss": 0.1650390625,
      "train_physics_loss": 0.0
    },
    {
      "step": 17850,
      "train_flow_matching_loss": 0.1591796875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.1239318698644638,
      "learning_rate": 5e-05,
      "loss": 0.1309,
      "step": 17860
    },
    {
      "step": 17860,
      "train_flow_matching_loss": 0.1728515625,
      "train_physics_loss": 0.0
    },
    {
      "step": 17860,
      "train_flow_matching_loss": 0.11669921875,
      "train_physics_loss": 0.0
    },
    {
      "step": 17860,
      "train_flow_matching_loss": 0.1669921875,
      "train_physics_loss": 0.0
    },
    {
      "step": 17860,
      "train_flow_matching_loss": 0.1181640625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.07646811753511429,
      "learning_rate": 5e-05,
      "loss": 0.1338,
      "step": 17870
    },
    {
      "step": 17870,
      "train_flow_matching_loss": 0.1416015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 17870,
      "train_flow_matching_loss": 0.1171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 17870,
      "train_flow_matching_loss": 0.1513671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 17870,
      "train_flow_matching_loss": 0.10791015625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09335806965827942,
      "learning_rate": 5e-05,
      "loss": 0.1336,
      "step": 17880
    },
    {
      "step": 17880,
      "train_flow_matching_loss": 0.1806640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 17880,
      "train_flow_matching_loss": 0.1484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 17880,
      "train_flow_matching_loss": 0.09765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 17880,
      "train_flow_matching_loss": 0.1748046875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08875934779644012,
      "learning_rate": 5e-05,
      "loss": 0.1323,
      "step": 17890
    },
    {
      "step": 17890,
      "train_flow_matching_loss": 0.1513671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 17890,
      "train_flow_matching_loss": 0.1923828125,
      "train_physics_loss": 0.0
    },
    {
      "step": 17890,
      "train_flow_matching_loss": 0.142578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 17890,
      "train_flow_matching_loss": 0.1357421875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.07982043921947479,
      "learning_rate": 5e-05,
      "loss": 0.1348,
      "step": 17900
    },
    {
      "step": 17900,
      "train_flow_matching_loss": 0.162109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 17900,
      "train_flow_matching_loss": 0.1611328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 17900,
      "train_flow_matching_loss": 0.1884765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 17900,
      "train_flow_matching_loss": 0.1455078125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09680137038230896,
      "learning_rate": 5e-05,
      "loss": 0.1309,
      "step": 17910
    },
    {
      "step": 17910,
      "train_flow_matching_loss": 0.126953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 17910,
      "train_flow_matching_loss": 0.171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 17910,
      "train_flow_matching_loss": 0.150390625,
      "train_physics_loss": 0.0
    },
    {
      "step": 17910,
      "train_flow_matching_loss": 0.2041015625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09879349172115326,
      "learning_rate": 5e-05,
      "loss": 0.1366,
      "step": 17920
    },
    {
      "step": 17920,
      "train_flow_matching_loss": 0.212890625,
      "train_physics_loss": 0.0
    },
    {
      "step": 17920,
      "train_flow_matching_loss": 0.154296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 17920,
      "train_flow_matching_loss": 0.1513671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 17920,
      "train_flow_matching_loss": 0.1533203125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.0889023020863533,
      "learning_rate": 5e-05,
      "loss": 0.1334,
      "step": 17930
    },
    {
      "step": 17930,
      "train_flow_matching_loss": 0.06103515625,
      "train_physics_loss": 0.32284238934516907
    },
    {
      "step": 17930,
      "train_flow_matching_loss": 0.11376953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 17930,
      "train_flow_matching_loss": 0.259765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 17930,
      "train_flow_matching_loss": 0.1044921875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09735361486673355,
      "learning_rate": 5e-05,
      "loss": 0.1351,
      "step": 17940
    },
    {
      "step": 17940,
      "train_flow_matching_loss": 0.06201171875,
      "train_physics_loss": 0.3428952395915985
    },
    {
      "step": 17940,
      "train_flow_matching_loss": 0.09619140625,
      "train_physics_loss": 0.0
    },
    {
      "step": 17940,
      "train_flow_matching_loss": 0.1435546875,
      "train_physics_loss": 0.0
    },
    {
      "step": 17940,
      "train_flow_matching_loss": 0.1279296875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09791313111782074,
      "learning_rate": 5e-05,
      "loss": 0.1325,
      "step": 17950
    },
    {
      "step": 17950,
      "train_flow_matching_loss": 0.056640625,
      "train_physics_loss": 0.3909042477607727
    },
    {
      "step": 17950,
      "train_flow_matching_loss": 0.1845703125,
      "train_physics_loss": 0.0
    },
    {
      "step": 17950,
      "train_flow_matching_loss": 0.1689453125,
      "train_physics_loss": 0.0
    },
    {
      "step": 17950,
      "train_flow_matching_loss": 0.2431640625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09259840101003647,
      "learning_rate": 5e-05,
      "loss": 0.1334,
      "step": 17960
    },
    {
      "step": 17960,
      "train_flow_matching_loss": 0.0279541015625,
      "train_physics_loss": 0.3104954659938812
    },
    {
      "step": 17960,
      "train_flow_matching_loss": 0.1083984375,
      "train_physics_loss": 0.0
    },
    {
      "step": 17960,
      "train_flow_matching_loss": 0.11376953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 17960,
      "train_flow_matching_loss": 0.1533203125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11520138382911682,
      "learning_rate": 5e-05,
      "loss": 0.1356,
      "step": 17970
    },
    {
      "step": 17970,
      "train_flow_matching_loss": 0.050537109375,
      "train_physics_loss": 0.3368012309074402
    },
    {
      "step": 17970,
      "train_flow_matching_loss": 0.1611328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 17970,
      "train_flow_matching_loss": 0.1728515625,
      "train_physics_loss": 0.0
    },
    {
      "step": 17970,
      "train_flow_matching_loss": 0.1259765625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 4.0141425132751465,
      "learning_rate": 5e-05,
      "loss": 0.1297,
      "step": 17980
    },
    {
      "step": 17980,
      "train_flow_matching_loss": 0.054931640625,
      "train_physics_loss": 0.30798977613449097
    },
    {
      "step": 17980,
      "train_flow_matching_loss": 0.1357421875,
      "train_physics_loss": 0.0
    },
    {
      "step": 17980,
      "train_flow_matching_loss": 0.1259765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 17980,
      "train_flow_matching_loss": 0.2158203125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10565757006406784,
      "learning_rate": 5e-05,
      "loss": 0.1337,
      "step": 17990
    },
    {
      "step": 17990,
      "train_flow_matching_loss": 0.056396484375,
      "train_physics_loss": 0.30095967650413513
    },
    {
      "step": 17990,
      "train_flow_matching_loss": 0.169921875,
      "train_physics_loss": 0.0
    },
    {
      "step": 17990,
      "train_flow_matching_loss": 0.1650390625,
      "train_physics_loss": 0.0
    },
    {
      "step": 17990,
      "train_flow_matching_loss": 0.1376953125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09401781111955643,
      "learning_rate": 5e-05,
      "loss": 0.1361,
      "step": 18000
    },
    {
      "step": 18000,
      "train_flow_matching_loss": 0.0810546875,
      "train_physics_loss": 0.330315500497818
    },
    {
      "step": 18000,
      "train_flow_matching_loss": 0.12255859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 18000,
      "train_flow_matching_loss": 0.1103515625,
      "train_physics_loss": 0.0
    },
    {
      "step": 18000,
      "train_flow_matching_loss": 0.1513671875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.12226025760173798,
      "learning_rate": 5e-05,
      "loss": 0.1315,
      "step": 18010
    },
    {
      "step": 18010,
      "train_flow_matching_loss": 0.1416015625,
      "train_physics_loss": 0.26987364888191223
    },
    {
      "step": 18010,
      "train_flow_matching_loss": 0.10400390625,
      "train_physics_loss": 0.3640320897102356
    },
    {
      "step": 18010,
      "train_flow_matching_loss": 0.208984375,
      "train_physics_loss": 0.0
    },
    {
      "step": 18010,
      "train_flow_matching_loss": 0.1875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.0791301354765892,
      "learning_rate": 5e-05,
      "loss": 0.1363,
      "step": 18020
    },
    {
      "step": 18020,
      "train_flow_matching_loss": 0.051513671875,
      "train_physics_loss": 0.29508012533187866
    },
    {
      "step": 18020,
      "train_flow_matching_loss": 0.046630859375,
      "train_physics_loss": 0.3230777084827423
    },
    {
      "step": 18020,
      "train_flow_matching_loss": 0.1279296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 18020,
      "train_flow_matching_loss": 0.134765625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10205690562725067,
      "learning_rate": 5e-05,
      "loss": 0.1354,
      "step": 18030
    },
    {
      "step": 18030,
      "train_flow_matching_loss": 0.03857421875,
      "train_physics_loss": 0.31989341974258423
    },
    {
      "step": 18030,
      "train_flow_matching_loss": 0.06591796875,
      "train_physics_loss": 0.33373454213142395
    },
    {
      "step": 18030,
      "train_flow_matching_loss": 0.1494140625,
      "train_physics_loss": 0.0
    },
    {
      "step": 18030,
      "train_flow_matching_loss": 0.1494140625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.1398901343345642,
      "learning_rate": 5e-05,
      "loss": 0.1307,
      "step": 18040
    },
    {
      "step": 18040,
      "train_flow_matching_loss": 0.045166015625,
      "train_physics_loss": 0.3402370810508728
    },
    {
      "step": 18040,
      "train_flow_matching_loss": 0.06201171875,
      "train_physics_loss": 0.31398147344589233
    },
    {
      "step": 18040,
      "train_flow_matching_loss": 0.140625,
      "train_physics_loss": 0.0
    },
    {
      "step": 18040,
      "train_flow_matching_loss": 0.10693359375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11763478815555573,
      "learning_rate": 5e-05,
      "loss": 0.1357,
      "step": 18050
    },
    {
      "step": 18050,
      "train_flow_matching_loss": 0.07568359375,
      "train_physics_loss": 0.3460127115249634
    },
    {
      "step": 18050,
      "train_flow_matching_loss": 0.05224609375,
      "train_physics_loss": 0.37353721261024475
    },
    {
      "step": 18050,
      "train_flow_matching_loss": 0.07666015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 18050,
      "train_flow_matching_loss": 0.12451171875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08421345800161362,
      "learning_rate": 5e-05,
      "loss": 0.1356,
      "step": 18060
    },
    {
      "step": 18060,
      "train_flow_matching_loss": 0.033447265625,
      "train_physics_loss": 0.3167850375175476
    },
    {
      "step": 18060,
      "train_flow_matching_loss": 0.052001953125,
      "train_physics_loss": 0.38298365473747253
    },
    {
      "step": 18060,
      "train_flow_matching_loss": 0.07080078125,
      "train_physics_loss": 0.3769381642341614
    },
    {
      "step": 18060,
      "train_flow_matching_loss": 0.1357421875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09289979934692383,
      "learning_rate": 5e-05,
      "loss": 0.1304,
      "step": 18070
    },
    {
      "step": 18070,
      "train_flow_matching_loss": 0.048828125,
      "train_physics_loss": 0.3350852131843567
    },
    {
      "step": 18070,
      "train_flow_matching_loss": 0.0966796875,
      "train_physics_loss": 0.29848819971084595
    },
    {
      "step": 18070,
      "train_flow_matching_loss": 0.10986328125,
      "train_physics_loss": 0.29539331793785095
    },
    {
      "step": 18070,
      "train_flow_matching_loss": 0.126953125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09125763922929764,
      "learning_rate": 5e-05,
      "loss": 0.134,
      "step": 18080
    },
    {
      "step": 18080,
      "train_flow_matching_loss": 0.07275390625,
      "train_physics_loss": 0.5094376802444458
    },
    {
      "step": 18080,
      "train_flow_matching_loss": 0.07275390625,
      "train_physics_loss": 0.3411327600479126
    },
    {
      "step": 18080,
      "train_flow_matching_loss": 0.034423828125,
      "train_physics_loss": 0.3148585259914398
    },
    {
      "step": 18080,
      "train_flow_matching_loss": 0.2099609375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10054974257946014,
      "learning_rate": 5e-05,
      "loss": 0.1296,
      "step": 18090
    },
    {
      "step": 18090,
      "train_flow_matching_loss": 0.1826171875,
      "train_physics_loss": 0.5056754350662231
    },
    {
      "step": 18090,
      "train_flow_matching_loss": 0.05712890625,
      "train_physics_loss": 0.28504490852355957
    },
    {
      "step": 18090,
      "train_flow_matching_loss": 0.032958984375,
      "train_physics_loss": 0.3462924361228943
    },
    {
      "step": 18090,
      "train_flow_matching_loss": 0.2021484375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09241925179958344,
      "learning_rate": 5e-05,
      "loss": 0.129,
      "step": 18100
    },
    {
      "step": 18100,
      "train_flow_matching_loss": 0.07373046875,
      "train_physics_loss": 0.32235580682754517
    },
    {
      "step": 18100,
      "train_flow_matching_loss": 0.06591796875,
      "train_physics_loss": 0.38853731751441956
    },
    {
      "step": 18100,
      "train_flow_matching_loss": 0.041748046875,
      "train_physics_loss": 0.342053085565567
    },
    {
      "step": 18100,
      "train_flow_matching_loss": 0.134765625,
      "train_physics_loss": 0.3371020555496216
    },
    {
      "grad_norm": 0.0926603153347969,
      "learning_rate": 5e-05,
      "loss": 0.1343,
      "step": 18110
    },
    {
      "step": 18110,
      "train_flow_matching_loss": 0.05712890625,
      "train_physics_loss": 0.3391728401184082
    },
    {
      "step": 18110,
      "train_flow_matching_loss": 0.04443359375,
      "train_physics_loss": 0.3564758002758026
    },
    {
      "step": 18110,
      "train_flow_matching_loss": 0.0322265625,
      "train_physics_loss": 0.3072192966938019
    },
    {
      "step": 18110,
      "train_flow_matching_loss": 0.050048828125,
      "train_physics_loss": 0.3404158055782318
    },
    {
      "grad_norm": 0.07922328263521194,
      "learning_rate": 5e-05,
      "loss": 0.1293,
      "step": 18120
    },
    {
      "step": 18120,
      "train_flow_matching_loss": 0.04541015625,
      "train_physics_loss": 0.31478598713874817
    },
    {
      "step": 18120,
      "train_flow_matching_loss": 0.033203125,
      "train_physics_loss": 0.34855836629867554
    },
    {
      "step": 18120,
      "train_flow_matching_loss": 0.017822265625,
      "train_physics_loss": 0.2978883683681488
    },
    {
      "step": 18120,
      "train_flow_matching_loss": 0.11083984375,
      "train_physics_loss": 0.32738280296325684
    },
    {
      "grad_norm": 0.07825953513383865,
      "learning_rate": 5e-05,
      "loss": 0.1267,
      "step": 18130
    },
    {
      "step": 18130,
      "train_flow_matching_loss": 0.04052734375,
      "train_physics_loss": 0.3588046729564667
    },
    {
      "step": 18130,
      "train_flow_matching_loss": 0.048095703125,
      "train_physics_loss": 0.34229719638824463
    },
    {
      "step": 18130,
      "train_flow_matching_loss": 0.03466796875,
      "train_physics_loss": 0.28800222277641296
    },
    {
      "step": 18130,
      "train_flow_matching_loss": 0.041015625,
      "train_physics_loss": 0.32162439823150635
    },
    {
      "grad_norm": 0.11551392078399658,
      "learning_rate": 5e-05,
      "loss": 0.1367,
      "step": 18140
    },
    {
      "step": 18140,
      "train_flow_matching_loss": 0.05126953125,
      "train_physics_loss": 0.3341943621635437
    },
    {
      "step": 18140,
      "train_flow_matching_loss": 0.06884765625,
      "train_physics_loss": 0.3751731216907501
    },
    {
      "step": 18140,
      "train_flow_matching_loss": 0.06103515625,
      "train_physics_loss": 0.3558860421180725
    },
    {
      "step": 18140,
      "train_flow_matching_loss": 0.08837890625,
      "train_physics_loss": 0.37147057056427
    },
    {
      "grad_norm": 0.08297920972108841,
      "learning_rate": 5e-05,
      "loss": 0.1309,
      "step": 18150
    },
    {
      "step": 18150,
      "train_flow_matching_loss": 0.0859375,
      "train_physics_loss": 0.4168417155742645
    },
    {
      "step": 18150,
      "train_flow_matching_loss": 0.035400390625,
      "train_physics_loss": 0.30186179280281067
    },
    {
      "step": 18150,
      "train_flow_matching_loss": 0.036865234375,
      "train_physics_loss": 0.33605584502220154
    },
    {
      "step": 18150,
      "train_flow_matching_loss": 0.07080078125,
      "train_physics_loss": 0.3200015723705292
    },
    {
      "grad_norm": 0.08606477826833725,
      "learning_rate": 5e-05,
      "loss": 0.1341,
      "step": 18160
    },
    {
      "step": 18160,
      "train_flow_matching_loss": 0.09765625,
      "train_physics_loss": 0.36201533675193787
    },
    {
      "step": 18160,
      "train_flow_matching_loss": 0.04052734375,
      "train_physics_loss": 0.3024548292160034
    },
    {
      "step": 18160,
      "train_flow_matching_loss": 0.0245361328125,
      "train_physics_loss": 0.30732613801956177
    },
    {
      "step": 18160,
      "train_flow_matching_loss": 0.046142578125,
      "train_physics_loss": 0.2880156636238098
    },
    {
      "grad_norm": 0.13034169375896454,
      "learning_rate": 5e-05,
      "loss": 0.1276,
      "step": 18170
    },
    {
      "step": 18170,
      "train_flow_matching_loss": 0.0634765625,
      "train_physics_loss": 0.3751235008239746
    },
    {
      "step": 18170,
      "train_flow_matching_loss": 0.036865234375,
      "train_physics_loss": 0.32883697748184204
    },
    {
      "step": 18170,
      "train_flow_matching_loss": 0.024169921875,
      "train_physics_loss": 0.3384617865085602
    },
    {
      "step": 18170,
      "train_flow_matching_loss": 0.044677734375,
      "train_physics_loss": 0.27261972427368164
    },
    {
      "grad_norm": 0.13166411221027374,
      "learning_rate": 5e-05,
      "loss": 0.1327,
      "step": 18180
    },
    {
      "step": 18180,
      "train_flow_matching_loss": 0.033447265625,
      "train_physics_loss": 0.302494615316391
    },
    {
      "step": 18180,
      "train_flow_matching_loss": 0.05078125,
      "train_physics_loss": 0.35775452852249146
    },
    {
      "step": 18180,
      "train_flow_matching_loss": 0.037109375,
      "train_physics_loss": 0.3126569390296936
    },
    {
      "step": 18180,
      "train_flow_matching_loss": 0.0673828125,
      "train_physics_loss": 0.32498371601104736
    },
    {
      "grad_norm": 0.11533520370721817,
      "learning_rate": 5e-05,
      "loss": 0.1304,
      "step": 18190
    },
    {
      "step": 18190,
      "train_flow_matching_loss": 0.03759765625,
      "train_physics_loss": 0.32964399456977844
    },
    {
      "step": 18190,
      "train_flow_matching_loss": 0.0283203125,
      "train_physics_loss": 0.33010977506637573
    },
    {
      "step": 18190,
      "train_flow_matching_loss": 0.078125,
      "train_physics_loss": 0.34053459763526917
    },
    {
      "step": 18190,
      "train_flow_matching_loss": 0.05078125,
      "train_physics_loss": 0.34312617778778076
    },
    {
      "grad_norm": 0.12560246884822845,
      "learning_rate": 5e-05,
      "loss": 0.1255,
      "step": 18200
    },
    {
      "step": 18200,
      "train_flow_matching_loss": 0.0311279296875,
      "train_physics_loss": 0.30415478348731995
    },
    {
      "step": 18200,
      "train_flow_matching_loss": 0.0546875,
      "train_physics_loss": 0.33161064982414246
    },
    {
      "step": 18200,
      "train_flow_matching_loss": 0.0291748046875,
      "train_physics_loss": 0.3132054805755615
    },
    {
      "step": 18200,
      "train_flow_matching_loss": 0.078125,
      "train_physics_loss": 0.31178921461105347
    },
    {
      "grad_norm": 0.08022334426641464,
      "learning_rate": 5e-05,
      "loss": 0.1257,
      "step": 18210
    },
    {
      "step": 18210,
      "train_flow_matching_loss": 0.06884765625,
      "train_physics_loss": 0.3210390508174896
    },
    {
      "step": 18210,
      "train_flow_matching_loss": 0.0966796875,
      "train_physics_loss": 0.32944804430007935
    },
    {
      "step": 18210,
      "train_flow_matching_loss": 0.032470703125,
      "train_physics_loss": 0.33637863397598267
    },
    {
      "step": 18210,
      "train_flow_matching_loss": 0.0693359375,
      "train_physics_loss": 0.3575533628463745
    },
    {
      "grad_norm": 0.0667281523346901,
      "learning_rate": 5e-05,
      "loss": 0.1308,
      "step": 18220
    },
    {
      "step": 18220,
      "train_flow_matching_loss": 0.029052734375,
      "train_physics_loss": 0.4062136113643646
    },
    {
      "step": 18220,
      "train_flow_matching_loss": 0.07861328125,
      "train_physics_loss": 0.3563290536403656
    },
    {
      "step": 18220,
      "train_flow_matching_loss": 0.050537109375,
      "train_physics_loss": 0.337047815322876
    },
    {
      "step": 18220,
      "train_flow_matching_loss": 0.04296875,
      "train_physics_loss": 0.3418194651603699
    },
    {
      "grad_norm": 0.0975741520524025,
      "learning_rate": 5e-05,
      "loss": 0.1325,
      "step": 18230
    },
    {
      "step": 18230,
      "train_flow_matching_loss": 0.03564453125,
      "train_physics_loss": 0.39680802822113037
    },
    {
      "step": 18230,
      "train_flow_matching_loss": 0.031982421875,
      "train_physics_loss": 0.2901562750339508
    },
    {
      "step": 18230,
      "train_flow_matching_loss": 0.0267333984375,
      "train_physics_loss": 0.2476121485233307
    },
    {
      "step": 18230,
      "train_flow_matching_loss": 0.037841796875,
      "train_physics_loss": 0.34006425738334656
    },
    {
      "grad_norm": 0.11172191798686981,
      "learning_rate": 5e-05,
      "loss": 0.1335,
      "step": 18240
    },
    {
      "step": 18240,
      "train_flow_matching_loss": 0.126953125,
      "train_physics_loss": 0.33112096786499023
    },
    {
      "step": 18240,
      "train_flow_matching_loss": 0.059814453125,
      "train_physics_loss": 0.34354984760284424
    },
    {
      "step": 18240,
      "train_flow_matching_loss": 0.07275390625,
      "train_physics_loss": 0.34810101985931396
    },
    {
      "step": 18240,
      "train_flow_matching_loss": 0.04638671875,
      "train_physics_loss": 0.3651706874370575
    },
    {
      "grad_norm": 0.08275146037340164,
      "learning_rate": 5e-05,
      "loss": 0.1343,
      "step": 18250
    },
    {
      "step": 18250,
      "train_flow_matching_loss": 0.04833984375,
      "train_physics_loss": 0.2974013388156891
    },
    {
      "step": 18250,
      "train_flow_matching_loss": 0.052490234375,
      "train_physics_loss": 0.34609630703926086
    },
    {
      "step": 18250,
      "train_flow_matching_loss": 0.0732421875,
      "train_physics_loss": 0.36346596479415894
    },
    {
      "step": 18250,
      "train_flow_matching_loss": 0.0693359375,
      "train_physics_loss": 0.3453141152858734
    },
    {
      "grad_norm": 0.10447540879249573,
      "learning_rate": 5e-05,
      "loss": 0.1266,
      "step": 18260
    },
    {
      "step": 18260,
      "train_flow_matching_loss": 0.048095703125,
      "train_physics_loss": 0.33112767338752747
    },
    {
      "step": 18260,
      "train_flow_matching_loss": 0.046630859375,
      "train_physics_loss": 0.3167753517627716
    },
    {
      "step": 18260,
      "train_flow_matching_loss": 0.036865234375,
      "train_physics_loss": 0.35144108533859253
    },
    {
      "step": 18260,
      "train_flow_matching_loss": 0.0810546875,
      "train_physics_loss": 0.31314000487327576
    },
    {
      "grad_norm": 0.18721935153007507,
      "learning_rate": 5e-05,
      "loss": 0.1329,
      "step": 18270
    },
    {
      "step": 18270,
      "train_flow_matching_loss": 0.052978515625,
      "train_physics_loss": 0.3226096034049988
    },
    {
      "step": 18270,
      "train_flow_matching_loss": 0.07177734375,
      "train_physics_loss": 0.0
    },
    {
      "step": 18270,
      "train_flow_matching_loss": 0.0262451171875,
      "train_physics_loss": 0.30079954862594604
    },
    {
      "step": 18270,
      "train_flow_matching_loss": 0.0546875,
      "train_physics_loss": 0.3140357732772827
    },
    {
      "grad_norm": 0.2424168884754181,
      "learning_rate": 5e-05,
      "loss": 0.1277,
      "step": 18280
    },
    {
      "step": 18280,
      "train_flow_matching_loss": 0.053466796875,
      "train_physics_loss": 0.2770399749279022
    },
    {
      "step": 18280,
      "train_flow_matching_loss": 0.11083984375,
      "train_physics_loss": 0.0
    },
    {
      "step": 18280,
      "train_flow_matching_loss": 0.0634765625,
      "train_physics_loss": 0.34021300077438354
    },
    {
      "step": 18280,
      "train_flow_matching_loss": 0.09375,
      "train_physics_loss": 0.46880990266799927
    },
    {
      "grad_norm": 0.23120538890361786,
      "learning_rate": 5e-05,
      "loss": 0.1356,
      "step": 18290
    },
    {
      "step": 18290,
      "train_flow_matching_loss": 0.058837890625,
      "train_physics_loss": 0.33638277649879456
    },
    {
      "step": 18290,
      "train_flow_matching_loss": 0.08984375,
      "train_physics_loss": 0.0
    },
    {
      "step": 18290,
      "train_flow_matching_loss": 0.1181640625,
      "train_physics_loss": 0.3812228739261627
    },
    {
      "step": 18290,
      "train_flow_matching_loss": 0.0250244140625,
      "train_physics_loss": 0.30806922912597656
    },
    {
      "grad_norm": 0.09055855870246887,
      "learning_rate": 5e-05,
      "loss": 0.1329,
      "step": 18300
    },
    {
      "step": 18300,
      "train_flow_matching_loss": 0.03955078125,
      "train_physics_loss": 0.27464020252227783
    },
    {
      "step": 18300,
      "train_flow_matching_loss": 0.130859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 18300,
      "train_flow_matching_loss": 0.056396484375,
      "train_physics_loss": 0.3532674312591553
    },
    {
      "step": 18300,
      "train_flow_matching_loss": 0.0703125,
      "train_physics_loss": 0.3174883723258972
    },
    {
      "grad_norm": 0.08472588658332825,
      "learning_rate": 5e-05,
      "loss": 0.1276,
      "step": 18310
    },
    {
      "step": 18310,
      "train_flow_matching_loss": 0.04248046875,
      "train_physics_loss": 0.3145330548286438
    },
    {
      "step": 18310,
      "train_flow_matching_loss": 0.126953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 18310,
      "train_flow_matching_loss": 0.059326171875,
      "train_physics_loss": 0.32495182752609253
    },
    {
      "step": 18310,
      "train_flow_matching_loss": 0.028564453125,
      "train_physics_loss": 0.3268468976020813
    },
    {
      "grad_norm": 0.42268040776252747,
      "learning_rate": 5e-05,
      "loss": 0.1258,
      "step": 18320
    },
    {
      "step": 18320,
      "train_flow_matching_loss": 0.06396484375,
      "train_physics_loss": 0.32521939277648926
    },
    {
      "step": 18320,
      "train_flow_matching_loss": 0.142578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 18320,
      "train_flow_matching_loss": 0.1259765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 18320,
      "train_flow_matching_loss": 0.042236328125,
      "train_physics_loss": 0.37148308753967285
    },
    {
      "grad_norm": 0.06807605177164078,
      "learning_rate": 5e-05,
      "loss": 0.1292,
      "step": 18330
    },
    {
      "step": 18330,
      "train_flow_matching_loss": 0.07568359375,
      "train_physics_loss": 0.3781163990497589
    },
    {
      "step": 18330,
      "train_flow_matching_loss": 0.10693359375,
      "train_physics_loss": 0.0
    },
    {
      "step": 18330,
      "train_flow_matching_loss": 0.11962890625,
      "train_physics_loss": 0.0
    },
    {
      "step": 18330,
      "train_flow_matching_loss": 0.05712890625,
      "train_physics_loss": 0.3210095167160034
    },
    {
      "grad_norm": 0.07890678197145462,
      "learning_rate": 5e-05,
      "loss": 0.1266,
      "step": 18340
    },
    {
      "step": 18340,
      "train_flow_matching_loss": 0.08056640625,
      "train_physics_loss": 0.36592310667037964
    },
    {
      "step": 18340,
      "train_flow_matching_loss": 0.1162109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 18340,
      "train_flow_matching_loss": 0.12158203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 18340,
      "train_flow_matching_loss": 0.08154296875,
      "train_physics_loss": 0.302435964345932
    },
    {
      "grad_norm": 0.09476926922798157,
      "learning_rate": 5e-05,
      "loss": 0.1244,
      "step": 18350
    },
    {
      "step": 18350,
      "train_flow_matching_loss": 0.07763671875,
      "train_physics_loss": 0.25088855624198914
    },
    {
      "step": 18350,
      "train_flow_matching_loss": 0.275390625,
      "train_physics_loss": 0.0
    },
    {
      "step": 18350,
      "train_flow_matching_loss": 0.12890625,
      "train_physics_loss": 0.0
    },
    {
      "step": 18350,
      "train_flow_matching_loss": 0.11865234375,
      "train_physics_loss": 0.345683753490448
    },
    {
      "grad_norm": 0.14190445840358734,
      "learning_rate": 5e-05,
      "loss": 0.1287,
      "step": 18360
    },
    {
      "step": 18360,
      "train_flow_matching_loss": 0.0771484375,
      "train_physics_loss": 0.34731435775756836
    },
    {
      "step": 18360,
      "train_flow_matching_loss": 0.11181640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 18360,
      "train_flow_matching_loss": 0.130859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 18360,
      "train_flow_matching_loss": 0.169921875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10755493491888046,
      "learning_rate": 5e-05,
      "loss": 0.1403,
      "step": 18370
    },
    {
      "step": 18370,
      "train_flow_matching_loss": 0.03466796875,
      "train_physics_loss": 0.3753359317779541
    },
    {
      "step": 18370,
      "train_flow_matching_loss": 0.2099609375,
      "train_physics_loss": 0.0
    },
    {
      "step": 18370,
      "train_flow_matching_loss": 0.10693359375,
      "train_physics_loss": 0.0
    },
    {
      "step": 18370,
      "train_flow_matching_loss": 0.146484375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08846788853406906,
      "learning_rate": 5e-05,
      "loss": 0.1286,
      "step": 18380
    },
    {
      "step": 18380,
      "train_flow_matching_loss": 0.045654296875,
      "train_physics_loss": 0.2722175121307373
    },
    {
      "step": 18380,
      "train_flow_matching_loss": 0.10400390625,
      "train_physics_loss": 0.0
    },
    {
      "step": 18380,
      "train_flow_matching_loss": 0.12255859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 18380,
      "train_flow_matching_loss": 0.154296875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10588865727186203,
      "learning_rate": 5e-05,
      "loss": 0.1236,
      "step": 18390
    },
    {
      "step": 18390,
      "train_flow_matching_loss": 0.0634765625,
      "train_physics_loss": 0.3337666392326355
    },
    {
      "step": 18390,
      "train_flow_matching_loss": 0.11181640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 18390,
      "train_flow_matching_loss": 0.12353515625,
      "train_physics_loss": 0.0
    },
    {
      "step": 18390,
      "train_flow_matching_loss": 0.1259765625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.06917351484298706,
      "learning_rate": 5e-05,
      "loss": 0.129,
      "step": 18400
    },
    {
      "step": 18400,
      "train_flow_matching_loss": 0.09716796875,
      "train_physics_loss": 0.0
    },
    {
      "step": 18400,
      "train_flow_matching_loss": 0.287109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 18400,
      "train_flow_matching_loss": 0.12451171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 18400,
      "train_flow_matching_loss": 0.146484375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10735209286212921,
      "learning_rate": 5e-05,
      "loss": 0.1321,
      "step": 18410
    },
    {
      "step": 18410,
      "train_flow_matching_loss": 0.1005859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 18410,
      "train_flow_matching_loss": 0.10888671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 18410,
      "train_flow_matching_loss": 0.1767578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 18410,
      "train_flow_matching_loss": 0.08349609375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.0799664705991745,
      "learning_rate": 5e-05,
      "loss": 0.1302,
      "step": 18420
    },
    {
      "step": 18420,
      "train_flow_matching_loss": 0.0634765625,
      "train_physics_loss": 0.37940990924835205
    },
    {
      "step": 18420,
      "train_flow_matching_loss": 0.1416015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 18420,
      "train_flow_matching_loss": 0.11669921875,
      "train_physics_loss": 0.0
    },
    {
      "step": 18420,
      "train_flow_matching_loss": 0.1162109375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11626937985420227,
      "learning_rate": 5e-05,
      "loss": 0.1301,
      "step": 18430
    },
    {
      "step": 18430,
      "train_flow_matching_loss": 0.0673828125,
      "train_physics_loss": 0.49627912044525146
    },
    {
      "step": 18430,
      "train_flow_matching_loss": 0.1650390625,
      "train_physics_loss": 0.0
    },
    {
      "step": 18430,
      "train_flow_matching_loss": 0.10546875,
      "train_physics_loss": 0.0
    },
    {
      "step": 18430,
      "train_flow_matching_loss": 0.11328125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.15231822431087494,
      "learning_rate": 5e-05,
      "loss": 0.1348,
      "step": 18440
    },
    {
      "step": 18440,
      "train_flow_matching_loss": 0.0673828125,
      "train_physics_loss": 0.3353252112865448
    },
    {
      "step": 18440,
      "train_flow_matching_loss": 0.09619140625,
      "train_physics_loss": 0.0
    },
    {
      "step": 18440,
      "train_flow_matching_loss": 0.1337890625,
      "train_physics_loss": 0.0
    },
    {
      "step": 18440,
      "train_flow_matching_loss": 0.1015625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08600103110074997,
      "learning_rate": 5e-05,
      "loss": 0.1272,
      "step": 18450
    },
    {
      "step": 18450,
      "train_flow_matching_loss": 0.169921875,
      "train_physics_loss": 0.0
    },
    {
      "step": 18450,
      "train_flow_matching_loss": 0.2373046875,
      "train_physics_loss": 0.0
    },
    {
      "step": 18450,
      "train_flow_matching_loss": 0.06787109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 18450,
      "train_flow_matching_loss": 0.04931640625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.13504375517368317,
      "learning_rate": 5e-05,
      "loss": 0.1329,
      "step": 18460
    },
    {
      "step": 18460,
      "train_flow_matching_loss": 0.09521484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 18460,
      "train_flow_matching_loss": 0.150390625,
      "train_physics_loss": 0.0
    },
    {
      "step": 18460,
      "train_flow_matching_loss": 0.10107421875,
      "train_physics_loss": 0.0
    },
    {
      "step": 18460,
      "train_flow_matching_loss": 0.1494140625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08704038709402084,
      "learning_rate": 5e-05,
      "loss": 0.1288,
      "step": 18470
    },
    {
      "step": 18470,
      "train_flow_matching_loss": 0.146484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 18470,
      "train_flow_matching_loss": 0.1845703125,
      "train_physics_loss": 0.0
    },
    {
      "step": 18470,
      "train_flow_matching_loss": 0.11181640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 18470,
      "train_flow_matching_loss": 0.1318359375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.43180185556411743,
      "learning_rate": 5e-05,
      "loss": 0.1281,
      "step": 18480
    },
    {
      "step": 18480,
      "train_flow_matching_loss": 0.1572265625,
      "train_physics_loss": 0.0
    },
    {
      "step": 18480,
      "train_flow_matching_loss": 0.11376953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 18480,
      "train_flow_matching_loss": 0.125,
      "train_physics_loss": 0.0
    },
    {
      "step": 18480,
      "train_flow_matching_loss": 0.1123046875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08936145901679993,
      "learning_rate": 5e-05,
      "loss": 0.133,
      "step": 18490
    },
    {
      "step": 18490,
      "train_flow_matching_loss": 0.1103515625,
      "train_physics_loss": 0.0
    },
    {
      "step": 18490,
      "train_flow_matching_loss": 0.142578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 18490,
      "train_flow_matching_loss": 0.1337890625,
      "train_physics_loss": 0.0
    },
    {
      "step": 18490,
      "train_flow_matching_loss": 0.1357421875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10727988183498383,
      "learning_rate": 5e-05,
      "loss": 0.1273,
      "step": 18500
    },
    {
      "step": 18500,
      "train_flow_matching_loss": 0.1103515625,
      "train_physics_loss": 0.0
    },
    {
      "step": 18500,
      "train_flow_matching_loss": 0.1513671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 18500,
      "train_flow_matching_loss": 0.11767578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 18500,
      "train_flow_matching_loss": 0.154296875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08504520356655121,
      "learning_rate": 5e-05,
      "loss": 0.1327,
      "step": 18510
    },
    {
      "step": 18510,
      "train_flow_matching_loss": 0.10400390625,
      "train_physics_loss": 0.0
    },
    {
      "step": 18510,
      "train_flow_matching_loss": 0.1318359375,
      "train_physics_loss": 0.0
    },
    {
      "step": 18510,
      "train_flow_matching_loss": 0.1376953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 18510,
      "train_flow_matching_loss": 0.1005859375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09852761030197144,
      "learning_rate": 5e-05,
      "loss": 0.1314,
      "step": 18520
    },
    {
      "step": 18520,
      "train_flow_matching_loss": 0.126953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 18520,
      "train_flow_matching_loss": 0.130859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 18520,
      "train_flow_matching_loss": 0.1611328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 18520,
      "train_flow_matching_loss": 0.158203125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08997298032045364,
      "learning_rate": 5e-05,
      "loss": 0.1303,
      "step": 18530
    },
    {
      "step": 18530,
      "train_flow_matching_loss": 0.1298828125,
      "train_physics_loss": 0.0
    },
    {
      "step": 18530,
      "train_flow_matching_loss": 0.10595703125,
      "train_physics_loss": 0.0
    },
    {
      "step": 18530,
      "train_flow_matching_loss": 0.1015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 18530,
      "train_flow_matching_loss": 0.059814453125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08902885764837265,
      "learning_rate": 5e-05,
      "loss": 0.1379,
      "step": 18540
    },
    {
      "step": 18540,
      "train_flow_matching_loss": 0.142578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 18540,
      "train_flow_matching_loss": 0.1083984375,
      "train_physics_loss": 0.0
    },
    {
      "step": 18540,
      "train_flow_matching_loss": 0.1171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 18540,
      "train_flow_matching_loss": 0.10888671875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.07513929158449173,
      "learning_rate": 5e-05,
      "loss": 0.1273,
      "step": 18550
    },
    {
      "step": 18550,
      "train_flow_matching_loss": 0.1728515625,
      "train_physics_loss": 0.0
    },
    {
      "step": 18550,
      "train_flow_matching_loss": 0.1396484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 18550,
      "train_flow_matching_loss": 0.06298828125,
      "train_physics_loss": 0.0
    },
    {
      "step": 18550,
      "train_flow_matching_loss": 0.1298828125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.12164044380187988,
      "learning_rate": 5e-05,
      "loss": 0.1323,
      "step": 18560
    },
    {
      "step": 18560,
      "train_flow_matching_loss": 0.1259765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 18560,
      "train_flow_matching_loss": 0.08837890625,
      "train_physics_loss": 0.0
    },
    {
      "step": 18560,
      "train_flow_matching_loss": 0.181640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 18560,
      "train_flow_matching_loss": 0.1708984375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09364259988069534,
      "learning_rate": 5e-05,
      "loss": 0.1328,
      "step": 18570
    },
    {
      "step": 18570,
      "train_flow_matching_loss": 0.06982421875,
      "train_physics_loss": 0.0
    },
    {
      "step": 18570,
      "train_flow_matching_loss": 0.0869140625,
      "train_physics_loss": 0.0
    },
    {
      "step": 18570,
      "train_flow_matching_loss": 0.154296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 18570,
      "train_flow_matching_loss": 0.06298828125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08536321669816971,
      "learning_rate": 5e-05,
      "loss": 0.1267,
      "step": 18580
    },
    {
      "step": 18580,
      "train_flow_matching_loss": 0.212890625,
      "train_physics_loss": 0.0
    },
    {
      "step": 18580,
      "train_flow_matching_loss": 0.1328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 18580,
      "train_flow_matching_loss": 0.06103515625,
      "train_physics_loss": 0.3385253846645355
    },
    {
      "step": 18580,
      "train_flow_matching_loss": 0.09228515625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09906045347452164,
      "learning_rate": 5e-05,
      "loss": 0.1358,
      "step": 18590
    },
    {
      "step": 18590,
      "train_flow_matching_loss": 0.1240234375,
      "train_physics_loss": 0.0
    },
    {
      "step": 18590,
      "train_flow_matching_loss": 0.171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 18590,
      "train_flow_matching_loss": 0.04443359375,
      "train_physics_loss": 0.3356209099292755
    },
    {
      "step": 18590,
      "train_flow_matching_loss": 0.1533203125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09028887748718262,
      "learning_rate": 5e-05,
      "loss": 0.1349,
      "step": 18600
    },
    {
      "step": 18600,
      "train_flow_matching_loss": 0.11328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 18600,
      "train_flow_matching_loss": 0.1884765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 18600,
      "train_flow_matching_loss": 0.06787109375,
      "train_physics_loss": 0.35776689648628235
    },
    {
      "step": 18600,
      "train_flow_matching_loss": 0.08984375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.07905478030443192,
      "learning_rate": 5e-05,
      "loss": 0.1322,
      "step": 18610
    },
    {
      "step": 18610,
      "train_flow_matching_loss": 0.1708984375,
      "train_physics_loss": 0.0
    },
    {
      "step": 18610,
      "train_flow_matching_loss": 0.11328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 18610,
      "train_flow_matching_loss": 0.0693359375,
      "train_physics_loss": 0.37511491775512695
    },
    {
      "step": 18610,
      "train_flow_matching_loss": 0.1376953125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09701826423406601,
      "learning_rate": 5e-05,
      "loss": 0.1247,
      "step": 18620
    },
    {
      "step": 18620,
      "train_flow_matching_loss": 0.07177734375,
      "train_physics_loss": 0.0
    },
    {
      "step": 18620,
      "train_flow_matching_loss": 0.10107421875,
      "train_physics_loss": 0.0
    },
    {
      "step": 18620,
      "train_flow_matching_loss": 0.046142578125,
      "train_physics_loss": 0.30544671416282654
    },
    {
      "step": 18620,
      "train_flow_matching_loss": 0.10009765625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09038685262203217,
      "learning_rate": 5e-05,
      "loss": 0.1332,
      "step": 18630
    },
    {
      "step": 18630,
      "train_flow_matching_loss": 0.1337890625,
      "train_physics_loss": 0.0
    },
    {
      "step": 18630,
      "train_flow_matching_loss": 0.1279296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 18630,
      "train_flow_matching_loss": 0.05712890625,
      "train_physics_loss": 0.34902989864349365
    },
    {
      "step": 18630,
      "train_flow_matching_loss": 0.1494140625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10533542931079865,
      "learning_rate": 5e-05,
      "loss": 0.1339,
      "step": 18640
    },
    {
      "step": 18640,
      "train_flow_matching_loss": 0.0859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 18640,
      "train_flow_matching_loss": 0.1064453125,
      "train_physics_loss": 0.0
    },
    {
      "step": 18640,
      "train_flow_matching_loss": 0.05810546875,
      "train_physics_loss": 0.4223947525024414
    },
    {
      "step": 18640,
      "train_flow_matching_loss": 0.1318359375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08129963278770447,
      "learning_rate": 5e-05,
      "loss": 0.1282,
      "step": 18650
    },
    {
      "step": 18650,
      "train_flow_matching_loss": 0.1513671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 18650,
      "train_flow_matching_loss": 0.09814453125,
      "train_physics_loss": 0.0
    },
    {
      "step": 18650,
      "train_flow_matching_loss": 0.040283203125,
      "train_physics_loss": 0.403962105512619
    },
    {
      "step": 18650,
      "train_flow_matching_loss": 0.1259765625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08671332895755768,
      "learning_rate": 5e-05,
      "loss": 0.132,
      "step": 18660
    },
    {
      "step": 18660,
      "train_flow_matching_loss": 0.1513671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 18660,
      "train_flow_matching_loss": 0.11376953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 18660,
      "train_flow_matching_loss": 0.05517578125,
      "train_physics_loss": 0.35477229952812195
    },
    {
      "step": 18660,
      "train_flow_matching_loss": 0.1416015625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.3163650929927826,
      "learning_rate": 5e-05,
      "loss": 0.141,
      "step": 18670
    },
    {
      "step": 18670,
      "train_flow_matching_loss": 0.1376953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 18670,
      "train_flow_matching_loss": 0.130859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 18670,
      "train_flow_matching_loss": 0.07421875,
      "train_physics_loss": 0.34135034680366516
    },
    {
      "step": 18670,
      "train_flow_matching_loss": 0.12255859375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.28835529088974,
      "learning_rate": 5e-05,
      "loss": 0.13,
      "step": 18680
    },
    {
      "step": 18680,
      "train_flow_matching_loss": 0.162109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 18680,
      "train_flow_matching_loss": 0.1259765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 18680,
      "train_flow_matching_loss": 0.0927734375,
      "train_physics_loss": 0.0
    },
    {
      "step": 18680,
      "train_flow_matching_loss": 0.1796875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.13702280819416046,
      "learning_rate": 5e-05,
      "loss": 0.1346,
      "step": 18690
    },
    {
      "step": 18690,
      "train_flow_matching_loss": 0.142578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 18690,
      "train_flow_matching_loss": 0.205078125,
      "train_physics_loss": 0.0
    },
    {
      "step": 18690,
      "train_flow_matching_loss": 0.05810546875,
      "train_physics_loss": 0.3687276840209961
    },
    {
      "step": 18690,
      "train_flow_matching_loss": 0.126953125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09481809288263321,
      "learning_rate": 5e-05,
      "loss": 0.1301,
      "step": 18700
    },
    {
      "step": 18700,
      "train_flow_matching_loss": 0.0693359375,
      "train_physics_loss": 0.39195486903190613
    },
    {
      "step": 18700,
      "train_flow_matching_loss": 0.12158203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 18700,
      "train_flow_matching_loss": 0.11279296875,
      "train_physics_loss": 0.4397767186164856
    },
    {
      "step": 18700,
      "train_flow_matching_loss": 0.1416015625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10089970380067825,
      "learning_rate": 5e-05,
      "loss": 0.1316,
      "step": 18710
    },
    {
      "step": 18710,
      "train_flow_matching_loss": 0.04443359375,
      "train_physics_loss": 0.3121899664402008
    },
    {
      "step": 18710,
      "train_flow_matching_loss": 0.1640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 18710,
      "train_flow_matching_loss": 0.03662109375,
      "train_physics_loss": 0.3387618660926819
    },
    {
      "step": 18710,
      "train_flow_matching_loss": 0.11767578125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11816253513097763,
      "learning_rate": 5e-05,
      "loss": 0.1303,
      "step": 18720
    },
    {
      "step": 18720,
      "train_flow_matching_loss": 0.043212890625,
      "train_physics_loss": 0.4720318913459778
    },
    {
      "step": 18720,
      "train_flow_matching_loss": 0.236328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 18720,
      "train_flow_matching_loss": 0.0791015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 18720,
      "train_flow_matching_loss": 0.1630859375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09995879977941513,
      "learning_rate": 5e-05,
      "loss": 0.1259,
      "step": 18730
    },
    {
      "step": 18730,
      "train_flow_matching_loss": 0.06201171875,
      "train_physics_loss": 0.35156160593032837
    },
    {
      "step": 18730,
      "train_flow_matching_loss": 0.08447265625,
      "train_physics_loss": 0.0
    },
    {
      "step": 18730,
      "train_flow_matching_loss": 0.06787109375,
      "train_physics_loss": 0.36317911744117737
    },
    {
      "step": 18730,
      "train_flow_matching_loss": 0.09716796875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.1002218946814537,
      "learning_rate": 5e-05,
      "loss": 0.1283,
      "step": 18740
    },
    {
      "step": 18740,
      "train_flow_matching_loss": 0.058837890625,
      "train_physics_loss": 0.33933889865875244
    },
    {
      "step": 18740,
      "train_flow_matching_loss": 0.130859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 18740,
      "train_flow_matching_loss": 0.06103515625,
      "train_physics_loss": 0.39909252524375916
    },
    {
      "step": 18740,
      "train_flow_matching_loss": 0.1181640625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08038903772830963,
      "learning_rate": 5e-05,
      "loss": 0.127,
      "step": 18750
    },
    {
      "step": 18750,
      "train_flow_matching_loss": 0.05322265625,
      "train_physics_loss": 0.3084256947040558
    },
    {
      "step": 18750,
      "train_flow_matching_loss": 0.1513671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 18750,
      "train_flow_matching_loss": 0.0751953125,
      "train_physics_loss": 0.38618478178977966
    },
    {
      "step": 18750,
      "train_flow_matching_loss": 0.1611328125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.07439307868480682,
      "learning_rate": 5e-05,
      "loss": 0.1281,
      "step": 18760
    },
    {
      "step": 18760,
      "train_flow_matching_loss": 0.1123046875,
      "train_physics_loss": 0.44434893131256104
    },
    {
      "step": 18760,
      "train_flow_matching_loss": 0.177734375,
      "train_physics_loss": 0.0
    },
    {
      "step": 18760,
      "train_flow_matching_loss": 0.07470703125,
      "train_physics_loss": 0.40967532992362976
    },
    {
      "step": 18760,
      "train_flow_matching_loss": 0.15625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.1296054571866989,
      "learning_rate": 5e-05,
      "loss": 0.1299,
      "step": 18770
    },
    {
      "step": 18770,
      "train_flow_matching_loss": 0.06103515625,
      "train_physics_loss": 0.332379013299942
    },
    {
      "step": 18770,
      "train_flow_matching_loss": 0.1845703125,
      "train_physics_loss": 0.0
    },
    {
      "step": 18770,
      "train_flow_matching_loss": 0.048095703125,
      "train_physics_loss": 0.3274034261703491
    },
    {
      "step": 18770,
      "train_flow_matching_loss": 0.12060546875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09673765301704407,
      "learning_rate": 5e-05,
      "loss": 0.1296,
      "step": 18780
    },
    {
      "step": 18780,
      "train_flow_matching_loss": 0.05419921875,
      "train_physics_loss": 0.36739906668663025
    },
    {
      "step": 18780,
      "train_flow_matching_loss": 0.11767578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 18780,
      "train_flow_matching_loss": 0.05517578125,
      "train_physics_loss": 0.3736492097377777
    },
    {
      "step": 18780,
      "train_flow_matching_loss": 0.140625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09767428040504456,
      "learning_rate": 5e-05,
      "loss": 0.1301,
      "step": 18790
    },
    {
      "step": 18790,
      "train_flow_matching_loss": 0.0294189453125,
      "train_physics_loss": 0.2921687662601471
    },
    {
      "step": 18790,
      "train_flow_matching_loss": 0.1748046875,
      "train_physics_loss": 0.0
    },
    {
      "step": 18790,
      "train_flow_matching_loss": 0.138671875,
      "train_physics_loss": 0.3748520314693451
    },
    {
      "step": 18790,
      "train_flow_matching_loss": 0.1611328125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08058846741914749,
      "learning_rate": 5e-05,
      "loss": 0.1331,
      "step": 18800
    },
    {
      "step": 18800,
      "train_flow_matching_loss": 0.058837890625,
      "train_physics_loss": 0.3030204176902771
    },
    {
      "step": 18800,
      "train_flow_matching_loss": 0.130859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 18800,
      "train_flow_matching_loss": 0.09765625,
      "train_physics_loss": 0.3402920663356781
    },
    {
      "step": 18800,
      "train_flow_matching_loss": 0.1337890625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.13402357697486877,
      "learning_rate": 5e-05,
      "loss": 0.1369,
      "step": 18810
    },
    {
      "step": 18810,
      "train_flow_matching_loss": 0.036865234375,
      "train_physics_loss": 0.4011780619621277
    },
    {
      "step": 18810,
      "train_flow_matching_loss": 0.2255859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 18810,
      "train_flow_matching_loss": 0.033447265625,
      "train_physics_loss": 0.2929833233356476
    },
    {
      "step": 18810,
      "train_flow_matching_loss": 0.11669921875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.07893281430006027,
      "learning_rate": 5e-05,
      "loss": 0.1305,
      "step": 18820
    },
    {
      "step": 18820,
      "train_flow_matching_loss": 0.061767578125,
      "train_physics_loss": 0.3593742847442627
    },
    {
      "step": 18820,
      "train_flow_matching_loss": 0.078125,
      "train_physics_loss": 0.0
    },
    {
      "step": 18820,
      "train_flow_matching_loss": 0.046630859375,
      "train_physics_loss": 0.3862965404987335
    },
    {
      "step": 18820,
      "train_flow_matching_loss": 0.1689453125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.12443750351667404,
      "learning_rate": 5e-05,
      "loss": 0.1332,
      "step": 18830
    },
    {
      "step": 18830,
      "train_flow_matching_loss": 0.055419921875,
      "train_physics_loss": 0.29634296894073486
    },
    {
      "step": 18830,
      "train_flow_matching_loss": 0.1318359375,
      "train_physics_loss": 0.0
    },
    {
      "step": 18830,
      "train_flow_matching_loss": 0.05859375,
      "train_physics_loss": 0.30967196822166443
    },
    {
      "step": 18830,
      "train_flow_matching_loss": 0.154296875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.07340624183416367,
      "learning_rate": 5e-05,
      "loss": 0.1297,
      "step": 18840
    },
    {
      "step": 18840,
      "train_flow_matching_loss": 0.0303955078125,
      "train_physics_loss": 0.3654380440711975
    },
    {
      "step": 18840,
      "train_flow_matching_loss": 0.1337890625,
      "train_physics_loss": 0.0
    },
    {
      "step": 18840,
      "train_flow_matching_loss": 0.2099609375,
      "train_physics_loss": 0.0
    },
    {
      "step": 18840,
      "train_flow_matching_loss": 0.09716796875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.07173030078411102,
      "learning_rate": 5e-05,
      "loss": 0.1264,
      "step": 18850
    },
    {
      "step": 18850,
      "train_flow_matching_loss": 0.0546875,
      "train_physics_loss": 0.3277043104171753
    },
    {
      "step": 18850,
      "train_flow_matching_loss": 0.1708984375,
      "train_physics_loss": 0.0
    },
    {
      "step": 18850,
      "train_flow_matching_loss": 0.1416015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 18850,
      "train_flow_matching_loss": 0.1123046875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.16047878563404083,
      "learning_rate": 5e-05,
      "loss": 0.128,
      "step": 18860
    },
    {
      "step": 18860,
      "train_flow_matching_loss": 0.055419921875,
      "train_physics_loss": 0.3572234809398651
    },
    {
      "step": 18860,
      "train_flow_matching_loss": 0.125,
      "train_physics_loss": 0.0
    },
    {
      "step": 18860,
      "train_flow_matching_loss": 0.1748046875,
      "train_physics_loss": 0.0
    },
    {
      "step": 18860,
      "train_flow_matching_loss": 0.045166015625,
      "train_physics_loss": 0.28350314497947693
    },
    {
      "grad_norm": 0.10053115338087082,
      "learning_rate": 5e-05,
      "loss": 0.131,
      "step": 18870
    },
    {
      "step": 18870,
      "train_flow_matching_loss": 0.04052734375,
      "train_physics_loss": 0.2981068193912506
    },
    {
      "step": 18870,
      "train_flow_matching_loss": 0.0966796875,
      "train_physics_loss": 0.0
    },
    {
      "step": 18870,
      "train_flow_matching_loss": 0.10546875,
      "train_physics_loss": 0.0
    },
    {
      "step": 18870,
      "train_flow_matching_loss": 0.07421875,
      "train_physics_loss": 0.2879320979118347
    },
    {
      "grad_norm": 0.08867619931697845,
      "learning_rate": 5e-05,
      "loss": 0.1309,
      "step": 18880
    },
    {
      "step": 18880,
      "train_flow_matching_loss": 0.0186767578125,
      "train_physics_loss": 0.3800048828125
    },
    {
      "step": 18880,
      "train_flow_matching_loss": 0.1259765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 18880,
      "train_flow_matching_loss": 0.1474609375,
      "train_physics_loss": 0.0
    },
    {
      "step": 18880,
      "train_flow_matching_loss": 0.07373046875,
      "train_physics_loss": 0.3218413293361664
    },
    {
      "grad_norm": 0.08650654554367065,
      "learning_rate": 5e-05,
      "loss": 0.128,
      "step": 18890
    },
    {
      "step": 18890,
      "train_flow_matching_loss": 0.040283203125,
      "train_physics_loss": 0.31207773089408875
    },
    {
      "step": 18890,
      "train_flow_matching_loss": 0.138671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 18890,
      "train_flow_matching_loss": 0.1416015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 18890,
      "train_flow_matching_loss": 0.045654296875,
      "train_physics_loss": 0.2935555875301361
    },
    {
      "grad_norm": 0.08972979336977005,
      "learning_rate": 5e-05,
      "loss": 0.1309,
      "step": 18900
    },
    {
      "step": 18900,
      "train_flow_matching_loss": 0.017822265625,
      "train_physics_loss": 0.31547337770462036
    },
    {
      "step": 18900,
      "train_flow_matching_loss": 0.10986328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 18900,
      "train_flow_matching_loss": 0.1220703125,
      "train_physics_loss": 0.0
    },
    {
      "step": 18900,
      "train_flow_matching_loss": 0.06005859375,
      "train_physics_loss": 0.31041163206100464
    },
    {
      "grad_norm": 0.12272243201732635,
      "learning_rate": 5e-05,
      "loss": 0.1304,
      "step": 18910
    },
    {
      "step": 18910,
      "train_flow_matching_loss": 0.0830078125,
      "train_physics_loss": 0.36124569177627563
    },
    {
      "step": 18910,
      "train_flow_matching_loss": 0.1142578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 18910,
      "train_flow_matching_loss": 0.12890625,
      "train_physics_loss": 0.0
    },
    {
      "step": 18910,
      "train_flow_matching_loss": 0.0247802734375,
      "train_physics_loss": 0.3575305938720703
    },
    {
      "grad_norm": 0.1358935683965683,
      "learning_rate": 5e-05,
      "loss": 0.1352,
      "step": 18920
    },
    {
      "step": 18920,
      "train_flow_matching_loss": 0.078125,
      "train_physics_loss": 0.38249728083610535
    },
    {
      "step": 18920,
      "train_flow_matching_loss": 0.109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 18920,
      "train_flow_matching_loss": 0.126953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 18920,
      "train_flow_matching_loss": 0.057861328125,
      "train_physics_loss": 0.33601146936416626
    },
    {
      "grad_norm": 0.1088884174823761,
      "learning_rate": 5e-05,
      "loss": 0.1312,
      "step": 18930
    },
    {
      "step": 18930,
      "train_flow_matching_loss": 0.09033203125,
      "train_physics_loss": 0.33142080903053284
    },
    {
      "step": 18930,
      "train_flow_matching_loss": 0.2001953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 18930,
      "train_flow_matching_loss": 0.140625,
      "train_physics_loss": 0.0
    },
    {
      "step": 18930,
      "train_flow_matching_loss": 0.04248046875,
      "train_physics_loss": 0.28864890336990356
    },
    {
      "grad_norm": 0.135321244597435,
      "learning_rate": 5e-05,
      "loss": 0.135,
      "step": 18940
    },
    {
      "step": 18940,
      "train_flow_matching_loss": 0.04931640625,
      "train_physics_loss": 0.28104764223098755
    },
    {
      "step": 18940,
      "train_flow_matching_loss": 0.12451171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 18940,
      "train_flow_matching_loss": 0.2197265625,
      "train_physics_loss": 0.0
    },
    {
      "step": 18940,
      "train_flow_matching_loss": 0.03857421875,
      "train_physics_loss": 0.33870765566825867
    },
    {
      "grad_norm": 0.15796993672847748,
      "learning_rate": 5e-05,
      "loss": 0.1282,
      "step": 18950
    },
    {
      "step": 18950,
      "train_flow_matching_loss": 0.12060546875,
      "train_physics_loss": 0.3522789180278778
    },
    {
      "step": 18950,
      "train_flow_matching_loss": 0.2138671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 18950,
      "train_flow_matching_loss": 0.2431640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 18950,
      "train_flow_matching_loss": 0.050537109375,
      "train_physics_loss": 0.3763311803340912
    },
    {
      "grad_norm": 0.0894884541630745,
      "learning_rate": 5e-05,
      "loss": 0.1345,
      "step": 18960
    },
    {
      "step": 18960,
      "train_flow_matching_loss": 0.1220703125,
      "train_physics_loss": 0.0
    },
    {
      "step": 18960,
      "train_flow_matching_loss": 0.10546875,
      "train_physics_loss": 0.0
    },
    {
      "step": 18960,
      "train_flow_matching_loss": 0.1640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 18960,
      "train_flow_matching_loss": 0.09423828125,
      "train_physics_loss": 0.44436439871788025
    },
    {
      "grad_norm": 0.09166064113378525,
      "learning_rate": 5e-05,
      "loss": 0.132,
      "step": 18970
    },
    {
      "step": 18970,
      "train_flow_matching_loss": 0.10302734375,
      "train_physics_loss": 0.0
    },
    {
      "step": 18970,
      "train_flow_matching_loss": 0.09619140625,
      "train_physics_loss": 0.0
    },
    {
      "step": 18970,
      "train_flow_matching_loss": 0.208984375,
      "train_physics_loss": 0.0
    },
    {
      "step": 18970,
      "train_flow_matching_loss": 0.07763671875,
      "train_physics_loss": 0.38796481490135193
    },
    {
      "grad_norm": 0.09074891358613968,
      "learning_rate": 5e-05,
      "loss": 0.1315,
      "step": 18980
    },
    {
      "step": 18980,
      "train_flow_matching_loss": 0.2060546875,
      "train_physics_loss": 0.0
    },
    {
      "step": 18980,
      "train_flow_matching_loss": 0.154296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 18980,
      "train_flow_matching_loss": 0.1318359375,
      "train_physics_loss": 0.0
    },
    {
      "step": 18980,
      "train_flow_matching_loss": 0.041015625,
      "train_physics_loss": 0.3204065263271332
    },
    {
      "grad_norm": 0.17209993302822113,
      "learning_rate": 5e-05,
      "loss": 0.1284,
      "step": 18990
    },
    {
      "step": 18990,
      "train_flow_matching_loss": 0.0986328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 18990,
      "train_flow_matching_loss": 0.134765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 18990,
      "train_flow_matching_loss": 0.12451171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 18990,
      "train_flow_matching_loss": 0.044921875,
      "train_physics_loss": 0.3012181222438812
    },
    {
      "grad_norm": 0.07889437675476074,
      "learning_rate": 5e-05,
      "loss": 0.1291,
      "step": 19000
    },
    {
      "step": 19000,
      "train_flow_matching_loss": 0.10693359375,
      "train_physics_loss": 0.0
    },
    {
      "step": 19000,
      "train_flow_matching_loss": 0.1474609375,
      "train_physics_loss": 0.0
    },
    {
      "step": 19000,
      "train_flow_matching_loss": 0.2001953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 19000,
      "train_flow_matching_loss": 0.08837890625,
      "train_physics_loss": 0.34162822365760803
    },
    {
      "grad_norm": 0.09204944968223572,
      "learning_rate": 5e-05,
      "loss": 0.1286,
      "step": 19010
    },
    {
      "step": 19010,
      "train_flow_matching_loss": 0.181640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 19010,
      "train_flow_matching_loss": 0.181640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 19010,
      "train_flow_matching_loss": 0.0986328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 19010,
      "train_flow_matching_loss": 0.05224609375,
      "train_physics_loss": 0.3657032251358032
    },
    {
      "grad_norm": 0.11182204633951187,
      "learning_rate": 5e-05,
      "loss": 0.1306,
      "step": 19020
    },
    {
      "step": 19020,
      "train_flow_matching_loss": 0.09423828125,
      "train_physics_loss": 0.0
    },
    {
      "step": 19020,
      "train_flow_matching_loss": 0.142578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 19020,
      "train_flow_matching_loss": 0.1572265625,
      "train_physics_loss": 0.0
    },
    {
      "step": 19020,
      "train_flow_matching_loss": 0.035888671875,
      "train_physics_loss": 0.33849766850471497
    },
    {
      "grad_norm": 0.15601509809494019,
      "learning_rate": 5e-05,
      "loss": 0.1345,
      "step": 19030
    },
    {
      "step": 19030,
      "train_flow_matching_loss": 0.1640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 19030,
      "train_flow_matching_loss": 0.12255859375,
      "train_physics_loss": 0.3366032838821411
    },
    {
      "step": 19030,
      "train_flow_matching_loss": 0.1298828125,
      "train_physics_loss": 0.0
    },
    {
      "step": 19030,
      "train_flow_matching_loss": 0.05224609375,
      "train_physics_loss": 0.30463260412216187
    },
    {
      "grad_norm": 0.1279507577419281,
      "learning_rate": 5e-05,
      "loss": 0.1301,
      "step": 19040
    },
    {
      "step": 19040,
      "train_flow_matching_loss": 0.08251953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 19040,
      "train_flow_matching_loss": 0.056396484375,
      "train_physics_loss": 0.319937527179718
    },
    {
      "step": 19040,
      "train_flow_matching_loss": 0.11279296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 19040,
      "train_flow_matching_loss": 0.041259765625,
      "train_physics_loss": 0.33570170402526855
    },
    {
      "grad_norm": 0.08695180714130402,
      "learning_rate": 5e-05,
      "loss": 0.1276,
      "step": 19050
    },
    {
      "step": 19050,
      "train_flow_matching_loss": 0.130859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 19050,
      "train_flow_matching_loss": 0.03173828125,
      "train_physics_loss": 0.30502063035964966
    },
    {
      "step": 19050,
      "train_flow_matching_loss": 0.0986328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 19050,
      "train_flow_matching_loss": 0.0703125,
      "train_physics_loss": 0.32298725843429565
    },
    {
      "grad_norm": 0.10247587412595749,
      "learning_rate": 5e-05,
      "loss": 0.1238,
      "step": 19060
    },
    {
      "step": 19060,
      "train_flow_matching_loss": 0.1982421875,
      "train_physics_loss": 0.0
    },
    {
      "step": 19060,
      "train_flow_matching_loss": 0.0869140625,
      "train_physics_loss": 0.317959189414978
    },
    {
      "step": 19060,
      "train_flow_matching_loss": 0.1298828125,
      "train_physics_loss": 0.0
    },
    {
      "step": 19060,
      "train_flow_matching_loss": 0.044921875,
      "train_physics_loss": 0.3330405652523041
    },
    {
      "grad_norm": 0.12505286931991577,
      "learning_rate": 5e-05,
      "loss": 0.13,
      "step": 19070
    },
    {
      "step": 19070,
      "train_flow_matching_loss": 0.126953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 19070,
      "train_flow_matching_loss": 0.037109375,
      "train_physics_loss": 0.33227503299713135
    },
    {
      "step": 19070,
      "train_flow_matching_loss": 0.1455078125,
      "train_physics_loss": 0.0
    },
    {
      "step": 19070,
      "train_flow_matching_loss": 0.12109375,
      "train_physics_loss": 0.5324214100837708
    },
    {
      "grad_norm": 0.07465114444494247,
      "learning_rate": 5e-05,
      "loss": 0.1252,
      "step": 19080
    },
    {
      "step": 19080,
      "train_flow_matching_loss": 0.166015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 19080,
      "train_flow_matching_loss": 0.0654296875,
      "train_physics_loss": 0.3011881709098816
    },
    {
      "step": 19080,
      "train_flow_matching_loss": 0.11767578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 19080,
      "train_flow_matching_loss": 0.0546875,
      "train_physics_loss": 0.3493190407752991
    },
    {
      "grad_norm": 0.10711951553821564,
      "learning_rate": 5e-05,
      "loss": 0.1273,
      "step": 19090
    },
    {
      "step": 19090,
      "train_flow_matching_loss": 0.154296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 19090,
      "train_flow_matching_loss": 0.035400390625,
      "train_physics_loss": 0.2770440876483917
    },
    {
      "step": 19090,
      "train_flow_matching_loss": 0.126953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 19090,
      "train_flow_matching_loss": 0.029052734375,
      "train_physics_loss": 0.31281527876853943
    },
    {
      "grad_norm": 0.2777528762817383,
      "learning_rate": 5e-05,
      "loss": 0.1307,
      "step": 19100
    },
    {
      "step": 19100,
      "train_flow_matching_loss": 0.109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 19100,
      "train_flow_matching_loss": 0.056640625,
      "train_physics_loss": 0.33666670322418213
    },
    {
      "step": 19100,
      "train_flow_matching_loss": 0.2021484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 19100,
      "train_flow_matching_loss": 0.07568359375,
      "train_physics_loss": 0.3711461126804352
    },
    {
      "grad_norm": 0.08686476945877075,
      "learning_rate": 5e-05,
      "loss": 0.1285,
      "step": 19110
    },
    {
      "step": 19110,
      "train_flow_matching_loss": 0.171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 19110,
      "train_flow_matching_loss": 0.048583984375,
      "train_physics_loss": 0.328230619430542
    },
    {
      "step": 19110,
      "train_flow_matching_loss": 0.2373046875,
      "train_physics_loss": 0.0
    },
    {
      "step": 19110,
      "train_flow_matching_loss": 0.036865234375,
      "train_physics_loss": 0.32729223370552063
    },
    {
      "grad_norm": 0.0948452576994896,
      "learning_rate": 5e-05,
      "loss": 0.1234,
      "step": 19120
    },
    {
      "step": 19120,
      "train_flow_matching_loss": 0.1689453125,
      "train_physics_loss": 0.0
    },
    {
      "step": 19120,
      "train_flow_matching_loss": 0.0390625,
      "train_physics_loss": 0.2915063798427582
    },
    {
      "step": 19120,
      "train_flow_matching_loss": 0.189453125,
      "train_physics_loss": 0.0
    },
    {
      "step": 19120,
      "train_flow_matching_loss": 0.1474609375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08585143089294434,
      "learning_rate": 5e-05,
      "loss": 0.1319,
      "step": 19130
    },
    {
      "step": 19130,
      "train_flow_matching_loss": 0.12060546875,
      "train_physics_loss": 0.0
    },
    {
      "step": 19130,
      "train_flow_matching_loss": 0.052978515625,
      "train_physics_loss": 0.3388286530971527
    },
    {
      "step": 19130,
      "train_flow_matching_loss": 0.1962890625,
      "train_physics_loss": 0.0
    },
    {
      "step": 19130,
      "train_flow_matching_loss": 0.1787109375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09292169660329819,
      "learning_rate": 5e-05,
      "loss": 0.129,
      "step": 19140
    },
    {
      "step": 19140,
      "train_flow_matching_loss": 0.146484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 19140,
      "train_flow_matching_loss": 0.044677734375,
      "train_physics_loss": 0.2837880849838257
    },
    {
      "step": 19140,
      "train_flow_matching_loss": 0.2353515625,
      "train_physics_loss": 0.0
    },
    {
      "step": 19140,
      "train_flow_matching_loss": 0.2197265625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10796693712472916,
      "learning_rate": 5e-05,
      "loss": 0.1307,
      "step": 19150
    },
    {
      "step": 19150,
      "train_flow_matching_loss": 0.1669921875,
      "train_physics_loss": 0.0
    },
    {
      "step": 19150,
      "train_flow_matching_loss": 0.032958984375,
      "train_physics_loss": 0.31994521617889404
    },
    {
      "step": 19150,
      "train_flow_matching_loss": 0.11865234375,
      "train_physics_loss": 0.0
    },
    {
      "step": 19150,
      "train_flow_matching_loss": 0.1533203125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10623610019683838,
      "learning_rate": 5e-05,
      "loss": 0.133,
      "step": 19160
    },
    {
      "step": 19160,
      "train_flow_matching_loss": 0.1630859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 19160,
      "train_flow_matching_loss": 0.04052734375,
      "train_physics_loss": 0.3446671664714813
    },
    {
      "step": 19160,
      "train_flow_matching_loss": 0.1474609375,
      "train_physics_loss": 0.0
    },
    {
      "step": 19160,
      "train_flow_matching_loss": 0.09814453125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.07598108798265457,
      "learning_rate": 5e-05,
      "loss": 0.1333,
      "step": 19170
    },
    {
      "step": 19170,
      "train_flow_matching_loss": 0.09912109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 19170,
      "train_flow_matching_loss": 0.08203125,
      "train_physics_loss": 0.29925182461738586
    },
    {
      "step": 19170,
      "train_flow_matching_loss": 0.162109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 19170,
      "train_flow_matching_loss": 0.138671875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.1070551797747612,
      "learning_rate": 5e-05,
      "loss": 0.1287,
      "step": 19180
    },
    {
      "step": 19180,
      "train_flow_matching_loss": 0.2021484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 19180,
      "train_flow_matching_loss": 0.03369140625,
      "train_physics_loss": 0.3478457033634186
    },
    {
      "step": 19180,
      "train_flow_matching_loss": 0.2119140625,
      "train_physics_loss": 0.0
    },
    {
      "step": 19180,
      "train_flow_matching_loss": 0.1552734375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.12280179560184479,
      "learning_rate": 5e-05,
      "loss": 0.1324,
      "step": 19190
    },
    {
      "step": 19190,
      "train_flow_matching_loss": 0.208984375,
      "train_physics_loss": 0.0
    },
    {
      "step": 19190,
      "train_flow_matching_loss": 0.061279296875,
      "train_physics_loss": 0.3721983730792999
    },
    {
      "step": 19190,
      "train_flow_matching_loss": 0.1796875,
      "train_physics_loss": 0.0
    },
    {
      "step": 19190,
      "train_flow_matching_loss": 0.0810546875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.14888052642345428,
      "learning_rate": 5e-05,
      "loss": 0.1284,
      "step": 19200
    },
    {
      "step": 19200,
      "train_flow_matching_loss": 0.12353515625,
      "train_physics_loss": 0.0
    },
    {
      "step": 19200,
      "train_flow_matching_loss": 0.052490234375,
      "train_physics_loss": 0.3154962956905365
    },
    {
      "step": 19200,
      "train_flow_matching_loss": 0.11083984375,
      "train_physics_loss": 0.0
    },
    {
      "step": 19200,
      "train_flow_matching_loss": 0.189453125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09199156612157822,
      "learning_rate": 5e-05,
      "loss": 0.1281,
      "step": 19210
    },
    {
      "step": 19210,
      "train_flow_matching_loss": 0.06787109375,
      "train_physics_loss": 0.2984408140182495
    },
    {
      "step": 19210,
      "train_flow_matching_loss": 0.050537109375,
      "train_physics_loss": 0.3189567029476166
    },
    {
      "step": 19210,
      "train_flow_matching_loss": 0.12451171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 19210,
      "train_flow_matching_loss": 0.15625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.1505931317806244,
      "learning_rate": 5e-05,
      "loss": 0.1295,
      "step": 19220
    },
    {
      "step": 19220,
      "train_flow_matching_loss": 0.07373046875,
      "train_physics_loss": 0.27707165479660034
    },
    {
      "step": 19220,
      "train_flow_matching_loss": 0.048828125,
      "train_physics_loss": 0.3176329731941223
    },
    {
      "step": 19220,
      "train_flow_matching_loss": 0.1357421875,
      "train_physics_loss": 0.0
    },
    {
      "step": 19220,
      "train_flow_matching_loss": 0.2158203125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08658494055271149,
      "learning_rate": 5e-05,
      "loss": 0.1329,
      "step": 19230
    },
    {
      "step": 19230,
      "train_flow_matching_loss": 0.05712890625,
      "train_physics_loss": 0.27399569749832153
    },
    {
      "step": 19230,
      "train_flow_matching_loss": 0.0306396484375,
      "train_physics_loss": 0.25205257534980774
    },
    {
      "step": 19230,
      "train_flow_matching_loss": 0.1787109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 19230,
      "train_flow_matching_loss": 0.07421875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.07440600544214249,
      "learning_rate": 5e-05,
      "loss": 0.1287,
      "step": 19240
    },
    {
      "step": 19240,
      "train_flow_matching_loss": 0.0703125,
      "train_physics_loss": 0.3606649339199066
    },
    {
      "step": 19240,
      "train_flow_matching_loss": 0.06591796875,
      "train_physics_loss": 0.3489263355731964
    },
    {
      "step": 19240,
      "train_flow_matching_loss": 0.1884765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 19240,
      "train_flow_matching_loss": 0.0732421875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.0989765003323555,
      "learning_rate": 5e-05,
      "loss": 0.1315,
      "step": 19250
    },
    {
      "step": 19250,
      "train_flow_matching_loss": 0.03955078125,
      "train_physics_loss": 0.28950637578964233
    },
    {
      "step": 19250,
      "train_flow_matching_loss": 0.03271484375,
      "train_physics_loss": 0.3318045735359192
    },
    {
      "step": 19250,
      "train_flow_matching_loss": 0.263671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 19250,
      "train_flow_matching_loss": 0.11181640625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.07969378679990768,
      "learning_rate": 5e-05,
      "loss": 0.128,
      "step": 19260
    },
    {
      "step": 19260,
      "train_flow_matching_loss": 0.046630859375,
      "train_physics_loss": 0.29697707295417786
    },
    {
      "step": 19260,
      "train_flow_matching_loss": 0.06884765625,
      "train_physics_loss": 0.5108216404914856
    },
    {
      "step": 19260,
      "train_flow_matching_loss": 0.1396484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 19260,
      "train_flow_matching_loss": 0.1279296875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.12049658596515656,
      "learning_rate": 5e-05,
      "loss": 0.1291,
      "step": 19270
    },
    {
      "step": 19270,
      "train_flow_matching_loss": 0.0859375,
      "train_physics_loss": 0.31780505180358887
    },
    {
      "step": 19270,
      "train_flow_matching_loss": 0.039794921875,
      "train_physics_loss": 0.2933861017227173
    },
    {
      "step": 19270,
      "train_flow_matching_loss": 0.1611328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 19270,
      "train_flow_matching_loss": 0.0791015625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.07402755320072174,
      "learning_rate": 5e-05,
      "loss": 0.1272,
      "step": 19280
    },
    {
      "step": 19280,
      "train_flow_matching_loss": 0.1064453125,
      "train_physics_loss": 0.3947620689868927
    },
    {
      "step": 19280,
      "train_flow_matching_loss": 0.0693359375,
      "train_physics_loss": 0.33263054490089417
    },
    {
      "step": 19280,
      "train_flow_matching_loss": 0.11279296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 19280,
      "train_flow_matching_loss": 0.11376953125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08965452015399933,
      "learning_rate": 5e-05,
      "loss": 0.132,
      "step": 19290
    },
    {
      "step": 19290,
      "train_flow_matching_loss": 0.08349609375,
      "train_physics_loss": 0.25828659534454346
    },
    {
      "step": 19290,
      "train_flow_matching_loss": 0.06884765625,
      "train_physics_loss": 0.3785010278224945
    },
    {
      "step": 19290,
      "train_flow_matching_loss": 0.2470703125,
      "train_physics_loss": 0.0
    },
    {
      "step": 19290,
      "train_flow_matching_loss": 0.0927734375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10019315779209137,
      "learning_rate": 5e-05,
      "loss": 0.1285,
      "step": 19300
    },
    {
      "step": 19300,
      "train_flow_matching_loss": 0.057373046875,
      "train_physics_loss": 0.3002144694328308
    },
    {
      "step": 19300,
      "train_flow_matching_loss": 0.05029296875,
      "train_physics_loss": 0.32616719603538513
    },
    {
      "step": 19300,
      "train_flow_matching_loss": 0.1845703125,
      "train_physics_loss": 0.0
    },
    {
      "step": 19300,
      "train_flow_matching_loss": 0.1640625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10948819667100906,
      "learning_rate": 5e-05,
      "loss": 0.1264,
      "step": 19310
    },
    {
      "step": 19310,
      "train_flow_matching_loss": 0.043701171875,
      "train_physics_loss": 0.3101278245449066
    },
    {
      "step": 19310,
      "train_flow_matching_loss": 0.05810546875,
      "train_physics_loss": 0.3141942620277405
    },
    {
      "step": 19310,
      "train_flow_matching_loss": 0.232421875,
      "train_physics_loss": 0.0
    },
    {
      "step": 19310,
      "train_flow_matching_loss": 0.10986328125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.13967056572437286,
      "learning_rate": 5e-05,
      "loss": 0.127,
      "step": 19320
    },
    {
      "step": 19320,
      "train_flow_matching_loss": 0.0546875,
      "train_physics_loss": 0.3135095238685608
    },
    {
      "step": 19320,
      "train_flow_matching_loss": 0.03564453125,
      "train_physics_loss": 0.2609597146511078
    },
    {
      "step": 19320,
      "train_flow_matching_loss": 0.134765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 19320,
      "train_flow_matching_loss": 0.10107421875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10390494763851166,
      "learning_rate": 5e-05,
      "loss": 0.1304,
      "step": 19330
    },
    {
      "step": 19330,
      "train_flow_matching_loss": 0.09033203125,
      "train_physics_loss": 0.36185911297798157
    },
    {
      "step": 19330,
      "train_flow_matching_loss": 0.0634765625,
      "train_physics_loss": 0.3361080288887024
    },
    {
      "step": 19330,
      "train_flow_matching_loss": 0.1513671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 19330,
      "train_flow_matching_loss": 0.1572265625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.06625946611166,
      "learning_rate": 5e-05,
      "loss": 0.1298,
      "step": 19340
    },
    {
      "step": 19340,
      "train_flow_matching_loss": 0.033447265625,
      "train_physics_loss": 0.3123924136161804
    },
    {
      "step": 19340,
      "train_flow_matching_loss": 0.05126953125,
      "train_physics_loss": 0.30801719427108765
    },
    {
      "step": 19340,
      "train_flow_matching_loss": 0.1337890625,
      "train_physics_loss": 0.0
    },
    {
      "step": 19340,
      "train_flow_matching_loss": 0.1708984375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.29001739621162415,
      "learning_rate": 5e-05,
      "loss": 0.1277,
      "step": 19350
    },
    {
      "step": 19350,
      "train_flow_matching_loss": 0.09375,
      "train_physics_loss": 0.26826339960098267
    },
    {
      "step": 19350,
      "train_flow_matching_loss": 0.045654296875,
      "train_physics_loss": 0.2966168522834778
    },
    {
      "step": 19350,
      "train_flow_matching_loss": 0.1279296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 19350,
      "train_flow_matching_loss": 0.171875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10532324016094208,
      "learning_rate": 5e-05,
      "loss": 0.1238,
      "step": 19360
    },
    {
      "step": 19360,
      "train_flow_matching_loss": 0.061767578125,
      "train_physics_loss": 0.28064584732055664
    },
    {
      "step": 19360,
      "train_flow_matching_loss": 0.0869140625,
      "train_physics_loss": 0.28754186630249023
    },
    {
      "step": 19360,
      "train_flow_matching_loss": 0.08935546875,
      "train_physics_loss": 0.0
    },
    {
      "step": 19360,
      "train_flow_matching_loss": 0.126953125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10724686831235886,
      "learning_rate": 5e-05,
      "loss": 0.1286,
      "step": 19370
    },
    {
      "step": 19370,
      "train_flow_matching_loss": 0.029052734375,
      "train_physics_loss": 0.27861493825912476
    },
    {
      "step": 19370,
      "train_flow_matching_loss": 0.11083984375,
      "train_physics_loss": 0.30939534306526184
    },
    {
      "step": 19370,
      "train_flow_matching_loss": 0.11376953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 19370,
      "train_flow_matching_loss": 0.1064453125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09379719942808151,
      "learning_rate": 5e-05,
      "loss": 0.1278,
      "step": 19380
    },
    {
      "step": 19380,
      "train_flow_matching_loss": 0.06494140625,
      "train_physics_loss": 0.2989937663078308
    },
    {
      "step": 19380,
      "train_flow_matching_loss": 0.0242919921875,
      "train_physics_loss": 0.29425692558288574
    },
    {
      "step": 19380,
      "train_flow_matching_loss": 0.140625,
      "train_physics_loss": 0.0
    },
    {
      "step": 19380,
      "train_flow_matching_loss": 0.10791015625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.12915897369384766,
      "learning_rate": 5e-05,
      "loss": 0.1312,
      "step": 19390
    },
    {
      "step": 19390,
      "train_flow_matching_loss": 0.033447265625,
      "train_physics_loss": 0.28470519185066223
    },
    {
      "step": 19390,
      "train_flow_matching_loss": 0.0306396484375,
      "train_physics_loss": 0.29665225744247437
    },
    {
      "step": 19390,
      "train_flow_matching_loss": 0.11376953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 19390,
      "train_flow_matching_loss": 0.12890625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.12089624255895615,
      "learning_rate": 5e-05,
      "loss": 0.1312,
      "step": 19400
    },
    {
      "step": 19400,
      "train_flow_matching_loss": 0.09423828125,
      "train_physics_loss": 0.3136645555496216
    },
    {
      "step": 19400,
      "train_flow_matching_loss": 0.031982421875,
      "train_physics_loss": 0.0
    },
    {
      "step": 19400,
      "train_flow_matching_loss": 0.0888671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 19400,
      "train_flow_matching_loss": 0.09326171875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.12533995509147644,
      "learning_rate": 5e-05,
      "loss": 0.1317,
      "step": 19410
    },
    {
      "step": 19410,
      "train_flow_matching_loss": 0.04736328125,
      "train_physics_loss": 0.30714860558509827
    },
    {
      "step": 19410,
      "train_flow_matching_loss": 0.06396484375,
      "train_physics_loss": 0.34929776191711426
    },
    {
      "step": 19410,
      "train_flow_matching_loss": 0.11572265625,
      "train_physics_loss": 0.0
    },
    {
      "step": 19410,
      "train_flow_matching_loss": 0.1005859375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10702484101057053,
      "learning_rate": 5e-05,
      "loss": 0.1265,
      "step": 19420
    },
    {
      "step": 19420,
      "train_flow_matching_loss": 0.06982421875,
      "train_physics_loss": 0.3225795328617096
    },
    {
      "step": 19420,
      "train_flow_matching_loss": 0.0986328125,
      "train_physics_loss": 0.2923518121242523
    },
    {
      "step": 19420,
      "train_flow_matching_loss": 0.109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 19420,
      "train_flow_matching_loss": 0.134765625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.07932891696691513,
      "learning_rate": 5e-05,
      "loss": 0.1294,
      "step": 19430
    },
    {
      "step": 19430,
      "train_flow_matching_loss": 0.06494140625,
      "train_physics_loss": 0.29992932081222534
    },
    {
      "step": 19430,
      "train_flow_matching_loss": 0.03173828125,
      "train_physics_loss": 0.2849593758583069
    },
    {
      "step": 19430,
      "train_flow_matching_loss": 0.11962890625,
      "train_physics_loss": 0.0
    },
    {
      "step": 19430,
      "train_flow_matching_loss": 0.1591796875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10604972392320633,
      "learning_rate": 5e-05,
      "loss": 0.1298,
      "step": 19440
    },
    {
      "step": 19440,
      "train_flow_matching_loss": 0.03173828125,
      "train_physics_loss": 0.31248536705970764
    },
    {
      "step": 19440,
      "train_flow_matching_loss": 0.0625,
      "train_physics_loss": 0.28008535504341125
    },
    {
      "step": 19440,
      "train_flow_matching_loss": 0.1201171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 19440,
      "train_flow_matching_loss": 0.1953125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.0616324320435524,
      "learning_rate": 5e-05,
      "loss": 0.1279,
      "step": 19450
    },
    {
      "step": 19450,
      "train_flow_matching_loss": 0.09912109375,
      "train_physics_loss": 0.5997318625450134
    },
    {
      "step": 19450,
      "train_flow_matching_loss": 0.042236328125,
      "train_physics_loss": 0.2945595383644104
    },
    {
      "step": 19450,
      "train_flow_matching_loss": 0.06787109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 19450,
      "train_flow_matching_loss": 0.07470703125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09626311808824539,
      "learning_rate": 5e-05,
      "loss": 0.1239,
      "step": 19460
    },
    {
      "step": 19460,
      "train_flow_matching_loss": 0.11474609375,
      "train_physics_loss": 0.0
    },
    {
      "step": 19460,
      "train_flow_matching_loss": 0.043701171875,
      "train_physics_loss": 0.2812406122684479
    },
    {
      "step": 19460,
      "train_flow_matching_loss": 0.1748046875,
      "train_physics_loss": 0.0
    },
    {
      "step": 19460,
      "train_flow_matching_loss": 0.07080078125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.17510837316513062,
      "learning_rate": 5e-05,
      "loss": 0.132,
      "step": 19470
    },
    {
      "step": 19470,
      "train_flow_matching_loss": 0.11767578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 19470,
      "train_flow_matching_loss": 0.0771484375,
      "train_physics_loss": 0.3030546009540558
    },
    {
      "step": 19470,
      "train_flow_matching_loss": 0.10302734375,
      "train_physics_loss": 0.0
    },
    {
      "step": 19470,
      "train_flow_matching_loss": 0.1484375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.07277210056781769,
      "learning_rate": 5e-05,
      "loss": 0.1346,
      "step": 19480
    },
    {
      "step": 19480,
      "train_flow_matching_loss": 0.10595703125,
      "train_physics_loss": 0.0
    },
    {
      "step": 19480,
      "train_flow_matching_loss": 0.13671875,
      "train_physics_loss": 0.3488875925540924
    },
    {
      "step": 19480,
      "train_flow_matching_loss": 0.1201171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 19480,
      "train_flow_matching_loss": 0.1396484375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11700209230184555,
      "learning_rate": 5e-05,
      "loss": 0.1351,
      "step": 19490
    },
    {
      "step": 19490,
      "train_flow_matching_loss": 0.1298828125,
      "train_physics_loss": 0.0
    },
    {
      "step": 19490,
      "train_flow_matching_loss": 0.04248046875,
      "train_physics_loss": 0.33849233388900757
    },
    {
      "step": 19490,
      "train_flow_matching_loss": 0.1337890625,
      "train_physics_loss": 0.0
    },
    {
      "step": 19490,
      "train_flow_matching_loss": 0.12890625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08202783763408661,
      "learning_rate": 5e-05,
      "loss": 0.1286,
      "step": 19500
    },
    {
      "step": 19500,
      "train_flow_matching_loss": 0.130859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 19500,
      "train_flow_matching_loss": 0.045654296875,
      "train_physics_loss": 0.297778457403183
    },
    {
      "step": 19500,
      "train_flow_matching_loss": 0.10693359375,
      "train_physics_loss": 0.0
    },
    {
      "step": 19500,
      "train_flow_matching_loss": 0.2412109375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.07192318886518478,
      "learning_rate": 5e-05,
      "loss": 0.1295,
      "step": 19510
    },
    {
      "step": 19510,
      "train_flow_matching_loss": 0.1025390625,
      "train_physics_loss": 0.0
    },
    {
      "step": 19510,
      "train_flow_matching_loss": 0.055908203125,
      "train_physics_loss": 0.3483545184135437
    },
    {
      "step": 19510,
      "train_flow_matching_loss": 0.10546875,
      "train_physics_loss": 0.0
    },
    {
      "step": 19510,
      "train_flow_matching_loss": 0.169921875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.12633773684501648,
      "learning_rate": 5e-05,
      "loss": 0.1288,
      "step": 19520
    },
    {
      "step": 19520,
      "train_flow_matching_loss": 0.1123046875,
      "train_physics_loss": 0.0
    },
    {
      "step": 19520,
      "train_flow_matching_loss": 0.062255859375,
      "train_physics_loss": 0.2965942621231079
    },
    {
      "step": 19520,
      "train_flow_matching_loss": 0.12451171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 19520,
      "train_flow_matching_loss": 0.2216796875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09152818471193314,
      "learning_rate": 5e-05,
      "loss": 0.127,
      "step": 19530
    },
    {
      "step": 19530,
      "train_flow_matching_loss": 0.1455078125,
      "train_physics_loss": 0.0
    },
    {
      "step": 19530,
      "train_flow_matching_loss": 0.033447265625,
      "train_physics_loss": 0.30143287777900696
    },
    {
      "step": 19530,
      "train_flow_matching_loss": 0.1494140625,
      "train_physics_loss": 0.0
    },
    {
      "step": 19530,
      "train_flow_matching_loss": 0.11962890625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09527971595525742,
      "learning_rate": 5e-05,
      "loss": 0.1271,
      "step": 19540
    },
    {
      "step": 19540,
      "train_flow_matching_loss": 0.0693359375,
      "train_physics_loss": 0.0
    },
    {
      "step": 19540,
      "train_flow_matching_loss": 0.054443359375,
      "train_physics_loss": 0.37300634384155273
    },
    {
      "step": 19540,
      "train_flow_matching_loss": 0.109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 19540,
      "train_flow_matching_loss": 0.11767578125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09501704573631287,
      "learning_rate": 5e-05,
      "loss": 0.131,
      "step": 19550
    },
    {
      "step": 19550,
      "train_flow_matching_loss": 0.1357421875,
      "train_physics_loss": 0.0
    },
    {
      "step": 19550,
      "train_flow_matching_loss": 0.1845703125,
      "train_physics_loss": 0.0
    },
    {
      "step": 19550,
      "train_flow_matching_loss": 0.171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 19550,
      "train_flow_matching_loss": 0.1845703125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.2357167899608612,
      "learning_rate": 5e-05,
      "loss": 0.1325,
      "step": 19560
    },
    {
      "step": 19560,
      "train_flow_matching_loss": 0.1591796875,
      "train_physics_loss": 0.0
    },
    {
      "step": 19560,
      "train_flow_matching_loss": 0.09228515625,
      "train_physics_loss": 0.0
    },
    {
      "step": 19560,
      "train_flow_matching_loss": 0.0693359375,
      "train_physics_loss": 0.0
    },
    {
      "step": 19560,
      "train_flow_matching_loss": 0.150390625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.0940583199262619,
      "learning_rate": 5e-05,
      "loss": 0.134,
      "step": 19570
    },
    {
      "step": 19570,
      "train_flow_matching_loss": 0.2021484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 19570,
      "train_flow_matching_loss": 0.09375,
      "train_physics_loss": 0.0
    },
    {
      "step": 19570,
      "train_flow_matching_loss": 0.1318359375,
      "train_physics_loss": 0.0
    },
    {
      "step": 19570,
      "train_flow_matching_loss": 0.1455078125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10396935045719147,
      "learning_rate": 5e-05,
      "loss": 0.1323,
      "step": 19580
    },
    {
      "step": 19580,
      "train_flow_matching_loss": 0.1865234375,
      "train_physics_loss": 0.0
    },
    {
      "step": 19580,
      "train_flow_matching_loss": 0.1533203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 19580,
      "train_flow_matching_loss": 0.1318359375,
      "train_physics_loss": 0.0
    },
    {
      "step": 19580,
      "train_flow_matching_loss": 0.125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11999017745256424,
      "learning_rate": 5e-05,
      "loss": 0.1264,
      "step": 19590
    },
    {
      "step": 19590,
      "train_flow_matching_loss": 0.11376953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 19590,
      "train_flow_matching_loss": 0.13671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 19590,
      "train_flow_matching_loss": 0.0791015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 19590,
      "train_flow_matching_loss": 0.11865234375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.1142643541097641,
      "learning_rate": 5e-05,
      "loss": 0.1296,
      "step": 19600
    },
    {
      "step": 19600,
      "train_flow_matching_loss": 0.08349609375,
      "train_physics_loss": 0.0
    },
    {
      "step": 19600,
      "train_flow_matching_loss": 0.1201171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 19600,
      "train_flow_matching_loss": 0.12451171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 19600,
      "train_flow_matching_loss": 0.138671875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08363062143325806,
      "learning_rate": 5e-05,
      "loss": 0.1244,
      "step": 19610
    },
    {
      "step": 19610,
      "train_flow_matching_loss": 0.2197265625,
      "train_physics_loss": 0.0
    },
    {
      "step": 19610,
      "train_flow_matching_loss": 0.2197265625,
      "train_physics_loss": 0.0
    },
    {
      "step": 19610,
      "train_flow_matching_loss": 0.134765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 19610,
      "train_flow_matching_loss": 0.08984375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08910251408815384,
      "learning_rate": 5e-05,
      "loss": 0.1279,
      "step": 19620
    },
    {
      "step": 19620,
      "train_flow_matching_loss": 0.1259765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 19620,
      "train_flow_matching_loss": 0.134765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 19620,
      "train_flow_matching_loss": 0.09814453125,
      "train_physics_loss": 0.0
    },
    {
      "step": 19620,
      "train_flow_matching_loss": 0.228515625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08401603996753693,
      "learning_rate": 5e-05,
      "loss": 0.1256,
      "step": 19630
    },
    {
      "step": 19630,
      "train_flow_matching_loss": 0.173828125,
      "train_physics_loss": 0.0
    },
    {
      "step": 19630,
      "train_flow_matching_loss": 0.1298828125,
      "train_physics_loss": 0.0
    },
    {
      "step": 19630,
      "train_flow_matching_loss": 0.185546875,
      "train_physics_loss": 0.0
    },
    {
      "step": 19630,
      "train_flow_matching_loss": 0.08154296875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.07515764981508255,
      "learning_rate": 5e-05,
      "loss": 0.1269,
      "step": 19640
    },
    {
      "step": 19640,
      "train_flow_matching_loss": 0.1083984375,
      "train_physics_loss": 0.0
    },
    {
      "step": 19640,
      "train_flow_matching_loss": 0.142578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 19640,
      "train_flow_matching_loss": 0.1416015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 19640,
      "train_flow_matching_loss": 0.1376953125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08127282559871674,
      "learning_rate": 5e-05,
      "loss": 0.126,
      "step": 19650
    },
    {
      "step": 19650,
      "train_flow_matching_loss": 0.1298828125,
      "train_physics_loss": 0.0
    },
    {
      "step": 19650,
      "train_flow_matching_loss": 0.0830078125,
      "train_physics_loss": 0.0
    },
    {
      "step": 19650,
      "train_flow_matching_loss": 0.1318359375,
      "train_physics_loss": 0.0
    },
    {
      "step": 19650,
      "train_flow_matching_loss": 0.189453125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10696335881948471,
      "learning_rate": 5e-05,
      "loss": 0.1332,
      "step": 19660
    },
    {
      "step": 19660,
      "train_flow_matching_loss": 0.11376953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 19660,
      "train_flow_matching_loss": 0.11572265625,
      "train_physics_loss": 0.0
    },
    {
      "step": 19660,
      "train_flow_matching_loss": 0.171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 19660,
      "train_flow_matching_loss": 0.1162109375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09350096434354782,
      "learning_rate": 5e-05,
      "loss": 0.1303,
      "step": 19670
    },
    {
      "step": 19670,
      "train_flow_matching_loss": 0.09375,
      "train_physics_loss": 0.0
    },
    {
      "step": 19670,
      "train_flow_matching_loss": 0.1884765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 19670,
      "train_flow_matching_loss": 0.11328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 19670,
      "train_flow_matching_loss": 0.06787109375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.06451351940631866,
      "learning_rate": 5e-05,
      "loss": 0.1244,
      "step": 19680
    },
    {
      "step": 19680,
      "train_flow_matching_loss": 0.1455078125,
      "train_physics_loss": 0.0
    },
    {
      "step": 19680,
      "train_flow_matching_loss": 0.1474609375,
      "train_physics_loss": 0.0
    },
    {
      "step": 19680,
      "train_flow_matching_loss": 0.1572265625,
      "train_physics_loss": 0.0
    },
    {
      "step": 19680,
      "train_flow_matching_loss": 0.09423828125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.07349345833063126,
      "learning_rate": 5e-05,
      "loss": 0.1244,
      "step": 19690
    },
    {
      "step": 19690,
      "train_flow_matching_loss": 0.1455078125,
      "train_physics_loss": 0.0
    },
    {
      "step": 19690,
      "train_flow_matching_loss": 0.1875,
      "train_physics_loss": 0.0
    },
    {
      "step": 19690,
      "train_flow_matching_loss": 0.1259765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 19690,
      "train_flow_matching_loss": 0.1240234375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.07131464034318924,
      "learning_rate": 5e-05,
      "loss": 0.1328,
      "step": 19700
    },
    {
      "step": 19700,
      "train_flow_matching_loss": 0.0888671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 19700,
      "train_flow_matching_loss": 0.134765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 19700,
      "train_flow_matching_loss": 0.1328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 19700,
      "train_flow_matching_loss": 0.154296875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.35199904441833496,
      "learning_rate": 5e-05,
      "loss": 0.1277,
      "step": 19710
    },
    {
      "step": 19710,
      "train_flow_matching_loss": 0.126953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 19710,
      "train_flow_matching_loss": 0.2119140625,
      "train_physics_loss": 0.0
    },
    {
      "step": 19710,
      "train_flow_matching_loss": 0.130859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 19710,
      "train_flow_matching_loss": 0.1494140625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11295698583126068,
      "learning_rate": 5e-05,
      "loss": 0.1304,
      "step": 19720
    },
    {
      "step": 19720,
      "train_flow_matching_loss": 0.08349609375,
      "train_physics_loss": 0.0
    },
    {
      "step": 19720,
      "train_flow_matching_loss": 0.10498046875,
      "train_physics_loss": 0.0
    },
    {
      "step": 19720,
      "train_flow_matching_loss": 0.09765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 19720,
      "train_flow_matching_loss": 0.1044921875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.4102761447429657,
      "learning_rate": 5e-05,
      "loss": 0.1299,
      "step": 19730
    },
    {
      "step": 19730,
      "train_flow_matching_loss": 0.09130859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 19730,
      "train_flow_matching_loss": 0.08837890625,
      "train_physics_loss": 0.0
    },
    {
      "step": 19730,
      "train_flow_matching_loss": 0.1796875,
      "train_physics_loss": 0.0
    },
    {
      "step": 19730,
      "train_flow_matching_loss": 0.10400390625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.07317771017551422,
      "learning_rate": 5e-05,
      "loss": 0.1302,
      "step": 19740
    },
    {
      "step": 19740,
      "train_flow_matching_loss": 0.1748046875,
      "train_physics_loss": 0.0
    },
    {
      "step": 19740,
      "train_flow_matching_loss": 0.10791015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 19740,
      "train_flow_matching_loss": 0.203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 19740,
      "train_flow_matching_loss": 0.1572265625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10694874823093414,
      "learning_rate": 5e-05,
      "loss": 0.1294,
      "step": 19750
    },
    {
      "step": 19750,
      "train_flow_matching_loss": 0.162109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 19750,
      "train_flow_matching_loss": 0.150390625,
      "train_physics_loss": 0.0
    },
    {
      "step": 19750,
      "train_flow_matching_loss": 0.09130859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 19750,
      "train_flow_matching_loss": 0.1240234375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09931919723749161,
      "learning_rate": 5e-05,
      "loss": 0.1284,
      "step": 19760
    },
    {
      "step": 19760,
      "train_flow_matching_loss": 0.1259765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 19760,
      "train_flow_matching_loss": 0.248046875,
      "train_physics_loss": 0.0
    },
    {
      "step": 19760,
      "train_flow_matching_loss": 0.2216796875,
      "train_physics_loss": 0.0
    },
    {
      "step": 19760,
      "train_flow_matching_loss": 0.1103515625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11073057353496552,
      "learning_rate": 5e-05,
      "loss": 0.1301,
      "step": 19770
    },
    {
      "step": 19770,
      "train_flow_matching_loss": 0.061279296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 19770,
      "train_flow_matching_loss": 0.10595703125,
      "train_physics_loss": 0.0
    },
    {
      "step": 19770,
      "train_flow_matching_loss": 0.11279296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 19770,
      "train_flow_matching_loss": 0.142578125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08588556945323944,
      "learning_rate": 5e-05,
      "loss": 0.1313,
      "step": 19780
    },
    {
      "step": 19780,
      "train_flow_matching_loss": 0.1279296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 19780,
      "train_flow_matching_loss": 0.1416015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 19780,
      "train_flow_matching_loss": 0.130859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 19780,
      "train_flow_matching_loss": 0.12158203125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11533913761377335,
      "learning_rate": 5e-05,
      "loss": 0.1327,
      "step": 19790
    },
    {
      "step": 19790,
      "train_flow_matching_loss": 0.1787109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 19790,
      "train_flow_matching_loss": 0.1201171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 19790,
      "train_flow_matching_loss": 0.1337890625,
      "train_physics_loss": 0.0
    },
    {
      "step": 19790,
      "train_flow_matching_loss": 0.1396484375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.07544532418251038,
      "learning_rate": 5e-05,
      "loss": 0.1264,
      "step": 19800
    },
    {
      "step": 19800,
      "train_flow_matching_loss": 0.1708984375,
      "train_physics_loss": 0.0
    },
    {
      "step": 19800,
      "train_flow_matching_loss": 0.1181640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 19800,
      "train_flow_matching_loss": 0.236328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 19800,
      "train_flow_matching_loss": 0.11962890625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09317775070667267,
      "learning_rate": 5e-05,
      "loss": 0.1282,
      "step": 19810
    },
    {
      "step": 19810,
      "train_flow_matching_loss": 0.1298828125,
      "train_physics_loss": 0.0
    },
    {
      "step": 19810,
      "train_flow_matching_loss": 0.12060546875,
      "train_physics_loss": 0.0
    },
    {
      "step": 19810,
      "train_flow_matching_loss": 0.1103515625,
      "train_physics_loss": 0.0
    },
    {
      "step": 19810,
      "train_flow_matching_loss": 0.08203125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.07525462657213211,
      "learning_rate": 5e-05,
      "loss": 0.1262,
      "step": 19820
    },
    {
      "step": 19820,
      "train_flow_matching_loss": 0.08203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 19820,
      "train_flow_matching_loss": 0.10400390625,
      "train_physics_loss": 0.0
    },
    {
      "step": 19820,
      "train_flow_matching_loss": 0.2021484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 19820,
      "train_flow_matching_loss": 0.08837890625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.17168712615966797,
      "learning_rate": 5e-05,
      "loss": 0.1267,
      "step": 19830
    },
    {
      "step": 19830,
      "train_flow_matching_loss": 0.10302734375,
      "train_physics_loss": 0.0
    },
    {
      "step": 19830,
      "train_flow_matching_loss": 0.08544921875,
      "train_physics_loss": 0.0
    },
    {
      "step": 19830,
      "train_flow_matching_loss": 0.09033203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 19830,
      "train_flow_matching_loss": 0.1455078125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.0935906320810318,
      "learning_rate": 5e-05,
      "loss": 0.1294,
      "step": 19840
    },
    {
      "step": 19840,
      "train_flow_matching_loss": 0.11474609375,
      "train_physics_loss": 0.0
    },
    {
      "step": 19840,
      "train_flow_matching_loss": 0.173828125,
      "train_physics_loss": 0.0
    },
    {
      "step": 19840,
      "train_flow_matching_loss": 0.11865234375,
      "train_physics_loss": 0.0
    },
    {
      "step": 19840,
      "train_flow_matching_loss": 0.10205078125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09024344384670258,
      "learning_rate": 5e-05,
      "loss": 0.1333,
      "step": 19850
    },
    {
      "step": 19850,
      "train_flow_matching_loss": 0.11474609375,
      "train_physics_loss": 0.0
    },
    {
      "step": 19850,
      "train_flow_matching_loss": 0.1044921875,
      "train_physics_loss": 0.0
    },
    {
      "step": 19850,
      "train_flow_matching_loss": 0.1279296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 19850,
      "train_flow_matching_loss": 0.1513671875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10273157805204391,
      "learning_rate": 5e-05,
      "loss": 0.1305,
      "step": 19860
    },
    {
      "step": 19860,
      "train_flow_matching_loss": 0.142578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 19860,
      "train_flow_matching_loss": 0.09521484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 19860,
      "train_flow_matching_loss": 0.1416015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 19860,
      "train_flow_matching_loss": 0.09716796875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10484370589256287,
      "learning_rate": 5e-05,
      "loss": 0.1261,
      "step": 19870
    },
    {
      "step": 19870,
      "train_flow_matching_loss": 0.12158203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 19870,
      "train_flow_matching_loss": 0.134765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 19870,
      "train_flow_matching_loss": 0.1328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 19870,
      "train_flow_matching_loss": 0.0869140625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.06503795832395554,
      "learning_rate": 5e-05,
      "loss": 0.1267,
      "step": 19880
    },
    {
      "step": 19880,
      "train_flow_matching_loss": 0.1728515625,
      "train_physics_loss": 0.0
    },
    {
      "step": 19880,
      "train_flow_matching_loss": 0.0810546875,
      "train_physics_loss": 0.0
    },
    {
      "step": 19880,
      "train_flow_matching_loss": 0.12060546875,
      "train_physics_loss": 0.0
    },
    {
      "step": 19880,
      "train_flow_matching_loss": 0.1201171875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09380993992090225,
      "learning_rate": 5e-05,
      "loss": 0.1294,
      "step": 19890
    },
    {
      "step": 19890,
      "train_flow_matching_loss": 0.2001953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 19890,
      "train_flow_matching_loss": 0.189453125,
      "train_physics_loss": 0.0
    },
    {
      "step": 19890,
      "train_flow_matching_loss": 0.10791015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 19890,
      "train_flow_matching_loss": 0.1640625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.0812254399061203,
      "learning_rate": 5e-05,
      "loss": 0.128,
      "step": 19900
    },
    {
      "step": 19900,
      "train_flow_matching_loss": 0.1689453125,
      "train_physics_loss": 0.0
    },
    {
      "step": 19900,
      "train_flow_matching_loss": 0.1181640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 19900,
      "train_flow_matching_loss": 0.1416015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 19900,
      "train_flow_matching_loss": 0.2314453125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08629434555768967,
      "learning_rate": 5e-05,
      "loss": 0.1255,
      "step": 19910
    },
    {
      "step": 19910,
      "train_flow_matching_loss": 0.08642578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 19910,
      "train_flow_matching_loss": 0.1025390625,
      "train_physics_loss": 0.0
    },
    {
      "step": 19910,
      "train_flow_matching_loss": 0.1376953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 19910,
      "train_flow_matching_loss": 0.130859375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11714515835046768,
      "learning_rate": 5e-05,
      "loss": 0.1255,
      "step": 19920
    },
    {
      "step": 19920,
      "train_flow_matching_loss": 0.08349609375,
      "train_physics_loss": 0.0
    },
    {
      "step": 19920,
      "train_flow_matching_loss": 0.11865234375,
      "train_physics_loss": 0.0
    },
    {
      "step": 19920,
      "train_flow_matching_loss": 0.1337890625,
      "train_physics_loss": 0.0
    },
    {
      "step": 19920,
      "train_flow_matching_loss": 0.134765625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09671320766210556,
      "learning_rate": 5e-05,
      "loss": 0.1222,
      "step": 19930
    },
    {
      "step": 19930,
      "train_flow_matching_loss": 0.1748046875,
      "train_physics_loss": 0.0
    },
    {
      "step": 19930,
      "train_flow_matching_loss": 0.07177734375,
      "train_physics_loss": 0.0
    },
    {
      "step": 19930,
      "train_flow_matching_loss": 0.1357421875,
      "train_physics_loss": 0.0
    },
    {
      "step": 19930,
      "train_flow_matching_loss": 0.185546875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10591620206832886,
      "learning_rate": 5e-05,
      "loss": 0.1245,
      "step": 19940
    },
    {
      "step": 19940,
      "train_flow_matching_loss": 0.171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 19940,
      "train_flow_matching_loss": 0.1083984375,
      "train_physics_loss": 0.0
    },
    {
      "step": 19940,
      "train_flow_matching_loss": 0.1728515625,
      "train_physics_loss": 0.0
    },
    {
      "step": 19940,
      "train_flow_matching_loss": 0.1416015625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09047140181064606,
      "learning_rate": 5e-05,
      "loss": 0.1309,
      "step": 19950
    },
    {
      "step": 19950,
      "train_flow_matching_loss": 0.1669921875,
      "train_physics_loss": 0.0
    },
    {
      "step": 19950,
      "train_flow_matching_loss": 0.09228515625,
      "train_physics_loss": 0.0
    },
    {
      "step": 19950,
      "train_flow_matching_loss": 0.06494140625,
      "train_physics_loss": 0.0
    },
    {
      "step": 19950,
      "train_flow_matching_loss": 0.1875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08316781371831894,
      "learning_rate": 5e-05,
      "loss": 0.1232,
      "step": 19960
    },
    {
      "step": 19960,
      "train_flow_matching_loss": 0.1904296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 19960,
      "train_flow_matching_loss": 0.1259765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 19960,
      "train_flow_matching_loss": 0.11572265625,
      "train_physics_loss": 0.0
    },
    {
      "step": 19960,
      "train_flow_matching_loss": 0.162109375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08039841055870056,
      "learning_rate": 5e-05,
      "loss": 0.1287,
      "step": 19970
    },
    {
      "step": 19970,
      "train_flow_matching_loss": 0.142578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 19970,
      "train_flow_matching_loss": 0.09423828125,
      "train_physics_loss": 0.0
    },
    {
      "step": 19970,
      "train_flow_matching_loss": 0.2333984375,
      "train_physics_loss": 0.0
    },
    {
      "step": 19970,
      "train_flow_matching_loss": 0.177734375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.0939137190580368,
      "learning_rate": 5e-05,
      "loss": 0.1248,
      "step": 19980
    },
    {
      "step": 19980,
      "train_flow_matching_loss": 0.1484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 19980,
      "train_flow_matching_loss": 0.1181640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 19980,
      "train_flow_matching_loss": 0.09521484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 19980,
      "train_flow_matching_loss": 0.205078125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08797077089548111,
      "learning_rate": 5e-05,
      "loss": 0.1276,
      "step": 19990
    },
    {
      "step": 19990,
      "train_flow_matching_loss": 0.271484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 19990,
      "train_flow_matching_loss": 0.045654296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 19990,
      "train_flow_matching_loss": 0.10595703125,
      "train_physics_loss": 0.0
    },
    {
      "step": 19990,
      "train_flow_matching_loss": 0.1728515625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.3165603578090668,
      "learning_rate": 5e-05,
      "loss": 0.1251,
      "step": 20000
    },
    {
      "step": 20000,
      "train_flow_matching_loss": 0.2470703125,
      "train_physics_loss": 0.0
    },
    {
      "step": 20000,
      "train_flow_matching_loss": 0.06396484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 20000,
      "train_flow_matching_loss": 0.1416015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 20000,
      "train_flow_matching_loss": 0.162109375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09259036928415298,
      "learning_rate": 5e-05,
      "loss": 0.1279,
      "step": 20010
    },
    {
      "step": 20010,
      "train_flow_matching_loss": 0.201171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 20010,
      "train_flow_matching_loss": 0.1318359375,
      "train_physics_loss": 0.0
    },
    {
      "step": 20010,
      "train_flow_matching_loss": 0.0927734375,
      "train_physics_loss": 0.0
    },
    {
      "step": 20010,
      "train_flow_matching_loss": 0.1064453125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.12280681729316711,
      "learning_rate": 5e-05,
      "loss": 0.1248,
      "step": 20020
    },
    {
      "step": 20020,
      "train_flow_matching_loss": 0.15625,
      "train_physics_loss": 0.0
    },
    {
      "step": 20020,
      "train_flow_matching_loss": 0.1650390625,
      "train_physics_loss": 0.0
    },
    {
      "step": 20020,
      "train_flow_matching_loss": 0.1708984375,
      "train_physics_loss": 0.0
    },
    {
      "step": 20020,
      "train_flow_matching_loss": 0.15625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.06495514512062073,
      "learning_rate": 5e-05,
      "loss": 0.1264,
      "step": 20030
    },
    {
      "step": 20030,
      "train_flow_matching_loss": 0.255859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 20030,
      "train_flow_matching_loss": 0.0908203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 20030,
      "train_flow_matching_loss": 0.146484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 20030,
      "train_flow_matching_loss": 0.1875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09279527515172958,
      "learning_rate": 5e-05,
      "loss": 0.1288,
      "step": 20040
    },
    {
      "step": 20040,
      "train_flow_matching_loss": 0.166015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 20040,
      "train_flow_matching_loss": 0.1083984375,
      "train_physics_loss": 0.0
    },
    {
      "step": 20040,
      "train_flow_matching_loss": 0.142578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 20040,
      "train_flow_matching_loss": 0.25390625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.18214595317840576,
      "learning_rate": 5e-05,
      "loss": 0.1238,
      "step": 20050
    },
    {
      "step": 20050,
      "train_flow_matching_loss": 0.240234375,
      "train_physics_loss": 0.0
    },
    {
      "step": 20050,
      "train_flow_matching_loss": 0.1435546875,
      "train_physics_loss": 0.0
    },
    {
      "step": 20050,
      "train_flow_matching_loss": 0.08544921875,
      "train_physics_loss": 0.0
    },
    {
      "step": 20050,
      "train_flow_matching_loss": 0.1181640625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.12867064774036407,
      "learning_rate": 5e-05,
      "loss": 0.1293,
      "step": 20060
    },
    {
      "step": 20060,
      "train_flow_matching_loss": 0.25390625,
      "train_physics_loss": 0.0
    },
    {
      "step": 20060,
      "train_flow_matching_loss": 0.1376953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 20060,
      "train_flow_matching_loss": 0.1005859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 20060,
      "train_flow_matching_loss": 0.1337890625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11773832142353058,
      "learning_rate": 5e-05,
      "loss": 0.1297,
      "step": 20070
    },
    {
      "step": 20070,
      "train_flow_matching_loss": 0.224609375,
      "train_physics_loss": 0.0
    },
    {
      "step": 20070,
      "train_flow_matching_loss": 0.1689453125,
      "train_physics_loss": 0.0
    },
    {
      "step": 20070,
      "train_flow_matching_loss": 0.142578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 20070,
      "train_flow_matching_loss": 0.142578125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.07548905909061432,
      "learning_rate": 5e-05,
      "loss": 0.1242,
      "step": 20080
    },
    {
      "step": 20080,
      "train_flow_matching_loss": 0.1904296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 20080,
      "train_flow_matching_loss": 0.1708984375,
      "train_physics_loss": 0.0
    },
    {
      "step": 20080,
      "train_flow_matching_loss": 0.09375,
      "train_physics_loss": 0.0
    },
    {
      "step": 20080,
      "train_flow_matching_loss": 0.1796875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08007221668958664,
      "learning_rate": 5e-05,
      "loss": 0.1273,
      "step": 20090
    },
    {
      "step": 20090,
      "train_flow_matching_loss": 0.1318359375,
      "train_physics_loss": 0.0
    },
    {
      "step": 20090,
      "train_flow_matching_loss": 0.193359375,
      "train_physics_loss": 0.0
    },
    {
      "step": 20090,
      "train_flow_matching_loss": 0.1015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 20090,
      "train_flow_matching_loss": 0.12255859375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.5181427597999573,
      "learning_rate": 5e-05,
      "loss": 0.1262,
      "step": 20100
    },
    {
      "step": 20100,
      "train_flow_matching_loss": 0.140625,
      "train_physics_loss": 0.0
    },
    {
      "step": 20100,
      "train_flow_matching_loss": 0.10986328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 20100,
      "train_flow_matching_loss": 0.1044921875,
      "train_physics_loss": 0.0
    },
    {
      "step": 20100,
      "train_flow_matching_loss": 0.1240234375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.12422988563776016,
      "learning_rate": 5e-05,
      "loss": 0.1249,
      "step": 20110
    },
    {
      "step": 20110,
      "train_flow_matching_loss": 0.1796875,
      "train_physics_loss": 0.0
    },
    {
      "step": 20110,
      "train_flow_matching_loss": 0.1689453125,
      "train_physics_loss": 0.0
    },
    {
      "step": 20110,
      "train_flow_matching_loss": 0.10693359375,
      "train_physics_loss": 0.0
    },
    {
      "step": 20110,
      "train_flow_matching_loss": 0.158203125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.07712562382221222,
      "learning_rate": 5e-05,
      "loss": 0.1255,
      "step": 20120
    },
    {
      "step": 20120,
      "train_flow_matching_loss": 0.1669921875,
      "train_physics_loss": 0.0
    },
    {
      "step": 20120,
      "train_flow_matching_loss": 0.1845703125,
      "train_physics_loss": 0.0
    },
    {
      "step": 20120,
      "train_flow_matching_loss": 0.0908203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 20120,
      "train_flow_matching_loss": 0.1904296875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.17336852848529816,
      "learning_rate": 5e-05,
      "loss": 0.1258,
      "step": 20130
    },
    {
      "step": 20130,
      "train_flow_matching_loss": 0.11865234375,
      "train_physics_loss": 0.0
    },
    {
      "step": 20130,
      "train_flow_matching_loss": 0.173828125,
      "train_physics_loss": 0.0
    },
    {
      "step": 20130,
      "train_flow_matching_loss": 0.134765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 20130,
      "train_flow_matching_loss": 0.1630859375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.07590607553720474,
      "learning_rate": 5e-05,
      "loss": 0.1311,
      "step": 20140
    },
    {
      "step": 20140,
      "train_flow_matching_loss": 0.240234375,
      "train_physics_loss": 0.0
    },
    {
      "step": 20140,
      "train_flow_matching_loss": 0.1650390625,
      "train_physics_loss": 0.0
    },
    {
      "step": 20140,
      "train_flow_matching_loss": 0.08837890625,
      "train_physics_loss": 0.0
    },
    {
      "step": 20140,
      "train_flow_matching_loss": 0.193359375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10121916234493256,
      "learning_rate": 5e-05,
      "loss": 0.1302,
      "step": 20150
    },
    {
      "step": 20150,
      "train_flow_matching_loss": 0.2080078125,
      "train_physics_loss": 0.0
    },
    {
      "step": 20150,
      "train_flow_matching_loss": 0.1767578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 20150,
      "train_flow_matching_loss": 0.138671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 20150,
      "train_flow_matching_loss": 0.1220703125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09048888087272644,
      "learning_rate": 5e-05,
      "loss": 0.1314,
      "step": 20160
    },
    {
      "step": 20160,
      "train_flow_matching_loss": 0.1455078125,
      "train_physics_loss": 0.0
    },
    {
      "step": 20160,
      "train_flow_matching_loss": 0.142578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 20160,
      "train_flow_matching_loss": 0.1162109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 20160,
      "train_flow_matching_loss": 0.1796875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10542362928390503,
      "learning_rate": 5e-05,
      "loss": 0.128,
      "step": 20170
    },
    {
      "step": 20170,
      "train_flow_matching_loss": 0.2099609375,
      "train_physics_loss": 0.0
    },
    {
      "step": 20170,
      "train_flow_matching_loss": 0.1259765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 20170,
      "train_flow_matching_loss": 0.0927734375,
      "train_physics_loss": 0.0
    },
    {
      "step": 20170,
      "train_flow_matching_loss": 0.1513671875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.40084701776504517,
      "learning_rate": 5e-05,
      "loss": 0.1322,
      "step": 20180
    },
    {
      "step": 20180,
      "train_flow_matching_loss": 0.1953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 20180,
      "train_flow_matching_loss": 0.1376953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 20180,
      "train_flow_matching_loss": 0.12109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 20180,
      "train_flow_matching_loss": 0.173828125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08118773996829987,
      "learning_rate": 5e-05,
      "loss": 0.1312,
      "step": 20190
    },
    {
      "step": 20190,
      "train_flow_matching_loss": 0.232421875,
      "train_physics_loss": 0.0
    },
    {
      "step": 20190,
      "train_flow_matching_loss": 0.11181640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 20190,
      "train_flow_matching_loss": 0.109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 20190,
      "train_flow_matching_loss": 0.13671875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11063332855701447,
      "learning_rate": 5e-05,
      "loss": 0.1352,
      "step": 20200
    },
    {
      "step": 20200,
      "train_flow_matching_loss": 0.130859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 20200,
      "train_flow_matching_loss": 0.1513671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 20200,
      "train_flow_matching_loss": 0.1259765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 20200,
      "train_flow_matching_loss": 0.1005859375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08047238737344742,
      "learning_rate": 5e-05,
      "loss": 0.1322,
      "step": 20210
    },
    {
      "step": 20210,
      "train_flow_matching_loss": 0.2119140625,
      "train_physics_loss": 0.0
    },
    {
      "step": 20210,
      "train_flow_matching_loss": 0.09130859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 20210,
      "train_flow_matching_loss": 0.0810546875,
      "train_physics_loss": 0.0
    },
    {
      "step": 20210,
      "train_flow_matching_loss": 0.10595703125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08404053002595901,
      "learning_rate": 5e-05,
      "loss": 0.1303,
      "step": 20220
    },
    {
      "step": 20220,
      "train_flow_matching_loss": 0.158203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 20220,
      "train_flow_matching_loss": 0.043212890625,
      "train_physics_loss": 0.0
    },
    {
      "step": 20220,
      "train_flow_matching_loss": 0.1943359375,
      "train_physics_loss": 0.0
    },
    {
      "step": 20220,
      "train_flow_matching_loss": 0.1220703125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10916363447904587,
      "learning_rate": 5e-05,
      "loss": 0.128,
      "step": 20230
    },
    {
      "step": 20230,
      "train_flow_matching_loss": 0.181640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 20230,
      "train_flow_matching_loss": 0.251953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 20230,
      "train_flow_matching_loss": 0.1533203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 20230,
      "train_flow_matching_loss": 0.10791015625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.30368709564208984,
      "learning_rate": 5e-05,
      "loss": 0.1304,
      "step": 20240
    },
    {
      "step": 20240,
      "train_flow_matching_loss": 0.1474609375,
      "train_physics_loss": 0.0
    },
    {
      "step": 20240,
      "train_flow_matching_loss": 0.1005859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 20240,
      "train_flow_matching_loss": 0.11083984375,
      "train_physics_loss": 0.0
    },
    {
      "step": 20240,
      "train_flow_matching_loss": 0.10302734375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.21000070869922638,
      "learning_rate": 5e-05,
      "loss": 0.1275,
      "step": 20250
    },
    {
      "step": 20250,
      "train_flow_matching_loss": 0.27734375,
      "train_physics_loss": 0.0
    },
    {
      "step": 20250,
      "train_flow_matching_loss": 0.142578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 20250,
      "train_flow_matching_loss": 0.10107421875,
      "train_physics_loss": 0.0
    },
    {
      "step": 20250,
      "train_flow_matching_loss": 0.1494140625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.1255851835012436,
      "learning_rate": 5e-05,
      "loss": 0.1335,
      "step": 20260
    },
    {
      "step": 20260,
      "train_flow_matching_loss": 0.1396484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 20260,
      "train_flow_matching_loss": 0.11474609375,
      "train_physics_loss": 0.0
    },
    {
      "step": 20260,
      "train_flow_matching_loss": 0.10693359375,
      "train_physics_loss": 0.0
    },
    {
      "step": 20260,
      "train_flow_matching_loss": 0.138671875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08204317837953568,
      "learning_rate": 5e-05,
      "loss": 0.1291,
      "step": 20270
    },
    {
      "step": 20270,
      "train_flow_matching_loss": 0.1103515625,
      "train_physics_loss": 0.0
    },
    {
      "step": 20270,
      "train_flow_matching_loss": 0.162109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 20270,
      "train_flow_matching_loss": 0.1591796875,
      "train_physics_loss": 0.0
    },
    {
      "step": 20270,
      "train_flow_matching_loss": 0.15625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.1646389216184616,
      "learning_rate": 5e-05,
      "loss": 0.1266,
      "step": 20280
    },
    {
      "step": 20280,
      "train_flow_matching_loss": 0.1318359375,
      "train_physics_loss": 0.0
    },
    {
      "step": 20280,
      "train_flow_matching_loss": 0.06298828125,
      "train_physics_loss": 0.0
    },
    {
      "step": 20280,
      "train_flow_matching_loss": 0.11279296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 20280,
      "train_flow_matching_loss": 0.1337890625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.12318049371242523,
      "learning_rate": 5e-05,
      "loss": 0.1275,
      "step": 20290
    },
    {
      "step": 20290,
      "train_flow_matching_loss": 0.2021484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 20290,
      "train_flow_matching_loss": 0.1240234375,
      "train_physics_loss": 0.0
    },
    {
      "step": 20290,
      "train_flow_matching_loss": 0.109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 20290,
      "train_flow_matching_loss": 0.203125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.1277288943529129,
      "learning_rate": 5e-05,
      "loss": 0.1292,
      "step": 20300
    },
    {
      "step": 20300,
      "train_flow_matching_loss": 0.10107421875,
      "train_physics_loss": 0.0
    },
    {
      "step": 20300,
      "train_flow_matching_loss": 0.146484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 20300,
      "train_flow_matching_loss": 0.12451171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 20300,
      "train_flow_matching_loss": 0.2021484375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.32255831360816956,
      "learning_rate": 5e-05,
      "loss": 0.1279,
      "step": 20310
    },
    {
      "step": 20310,
      "train_flow_matching_loss": 0.0927734375,
      "train_physics_loss": 0.0
    },
    {
      "step": 20310,
      "train_flow_matching_loss": 0.0966796875,
      "train_physics_loss": 0.0
    },
    {
      "step": 20310,
      "train_flow_matching_loss": 0.15625,
      "train_physics_loss": 0.0
    },
    {
      "step": 20310,
      "train_flow_matching_loss": 0.171875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.07316431403160095,
      "learning_rate": 5e-05,
      "loss": 0.1297,
      "step": 20320
    },
    {
      "step": 20320,
      "train_flow_matching_loss": 0.138671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 20320,
      "train_flow_matching_loss": 0.11572265625,
      "train_physics_loss": 0.0
    },
    {
      "step": 20320,
      "train_flow_matching_loss": 0.12255859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 20320,
      "train_flow_matching_loss": 0.1044921875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.07055330276489258,
      "learning_rate": 5e-05,
      "loss": 0.1295,
      "step": 20330
    },
    {
      "step": 20330,
      "train_flow_matching_loss": 0.1953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 20330,
      "train_flow_matching_loss": 0.2451171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 20330,
      "train_flow_matching_loss": 0.1044921875,
      "train_physics_loss": 0.0
    },
    {
      "step": 20330,
      "train_flow_matching_loss": 0.10205078125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09583035856485367,
      "learning_rate": 5e-05,
      "loss": 0.1343,
      "step": 20340
    },
    {
      "step": 20340,
      "train_flow_matching_loss": 0.1220703125,
      "train_physics_loss": 0.0
    },
    {
      "step": 20340,
      "train_flow_matching_loss": 0.08447265625,
      "train_physics_loss": 0.0
    },
    {
      "step": 20340,
      "train_flow_matching_loss": 0.07958984375,
      "train_physics_loss": 0.0
    },
    {
      "step": 20340,
      "train_flow_matching_loss": 0.275390625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10306263715028763,
      "learning_rate": 5e-05,
      "loss": 0.1297,
      "step": 20350
    },
    {
      "step": 20350,
      "train_flow_matching_loss": 0.134765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 20350,
      "train_flow_matching_loss": 0.169921875,
      "train_physics_loss": 0.0
    },
    {
      "step": 20350,
      "train_flow_matching_loss": 0.11181640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 20350,
      "train_flow_matching_loss": 0.1416015625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.0930974930524826,
      "learning_rate": 5e-05,
      "loss": 0.1296,
      "step": 20360
    },
    {
      "step": 20360,
      "train_flow_matching_loss": 0.146484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 20360,
      "train_flow_matching_loss": 0.15625,
      "train_physics_loss": 0.0
    },
    {
      "step": 20360,
      "train_flow_matching_loss": 0.1357421875,
      "train_physics_loss": 0.0
    },
    {
      "step": 20360,
      "train_flow_matching_loss": 0.08935546875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.1059870794415474,
      "learning_rate": 5e-05,
      "loss": 0.1263,
      "step": 20370
    },
    {
      "step": 20370,
      "train_flow_matching_loss": 0.142578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 20370,
      "train_flow_matching_loss": 0.1650390625,
      "train_physics_loss": 0.0
    },
    {
      "step": 20370,
      "train_flow_matching_loss": 0.1533203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 20370,
      "train_flow_matching_loss": 0.1611328125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.1039142981171608,
      "learning_rate": 5e-05,
      "loss": 0.1271,
      "step": 20380
    },
    {
      "step": 20380,
      "train_flow_matching_loss": 0.11328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 20380,
      "train_flow_matching_loss": 0.1552734375,
      "train_physics_loss": 0.0
    },
    {
      "step": 20380,
      "train_flow_matching_loss": 0.1259765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 20380,
      "train_flow_matching_loss": 0.07958984375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08498284220695496,
      "learning_rate": 5e-05,
      "loss": 0.1277,
      "step": 20390
    },
    {
      "step": 20390,
      "train_flow_matching_loss": 0.1376953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 20390,
      "train_flow_matching_loss": 0.1298828125,
      "train_physics_loss": 0.0
    },
    {
      "step": 20390,
      "train_flow_matching_loss": 0.142578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 20390,
      "train_flow_matching_loss": 0.11279296875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10912338644266129,
      "learning_rate": 5e-05,
      "loss": 0.1264,
      "step": 20400
    },
    {
      "step": 20400,
      "train_flow_matching_loss": 0.11474609375,
      "train_physics_loss": 0.0
    },
    {
      "step": 20400,
      "train_flow_matching_loss": 0.150390625,
      "train_physics_loss": 0.0
    },
    {
      "step": 20400,
      "train_flow_matching_loss": 0.1435546875,
      "train_physics_loss": 0.0
    },
    {
      "step": 20400,
      "train_flow_matching_loss": 0.1416015625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.0915747582912445,
      "learning_rate": 5e-05,
      "loss": 0.1289,
      "step": 20410
    },
    {
      "step": 20410,
      "train_flow_matching_loss": 0.0869140625,
      "train_physics_loss": 0.0
    },
    {
      "step": 20410,
      "train_flow_matching_loss": 0.205078125,
      "train_physics_loss": 0.0
    },
    {
      "step": 20410,
      "train_flow_matching_loss": 0.1552734375,
      "train_physics_loss": 0.0
    },
    {
      "step": 20410,
      "train_flow_matching_loss": 0.2060546875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08929888159036636,
      "learning_rate": 5e-05,
      "loss": 0.1257,
      "step": 20420
    },
    {
      "step": 20420,
      "train_flow_matching_loss": 0.130859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 20420,
      "train_flow_matching_loss": 0.1494140625,
      "train_physics_loss": 0.0
    },
    {
      "step": 20420,
      "train_flow_matching_loss": 0.181640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 20420,
      "train_flow_matching_loss": 0.1865234375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09058790653944016,
      "learning_rate": 5e-05,
      "loss": 0.127,
      "step": 20430
    },
    {
      "step": 20430,
      "train_flow_matching_loss": 0.12060546875,
      "train_physics_loss": 0.0
    },
    {
      "step": 20430,
      "train_flow_matching_loss": 0.11328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 20430,
      "train_flow_matching_loss": 0.1962890625,
      "train_physics_loss": 0.0
    },
    {
      "step": 20430,
      "train_flow_matching_loss": 0.2060546875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10175210237503052,
      "learning_rate": 5e-05,
      "loss": 0.1297,
      "step": 20440
    },
    {
      "step": 20440,
      "train_flow_matching_loss": 0.1396484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 20440,
      "train_flow_matching_loss": 0.12890625,
      "train_physics_loss": 0.0
    },
    {
      "step": 20440,
      "train_flow_matching_loss": 0.125,
      "train_physics_loss": 0.0
    },
    {
      "step": 20440,
      "train_flow_matching_loss": 0.287109375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09962885081768036,
      "learning_rate": 5e-05,
      "loss": 0.125,
      "step": 20450
    },
    {
      "step": 20450,
      "train_flow_matching_loss": 0.216796875,
      "train_physics_loss": 0.0
    },
    {
      "step": 20450,
      "train_flow_matching_loss": 0.146484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 20450,
      "train_flow_matching_loss": 0.1259765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 20450,
      "train_flow_matching_loss": 0.1865234375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09543739259243011,
      "learning_rate": 5e-05,
      "loss": 0.1269,
      "step": 20460
    },
    {
      "step": 20460,
      "train_flow_matching_loss": 0.11328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 20460,
      "train_flow_matching_loss": 0.1533203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 20460,
      "train_flow_matching_loss": 0.09375,
      "train_physics_loss": 0.0
    },
    {
      "step": 20460,
      "train_flow_matching_loss": 0.0966796875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11481272429227829,
      "learning_rate": 5e-05,
      "loss": 0.1274,
      "step": 20470
    },
    {
      "step": 20470,
      "train_flow_matching_loss": 0.1396484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 20470,
      "train_flow_matching_loss": 0.1650390625,
      "train_physics_loss": 0.0
    },
    {
      "step": 20470,
      "train_flow_matching_loss": 0.1337890625,
      "train_physics_loss": 0.0
    },
    {
      "step": 20470,
      "train_flow_matching_loss": 0.177734375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.0907912477850914,
      "learning_rate": 5e-05,
      "loss": 0.1234,
      "step": 20480
    },
    {
      "step": 20480,
      "train_flow_matching_loss": 0.10546875,
      "train_physics_loss": 0.0
    },
    {
      "step": 20480,
      "train_flow_matching_loss": 0.1787109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 20480,
      "train_flow_matching_loss": 0.1845703125,
      "train_physics_loss": 0.0
    },
    {
      "step": 20480,
      "train_flow_matching_loss": 0.2490234375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11995331197977066,
      "learning_rate": 5e-05,
      "loss": 0.1253,
      "step": 20490
    },
    {
      "step": 20490,
      "train_flow_matching_loss": 0.1533203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 20490,
      "train_flow_matching_loss": 0.1171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 20490,
      "train_flow_matching_loss": 0.11474609375,
      "train_physics_loss": 0.0
    },
    {
      "step": 20490,
      "train_flow_matching_loss": 0.11328125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10168752819299698,
      "learning_rate": 5e-05,
      "loss": 0.1231,
      "step": 20500
    },
    {
      "step": 20500,
      "train_flow_matching_loss": 0.1923828125,
      "train_physics_loss": 0.0
    },
    {
      "step": 20500,
      "train_flow_matching_loss": 0.1064453125,
      "train_physics_loss": 0.0
    },
    {
      "step": 20500,
      "train_flow_matching_loss": 0.09619140625,
      "train_physics_loss": 0.0
    },
    {
      "step": 20500,
      "train_flow_matching_loss": 0.1552734375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.07008050382137299,
      "learning_rate": 5e-05,
      "loss": 0.1316,
      "step": 20510
    },
    {
      "step": 20510,
      "train_flow_matching_loss": 0.125,
      "train_physics_loss": 0.0
    },
    {
      "step": 20510,
      "train_flow_matching_loss": 0.1533203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 20510,
      "train_flow_matching_loss": 0.1455078125,
      "train_physics_loss": 0.0
    },
    {
      "step": 20510,
      "train_flow_matching_loss": 0.1572265625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.14762862026691437,
      "learning_rate": 5e-05,
      "loss": 0.1256,
      "step": 20520
    },
    {
      "step": 20520,
      "train_flow_matching_loss": 0.1982421875,
      "train_physics_loss": 0.0
    },
    {
      "step": 20520,
      "train_flow_matching_loss": 0.09716796875,
      "train_physics_loss": 0.0
    },
    {
      "step": 20520,
      "train_flow_matching_loss": 0.2001953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 20520,
      "train_flow_matching_loss": 0.125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08353471755981445,
      "learning_rate": 5e-05,
      "loss": 0.1274,
      "step": 20530
    },
    {
      "step": 20530,
      "train_flow_matching_loss": 0.1083984375,
      "train_physics_loss": 0.0
    },
    {
      "step": 20530,
      "train_flow_matching_loss": 0.1923828125,
      "train_physics_loss": 0.0
    },
    {
      "step": 20530,
      "train_flow_matching_loss": 0.09716796875,
      "train_physics_loss": 0.0
    },
    {
      "step": 20530,
      "train_flow_matching_loss": 0.12451171875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.1112353652715683,
      "learning_rate": 5e-05,
      "loss": 0.1291,
      "step": 20540
    },
    {
      "step": 20540,
      "train_flow_matching_loss": 0.08544921875,
      "train_physics_loss": 0.0
    },
    {
      "step": 20540,
      "train_flow_matching_loss": 0.142578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 20540,
      "train_flow_matching_loss": 0.08251953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 20540,
      "train_flow_matching_loss": 0.2119140625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.06896704435348511,
      "learning_rate": 5e-05,
      "loss": 0.1238,
      "step": 20550
    },
    {
      "step": 20550,
      "train_flow_matching_loss": 0.08349609375,
      "train_physics_loss": 0.0
    },
    {
      "step": 20550,
      "train_flow_matching_loss": 0.1435546875,
      "train_physics_loss": 0.0
    },
    {
      "step": 20550,
      "train_flow_matching_loss": 0.193359375,
      "train_physics_loss": 0.0
    },
    {
      "step": 20550,
      "train_flow_matching_loss": 0.154296875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10464230924844742,
      "learning_rate": 5e-05,
      "loss": 0.1283,
      "step": 20560
    },
    {
      "step": 20560,
      "train_flow_matching_loss": 0.10400390625,
      "train_physics_loss": 0.0
    },
    {
      "step": 20560,
      "train_flow_matching_loss": 0.205078125,
      "train_physics_loss": 0.0
    },
    {
      "step": 20560,
      "train_flow_matching_loss": 0.1875,
      "train_physics_loss": 0.0
    },
    {
      "step": 20560,
      "train_flow_matching_loss": 0.2138671875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09093734622001648,
      "learning_rate": 5e-05,
      "loss": 0.1285,
      "step": 20570
    },
    {
      "step": 20570,
      "train_flow_matching_loss": 0.2275390625,
      "train_physics_loss": 0.0
    },
    {
      "step": 20570,
      "train_flow_matching_loss": 0.201171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 20570,
      "train_flow_matching_loss": 0.130859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 20570,
      "train_flow_matching_loss": 0.2421875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09558191150426865,
      "learning_rate": 5e-05,
      "loss": 0.1311,
      "step": 20580
    },
    {
      "step": 20580,
      "train_flow_matching_loss": 0.1240234375,
      "train_physics_loss": 0.0
    },
    {
      "step": 20580,
      "train_flow_matching_loss": 0.146484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 20580,
      "train_flow_matching_loss": 0.109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 20580,
      "train_flow_matching_loss": 0.228515625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.0986843854188919,
      "learning_rate": 5e-05,
      "loss": 0.1235,
      "step": 20590
    },
    {
      "step": 20590,
      "train_flow_matching_loss": 0.09765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 20590,
      "train_flow_matching_loss": 0.1865234375,
      "train_physics_loss": 0.0
    },
    {
      "step": 20590,
      "train_flow_matching_loss": 0.169921875,
      "train_physics_loss": 0.0
    },
    {
      "step": 20590,
      "train_flow_matching_loss": 0.07470703125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.1474790871143341,
      "learning_rate": 5e-05,
      "loss": 0.1304,
      "step": 20600
    },
    {
      "step": 20600,
      "train_flow_matching_loss": 0.146484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 20600,
      "train_flow_matching_loss": 0.0927734375,
      "train_physics_loss": 0.0
    },
    {
      "step": 20600,
      "train_flow_matching_loss": 0.09521484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 20600,
      "train_flow_matching_loss": 0.1611328125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11239366978406906,
      "learning_rate": 5e-05,
      "loss": 0.1315,
      "step": 20610
    },
    {
      "step": 20610,
      "train_flow_matching_loss": 0.1484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 20610,
      "train_flow_matching_loss": 0.173828125,
      "train_physics_loss": 0.0
    },
    {
      "step": 20610,
      "train_flow_matching_loss": 0.1845703125,
      "train_physics_loss": 0.0
    },
    {
      "step": 20610,
      "train_flow_matching_loss": 0.1630859375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.1154150739312172,
      "learning_rate": 5e-05,
      "loss": 0.1253,
      "step": 20620
    },
    {
      "step": 20620,
      "train_flow_matching_loss": 0.25390625,
      "train_physics_loss": 0.0
    },
    {
      "step": 20620,
      "train_flow_matching_loss": 0.1787109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 20620,
      "train_flow_matching_loss": 0.1416015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 20620,
      "train_flow_matching_loss": 0.185546875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09613265842199326,
      "learning_rate": 5e-05,
      "loss": 0.1253,
      "step": 20630
    },
    {
      "step": 20630,
      "train_flow_matching_loss": 0.1279296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 20630,
      "train_flow_matching_loss": 0.1513671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 20630,
      "train_flow_matching_loss": 0.13671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 20630,
      "train_flow_matching_loss": 0.125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08420579135417938,
      "learning_rate": 5e-05,
      "loss": 0.1257,
      "step": 20640
    },
    {
      "step": 20640,
      "train_flow_matching_loss": 0.2041015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 20640,
      "train_flow_matching_loss": 0.09228515625,
      "train_physics_loss": 0.0
    },
    {
      "step": 20640,
      "train_flow_matching_loss": 0.10009765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 20640,
      "train_flow_matching_loss": 0.11962890625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.07976025342941284,
      "learning_rate": 5e-05,
      "loss": 0.1284,
      "step": 20650
    },
    {
      "step": 20650,
      "train_flow_matching_loss": 0.1337890625,
      "train_physics_loss": 0.0
    },
    {
      "step": 20650,
      "train_flow_matching_loss": 0.1474609375,
      "train_physics_loss": 0.0
    },
    {
      "step": 20650,
      "train_flow_matching_loss": 0.146484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 20650,
      "train_flow_matching_loss": 0.244140625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08316649496555328,
      "learning_rate": 5e-05,
      "loss": 0.1304,
      "step": 20660
    },
    {
      "step": 20660,
      "train_flow_matching_loss": 0.1552734375,
      "train_physics_loss": 0.0
    },
    {
      "step": 20660,
      "train_flow_matching_loss": 0.09375,
      "train_physics_loss": 0.0
    },
    {
      "step": 20660,
      "train_flow_matching_loss": 0.1259765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 20660,
      "train_flow_matching_loss": 0.10400390625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10517945885658264,
      "learning_rate": 5e-05,
      "loss": 0.1297,
      "step": 20670
    },
    {
      "step": 20670,
      "train_flow_matching_loss": 0.0732421875,
      "train_physics_loss": 0.0
    },
    {
      "step": 20670,
      "train_flow_matching_loss": 0.1533203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 20670,
      "train_flow_matching_loss": 0.09375,
      "train_physics_loss": 0.0
    },
    {
      "step": 20670,
      "train_flow_matching_loss": 0.11669921875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.1161646842956543,
      "learning_rate": 5e-05,
      "loss": 0.1279,
      "step": 20680
    },
    {
      "step": 20680,
      "train_flow_matching_loss": 0.154296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 20680,
      "train_flow_matching_loss": 0.1513671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 20680,
      "train_flow_matching_loss": 0.10693359375,
      "train_physics_loss": 0.0
    },
    {
      "step": 20680,
      "train_flow_matching_loss": 0.201171875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.1026797890663147,
      "learning_rate": 5e-05,
      "loss": 0.1316,
      "step": 20690
    },
    {
      "step": 20690,
      "train_flow_matching_loss": 0.185546875,
      "train_physics_loss": 0.0
    },
    {
      "step": 20690,
      "train_flow_matching_loss": 0.1689453125,
      "train_physics_loss": 0.0
    },
    {
      "step": 20690,
      "train_flow_matching_loss": 0.10009765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 20690,
      "train_flow_matching_loss": 0.1748046875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11597111821174622,
      "learning_rate": 5e-05,
      "loss": 0.1288,
      "step": 20700
    },
    {
      "step": 20700,
      "train_flow_matching_loss": 0.09912109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 20700,
      "train_flow_matching_loss": 0.1201171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 20700,
      "train_flow_matching_loss": 0.10986328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 20700,
      "train_flow_matching_loss": 0.10888671875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.12162834405899048,
      "learning_rate": 5e-05,
      "loss": 0.1271,
      "step": 20710
    },
    {
      "step": 20710,
      "train_flow_matching_loss": 0.1787109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 20710,
      "train_flow_matching_loss": 0.07958984375,
      "train_physics_loss": 0.0
    },
    {
      "step": 20710,
      "train_flow_matching_loss": 0.1494140625,
      "train_physics_loss": 0.0
    },
    {
      "step": 20710,
      "train_flow_matching_loss": 0.08349609375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08602569997310638,
      "learning_rate": 5e-05,
      "loss": 0.1302,
      "step": 20720
    },
    {
      "step": 20720,
      "train_flow_matching_loss": 0.1533203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 20720,
      "train_flow_matching_loss": 0.12109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 20720,
      "train_flow_matching_loss": 0.0810546875,
      "train_physics_loss": 0.0
    },
    {
      "step": 20720,
      "train_flow_matching_loss": 0.1298828125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10859501361846924,
      "learning_rate": 5e-05,
      "loss": 0.1297,
      "step": 20730
    },
    {
      "step": 20730,
      "train_flow_matching_loss": 0.12158203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 20730,
      "train_flow_matching_loss": 0.1376953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 20730,
      "train_flow_matching_loss": 0.11328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 20730,
      "train_flow_matching_loss": 0.1259765625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11124545335769653,
      "learning_rate": 5e-05,
      "loss": 0.1265,
      "step": 20740
    },
    {
      "step": 20740,
      "train_flow_matching_loss": 0.1044921875,
      "train_physics_loss": 0.0
    },
    {
      "step": 20740,
      "train_flow_matching_loss": 0.06982421875,
      "train_physics_loss": 0.0
    },
    {
      "step": 20740,
      "train_flow_matching_loss": 0.171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 20740,
      "train_flow_matching_loss": 0.1982421875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11091964691877365,
      "learning_rate": 5e-05,
      "loss": 0.1274,
      "step": 20750
    },
    {
      "step": 20750,
      "train_flow_matching_loss": 0.1650390625,
      "train_physics_loss": 0.0
    },
    {
      "step": 20750,
      "train_flow_matching_loss": 0.189453125,
      "train_physics_loss": 0.0
    },
    {
      "step": 20750,
      "train_flow_matching_loss": 0.1005859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 20750,
      "train_flow_matching_loss": 0.1064453125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.0980566143989563,
      "learning_rate": 5e-05,
      "loss": 0.1265,
      "step": 20760
    },
    {
      "step": 20760,
      "train_flow_matching_loss": 0.1708984375,
      "train_physics_loss": 0.0
    },
    {
      "step": 20760,
      "train_flow_matching_loss": 0.1630859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 20760,
      "train_flow_matching_loss": 0.15625,
      "train_physics_loss": 0.0
    },
    {
      "step": 20760,
      "train_flow_matching_loss": 0.181640625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.4312306046485901,
      "learning_rate": 5e-05,
      "loss": 0.1262,
      "step": 20770
    },
    {
      "step": 20770,
      "train_flow_matching_loss": 0.171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 20770,
      "train_flow_matching_loss": 0.11474609375,
      "train_physics_loss": 0.0
    },
    {
      "step": 20770,
      "train_flow_matching_loss": 0.09912109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 20770,
      "train_flow_matching_loss": 0.0673828125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10818524658679962,
      "learning_rate": 5e-05,
      "loss": 0.1329,
      "step": 20780
    },
    {
      "step": 20780,
      "train_flow_matching_loss": 0.140625,
      "train_physics_loss": 0.0
    },
    {
      "step": 20780,
      "train_flow_matching_loss": 0.2392578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 20780,
      "train_flow_matching_loss": 0.11572265625,
      "train_physics_loss": 0.0
    },
    {
      "step": 20780,
      "train_flow_matching_loss": 0.0810546875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.0886693224310875,
      "learning_rate": 5e-05,
      "loss": 0.1297,
      "step": 20790
    },
    {
      "step": 20790,
      "train_flow_matching_loss": 0.1787109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 20790,
      "train_flow_matching_loss": 0.1416015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 20790,
      "train_flow_matching_loss": 0.10986328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 20790,
      "train_flow_matching_loss": 0.09619140625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09818050265312195,
      "learning_rate": 5e-05,
      "loss": 0.1277,
      "step": 20800
    },
    {
      "step": 20800,
      "train_flow_matching_loss": 0.2099609375,
      "train_physics_loss": 0.0
    },
    {
      "step": 20800,
      "train_flow_matching_loss": 0.140625,
      "train_physics_loss": 0.0
    },
    {
      "step": 20800,
      "train_flow_matching_loss": 0.10791015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 20800,
      "train_flow_matching_loss": 0.0986328125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09214509278535843,
      "learning_rate": 5e-05,
      "loss": 0.126,
      "step": 20810
    },
    {
      "step": 20810,
      "train_flow_matching_loss": 0.11474609375,
      "train_physics_loss": 0.0
    },
    {
      "step": 20810,
      "train_flow_matching_loss": 0.1484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 20810,
      "train_flow_matching_loss": 0.1201171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 20810,
      "train_flow_matching_loss": 0.12890625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09667468816041946,
      "learning_rate": 5e-05,
      "loss": 0.1249,
      "step": 20820
    },
    {
      "step": 20820,
      "train_flow_matching_loss": 0.2294921875,
      "train_physics_loss": 0.0
    },
    {
      "step": 20820,
      "train_flow_matching_loss": 0.1416015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 20820,
      "train_flow_matching_loss": 0.134765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 20820,
      "train_flow_matching_loss": 0.08984375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.14456573128700256,
      "learning_rate": 5e-05,
      "loss": 0.1239,
      "step": 20830
    },
    {
      "step": 20830,
      "train_flow_matching_loss": 0.1142578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 20830,
      "train_flow_matching_loss": 0.1396484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 20830,
      "train_flow_matching_loss": 0.1318359375,
      "train_physics_loss": 0.0
    },
    {
      "step": 20830,
      "train_flow_matching_loss": 0.0888671875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10262014716863632,
      "learning_rate": 5e-05,
      "loss": 0.1261,
      "step": 20840
    },
    {
      "step": 20840,
      "train_flow_matching_loss": 0.1650390625,
      "train_physics_loss": 0.0
    },
    {
      "step": 20840,
      "train_flow_matching_loss": 0.1162109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 20840,
      "train_flow_matching_loss": 0.11865234375,
      "train_physics_loss": 0.0
    },
    {
      "step": 20840,
      "train_flow_matching_loss": 0.0732421875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11498123407363892,
      "learning_rate": 5e-05,
      "loss": 0.1285,
      "step": 20850
    },
    {
      "step": 20850,
      "train_flow_matching_loss": 0.13671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 20850,
      "train_flow_matching_loss": 0.146484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 20850,
      "train_flow_matching_loss": 0.10546875,
      "train_physics_loss": 0.0
    },
    {
      "step": 20850,
      "train_flow_matching_loss": 0.158203125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09281987696886063,
      "learning_rate": 5e-05,
      "loss": 0.1233,
      "step": 20860
    },
    {
      "step": 20860,
      "train_flow_matching_loss": 0.0947265625,
      "train_physics_loss": 0.0
    },
    {
      "step": 20860,
      "train_flow_matching_loss": 0.1650390625,
      "train_physics_loss": 0.0
    },
    {
      "step": 20860,
      "train_flow_matching_loss": 0.1123046875,
      "train_physics_loss": 0.0
    },
    {
      "step": 20860,
      "train_flow_matching_loss": 0.138671875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.07653522491455078,
      "learning_rate": 5e-05,
      "loss": 0.1252,
      "step": 20870
    },
    {
      "step": 20870,
      "train_flow_matching_loss": 0.130859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 20870,
      "train_flow_matching_loss": 0.11474609375,
      "train_physics_loss": 0.0
    },
    {
      "step": 20870,
      "train_flow_matching_loss": 0.1513671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 20870,
      "train_flow_matching_loss": 0.150390625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.17063726484775543,
      "learning_rate": 5e-05,
      "loss": 0.1287,
      "step": 20880
    },
    {
      "step": 20880,
      "train_flow_matching_loss": 0.1259765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 20880,
      "train_flow_matching_loss": 0.1220703125,
      "train_physics_loss": 0.0
    },
    {
      "step": 20880,
      "train_flow_matching_loss": 0.0771484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 20880,
      "train_flow_matching_loss": 0.10693359375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.12911102175712585,
      "learning_rate": 5e-05,
      "loss": 0.1249,
      "step": 20890
    },
    {
      "step": 20890,
      "train_flow_matching_loss": 0.10205078125,
      "train_physics_loss": 0.0
    },
    {
      "step": 20890,
      "train_flow_matching_loss": 0.1279296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 20890,
      "train_flow_matching_loss": 0.08544921875,
      "train_physics_loss": 0.0
    },
    {
      "step": 20890,
      "train_flow_matching_loss": 0.0830078125,
      "train_physics_loss": 0.25749537348747253
    },
    {
      "grad_norm": 0.1565491259098053,
      "learning_rate": 5e-05,
      "loss": 0.1268,
      "step": 20900
    },
    {
      "step": 20900,
      "train_flow_matching_loss": 0.1455078125,
      "train_physics_loss": 0.0
    },
    {
      "step": 20900,
      "train_flow_matching_loss": 0.13671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 20900,
      "train_flow_matching_loss": 0.0673828125,
      "train_physics_loss": 0.0
    },
    {
      "step": 20900,
      "train_flow_matching_loss": 0.044189453125,
      "train_physics_loss": 0.26861372590065
    },
    {
      "grad_norm": 0.07472150027751923,
      "learning_rate": 5e-05,
      "loss": 0.123,
      "step": 20910
    },
    {
      "step": 20910,
      "train_flow_matching_loss": 0.1318359375,
      "train_physics_loss": 0.0
    },
    {
      "step": 20910,
      "train_flow_matching_loss": 0.1484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 20910,
      "train_flow_matching_loss": 0.1787109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 20910,
      "train_flow_matching_loss": 0.0274658203125,
      "train_physics_loss": 0.2594541311264038
    },
    {
      "grad_norm": 0.10739739239215851,
      "learning_rate": 5e-05,
      "loss": 0.1291,
      "step": 20920
    },
    {
      "step": 20920,
      "train_flow_matching_loss": 0.1748046875,
      "train_physics_loss": 0.0
    },
    {
      "step": 20920,
      "train_flow_matching_loss": 0.130859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 20920,
      "train_flow_matching_loss": 0.12158203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 20920,
      "train_flow_matching_loss": 0.05859375,
      "train_physics_loss": 0.23836806416511536
    },
    {
      "grad_norm": 0.14593981206417084,
      "learning_rate": 5e-05,
      "loss": 0.1224,
      "step": 20930
    },
    {
      "step": 20930,
      "train_flow_matching_loss": 0.09619140625,
      "train_physics_loss": 0.0
    },
    {
      "step": 20930,
      "train_flow_matching_loss": 0.154296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 20930,
      "train_flow_matching_loss": 0.0927734375,
      "train_physics_loss": 0.0
    },
    {
      "step": 20930,
      "train_flow_matching_loss": 0.038818359375,
      "train_physics_loss": 0.2669488787651062
    },
    {
      "grad_norm": 0.10379277914762497,
      "learning_rate": 5e-05,
      "loss": 0.1262,
      "step": 20940
    },
    {
      "step": 20940,
      "train_flow_matching_loss": 0.0849609375,
      "train_physics_loss": 0.0
    },
    {
      "step": 20940,
      "train_flow_matching_loss": 0.166015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 20940,
      "train_flow_matching_loss": 0.0859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 20940,
      "train_flow_matching_loss": 0.06884765625,
      "train_physics_loss": 0.28239837288856506
    },
    {
      "grad_norm": 0.11816776543855667,
      "learning_rate": 5e-05,
      "loss": 0.1235,
      "step": 20950
    },
    {
      "step": 20950,
      "train_flow_matching_loss": 0.0986328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 20950,
      "train_flow_matching_loss": 0.197265625,
      "train_physics_loss": 0.0
    },
    {
      "step": 20950,
      "train_flow_matching_loss": 0.125,
      "train_physics_loss": 0.0
    },
    {
      "step": 20950,
      "train_flow_matching_loss": 0.0244140625,
      "train_physics_loss": 0.29499417543411255
    },
    {
      "grad_norm": 0.09176024049520493,
      "learning_rate": 5e-05,
      "loss": 0.1276,
      "step": 20960
    },
    {
      "step": 20960,
      "train_flow_matching_loss": 0.1201171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 20960,
      "train_flow_matching_loss": 0.134765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 20960,
      "train_flow_matching_loss": 0.154296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 20960,
      "train_flow_matching_loss": 0.03759765625,
      "train_physics_loss": 0.28454074263572693
    },
    {
      "grad_norm": 0.12045302242040634,
      "learning_rate": 5e-05,
      "loss": 0.1306,
      "step": 20970
    },
    {
      "step": 20970,
      "train_flow_matching_loss": 0.1083984375,
      "train_physics_loss": 0.0
    },
    {
      "step": 20970,
      "train_flow_matching_loss": 0.1416015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 20970,
      "train_flow_matching_loss": 0.1591796875,
      "train_physics_loss": 0.0
    },
    {
      "step": 20970,
      "train_flow_matching_loss": 0.10986328125,
      "train_physics_loss": 0.4690864086151123
    },
    {
      "grad_norm": 0.10238298773765564,
      "learning_rate": 5e-05,
      "loss": 0.1283,
      "step": 20980
    },
    {
      "step": 20980,
      "train_flow_matching_loss": 0.138671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 20980,
      "train_flow_matching_loss": 0.2158203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 20980,
      "train_flow_matching_loss": 0.140625,
      "train_physics_loss": 0.0
    },
    {
      "step": 20980,
      "train_flow_matching_loss": 0.162109375,
      "train_physics_loss": 0.46092233061790466
    },
    {
      "grad_norm": 0.11378461867570877,
      "learning_rate": 5e-05,
      "loss": 0.1318,
      "step": 20990
    },
    {
      "step": 20990,
      "train_flow_matching_loss": 0.130859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 20990,
      "train_flow_matching_loss": 0.09765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 20990,
      "train_flow_matching_loss": 0.1875,
      "train_physics_loss": 0.0
    },
    {
      "step": 20990,
      "train_flow_matching_loss": 0.051513671875,
      "train_physics_loss": 0.3047192692756653
    },
    {
      "grad_norm": 0.08337420970201492,
      "learning_rate": 5e-05,
      "loss": 0.1338,
      "step": 21000
    },
    {
      "step": 21000,
      "train_flow_matching_loss": 0.07177734375,
      "train_physics_loss": 0.0
    },
    {
      "step": 21000,
      "train_flow_matching_loss": 0.185546875,
      "train_physics_loss": 0.0
    },
    {
      "step": 21000,
      "train_flow_matching_loss": 0.10693359375,
      "train_physics_loss": 0.0
    },
    {
      "step": 21000,
      "train_flow_matching_loss": 0.042724609375,
      "train_physics_loss": 0.2613271474838257
    },
    {
      "grad_norm": 0.08563712239265442,
      "learning_rate": 5e-05,
      "loss": 0.1251,
      "step": 21010
    },
    {
      "step": 21010,
      "train_flow_matching_loss": 0.054443359375,
      "train_physics_loss": 0.0
    },
    {
      "step": 21010,
      "train_flow_matching_loss": 0.2060546875,
      "train_physics_loss": 0.0
    },
    {
      "step": 21010,
      "train_flow_matching_loss": 0.11328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 21010,
      "train_flow_matching_loss": 0.05419921875,
      "train_physics_loss": 0.4263157546520233
    },
    {
      "grad_norm": 0.08289536833763123,
      "learning_rate": 5e-05,
      "loss": 0.1268,
      "step": 21020
    },
    {
      "step": 21020,
      "train_flow_matching_loss": 0.1484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 21020,
      "train_flow_matching_loss": 0.1533203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 21020,
      "train_flow_matching_loss": 0.251953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 21020,
      "train_flow_matching_loss": 0.049560546875,
      "train_physics_loss": 0.28587841987609863
    },
    {
      "grad_norm": 0.11456844955682755,
      "learning_rate": 5e-05,
      "loss": 0.1275,
      "step": 21030
    },
    {
      "step": 21030,
      "train_flow_matching_loss": 0.2255859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 21030,
      "train_flow_matching_loss": 0.171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 21030,
      "train_flow_matching_loss": 0.158203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 21030,
      "train_flow_matching_loss": 0.060302734375,
      "train_physics_loss": 0.27596911787986755
    },
    {
      "grad_norm": 0.11604553461074829,
      "learning_rate": 5e-05,
      "loss": 0.1271,
      "step": 21040
    },
    {
      "step": 21040,
      "train_flow_matching_loss": 0.1875,
      "train_physics_loss": 0.0
    },
    {
      "step": 21040,
      "train_flow_matching_loss": 0.203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 21040,
      "train_flow_matching_loss": 0.1005859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 21040,
      "train_flow_matching_loss": 0.0859375,
      "train_physics_loss": 0.34384965896606445
    },
    {
      "grad_norm": 0.12134195119142532,
      "learning_rate": 5e-05,
      "loss": 0.1332,
      "step": 21050
    },
    {
      "step": 21050,
      "train_flow_matching_loss": 0.10595703125,
      "train_physics_loss": 0.0
    },
    {
      "step": 21050,
      "train_flow_matching_loss": 0.150390625,
      "train_physics_loss": 0.0
    },
    {
      "step": 21050,
      "train_flow_matching_loss": 0.1669921875,
      "train_physics_loss": 0.0
    },
    {
      "step": 21050,
      "train_flow_matching_loss": 0.0322265625,
      "train_physics_loss": 0.29272231459617615
    },
    {
      "grad_norm": 0.2460578978061676,
      "learning_rate": 5e-05,
      "loss": 0.1326,
      "step": 21060
    },
    {
      "step": 21060,
      "train_flow_matching_loss": 0.1396484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 21060,
      "train_flow_matching_loss": 0.08837890625,
      "train_physics_loss": 0.0
    },
    {
      "step": 21060,
      "train_flow_matching_loss": 0.12353515625,
      "train_physics_loss": 0.0
    },
    {
      "step": 21060,
      "train_flow_matching_loss": 0.033203125,
      "train_physics_loss": 0.26948341727256775
    },
    {
      "grad_norm": 0.12032999843358994,
      "learning_rate": 5e-05,
      "loss": 0.1303,
      "step": 21070
    },
    {
      "step": 21070,
      "train_flow_matching_loss": 0.07958984375,
      "train_physics_loss": 0.0
    },
    {
      "step": 21070,
      "train_flow_matching_loss": 0.0927734375,
      "train_physics_loss": 0.27398136258125305
    },
    {
      "step": 21070,
      "train_flow_matching_loss": 0.0927734375,
      "train_physics_loss": 0.0
    },
    {
      "step": 21070,
      "train_flow_matching_loss": 0.036865234375,
      "train_physics_loss": 0.29473620653152466
    },
    {
      "grad_norm": 0.07419518381357193,
      "learning_rate": 5e-05,
      "loss": 0.1339,
      "step": 21080
    },
    {
      "step": 21080,
      "train_flow_matching_loss": 0.138671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 21080,
      "train_flow_matching_loss": 0.05908203125,
      "train_physics_loss": 0.32854512333869934
    },
    {
      "step": 21080,
      "train_flow_matching_loss": 0.0888671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 21080,
      "train_flow_matching_loss": 0.0771484375,
      "train_physics_loss": 0.4275137484073639
    },
    {
      "grad_norm": 0.12332549691200256,
      "learning_rate": 5e-05,
      "loss": 0.1279,
      "step": 21090
    },
    {
      "step": 21090,
      "train_flow_matching_loss": 0.1376953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 21090,
      "train_flow_matching_loss": 0.06591796875,
      "train_physics_loss": 0.33755993843078613
    },
    {
      "step": 21090,
      "train_flow_matching_loss": 0.1708984375,
      "train_physics_loss": 0.0
    },
    {
      "step": 21090,
      "train_flow_matching_loss": 0.09033203125,
      "train_physics_loss": 0.3295641839504242
    },
    {
      "grad_norm": 0.09794706851243973,
      "learning_rate": 5e-05,
      "loss": 0.1331,
      "step": 21100
    },
    {
      "step": 21100,
      "train_flow_matching_loss": 0.1767578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 21100,
      "train_flow_matching_loss": 0.09228515625,
      "train_physics_loss": 0.3841969668865204
    },
    {
      "step": 21100,
      "train_flow_matching_loss": 0.08740234375,
      "train_physics_loss": 0.0
    },
    {
      "step": 21100,
      "train_flow_matching_loss": 0.06494140625,
      "train_physics_loss": 0.31721872091293335
    },
    {
      "grad_norm": 0.09597492218017578,
      "learning_rate": 5e-05,
      "loss": 0.1259,
      "step": 21110
    },
    {
      "step": 21110,
      "train_flow_matching_loss": 0.1474609375,
      "train_physics_loss": 0.0
    },
    {
      "step": 21110,
      "train_flow_matching_loss": 0.0732421875,
      "train_physics_loss": 0.3564886450767517
    },
    {
      "step": 21110,
      "train_flow_matching_loss": 0.1552734375,
      "train_physics_loss": 0.0
    },
    {
      "step": 21110,
      "train_flow_matching_loss": 0.07080078125,
      "train_physics_loss": 0.303962767124176
    },
    {
      "grad_norm": 0.09635777026414871,
      "learning_rate": 5e-05,
      "loss": 0.128,
      "step": 21120
    },
    {
      "step": 21120,
      "train_flow_matching_loss": 0.1337890625,
      "train_physics_loss": 0.0
    },
    {
      "step": 21120,
      "train_flow_matching_loss": 0.04541015625,
      "train_physics_loss": 0.33888858556747437
    },
    {
      "step": 21120,
      "train_flow_matching_loss": 0.10791015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 21120,
      "train_flow_matching_loss": 0.04541015625,
      "train_physics_loss": 0.27697038650512695
    },
    {
      "grad_norm": 0.1099337562918663,
      "learning_rate": 5e-05,
      "loss": 0.1281,
      "step": 21130
    },
    {
      "step": 21130,
      "train_flow_matching_loss": 0.1337890625,
      "train_physics_loss": 0.0
    },
    {
      "step": 21130,
      "train_flow_matching_loss": 0.07373046875,
      "train_physics_loss": 0.43539750576019287
    },
    {
      "step": 21130,
      "train_flow_matching_loss": 0.1376953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 21130,
      "train_flow_matching_loss": 0.06982421875,
      "train_physics_loss": 0.3564274311065674
    },
    {
      "grad_norm": 0.11825357377529144,
      "learning_rate": 5e-05,
      "loss": 0.1299,
      "step": 21140
    },
    {
      "step": 21140,
      "train_flow_matching_loss": 0.08203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 21140,
      "train_flow_matching_loss": 0.06201171875,
      "train_physics_loss": 0.30095013976097107
    },
    {
      "step": 21140,
      "train_flow_matching_loss": 0.171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 21140,
      "train_flow_matching_loss": 0.12353515625,
      "train_physics_loss": 0.41028615832328796
    },
    {
      "grad_norm": 0.0977751612663269,
      "learning_rate": 5e-05,
      "loss": 0.1291,
      "step": 21150
    },
    {
      "step": 21150,
      "train_flow_matching_loss": 0.1220703125,
      "train_physics_loss": 0.0
    },
    {
      "step": 21150,
      "train_flow_matching_loss": 0.05908203125,
      "train_physics_loss": 0.3289493918418884
    },
    {
      "step": 21150,
      "train_flow_matching_loss": 0.1630859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 21150,
      "train_flow_matching_loss": 0.126953125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09295012801885605,
      "learning_rate": 5e-05,
      "loss": 0.1258,
      "step": 21160
    },
    {
      "step": 21160,
      "train_flow_matching_loss": 0.12353515625,
      "train_physics_loss": 0.0
    },
    {
      "step": 21160,
      "train_flow_matching_loss": 0.07470703125,
      "train_physics_loss": 0.3589010536670685
    },
    {
      "step": 21160,
      "train_flow_matching_loss": 0.060546875,
      "train_physics_loss": 0.0
    },
    {
      "step": 21160,
      "train_flow_matching_loss": 0.11865234375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.20597682893276215,
      "learning_rate": 5e-05,
      "loss": 0.1298,
      "step": 21170
    },
    {
      "step": 21170,
      "train_flow_matching_loss": 0.1357421875,
      "train_physics_loss": 0.0
    },
    {
      "step": 21170,
      "train_flow_matching_loss": 0.05078125,
      "train_physics_loss": 0.29500219225883484
    },
    {
      "step": 21170,
      "train_flow_matching_loss": 0.12060546875,
      "train_physics_loss": 0.0
    },
    {
      "step": 21170,
      "train_flow_matching_loss": 0.1044921875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.07190337777137756,
      "learning_rate": 5e-05,
      "loss": 0.1269,
      "step": 21180
    },
    {
      "step": 21180,
      "train_flow_matching_loss": 0.1298828125,
      "train_physics_loss": 0.0
    },
    {
      "step": 21180,
      "train_flow_matching_loss": 0.061279296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 21180,
      "train_flow_matching_loss": 0.0859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 21180,
      "train_flow_matching_loss": 0.0634765625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08987637609243393,
      "learning_rate": 5e-05,
      "loss": 0.1248,
      "step": 21190
    },
    {
      "step": 21190,
      "train_flow_matching_loss": 0.1298828125,
      "train_physics_loss": 0.0
    },
    {
      "step": 21190,
      "train_flow_matching_loss": 0.07470703125,
      "train_physics_loss": 0.32179638743400574
    },
    {
      "step": 21190,
      "train_flow_matching_loss": 0.12109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 21190,
      "train_flow_matching_loss": 0.0625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.13776163756847382,
      "learning_rate": 5e-05,
      "loss": 0.1302,
      "step": 21200
    },
    {
      "step": 21200,
      "train_flow_matching_loss": 0.2431640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 21200,
      "train_flow_matching_loss": 0.046875,
      "train_physics_loss": 0.30625173449516296
    },
    {
      "step": 21200,
      "train_flow_matching_loss": 0.1181640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 21200,
      "train_flow_matching_loss": 0.10205078125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10936693102121353,
      "learning_rate": 5e-05,
      "loss": 0.1258,
      "step": 21210
    },
    {
      "step": 21210,
      "train_flow_matching_loss": 0.177734375,
      "train_physics_loss": 0.0
    },
    {
      "step": 21210,
      "train_flow_matching_loss": 0.032958984375,
      "train_physics_loss": 0.30261746048927307
    },
    {
      "step": 21210,
      "train_flow_matching_loss": 0.1513671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 21210,
      "train_flow_matching_loss": 0.0908203125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09453091770410538,
      "learning_rate": 5e-05,
      "loss": 0.129,
      "step": 21220
    },
    {
      "step": 21220,
      "train_flow_matching_loss": 0.1435546875,
      "train_physics_loss": 0.0
    },
    {
      "step": 21220,
      "train_flow_matching_loss": 0.0693359375,
      "train_physics_loss": 0.3383055329322815
    },
    {
      "step": 21220,
      "train_flow_matching_loss": 0.142578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 21220,
      "train_flow_matching_loss": 0.2001953125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08894219994544983,
      "learning_rate": 5e-05,
      "loss": 0.1298,
      "step": 21230
    },
    {
      "step": 21230,
      "train_flow_matching_loss": 0.1884765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 21230,
      "train_flow_matching_loss": 0.0546875,
      "train_physics_loss": 0.27584195137023926
    },
    {
      "step": 21230,
      "train_flow_matching_loss": 0.09716796875,
      "train_physics_loss": 0.0
    },
    {
      "step": 21230,
      "train_flow_matching_loss": 0.1083984375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08083490282297134,
      "learning_rate": 5e-05,
      "loss": 0.1277,
      "step": 21240
    },
    {
      "step": 21240,
      "train_flow_matching_loss": 0.1728515625,
      "train_physics_loss": 0.0
    },
    {
      "step": 21240,
      "train_flow_matching_loss": 0.01708984375,
      "train_physics_loss": 0.2912556529045105
    },
    {
      "step": 21240,
      "train_flow_matching_loss": 0.1728515625,
      "train_physics_loss": 0.0
    },
    {
      "step": 21240,
      "train_flow_matching_loss": 0.06982421875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.07348308712244034,
      "learning_rate": 5e-05,
      "loss": 0.1274,
      "step": 21250
    },
    {
      "step": 21250,
      "train_flow_matching_loss": 0.1103515625,
      "train_physics_loss": 0.0
    },
    {
      "step": 21250,
      "train_flow_matching_loss": 0.09521484375,
      "train_physics_loss": 0.350796103477478
    },
    {
      "step": 21250,
      "train_flow_matching_loss": 0.1123046875,
      "train_physics_loss": 0.0
    },
    {
      "step": 21250,
      "train_flow_matching_loss": 0.1533203125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.16401244699954987,
      "learning_rate": 5e-05,
      "loss": 0.1279,
      "step": 21260
    },
    {
      "step": 21260,
      "train_flow_matching_loss": 0.177734375,
      "train_physics_loss": 0.0
    },
    {
      "step": 21260,
      "train_flow_matching_loss": 0.08251953125,
      "train_physics_loss": 0.3398217260837555
    },
    {
      "step": 21260,
      "train_flow_matching_loss": 0.1396484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 21260,
      "train_flow_matching_loss": 0.12060546875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.1198842003941536,
      "learning_rate": 5e-05,
      "loss": 0.1318,
      "step": 21270
    },
    {
      "step": 21270,
      "train_flow_matching_loss": 0.150390625,
      "train_physics_loss": 0.0
    },
    {
      "step": 21270,
      "train_flow_matching_loss": 0.07421875,
      "train_physics_loss": 0.3093824088573456
    },
    {
      "step": 21270,
      "train_flow_matching_loss": 0.1240234375,
      "train_physics_loss": 0.0
    },
    {
      "step": 21270,
      "train_flow_matching_loss": 0.1337890625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10110359638929367,
      "learning_rate": 5e-05,
      "loss": 0.1267,
      "step": 21280
    },
    {
      "step": 21280,
      "train_flow_matching_loss": 0.1259765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 21280,
      "train_flow_matching_loss": 0.03564453125,
      "train_physics_loss": 0.32453474402427673
    },
    {
      "step": 21280,
      "train_flow_matching_loss": 0.08935546875,
      "train_physics_loss": 0.0
    },
    {
      "step": 21280,
      "train_flow_matching_loss": 0.1171875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09711199253797531,
      "learning_rate": 5e-05,
      "loss": 0.1259,
      "step": 21290
    },
    {
      "step": 21290,
      "train_flow_matching_loss": 0.208984375,
      "train_physics_loss": 0.0
    },
    {
      "step": 21290,
      "train_flow_matching_loss": 0.07177734375,
      "train_physics_loss": 0.2758656442165375
    },
    {
      "step": 21290,
      "train_flow_matching_loss": 0.10986328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 21290,
      "train_flow_matching_loss": 0.10400390625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08594700694084167,
      "learning_rate": 5e-05,
      "loss": 0.1281,
      "step": 21300
    },
    {
      "step": 21300,
      "train_flow_matching_loss": 0.1689453125,
      "train_physics_loss": 0.0
    },
    {
      "step": 21300,
      "train_flow_matching_loss": 0.056396484375,
      "train_physics_loss": 0.3069036900997162
    },
    {
      "step": 21300,
      "train_flow_matching_loss": 0.1513671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 21300,
      "train_flow_matching_loss": 0.109375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11166267842054367,
      "learning_rate": 5e-05,
      "loss": 0.1175,
      "step": 21310
    },
    {
      "step": 21310,
      "train_flow_matching_loss": 0.126953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 21310,
      "train_flow_matching_loss": 0.043701171875,
      "train_physics_loss": 0.26282989978790283
    },
    {
      "step": 21310,
      "train_flow_matching_loss": 0.11572265625,
      "train_physics_loss": 0.0
    },
    {
      "step": 21310,
      "train_flow_matching_loss": 0.0810546875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.4857986569404602,
      "learning_rate": 5e-05,
      "loss": 0.1217,
      "step": 21320
    },
    {
      "step": 21320,
      "train_flow_matching_loss": 0.1416015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 21320,
      "train_flow_matching_loss": 0.04638671875,
      "train_physics_loss": 0.22154097259044647
    },
    {
      "step": 21320,
      "train_flow_matching_loss": 0.1181640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 21320,
      "train_flow_matching_loss": 0.162109375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09609729051589966,
      "learning_rate": 5e-05,
      "loss": 0.1254,
      "step": 21330
    },
    {
      "step": 21330,
      "train_flow_matching_loss": 0.2158203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 21330,
      "train_flow_matching_loss": 0.12158203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 21330,
      "train_flow_matching_loss": 0.07763671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 21330,
      "train_flow_matching_loss": 0.10986328125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.07565900683403015,
      "learning_rate": 5e-05,
      "loss": 0.125,
      "step": 21340
    },
    {
      "step": 21340,
      "train_flow_matching_loss": 0.1181640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 21340,
      "train_flow_matching_loss": 0.0751953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 21340,
      "train_flow_matching_loss": 0.08251953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 21340,
      "train_flow_matching_loss": 0.1064453125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.15952961146831512,
      "learning_rate": 5e-05,
      "loss": 0.1229,
      "step": 21350
    },
    {
      "step": 21350,
      "train_flow_matching_loss": 0.146484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 21350,
      "train_flow_matching_loss": 0.1201171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 21350,
      "train_flow_matching_loss": 0.12060546875,
      "train_physics_loss": 0.0
    },
    {
      "step": 21350,
      "train_flow_matching_loss": 0.07666015625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.12566159665584564,
      "learning_rate": 5e-05,
      "loss": 0.1265,
      "step": 21360
    },
    {
      "step": 21360,
      "train_flow_matching_loss": 0.1689453125,
      "train_physics_loss": 0.0
    },
    {
      "step": 21360,
      "train_flow_matching_loss": 0.09130859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 21360,
      "train_flow_matching_loss": 0.162109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 21360,
      "train_flow_matching_loss": 0.10107421875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.25710469484329224,
      "learning_rate": 5e-05,
      "loss": 0.1271,
      "step": 21370
    },
    {
      "step": 21370,
      "train_flow_matching_loss": 0.181640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 21370,
      "train_flow_matching_loss": 0.09033203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 21370,
      "train_flow_matching_loss": 0.1220703125,
      "train_physics_loss": 0.0
    },
    {
      "step": 21370,
      "train_flow_matching_loss": 0.0791015625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11278533190488815,
      "learning_rate": 5e-05,
      "loss": 0.1276,
      "step": 21380
    },
    {
      "step": 21380,
      "train_flow_matching_loss": 0.142578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 21380,
      "train_flow_matching_loss": 0.1552734375,
      "train_physics_loss": 0.0
    },
    {
      "step": 21380,
      "train_flow_matching_loss": 0.11474609375,
      "train_physics_loss": 0.0
    },
    {
      "step": 21380,
      "train_flow_matching_loss": 0.09619140625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.1457659900188446,
      "learning_rate": 5e-05,
      "loss": 0.1236,
      "step": 21390
    },
    {
      "step": 21390,
      "train_flow_matching_loss": 0.060791015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 21390,
      "train_flow_matching_loss": 0.177734375,
      "train_physics_loss": 0.0
    },
    {
      "step": 21390,
      "train_flow_matching_loss": 0.15625,
      "train_physics_loss": 0.0
    },
    {
      "step": 21390,
      "train_flow_matching_loss": 0.11767578125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.13790322840213776,
      "learning_rate": 5e-05,
      "loss": 0.128,
      "step": 21400
    },
    {
      "step": 21400,
      "train_flow_matching_loss": 0.1484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 21400,
      "train_flow_matching_loss": 0.126953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 21400,
      "train_flow_matching_loss": 0.09814453125,
      "train_physics_loss": 0.0
    },
    {
      "step": 21400,
      "train_flow_matching_loss": 0.1630859375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11768510192632675,
      "learning_rate": 5e-05,
      "loss": 0.126,
      "step": 21410
    },
    {
      "step": 21410,
      "train_flow_matching_loss": 0.1484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 21410,
      "train_flow_matching_loss": 0.15625,
      "train_physics_loss": 0.0
    },
    {
      "step": 21410,
      "train_flow_matching_loss": 0.09423828125,
      "train_physics_loss": 0.0
    },
    {
      "step": 21410,
      "train_flow_matching_loss": 0.12158203125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.0670999065041542,
      "learning_rate": 5e-05,
      "loss": 0.125,
      "step": 21420
    },
    {
      "step": 21420,
      "train_flow_matching_loss": 0.109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 21420,
      "train_flow_matching_loss": 0.1259765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 21420,
      "train_flow_matching_loss": 0.13671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 21420,
      "train_flow_matching_loss": 0.197265625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.084079809486866,
      "learning_rate": 5e-05,
      "loss": 0.1287,
      "step": 21430
    },
    {
      "step": 21430,
      "train_flow_matching_loss": 0.1484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 21430,
      "train_flow_matching_loss": 0.06005859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 21430,
      "train_flow_matching_loss": 0.1005859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 21430,
      "train_flow_matching_loss": 0.150390625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.1115536317229271,
      "learning_rate": 5e-05,
      "loss": 0.1294,
      "step": 21440
    },
    {
      "step": 21440,
      "train_flow_matching_loss": 0.1982421875,
      "train_physics_loss": 0.0
    },
    {
      "step": 21440,
      "train_flow_matching_loss": 0.1396484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 21440,
      "train_flow_matching_loss": 0.166015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 21440,
      "train_flow_matching_loss": 0.1943359375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08689635992050171,
      "learning_rate": 5e-05,
      "loss": 0.1298,
      "step": 21450
    },
    {
      "step": 21450,
      "train_flow_matching_loss": 0.1572265625,
      "train_physics_loss": 0.0
    },
    {
      "step": 21450,
      "train_flow_matching_loss": 0.08203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 21450,
      "train_flow_matching_loss": 0.09716796875,
      "train_physics_loss": 0.0
    },
    {
      "step": 21450,
      "train_flow_matching_loss": 0.1640625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08229362964630127,
      "learning_rate": 5e-05,
      "loss": 0.1263,
      "step": 21460
    },
    {
      "step": 21460,
      "train_flow_matching_loss": 0.09765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 21460,
      "train_flow_matching_loss": 0.08642578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 21460,
      "train_flow_matching_loss": 0.1728515625,
      "train_physics_loss": 0.0
    },
    {
      "step": 21460,
      "train_flow_matching_loss": 0.11962890625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09731732308864594,
      "learning_rate": 5e-05,
      "loss": 0.1257,
      "step": 21470
    },
    {
      "step": 21470,
      "train_flow_matching_loss": 0.09326171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 21470,
      "train_flow_matching_loss": 0.09912109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 21470,
      "train_flow_matching_loss": 0.2099609375,
      "train_physics_loss": 0.0
    },
    {
      "step": 21470,
      "train_flow_matching_loss": 0.07421875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.0797698050737381,
      "learning_rate": 5e-05,
      "loss": 0.1269,
      "step": 21480
    },
    {
      "step": 21480,
      "train_flow_matching_loss": 0.1650390625,
      "train_physics_loss": 0.0
    },
    {
      "step": 21480,
      "train_flow_matching_loss": 0.08544921875,
      "train_physics_loss": 0.0
    },
    {
      "step": 21480,
      "train_flow_matching_loss": 0.1357421875,
      "train_physics_loss": 0.0
    },
    {
      "step": 21480,
      "train_flow_matching_loss": 0.173828125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09389413893222809,
      "learning_rate": 5e-05,
      "loss": 0.1262,
      "step": 21490
    },
    {
      "step": 21490,
      "train_flow_matching_loss": 0.208984375,
      "train_physics_loss": 0.0
    },
    {
      "step": 21490,
      "train_flow_matching_loss": 0.1748046875,
      "train_physics_loss": 0.0
    },
    {
      "step": 21490,
      "train_flow_matching_loss": 0.203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 21490,
      "train_flow_matching_loss": 0.09033203125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09070222824811935,
      "learning_rate": 5e-05,
      "loss": 0.125,
      "step": 21500
    },
    {
      "step": 21500,
      "train_flow_matching_loss": 0.09619140625,
      "train_physics_loss": 0.0
    },
    {
      "step": 21500,
      "train_flow_matching_loss": 0.1259765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 21500,
      "train_flow_matching_loss": 0.1591796875,
      "train_physics_loss": 0.0
    },
    {
      "step": 21500,
      "train_flow_matching_loss": 0.1533203125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10537629574537277,
      "learning_rate": 5e-05,
      "loss": 0.1272,
      "step": 21510
    },
    {
      "step": 21510,
      "train_flow_matching_loss": 0.09765625,
      "train_physics_loss": 0.26612961292266846
    },
    {
      "step": 21510,
      "train_flow_matching_loss": 0.08837890625,
      "train_physics_loss": 0.0
    },
    {
      "step": 21510,
      "train_flow_matching_loss": 0.1748046875,
      "train_physics_loss": 0.0
    },
    {
      "step": 21510,
      "train_flow_matching_loss": 0.1015625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.105506531894207,
      "learning_rate": 5e-05,
      "loss": 0.129,
      "step": 21520
    },
    {
      "step": 21520,
      "train_flow_matching_loss": 0.06640625,
      "train_physics_loss": 0.24442842602729797
    },
    {
      "step": 21520,
      "train_flow_matching_loss": 0.142578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 21520,
      "train_flow_matching_loss": 0.10986328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 21520,
      "train_flow_matching_loss": 0.1142578125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.12619026005268097,
      "learning_rate": 5e-05,
      "loss": 0.1251,
      "step": 21530
    },
    {
      "step": 21530,
      "train_flow_matching_loss": 0.0888671875,
      "train_physics_loss": 0.2536155879497528
    },
    {
      "step": 21530,
      "train_flow_matching_loss": 0.150390625,
      "train_physics_loss": 0.0
    },
    {
      "step": 21530,
      "train_flow_matching_loss": 0.12158203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 21530,
      "train_flow_matching_loss": 0.1708984375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11474620550870895,
      "learning_rate": 5e-05,
      "loss": 0.1273,
      "step": 21540
    },
    {
      "step": 21540,
      "train_flow_matching_loss": 0.0224609375,
      "train_physics_loss": 0.2912366986274719
    },
    {
      "step": 21540,
      "train_flow_matching_loss": 0.1318359375,
      "train_physics_loss": 0.0
    },
    {
      "step": 21540,
      "train_flow_matching_loss": 0.1103515625,
      "train_physics_loss": 0.0
    },
    {
      "step": 21540,
      "train_flow_matching_loss": 0.1474609375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10033903270959854,
      "learning_rate": 5e-05,
      "loss": 0.1228,
      "step": 21550
    },
    {
      "step": 21550,
      "train_flow_matching_loss": 0.08203125,
      "train_physics_loss": 0.32210853695869446
    },
    {
      "step": 21550,
      "train_flow_matching_loss": 0.1025390625,
      "train_physics_loss": 0.0
    },
    {
      "step": 21550,
      "train_flow_matching_loss": 0.1875,
      "train_physics_loss": 0.0
    },
    {
      "step": 21550,
      "train_flow_matching_loss": 0.1171875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08290301263332367,
      "learning_rate": 5e-05,
      "loss": 0.1237,
      "step": 21560
    },
    {
      "step": 21560,
      "train_flow_matching_loss": 0.07568359375,
      "train_physics_loss": 0.33992746472358704
    },
    {
      "step": 21560,
      "train_flow_matching_loss": 0.1572265625,
      "train_physics_loss": 0.0
    },
    {
      "step": 21560,
      "train_flow_matching_loss": 0.11669921875,
      "train_physics_loss": 0.0
    },
    {
      "step": 21560,
      "train_flow_matching_loss": 0.078125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.12290524691343307,
      "learning_rate": 5e-05,
      "loss": 0.1258,
      "step": 21570
    },
    {
      "step": 21570,
      "train_flow_matching_loss": 0.0625,
      "train_physics_loss": 0.2710314691066742
    },
    {
      "step": 21570,
      "train_flow_matching_loss": 0.037353515625,
      "train_physics_loss": 0.0
    },
    {
      "step": 21570,
      "train_flow_matching_loss": 0.138671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 21570,
      "train_flow_matching_loss": 0.205078125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.14376506209373474,
      "learning_rate": 5e-05,
      "loss": 0.1259,
      "step": 21580
    },
    {
      "step": 21580,
      "train_flow_matching_loss": 0.044189453125,
      "train_physics_loss": 0.2562028169631958
    },
    {
      "step": 21580,
      "train_flow_matching_loss": 0.1376953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 21580,
      "train_flow_matching_loss": 0.1416015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 21580,
      "train_flow_matching_loss": 0.12353515625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11482765525579453,
      "learning_rate": 5e-05,
      "loss": 0.1336,
      "step": 21590
    },
    {
      "step": 21590,
      "train_flow_matching_loss": 0.0308837890625,
      "train_physics_loss": 0.2641782760620117
    },
    {
      "step": 21590,
      "train_flow_matching_loss": 0.205078125,
      "train_physics_loss": 0.0
    },
    {
      "step": 21590,
      "train_flow_matching_loss": 0.07373046875,
      "train_physics_loss": 0.0
    },
    {
      "step": 21590,
      "train_flow_matching_loss": 0.185546875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.7897699475288391,
      "learning_rate": 5e-05,
      "loss": 0.1247,
      "step": 21600
    },
    {
      "step": 21600,
      "train_flow_matching_loss": 0.07080078125,
      "train_physics_loss": 0.28587788343429565
    },
    {
      "step": 21600,
      "train_flow_matching_loss": 0.154296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 21600,
      "train_flow_matching_loss": 0.07275390625,
      "train_physics_loss": 0.0
    },
    {
      "step": 21600,
      "train_flow_matching_loss": 0.12109375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.0793444886803627,
      "learning_rate": 5e-05,
      "loss": 0.1274,
      "step": 21610
    },
    {
      "step": 21610,
      "train_flow_matching_loss": 0.057373046875,
      "train_physics_loss": 0.2475460171699524
    },
    {
      "step": 21610,
      "train_flow_matching_loss": 0.142578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 21610,
      "train_flow_matching_loss": 0.1474609375,
      "train_physics_loss": 0.0
    },
    {
      "step": 21610,
      "train_flow_matching_loss": 0.10302734375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.07815336436033249,
      "learning_rate": 5e-05,
      "loss": 0.1232,
      "step": 21620
    },
    {
      "step": 21620,
      "train_flow_matching_loss": 0.03515625,
      "train_physics_loss": 0.22570918500423431
    },
    {
      "step": 21620,
      "train_flow_matching_loss": 0.1171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 21620,
      "train_flow_matching_loss": 0.2119140625,
      "train_physics_loss": 0.0
    },
    {
      "step": 21620,
      "train_flow_matching_loss": 0.2470703125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.07328812777996063,
      "learning_rate": 5e-05,
      "loss": 0.1255,
      "step": 21630
    },
    {
      "step": 21630,
      "train_flow_matching_loss": 0.05712890625,
      "train_physics_loss": 0.28286051750183105
    },
    {
      "step": 21630,
      "train_flow_matching_loss": 0.060546875,
      "train_physics_loss": 0.0
    },
    {
      "step": 21630,
      "train_flow_matching_loss": 0.026611328125,
      "train_physics_loss": 0.26666033267974854
    },
    {
      "step": 21630,
      "train_flow_matching_loss": 0.1591796875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.13270853459835052,
      "learning_rate": 5e-05,
      "loss": 0.1268,
      "step": 21640
    },
    {
      "step": 21640,
      "train_flow_matching_loss": 0.05859375,
      "train_physics_loss": 0.2918247580528259
    },
    {
      "step": 21640,
      "train_flow_matching_loss": 0.1357421875,
      "train_physics_loss": 0.0
    },
    {
      "step": 21640,
      "train_flow_matching_loss": 0.036376953125,
      "train_physics_loss": 0.2533701956272125
    },
    {
      "step": 21640,
      "train_flow_matching_loss": 0.10693359375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10359709709882736,
      "learning_rate": 5e-05,
      "loss": 0.122,
      "step": 21650
    },
    {
      "step": 21650,
      "train_flow_matching_loss": 0.06396484375,
      "train_physics_loss": 0.3094198703765869
    },
    {
      "step": 21650,
      "train_flow_matching_loss": 0.1552734375,
      "train_physics_loss": 0.0
    },
    {
      "step": 21650,
      "train_flow_matching_loss": 0.060546875,
      "train_physics_loss": 0.2977331280708313
    },
    {
      "step": 21650,
      "train_flow_matching_loss": 0.1572265625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.0996469110250473,
      "learning_rate": 5e-05,
      "loss": 0.122,
      "step": 21660
    },
    {
      "step": 21660,
      "train_flow_matching_loss": 0.037353515625,
      "train_physics_loss": 0.2896789610385895
    },
    {
      "step": 21660,
      "train_flow_matching_loss": 0.1455078125,
      "train_physics_loss": 0.0
    },
    {
      "step": 21660,
      "train_flow_matching_loss": 0.0262451171875,
      "train_physics_loss": 0.27556371688842773
    },
    {
      "step": 21660,
      "train_flow_matching_loss": 0.031005859375,
      "train_physics_loss": 0.23955123126506805
    },
    {
      "grad_norm": 0.11077682673931122,
      "learning_rate": 5e-05,
      "loss": 0.1238,
      "step": 21670
    },
    {
      "step": 21670,
      "train_flow_matching_loss": 0.03759765625,
      "train_physics_loss": 0.24633073806762695
    },
    {
      "step": 21670,
      "train_flow_matching_loss": 0.2109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 21670,
      "train_flow_matching_loss": 0.1123046875,
      "train_physics_loss": 0.37198907136917114
    },
    {
      "step": 21670,
      "train_flow_matching_loss": 0.046630859375,
      "train_physics_loss": 0.247833251953125
    },
    {
      "grad_norm": 0.10922836512327194,
      "learning_rate": 5e-05,
      "loss": 0.1278,
      "step": 21680
    },
    {
      "step": 21680,
      "train_flow_matching_loss": 0.05810546875,
      "train_physics_loss": 0.22852583229541779
    },
    {
      "step": 21680,
      "train_flow_matching_loss": 0.15625,
      "train_physics_loss": 0.0
    },
    {
      "step": 21680,
      "train_flow_matching_loss": 0.056640625,
      "train_physics_loss": 0.2335052788257599
    },
    {
      "step": 21680,
      "train_flow_matching_loss": 0.054443359375,
      "train_physics_loss": 0.2554932236671448
    },
    {
      "grad_norm": 0.07936117798089981,
      "learning_rate": 5e-05,
      "loss": 0.1297,
      "step": 21690
    },
    {
      "step": 21690,
      "train_flow_matching_loss": 0.0810546875,
      "train_physics_loss": 0.314897358417511
    },
    {
      "step": 21690,
      "train_flow_matching_loss": 0.1044921875,
      "train_physics_loss": 0.0
    },
    {
      "step": 21690,
      "train_flow_matching_loss": 0.029541015625,
      "train_physics_loss": 0.247375026345253
    },
    {
      "step": 21690,
      "train_flow_matching_loss": 0.07763671875,
      "train_physics_loss": 0.270123153924942
    },
    {
      "grad_norm": 0.1465376764535904,
      "learning_rate": 5e-05,
      "loss": 0.1298,
      "step": 21700
    },
    {
      "step": 21700,
      "train_flow_matching_loss": 0.058837890625,
      "train_physics_loss": 0.2673220634460449
    },
    {
      "step": 21700,
      "train_flow_matching_loss": 0.1318359375,
      "train_physics_loss": 0.0
    },
    {
      "step": 21700,
      "train_flow_matching_loss": 0.07666015625,
      "train_physics_loss": 0.2653888165950775
    },
    {
      "step": 21700,
      "train_flow_matching_loss": 0.07861328125,
      "train_physics_loss": 0.3329503536224365
    },
    {
      "grad_norm": 0.12657997012138367,
      "learning_rate": 5e-05,
      "loss": 0.1237,
      "step": 21710
    },
    {
      "step": 21710,
      "train_flow_matching_loss": 0.04931640625,
      "train_physics_loss": 0.26527073979377747
    },
    {
      "step": 21710,
      "train_flow_matching_loss": 0.10009765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 21710,
      "train_flow_matching_loss": 0.033935546875,
      "train_physics_loss": 0.26859837770462036
    },
    {
      "step": 21710,
      "train_flow_matching_loss": 0.07275390625,
      "train_physics_loss": 0.39912858605384827
    },
    {
      "grad_norm": 0.10545649379491806,
      "learning_rate": 5e-05,
      "loss": 0.1278,
      "step": 21720
    },
    {
      "step": 21720,
      "train_flow_matching_loss": 0.08935546875,
      "train_physics_loss": 0.20674148201942444
    },
    {
      "step": 21720,
      "train_flow_matching_loss": 0.06640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 21720,
      "train_flow_matching_loss": 0.056640625,
      "train_physics_loss": 0.3267197906970978
    },
    {
      "step": 21720,
      "train_flow_matching_loss": 0.024169921875,
      "train_physics_loss": 0.22996960580348969
    },
    {
      "grad_norm": 0.10640794783830643,
      "learning_rate": 5e-05,
      "loss": 0.1278,
      "step": 21730
    },
    {
      "step": 21730,
      "train_flow_matching_loss": 0.06689453125,
      "train_physics_loss": 0.266981303691864
    },
    {
      "step": 21730,
      "train_flow_matching_loss": 0.07666015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 21730,
      "train_flow_matching_loss": 0.036865234375,
      "train_physics_loss": 0.21471193432807922
    },
    {
      "step": 21730,
      "train_flow_matching_loss": 0.047607421875,
      "train_physics_loss": 0.2938458323478699
    },
    {
      "grad_norm": 0.1112717017531395,
      "learning_rate": 5e-05,
      "loss": 0.1224,
      "step": 21740
    },
    {
      "step": 21740,
      "train_flow_matching_loss": 0.03369140625,
      "train_physics_loss": 0.2371121495962143
    },
    {
      "step": 21740,
      "train_flow_matching_loss": 0.1796875,
      "train_physics_loss": 0.0
    },
    {
      "step": 21740,
      "train_flow_matching_loss": 0.044677734375,
      "train_physics_loss": 0.22713297605514526
    },
    {
      "step": 21740,
      "train_flow_matching_loss": 0.0673828125,
      "train_physics_loss": 0.3093274235725403
    },
    {
      "grad_norm": 0.06621850281953812,
      "learning_rate": 5e-05,
      "loss": 0.1302,
      "step": 21750
    },
    {
      "step": 21750,
      "train_flow_matching_loss": 0.050537109375,
      "train_physics_loss": 0.26113903522491455
    },
    {
      "step": 21750,
      "train_flow_matching_loss": 0.1376953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 21750,
      "train_flow_matching_loss": 0.154296875,
      "train_physics_loss": 0.2693541646003723
    },
    {
      "step": 21750,
      "train_flow_matching_loss": 0.027099609375,
      "train_physics_loss": 0.2523050010204315
    },
    {
      "grad_norm": 0.08561121672391891,
      "learning_rate": 5e-05,
      "loss": 0.1301,
      "step": 21760
    },
    {
      "step": 21760,
      "train_flow_matching_loss": 0.0306396484375,
      "train_physics_loss": 0.2718781530857086
    },
    {
      "step": 21760,
      "train_flow_matching_loss": 0.08740234375,
      "train_physics_loss": 0.0
    },
    {
      "step": 21760,
      "train_flow_matching_loss": 0.0289306640625,
      "train_physics_loss": 0.2565782964229584
    },
    {
      "step": 21760,
      "train_flow_matching_loss": 0.046875,
      "train_physics_loss": 0.2679807245731354
    },
    {
      "grad_norm": 0.06779491156339645,
      "learning_rate": 5e-05,
      "loss": 0.1245,
      "step": 21770
    },
    {
      "step": 21770,
      "train_flow_matching_loss": 0.1474609375,
      "train_physics_loss": 0.0
    },
    {
      "step": 21770,
      "train_flow_matching_loss": 0.1728515625,
      "train_physics_loss": 0.0
    },
    {
      "step": 21770,
      "train_flow_matching_loss": 0.08349609375,
      "train_physics_loss": 0.24903704226016998
    },
    {
      "step": 21770,
      "train_flow_matching_loss": 0.11376953125,
      "train_physics_loss": 0.24045805633068085
    },
    {
      "grad_norm": 0.13696140050888062,
      "learning_rate": 5e-05,
      "loss": 0.1265,
      "step": 21780
    },
    {
      "step": 21780,
      "train_flow_matching_loss": 0.1513671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 21780,
      "train_flow_matching_loss": 0.15625,
      "train_physics_loss": 0.0
    },
    {
      "step": 21780,
      "train_flow_matching_loss": 0.049560546875,
      "train_physics_loss": 0.2602829337120056
    },
    {
      "step": 21780,
      "train_flow_matching_loss": 0.0810546875,
      "train_physics_loss": 0.3221845328807831
    },
    {
      "grad_norm": 0.09738263487815857,
      "learning_rate": 5e-05,
      "loss": 0.1282,
      "step": 21790
    },
    {
      "step": 21790,
      "train_flow_matching_loss": 0.150390625,
      "train_physics_loss": 0.0
    },
    {
      "step": 21790,
      "train_flow_matching_loss": 0.11669921875,
      "train_physics_loss": 0.0
    },
    {
      "step": 21790,
      "train_flow_matching_loss": 0.0810546875,
      "train_physics_loss": 0.35637736320495605
    },
    {
      "step": 21790,
      "train_flow_matching_loss": 0.047119140625,
      "train_physics_loss": 0.2153155505657196
    },
    {
      "grad_norm": 0.09018953144550323,
      "learning_rate": 5e-05,
      "loss": 0.1271,
      "step": 21800
    },
    {
      "step": 21800,
      "train_flow_matching_loss": 0.15625,
      "train_physics_loss": 0.0
    },
    {
      "step": 21800,
      "train_flow_matching_loss": 0.09912109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 21800,
      "train_flow_matching_loss": 0.031982421875,
      "train_physics_loss": 0.2541743814945221
    },
    {
      "step": 21800,
      "train_flow_matching_loss": 0.0751953125,
      "train_physics_loss": 0.32369279861450195
    },
    {
      "grad_norm": 0.09833764284849167,
      "learning_rate": 5e-05,
      "loss": 0.1302,
      "step": 21810
    },
    {
      "step": 21810,
      "train_flow_matching_loss": 0.2314453125,
      "train_physics_loss": 0.0
    },
    {
      "step": 21810,
      "train_flow_matching_loss": 0.1171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 21810,
      "train_flow_matching_loss": 0.04931640625,
      "train_physics_loss": 0.23100902140140533
    },
    {
      "step": 21810,
      "train_flow_matching_loss": 0.056640625,
      "train_physics_loss": 0.2786902189254761
    },
    {
      "grad_norm": 0.08740510046482086,
      "learning_rate": 5e-05,
      "loss": 0.1229,
      "step": 21820
    },
    {
      "step": 21820,
      "train_flow_matching_loss": 0.1484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 21820,
      "train_flow_matching_loss": 0.1357421875,
      "train_physics_loss": 0.0
    },
    {
      "step": 21820,
      "train_flow_matching_loss": 0.0927734375,
      "train_physics_loss": 0.3336538076400757
    },
    {
      "step": 21820,
      "train_flow_matching_loss": 0.08544921875,
      "train_physics_loss": 0.23204463720321655
    },
    {
      "grad_norm": 0.10140716284513474,
      "learning_rate": 5e-05,
      "loss": 0.1281,
      "step": 21830
    },
    {
      "step": 21830,
      "train_flow_matching_loss": 0.1484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 21830,
      "train_flow_matching_loss": 0.10107421875,
      "train_physics_loss": 0.0
    },
    {
      "step": 21830,
      "train_flow_matching_loss": 0.037841796875,
      "train_physics_loss": 0.2767086923122406
    },
    {
      "step": 21830,
      "train_flow_matching_loss": 0.08056640625,
      "train_physics_loss": 0.3172371983528137
    },
    {
      "grad_norm": 0.10769126564264297,
      "learning_rate": 5e-05,
      "loss": 0.1233,
      "step": 21840
    },
    {
      "step": 21840,
      "train_flow_matching_loss": 0.142578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 21840,
      "train_flow_matching_loss": 0.1904296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 21840,
      "train_flow_matching_loss": 0.047119140625,
      "train_physics_loss": 0.0
    },
    {
      "step": 21840,
      "train_flow_matching_loss": 0.046630859375,
      "train_physics_loss": 0.3356226980686188
    },
    {
      "grad_norm": 0.11426663398742676,
      "learning_rate": 5e-05,
      "loss": 0.1316,
      "step": 21850
    },
    {
      "step": 21850,
      "train_flow_matching_loss": 0.1376953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 21850,
      "train_flow_matching_loss": 0.0859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 21850,
      "train_flow_matching_loss": 0.0390625,
      "train_physics_loss": 0.27870112657546997
    },
    {
      "step": 21850,
      "train_flow_matching_loss": 0.05029296875,
      "train_physics_loss": 0.30936071276664734
    },
    {
      "grad_norm": 0.08967472612857819,
      "learning_rate": 5e-05,
      "loss": 0.1304,
      "step": 21860
    },
    {
      "step": 21860,
      "train_flow_matching_loss": 0.1298828125,
      "train_physics_loss": 0.0
    },
    {
      "step": 21860,
      "train_flow_matching_loss": 0.1943359375,
      "train_physics_loss": 0.0
    },
    {
      "step": 21860,
      "train_flow_matching_loss": 0.05859375,
      "train_physics_loss": 0.2157909870147705
    },
    {
      "step": 21860,
      "train_flow_matching_loss": 0.049072265625,
      "train_physics_loss": 0.22903452813625336
    },
    {
      "grad_norm": 0.1329660266637802,
      "learning_rate": 5e-05,
      "loss": 0.1312,
      "step": 21870
    },
    {
      "step": 21870,
      "train_flow_matching_loss": 0.1962890625,
      "train_physics_loss": 0.0
    },
    {
      "step": 21870,
      "train_flow_matching_loss": 0.10888671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 21870,
      "train_flow_matching_loss": 0.046875,
      "train_physics_loss": 0.26038289070129395
    },
    {
      "step": 21870,
      "train_flow_matching_loss": 0.026611328125,
      "train_physics_loss": 0.2536858916282654
    },
    {
      "grad_norm": 0.1252875179052353,
      "learning_rate": 5e-05,
      "loss": 0.1284,
      "step": 21880
    },
    {
      "step": 21880,
      "train_flow_matching_loss": 0.1328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 21880,
      "train_flow_matching_loss": 0.1826171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 21880,
      "train_flow_matching_loss": 0.057373046875,
      "train_physics_loss": 0.0
    },
    {
      "step": 21880,
      "train_flow_matching_loss": 0.0283203125,
      "train_physics_loss": 0.2787010669708252
    },
    {
      "grad_norm": 0.08264515548944473,
      "learning_rate": 5e-05,
      "loss": 0.1273,
      "step": 21890
    },
    {
      "step": 21890,
      "train_flow_matching_loss": 0.1494140625,
      "train_physics_loss": 0.0
    },
    {
      "step": 21890,
      "train_flow_matching_loss": 0.1103515625,
      "train_physics_loss": 0.0
    },
    {
      "step": 21890,
      "train_flow_matching_loss": 0.126953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 21890,
      "train_flow_matching_loss": 0.050048828125,
      "train_physics_loss": 0.2370782494544983
    },
    {
      "grad_norm": 0.15448464453220367,
      "learning_rate": 5e-05,
      "loss": 0.1257,
      "step": 21900
    },
    {
      "step": 21900,
      "train_flow_matching_loss": 0.142578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 21900,
      "train_flow_matching_loss": 0.08740234375,
      "train_physics_loss": 0.0
    },
    {
      "step": 21900,
      "train_flow_matching_loss": 0.1376953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 21900,
      "train_flow_matching_loss": 0.10107421875,
      "train_physics_loss": 0.27564841508865356
    },
    {
      "grad_norm": 0.10832514613866806,
      "learning_rate": 5e-05,
      "loss": 0.1295,
      "step": 21910
    },
    {
      "step": 21910,
      "train_flow_matching_loss": 0.1279296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 21910,
      "train_flow_matching_loss": 0.126953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 21910,
      "train_flow_matching_loss": 0.2001953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 21910,
      "train_flow_matching_loss": 0.1787109375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.1067936047911644,
      "learning_rate": 5e-05,
      "loss": 0.1246,
      "step": 21920
    },
    {
      "step": 21920,
      "train_flow_matching_loss": 0.1865234375,
      "train_physics_loss": 0.0
    },
    {
      "step": 21920,
      "train_flow_matching_loss": 0.0830078125,
      "train_physics_loss": 0.0
    },
    {
      "step": 21920,
      "train_flow_matching_loss": 0.1396484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 21920,
      "train_flow_matching_loss": 0.1865234375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.132022887468338,
      "learning_rate": 5e-05,
      "loss": 0.1327,
      "step": 21930
    },
    {
      "step": 21930,
      "train_flow_matching_loss": 0.2109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 21930,
      "train_flow_matching_loss": 0.12451171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 21930,
      "train_flow_matching_loss": 0.1953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 21930,
      "train_flow_matching_loss": 0.1650390625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10284681618213654,
      "learning_rate": 5e-05,
      "loss": 0.1276,
      "step": 21940
    },
    {
      "step": 21940,
      "train_flow_matching_loss": 0.15625,
      "train_physics_loss": 0.0
    },
    {
      "step": 21940,
      "train_flow_matching_loss": 0.1357421875,
      "train_physics_loss": 0.0
    },
    {
      "step": 21940,
      "train_flow_matching_loss": 0.1376953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 21940,
      "train_flow_matching_loss": 0.1298828125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10809893906116486,
      "learning_rate": 5e-05,
      "loss": 0.1261,
      "step": 21950
    },
    {
      "step": 21950,
      "train_flow_matching_loss": 0.1748046875,
      "train_physics_loss": 0.0
    },
    {
      "step": 21950,
      "train_flow_matching_loss": 0.1494140625,
      "train_physics_loss": 0.0
    },
    {
      "step": 21950,
      "train_flow_matching_loss": 0.1689453125,
      "train_physics_loss": 0.0
    },
    {
      "step": 21950,
      "train_flow_matching_loss": 0.2236328125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10528239607810974,
      "learning_rate": 5e-05,
      "loss": 0.1252,
      "step": 21960
    },
    {
      "step": 21960,
      "train_flow_matching_loss": 0.1328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 21960,
      "train_flow_matching_loss": 0.1279296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 21960,
      "train_flow_matching_loss": 0.0888671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 21960,
      "train_flow_matching_loss": 0.0927734375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.35241010785102844,
      "learning_rate": 5e-05,
      "loss": 0.1203,
      "step": 21970
    },
    {
      "step": 21970,
      "train_flow_matching_loss": 0.130859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 21970,
      "train_flow_matching_loss": 0.1484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 21970,
      "train_flow_matching_loss": 0.10595703125,
      "train_physics_loss": 0.0
    },
    {
      "step": 21970,
      "train_flow_matching_loss": 0.09228515625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.1373250037431717,
      "learning_rate": 5e-05,
      "loss": 0.1327,
      "step": 21980
    },
    {
      "step": 21980,
      "train_flow_matching_loss": 0.1474609375,
      "train_physics_loss": 0.0
    },
    {
      "step": 21980,
      "train_flow_matching_loss": 0.1201171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 21980,
      "train_flow_matching_loss": 0.1220703125,
      "train_physics_loss": 0.0
    },
    {
      "step": 21980,
      "train_flow_matching_loss": 0.11865234375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.19458377361297607,
      "learning_rate": 5e-05,
      "loss": 0.1212,
      "step": 21990
    },
    {
      "step": 21990,
      "train_flow_matching_loss": 0.12158203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 21990,
      "train_flow_matching_loss": 0.0888671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 21990,
      "train_flow_matching_loss": 0.10791015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 21990,
      "train_flow_matching_loss": 0.146484375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09592869877815247,
      "learning_rate": 5e-05,
      "loss": 0.1255,
      "step": 22000
    },
    {
      "step": 22000,
      "train_flow_matching_loss": 0.09765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 22000,
      "train_flow_matching_loss": 0.138671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 22000,
      "train_flow_matching_loss": 0.0927734375,
      "train_physics_loss": 0.0
    },
    {
      "step": 22000,
      "train_flow_matching_loss": 0.177734375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.23685869574546814,
      "learning_rate": 5e-05,
      "loss": 0.128,
      "step": 22010
    },
    {
      "step": 22010,
      "train_flow_matching_loss": 0.142578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 22010,
      "train_flow_matching_loss": 0.07861328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 22010,
      "train_flow_matching_loss": 0.1630859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 22010,
      "train_flow_matching_loss": 0.0908203125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.07494814693927765,
      "learning_rate": 5e-05,
      "loss": 0.126,
      "step": 22020
    },
    {
      "step": 22020,
      "train_flow_matching_loss": 0.201171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 22020,
      "train_flow_matching_loss": 0.1298828125,
      "train_physics_loss": 0.0
    },
    {
      "step": 22020,
      "train_flow_matching_loss": 0.11328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 22020,
      "train_flow_matching_loss": 0.1875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.18415598571300507,
      "learning_rate": 5e-05,
      "loss": 0.1271,
      "step": 22030
    },
    {
      "step": 22030,
      "train_flow_matching_loss": 0.05029296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 22030,
      "train_flow_matching_loss": 0.10009765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 22030,
      "train_flow_matching_loss": 0.09326171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 22030,
      "train_flow_matching_loss": 0.2119140625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.1154785081744194,
      "learning_rate": 5e-05,
      "loss": 0.1273,
      "step": 22040
    },
    {
      "step": 22040,
      "train_flow_matching_loss": 0.09521484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 22040,
      "train_flow_matching_loss": 0.09619140625,
      "train_physics_loss": 0.0
    },
    {
      "step": 22040,
      "train_flow_matching_loss": 0.138671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 22040,
      "train_flow_matching_loss": 0.1259765625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11347944289445877,
      "learning_rate": 5e-05,
      "loss": 0.127,
      "step": 22050
    },
    {
      "step": 22050,
      "train_flow_matching_loss": 0.181640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 22050,
      "train_flow_matching_loss": 0.2314453125,
      "train_physics_loss": 0.0
    },
    {
      "step": 22050,
      "train_flow_matching_loss": 0.1083984375,
      "train_physics_loss": 0.0
    },
    {
      "step": 22050,
      "train_flow_matching_loss": 0.1259765625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08656008541584015,
      "learning_rate": 5e-05,
      "loss": 0.1244,
      "step": 22060
    },
    {
      "step": 22060,
      "train_flow_matching_loss": 0.11083984375,
      "train_physics_loss": 0.0
    },
    {
      "step": 22060,
      "train_flow_matching_loss": 0.11572265625,
      "train_physics_loss": 0.0
    },
    {
      "step": 22060,
      "train_flow_matching_loss": 0.166015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 22060,
      "train_flow_matching_loss": 0.12255859375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.0710383728146553,
      "learning_rate": 5e-05,
      "loss": 0.1259,
      "step": 22070
    },
    {
      "step": 22070,
      "train_flow_matching_loss": 0.140625,
      "train_physics_loss": 0.0
    },
    {
      "step": 22070,
      "train_flow_matching_loss": 0.08251953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 22070,
      "train_flow_matching_loss": 0.138671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 22070,
      "train_flow_matching_loss": 0.201171875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11406462639570236,
      "learning_rate": 5e-05,
      "loss": 0.1282,
      "step": 22080
    },
    {
      "step": 22080,
      "train_flow_matching_loss": 0.12158203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 22080,
      "train_flow_matching_loss": 0.1328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 22080,
      "train_flow_matching_loss": 0.109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 22080,
      "train_flow_matching_loss": 0.07470703125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08382008224725723,
      "learning_rate": 5e-05,
      "loss": 0.1257,
      "step": 22090
    },
    {
      "step": 22090,
      "train_flow_matching_loss": 0.126953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 22090,
      "train_flow_matching_loss": 0.09228515625,
      "train_physics_loss": 0.0
    },
    {
      "step": 22090,
      "train_flow_matching_loss": 0.09814453125,
      "train_physics_loss": 0.0
    },
    {
      "step": 22090,
      "train_flow_matching_loss": 0.1435546875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.171194925904274,
      "learning_rate": 5e-05,
      "loss": 0.1275,
      "step": 22100
    },
    {
      "step": 22100,
      "train_flow_matching_loss": 0.11865234375,
      "train_physics_loss": 0.0
    },
    {
      "step": 22100,
      "train_flow_matching_loss": 0.11328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 22100,
      "train_flow_matching_loss": 0.1552734375,
      "train_physics_loss": 0.0
    },
    {
      "step": 22100,
      "train_flow_matching_loss": 0.07666015625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.0879741907119751,
      "learning_rate": 5e-05,
      "loss": 0.13,
      "step": 22110
    },
    {
      "step": 22110,
      "train_flow_matching_loss": 0.0947265625,
      "train_physics_loss": 0.0
    },
    {
      "step": 22110,
      "train_flow_matching_loss": 0.2197265625,
      "train_physics_loss": 0.0
    },
    {
      "step": 22110,
      "train_flow_matching_loss": 0.1611328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 22110,
      "train_flow_matching_loss": 0.1162109375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.15058942139148712,
      "learning_rate": 5e-05,
      "loss": 0.1224,
      "step": 22120
    },
    {
      "step": 22120,
      "train_flow_matching_loss": 0.1630859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 22120,
      "train_flow_matching_loss": 0.11865234375,
      "train_physics_loss": 0.0
    },
    {
      "step": 22120,
      "train_flow_matching_loss": 0.1708984375,
      "train_physics_loss": 0.0
    },
    {
      "step": 22120,
      "train_flow_matching_loss": 0.1474609375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08644179999828339,
      "learning_rate": 5e-05,
      "loss": 0.1244,
      "step": 22130
    },
    {
      "step": 22130,
      "train_flow_matching_loss": 0.1533203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 22130,
      "train_flow_matching_loss": 0.12109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 22130,
      "train_flow_matching_loss": 0.2353515625,
      "train_physics_loss": 0.0
    },
    {
      "step": 22130,
      "train_flow_matching_loss": 0.0771484375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09944161772727966,
      "learning_rate": 5e-05,
      "loss": 0.1236,
      "step": 22140
    },
    {
      "step": 22140,
      "train_flow_matching_loss": 0.1005859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 22140,
      "train_flow_matching_loss": 0.09423828125,
      "train_physics_loss": 0.0
    },
    {
      "step": 22140,
      "train_flow_matching_loss": 0.1083984375,
      "train_physics_loss": 0.0
    },
    {
      "step": 22140,
      "train_flow_matching_loss": 0.08349609375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.1635768860578537,
      "learning_rate": 5e-05,
      "loss": 0.1293,
      "step": 22150
    },
    {
      "step": 22150,
      "train_flow_matching_loss": 0.21875,
      "train_physics_loss": 0.0
    },
    {
      "step": 22150,
      "train_flow_matching_loss": 0.166015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 22150,
      "train_flow_matching_loss": 0.1044921875,
      "train_physics_loss": 0.0
    },
    {
      "step": 22150,
      "train_flow_matching_loss": 0.1611328125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.13345035910606384,
      "learning_rate": 5e-05,
      "loss": 0.1276,
      "step": 22160
    },
    {
      "step": 22160,
      "train_flow_matching_loss": 0.1435546875,
      "train_physics_loss": 0.0
    },
    {
      "step": 22160,
      "train_flow_matching_loss": 0.1259765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 22160,
      "train_flow_matching_loss": 0.1337890625,
      "train_physics_loss": 0.0
    },
    {
      "step": 22160,
      "train_flow_matching_loss": 0.06884765625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.07749021053314209,
      "learning_rate": 5e-05,
      "loss": 0.1222,
      "step": 22170
    },
    {
      "step": 22170,
      "train_flow_matching_loss": 0.1513671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 22170,
      "train_flow_matching_loss": 0.162109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 22170,
      "train_flow_matching_loss": 0.08056640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 22170,
      "train_flow_matching_loss": 0.1484375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09723736345767975,
      "learning_rate": 5e-05,
      "loss": 0.1289,
      "step": 22180
    },
    {
      "step": 22180,
      "train_flow_matching_loss": 0.171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 22180,
      "train_flow_matching_loss": 0.205078125,
      "train_physics_loss": 0.0
    },
    {
      "step": 22180,
      "train_flow_matching_loss": 0.11865234375,
      "train_physics_loss": 0.0
    },
    {
      "step": 22180,
      "train_flow_matching_loss": 0.126953125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.12553735077381134,
      "learning_rate": 5e-05,
      "loss": 0.1302,
      "step": 22190
    },
    {
      "step": 22190,
      "train_flow_matching_loss": 0.1396484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 22190,
      "train_flow_matching_loss": 0.10302734375,
      "train_physics_loss": 0.0
    },
    {
      "step": 22190,
      "train_flow_matching_loss": 0.1376953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 22190,
      "train_flow_matching_loss": 0.0771484375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.07999801635742188,
      "learning_rate": 5e-05,
      "loss": 0.1252,
      "step": 22200
    },
    {
      "step": 22200,
      "train_flow_matching_loss": 0.130859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 22200,
      "train_flow_matching_loss": 0.08251953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 22200,
      "train_flow_matching_loss": 0.13671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 22200,
      "train_flow_matching_loss": 0.0859375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.07457007467746735,
      "learning_rate": 5e-05,
      "loss": 0.1203,
      "step": 22210
    },
    {
      "step": 22210,
      "train_flow_matching_loss": 0.0927734375,
      "train_physics_loss": 0.0
    },
    {
      "step": 22210,
      "train_flow_matching_loss": 0.11474609375,
      "train_physics_loss": 0.0
    },
    {
      "step": 22210,
      "train_flow_matching_loss": 0.1357421875,
      "train_physics_loss": 0.0
    },
    {
      "step": 22210,
      "train_flow_matching_loss": 0.078125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.1072613000869751,
      "learning_rate": 5e-05,
      "loss": 0.1298,
      "step": 22220
    },
    {
      "step": 22220,
      "train_flow_matching_loss": 0.162109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 22220,
      "train_flow_matching_loss": 0.09716796875,
      "train_physics_loss": 0.0
    },
    {
      "step": 22220,
      "train_flow_matching_loss": 0.10546875,
      "train_physics_loss": 0.0
    },
    {
      "step": 22220,
      "train_flow_matching_loss": 0.06982421875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.2901853024959564,
      "learning_rate": 5e-05,
      "loss": 0.1246,
      "step": 22230
    },
    {
      "step": 22230,
      "train_flow_matching_loss": 0.11279296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 22230,
      "train_flow_matching_loss": 0.09814453125,
      "train_physics_loss": 0.0
    },
    {
      "step": 22230,
      "train_flow_matching_loss": 0.154296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 22230,
      "train_flow_matching_loss": 0.11279296875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.17554372549057007,
      "learning_rate": 5e-05,
      "loss": 0.1282,
      "step": 22240
    },
    {
      "step": 22240,
      "train_flow_matching_loss": 0.09228515625,
      "train_physics_loss": 0.0
    },
    {
      "step": 22240,
      "train_flow_matching_loss": 0.1494140625,
      "train_physics_loss": 0.0
    },
    {
      "step": 22240,
      "train_flow_matching_loss": 0.134765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 22240,
      "train_flow_matching_loss": 0.1591796875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08392155915498734,
      "learning_rate": 5e-05,
      "loss": 0.1303,
      "step": 22250
    },
    {
      "step": 22250,
      "train_flow_matching_loss": 0.11865234375,
      "train_physics_loss": 0.0
    },
    {
      "step": 22250,
      "train_flow_matching_loss": 0.189453125,
      "train_physics_loss": 0.0
    },
    {
      "step": 22250,
      "train_flow_matching_loss": 0.1572265625,
      "train_physics_loss": 0.0
    },
    {
      "step": 22250,
      "train_flow_matching_loss": 0.11328125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09064125269651413,
      "learning_rate": 5e-05,
      "loss": 0.1299,
      "step": 22260
    },
    {
      "step": 22260,
      "train_flow_matching_loss": 0.1279296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 22260,
      "train_flow_matching_loss": 0.1162109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 22260,
      "train_flow_matching_loss": 0.0849609375,
      "train_physics_loss": 0.0
    },
    {
      "step": 22260,
      "train_flow_matching_loss": 0.0869140625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.07555610686540604,
      "learning_rate": 5e-05,
      "loss": 0.1234,
      "step": 22270
    },
    {
      "step": 22270,
      "train_flow_matching_loss": 0.146484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 22270,
      "train_flow_matching_loss": 0.09130859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 22270,
      "train_flow_matching_loss": 0.1259765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 22270,
      "train_flow_matching_loss": 0.0986328125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11487894505262375,
      "learning_rate": 5e-05,
      "loss": 0.1258,
      "step": 22280
    },
    {
      "step": 22280,
      "train_flow_matching_loss": 0.162109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 22280,
      "train_flow_matching_loss": 0.059326171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 22280,
      "train_flow_matching_loss": 0.1318359375,
      "train_physics_loss": 0.0
    },
    {
      "step": 22280,
      "train_flow_matching_loss": 0.1064453125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.06923800706863403,
      "learning_rate": 5e-05,
      "loss": 0.1189,
      "step": 22290
    },
    {
      "step": 22290,
      "train_flow_matching_loss": 0.09326171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 22290,
      "train_flow_matching_loss": 0.1279296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 22290,
      "train_flow_matching_loss": 0.1689453125,
      "train_physics_loss": 0.0
    },
    {
      "step": 22290,
      "train_flow_matching_loss": 0.0634765625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09668474644422531,
      "learning_rate": 5e-05,
      "loss": 0.1213,
      "step": 22300
    },
    {
      "step": 22300,
      "train_flow_matching_loss": 0.0869140625,
      "train_physics_loss": 0.0
    },
    {
      "step": 22300,
      "train_flow_matching_loss": 0.173828125,
      "train_physics_loss": 0.0
    },
    {
      "step": 22300,
      "train_flow_matching_loss": 0.15625,
      "train_physics_loss": 0.0
    },
    {
      "step": 22300,
      "train_flow_matching_loss": 0.134765625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.46858471632003784,
      "learning_rate": 5e-05,
      "loss": 0.1256,
      "step": 22310
    },
    {
      "step": 22310,
      "train_flow_matching_loss": 0.10986328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 22310,
      "train_flow_matching_loss": 0.09912109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 22310,
      "train_flow_matching_loss": 0.158203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 22310,
      "train_flow_matching_loss": 0.1845703125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08640273660421371,
      "learning_rate": 5e-05,
      "loss": 0.129,
      "step": 22320
    },
    {
      "step": 22320,
      "train_flow_matching_loss": 0.1435546875,
      "train_physics_loss": 0.0
    },
    {
      "step": 22320,
      "train_flow_matching_loss": 0.1455078125,
      "train_physics_loss": 0.0
    },
    {
      "step": 22320,
      "train_flow_matching_loss": 0.134765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 22320,
      "train_flow_matching_loss": 0.09765625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.16220323741436005,
      "learning_rate": 5e-05,
      "loss": 0.1264,
      "step": 22330
    },
    {
      "step": 22330,
      "train_flow_matching_loss": 0.11083984375,
      "train_physics_loss": 0.0
    },
    {
      "step": 22330,
      "train_flow_matching_loss": 0.146484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 22330,
      "train_flow_matching_loss": 0.1845703125,
      "train_physics_loss": 0.0
    },
    {
      "step": 22330,
      "train_flow_matching_loss": 0.146484375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.12726712226867676,
      "learning_rate": 5e-05,
      "loss": 0.1276,
      "step": 22340
    },
    {
      "step": 22340,
      "train_flow_matching_loss": 0.1123046875,
      "train_physics_loss": 0.0
    },
    {
      "step": 22340,
      "train_flow_matching_loss": 0.11865234375,
      "train_physics_loss": 0.0
    },
    {
      "step": 22340,
      "train_flow_matching_loss": 0.07568359375,
      "train_physics_loss": 0.0
    },
    {
      "step": 22340,
      "train_flow_matching_loss": 0.1357421875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.1064981073141098,
      "learning_rate": 5e-05,
      "loss": 0.1251,
      "step": 22350
    },
    {
      "step": 22350,
      "train_flow_matching_loss": 0.062255859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 22350,
      "train_flow_matching_loss": 0.031982421875,
      "train_physics_loss": 0.2551891803741455
    },
    {
      "step": 22350,
      "train_flow_matching_loss": 0.09375,
      "train_physics_loss": 0.0
    },
    {
      "step": 22350,
      "train_flow_matching_loss": 0.1201171875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.07673086225986481,
      "learning_rate": 5e-05,
      "loss": 0.1223,
      "step": 22360
    },
    {
      "step": 22360,
      "train_flow_matching_loss": 0.11328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 22360,
      "train_flow_matching_loss": 0.05615234375,
      "train_physics_loss": 0.2057085782289505
    },
    {
      "step": 22360,
      "train_flow_matching_loss": 0.1416015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 22360,
      "train_flow_matching_loss": 0.10302734375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11218713223934174,
      "learning_rate": 5e-05,
      "loss": 0.1287,
      "step": 22370
    },
    {
      "step": 22370,
      "train_flow_matching_loss": 0.0771484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 22370,
      "train_flow_matching_loss": 0.07275390625,
      "train_physics_loss": 0.26433423161506653
    },
    {
      "step": 22370,
      "train_flow_matching_loss": 0.1484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 22370,
      "train_flow_matching_loss": 0.1533203125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.05654466524720192,
      "learning_rate": 5e-05,
      "loss": 0.1288,
      "step": 22380
    },
    {
      "step": 22380,
      "train_flow_matching_loss": 0.11083984375,
      "train_physics_loss": 0.0
    },
    {
      "step": 22380,
      "train_flow_matching_loss": 0.0771484375,
      "train_physics_loss": 0.4081283509731293
    },
    {
      "step": 22380,
      "train_flow_matching_loss": 0.10302734375,
      "train_physics_loss": 0.0
    },
    {
      "step": 22380,
      "train_flow_matching_loss": 0.1044921875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10510283708572388,
      "learning_rate": 5e-05,
      "loss": 0.1241,
      "step": 22390
    },
    {
      "step": 22390,
      "train_flow_matching_loss": 0.10546875,
      "train_physics_loss": 0.0
    },
    {
      "step": 22390,
      "train_flow_matching_loss": 0.031982421875,
      "train_physics_loss": 0.25902172923088074
    },
    {
      "step": 22390,
      "train_flow_matching_loss": 0.1552734375,
      "train_physics_loss": 0.0
    },
    {
      "step": 22390,
      "train_flow_matching_loss": 0.08056640625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10050676763057709,
      "learning_rate": 5e-05,
      "loss": 0.1252,
      "step": 22400
    },
    {
      "step": 22400,
      "train_flow_matching_loss": 0.1015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 22400,
      "train_flow_matching_loss": 0.0732421875,
      "train_physics_loss": 0.24001111090183258
    },
    {
      "step": 22400,
      "train_flow_matching_loss": 0.2119140625,
      "train_physics_loss": 0.0
    },
    {
      "step": 22400,
      "train_flow_matching_loss": 0.1005859375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.1304655373096466,
      "learning_rate": 5e-05,
      "loss": 0.1257,
      "step": 22410
    },
    {
      "step": 22410,
      "train_flow_matching_loss": 0.1796875,
      "train_physics_loss": 0.0
    },
    {
      "step": 22410,
      "train_flow_matching_loss": 0.0230712890625,
      "train_physics_loss": 0.24608367681503296
    },
    {
      "step": 22410,
      "train_flow_matching_loss": 0.142578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 22410,
      "train_flow_matching_loss": 0.036376953125,
      "train_physics_loss": 0.2928435802459717
    },
    {
      "grad_norm": 0.07706186920404434,
      "learning_rate": 5e-05,
      "loss": 0.1254,
      "step": 22420
    },
    {
      "step": 22420,
      "train_flow_matching_loss": 0.1435546875,
      "train_physics_loss": 0.0
    },
    {
      "step": 22420,
      "train_flow_matching_loss": 0.053955078125,
      "train_physics_loss": 0.29135555028915405
    },
    {
      "step": 22420,
      "train_flow_matching_loss": 0.10009765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 22420,
      "train_flow_matching_loss": 0.045654296875,
      "train_physics_loss": 0.2658615708351135
    },
    {
      "grad_norm": 0.11593072861433029,
      "learning_rate": 5e-05,
      "loss": 0.1255,
      "step": 22430
    },
    {
      "step": 22430,
      "train_flow_matching_loss": 0.181640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 22430,
      "train_flow_matching_loss": 0.07861328125,
      "train_physics_loss": 0.24261459708213806
    },
    {
      "step": 22430,
      "train_flow_matching_loss": 0.126953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 22430,
      "train_flow_matching_loss": 0.039794921875,
      "train_physics_loss": 0.2411905825138092
    },
    {
      "grad_norm": 0.09244447201490402,
      "learning_rate": 5e-05,
      "loss": 0.1232,
      "step": 22440
    },
    {
      "step": 22440,
      "train_flow_matching_loss": 0.0888671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 22440,
      "train_flow_matching_loss": 0.08154296875,
      "train_physics_loss": 0.24404989182949066
    },
    {
      "step": 22440,
      "train_flow_matching_loss": 0.10498046875,
      "train_physics_loss": 0.0
    },
    {
      "step": 22440,
      "train_flow_matching_loss": 0.04150390625,
      "train_physics_loss": 0.23208169639110565
    },
    {
      "grad_norm": 0.11561146378517151,
      "learning_rate": 5e-05,
      "loss": 0.1344,
      "step": 22450
    },
    {
      "step": 22450,
      "train_flow_matching_loss": 0.1513671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 22450,
      "train_flow_matching_loss": 0.0947265625,
      "train_physics_loss": 0.24569743871688843
    },
    {
      "step": 22450,
      "train_flow_matching_loss": 0.1962890625,
      "train_physics_loss": 0.0
    },
    {
      "step": 22450,
      "train_flow_matching_loss": 0.0751953125,
      "train_physics_loss": 0.2519969046115875
    },
    {
      "grad_norm": 0.09328745305538177,
      "learning_rate": 5e-05,
      "loss": 0.1308,
      "step": 22460
    },
    {
      "step": 22460,
      "train_flow_matching_loss": 0.1455078125,
      "train_physics_loss": 0.0
    },
    {
      "step": 22460,
      "train_flow_matching_loss": 0.045654296875,
      "train_physics_loss": 0.2705322206020355
    },
    {
      "step": 22460,
      "train_flow_matching_loss": 0.1376953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 22460,
      "train_flow_matching_loss": 0.034423828125,
      "train_physics_loss": 0.29090991616249084
    },
    {
      "grad_norm": 0.06385348737239838,
      "learning_rate": 5e-05,
      "loss": 0.1255,
      "step": 22470
    },
    {
      "step": 22470,
      "train_flow_matching_loss": 0.154296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 22470,
      "train_flow_matching_loss": 0.029296875,
      "train_physics_loss": 0.24049727618694305
    },
    {
      "step": 22470,
      "train_flow_matching_loss": 0.162109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 22470,
      "train_flow_matching_loss": 0.060546875,
      "train_physics_loss": 0.2313549965620041
    },
    {
      "grad_norm": 0.09568840265274048,
      "learning_rate": 5e-05,
      "loss": 0.1267,
      "step": 22480
    },
    {
      "step": 22480,
      "train_flow_matching_loss": 0.09814453125,
      "train_physics_loss": 0.0
    },
    {
      "step": 22480,
      "train_flow_matching_loss": 0.06787109375,
      "train_physics_loss": 0.2532395124435425
    },
    {
      "step": 22480,
      "train_flow_matching_loss": 0.1689453125,
      "train_physics_loss": 0.0
    },
    {
      "step": 22480,
      "train_flow_matching_loss": 0.044921875,
      "train_physics_loss": 0.2534564137458801
    },
    {
      "grad_norm": 0.118157759308815,
      "learning_rate": 5e-05,
      "loss": 0.1287,
      "step": 22490
    },
    {
      "step": 22490,
      "train_flow_matching_loss": 0.0830078125,
      "train_physics_loss": 0.0
    },
    {
      "step": 22490,
      "train_flow_matching_loss": 0.038330078125,
      "train_physics_loss": 0.24618767201900482
    },
    {
      "step": 22490,
      "train_flow_matching_loss": 0.2255859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 22490,
      "train_flow_matching_loss": 0.078125,
      "train_physics_loss": 0.2462048977613449
    },
    {
      "grad_norm": 0.07508119195699692,
      "learning_rate": 5e-05,
      "loss": 0.1274,
      "step": 22500
    },
    {
      "step": 22500,
      "train_flow_matching_loss": 0.1845703125,
      "train_physics_loss": 0.0
    },
    {
      "step": 22500,
      "train_flow_matching_loss": 0.07080078125,
      "train_physics_loss": 0.29105377197265625
    },
    {
      "step": 22500,
      "train_flow_matching_loss": 0.10400390625,
      "train_physics_loss": 0.0
    },
    {
      "step": 22500,
      "train_flow_matching_loss": 0.046142578125,
      "train_physics_loss": 0.2367914617061615
    },
    {
      "grad_norm": 0.08407912403345108,
      "learning_rate": 5e-05,
      "loss": 0.1206,
      "step": 22510
    },
    {
      "step": 22510,
      "train_flow_matching_loss": 0.126953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 22510,
      "train_flow_matching_loss": 0.11572265625,
      "train_physics_loss": 0.2779834270477295
    },
    {
      "step": 22510,
      "train_flow_matching_loss": 0.08203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 22510,
      "train_flow_matching_loss": 0.04248046875,
      "train_physics_loss": 0.28916624188423157
    },
    {
      "grad_norm": 0.12238790094852448,
      "learning_rate": 5e-05,
      "loss": 0.1245,
      "step": 22520
    },
    {
      "step": 22520,
      "train_flow_matching_loss": 0.046630859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 22520,
      "train_flow_matching_loss": 0.0869140625,
      "train_physics_loss": 0.3042736053466797
    },
    {
      "step": 22520,
      "train_flow_matching_loss": 0.2490234375,
      "train_physics_loss": 0.0
    },
    {
      "step": 22520,
      "train_flow_matching_loss": 0.0279541015625,
      "train_physics_loss": 0.2364944964647293
    },
    {
      "grad_norm": 0.0916934460401535,
      "learning_rate": 5e-05,
      "loss": 0.1285,
      "step": 22530
    },
    {
      "step": 22530,
      "train_flow_matching_loss": 0.1318359375,
      "train_physics_loss": 0.0
    },
    {
      "step": 22530,
      "train_flow_matching_loss": 0.040283203125,
      "train_physics_loss": 0.3004290461540222
    },
    {
      "step": 22530,
      "train_flow_matching_loss": 0.1611328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 22530,
      "train_flow_matching_loss": 0.044921875,
      "train_physics_loss": 0.2848287522792816
    },
    {
      "grad_norm": 0.12643811106681824,
      "learning_rate": 5e-05,
      "loss": 0.1224,
      "step": 22540
    },
    {
      "step": 22540,
      "train_flow_matching_loss": 0.0654296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 22540,
      "train_flow_matching_loss": 0.0252685546875,
      "train_physics_loss": 0.24198707938194275
    },
    {
      "step": 22540,
      "train_flow_matching_loss": 0.181640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 22540,
      "train_flow_matching_loss": 0.0693359375,
      "train_physics_loss": 0.2429845631122589
    },
    {
      "grad_norm": 0.08346502482891083,
      "learning_rate": 5e-05,
      "loss": 0.1242,
      "step": 22550
    },
    {
      "step": 22550,
      "train_flow_matching_loss": 0.1572265625,
      "train_physics_loss": 0.0
    },
    {
      "step": 22550,
      "train_flow_matching_loss": 0.06298828125,
      "train_physics_loss": 0.272624135017395
    },
    {
      "step": 22550,
      "train_flow_matching_loss": 0.142578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 22550,
      "train_flow_matching_loss": 0.06005859375,
      "train_physics_loss": 0.27862104773521423
    },
    {
      "grad_norm": 0.08347789198160172,
      "learning_rate": 5e-05,
      "loss": 0.1239,
      "step": 22560
    },
    {
      "step": 22560,
      "train_flow_matching_loss": 0.1904296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 22560,
      "train_flow_matching_loss": 0.0634765625,
      "train_physics_loss": 0.1877473145723343
    },
    {
      "step": 22560,
      "train_flow_matching_loss": 0.130859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 22560,
      "train_flow_matching_loss": 0.049560546875,
      "train_physics_loss": 0.2329821139574051
    },
    {
      "grad_norm": 0.08527850359678268,
      "learning_rate": 5e-05,
      "loss": 0.1258,
      "step": 22570
    },
    {
      "step": 22570,
      "train_flow_matching_loss": 0.154296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 22570,
      "train_flow_matching_loss": 0.060791015625,
      "train_physics_loss": 0.24255703389644623
    },
    {
      "step": 22570,
      "train_flow_matching_loss": 0.1279296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 22570,
      "train_flow_matching_loss": 0.045166015625,
      "train_physics_loss": 0.21518927812576294
    },
    {
      "grad_norm": 0.08486635237932205,
      "learning_rate": 5e-05,
      "loss": 0.1203,
      "step": 22580
    },
    {
      "step": 22580,
      "train_flow_matching_loss": 0.2412109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 22580,
      "train_flow_matching_loss": 0.049560546875,
      "train_physics_loss": 0.24034789204597473
    },
    {
      "step": 22580,
      "train_flow_matching_loss": 0.1669921875,
      "train_physics_loss": 0.0
    },
    {
      "step": 22580,
      "train_flow_matching_loss": 0.03173828125,
      "train_physics_loss": 0.2341415137052536
    },
    {
      "grad_norm": 0.15972229838371277,
      "learning_rate": 5e-05,
      "loss": 0.1211,
      "step": 22590
    },
    {
      "step": 22590,
      "train_flow_matching_loss": 0.1455078125,
      "train_physics_loss": 0.0
    },
    {
      "step": 22590,
      "train_flow_matching_loss": 0.048828125,
      "train_physics_loss": 0.2480248510837555
    },
    {
      "step": 22590,
      "train_flow_matching_loss": 0.09375,
      "train_physics_loss": 0.0
    },
    {
      "step": 22590,
      "train_flow_matching_loss": 0.051025390625,
      "train_physics_loss": 0.2543039917945862
    },
    {
      "grad_norm": 0.07387030869722366,
      "learning_rate": 5e-05,
      "loss": 0.1271,
      "step": 22600
    },
    {
      "step": 22600,
      "train_flow_matching_loss": 0.1728515625,
      "train_physics_loss": 0.0
    },
    {
      "step": 22600,
      "train_flow_matching_loss": 0.03125,
      "train_physics_loss": 0.25293073058128357
    },
    {
      "step": 22600,
      "train_flow_matching_loss": 0.10693359375,
      "train_physics_loss": 0.0
    },
    {
      "step": 22600,
      "train_flow_matching_loss": 0.0673828125,
      "train_physics_loss": 0.29160404205322266
    },
    {
      "grad_norm": 0.1229901909828186,
      "learning_rate": 5e-05,
      "loss": 0.1262,
      "step": 22610
    },
    {
      "step": 22610,
      "train_flow_matching_loss": 0.1787109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 22610,
      "train_flow_matching_loss": 0.055908203125,
      "train_physics_loss": 0.2315617948770523
    },
    {
      "step": 22610,
      "train_flow_matching_loss": 0.12353515625,
      "train_physics_loss": 0.0
    },
    {
      "step": 22610,
      "train_flow_matching_loss": 0.048583984375,
      "train_physics_loss": 0.262614905834198
    },
    {
      "grad_norm": 0.09783855825662613,
      "learning_rate": 5e-05,
      "loss": 0.1237,
      "step": 22620
    },
    {
      "step": 22620,
      "train_flow_matching_loss": 0.1689453125,
      "train_physics_loss": 0.0
    },
    {
      "step": 22620,
      "train_flow_matching_loss": 0.0712890625,
      "train_physics_loss": 0.23649050295352936
    },
    {
      "step": 22620,
      "train_flow_matching_loss": 0.11083984375,
      "train_physics_loss": 0.0
    },
    {
      "step": 22620,
      "train_flow_matching_loss": 0.039306640625,
      "train_physics_loss": 0.24216648936271667
    },
    {
      "grad_norm": 0.10240472108125687,
      "learning_rate": 5e-05,
      "loss": 0.119,
      "step": 22630
    },
    {
      "step": 22630,
      "train_flow_matching_loss": 0.08740234375,
      "train_physics_loss": 0.0
    },
    {
      "step": 22630,
      "train_flow_matching_loss": 0.0693359375,
      "train_physics_loss": 0.28459644317626953
    },
    {
      "step": 22630,
      "train_flow_matching_loss": 0.162109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 22630,
      "train_flow_matching_loss": 0.057373046875,
      "train_physics_loss": 0.2947852909564972
    },
    {
      "grad_norm": 0.10928166657686234,
      "learning_rate": 5e-05,
      "loss": 0.1234,
      "step": 22640
    },
    {
      "step": 22640,
      "train_flow_matching_loss": 0.10595703125,
      "train_physics_loss": 0.0
    },
    {
      "step": 22640,
      "train_flow_matching_loss": 0.06494140625,
      "train_physics_loss": 0.23729024827480316
    },
    {
      "step": 22640,
      "train_flow_matching_loss": 0.185546875,
      "train_physics_loss": 0.0
    },
    {
      "step": 22640,
      "train_flow_matching_loss": 0.09423828125,
      "train_physics_loss": 0.31086012721061707
    },
    {
      "grad_norm": 0.08604884892702103,
      "learning_rate": 5e-05,
      "loss": 0.1223,
      "step": 22650
    },
    {
      "step": 22650,
      "train_flow_matching_loss": 0.1767578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 22650,
      "train_flow_matching_loss": 0.0966796875,
      "train_physics_loss": 0.26788395643234253
    },
    {
      "step": 22650,
      "train_flow_matching_loss": 0.169921875,
      "train_physics_loss": 0.0
    },
    {
      "step": 22650,
      "train_flow_matching_loss": 0.045166015625,
      "train_physics_loss": 0.26068314909935
    },
    {
      "grad_norm": 0.11528705060482025,
      "learning_rate": 5e-05,
      "loss": 0.1254,
      "step": 22660
    },
    {
      "step": 22660,
      "train_flow_matching_loss": 0.10791015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 22660,
      "train_flow_matching_loss": 0.08740234375,
      "train_physics_loss": 0.32632070779800415
    },
    {
      "step": 22660,
      "train_flow_matching_loss": 0.1572265625,
      "train_physics_loss": 0.0
    },
    {
      "step": 22660,
      "train_flow_matching_loss": 0.06591796875,
      "train_physics_loss": 0.549412727355957
    },
    {
      "grad_norm": 0.09567806869745255,
      "learning_rate": 5e-05,
      "loss": 0.1269,
      "step": 22670
    },
    {
      "step": 22670,
      "train_flow_matching_loss": 0.1416015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 22670,
      "train_flow_matching_loss": 0.0732421875,
      "train_physics_loss": 0.2497406303882599
    },
    {
      "step": 22670,
      "train_flow_matching_loss": 0.1220703125,
      "train_physics_loss": 0.0
    },
    {
      "step": 22670,
      "train_flow_matching_loss": 0.0260009765625,
      "train_physics_loss": 0.24859283864498138
    },
    {
      "grad_norm": 0.07117068767547607,
      "learning_rate": 5e-05,
      "loss": 0.1226,
      "step": 22680
    },
    {
      "step": 22680,
      "train_flow_matching_loss": 0.1708984375,
      "train_physics_loss": 0.0
    },
    {
      "step": 22680,
      "train_flow_matching_loss": 0.06689453125,
      "train_physics_loss": 0.22078609466552734
    },
    {
      "step": 22680,
      "train_flow_matching_loss": 0.10107421875,
      "train_physics_loss": 0.0
    },
    {
      "step": 22680,
      "train_flow_matching_loss": 0.08740234375,
      "train_physics_loss": 0.3019354045391083
    },
    {
      "grad_norm": 0.105598084628582,
      "learning_rate": 5e-05,
      "loss": 0.1287,
      "step": 22690
    },
    {
      "step": 22690,
      "train_flow_matching_loss": 0.125,
      "train_physics_loss": 0.0
    },
    {
      "step": 22690,
      "train_flow_matching_loss": 0.046875,
      "train_physics_loss": 0.19471833109855652
    },
    {
      "step": 22690,
      "train_flow_matching_loss": 0.2080078125,
      "train_physics_loss": 0.0
    },
    {
      "step": 22690,
      "train_flow_matching_loss": 0.04931640625,
      "train_physics_loss": 0.22721821069717407
    },
    {
      "grad_norm": 0.11544430255889893,
      "learning_rate": 5e-05,
      "loss": 0.1263,
      "step": 22700
    },
    {
      "step": 22700,
      "train_flow_matching_loss": 0.1826171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 22700,
      "train_flow_matching_loss": 0.05517578125,
      "train_physics_loss": 0.3345640003681183
    },
    {
      "step": 22700,
      "train_flow_matching_loss": 0.154296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 22700,
      "train_flow_matching_loss": 0.1005859375,
      "train_physics_loss": 0.21512366831302643
    },
    {
      "grad_norm": 0.141015887260437,
      "learning_rate": 5e-05,
      "loss": 0.1286,
      "step": 22710
    },
    {
      "step": 22710,
      "train_flow_matching_loss": 0.1259765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 22710,
      "train_flow_matching_loss": 0.0703125,
      "train_physics_loss": 0.2265801578760147
    },
    {
      "step": 22710,
      "train_flow_matching_loss": 0.1923828125,
      "train_physics_loss": 0.0
    },
    {
      "step": 22710,
      "train_flow_matching_loss": 0.039306640625,
      "train_physics_loss": 0.23491007089614868
    },
    {
      "grad_norm": 0.08461853861808777,
      "learning_rate": 5e-05,
      "loss": 0.1235,
      "step": 22720
    },
    {
      "step": 22720,
      "train_flow_matching_loss": 0.2119140625,
      "train_physics_loss": 0.0
    },
    {
      "step": 22720,
      "train_flow_matching_loss": 0.07763671875,
      "train_physics_loss": 0.2604999840259552
    },
    {
      "step": 22720,
      "train_flow_matching_loss": 0.1298828125,
      "train_physics_loss": 0.0
    },
    {
      "step": 22720,
      "train_flow_matching_loss": 0.030029296875,
      "train_physics_loss": 0.2465624362230301
    },
    {
      "grad_norm": 0.09568346291780472,
      "learning_rate": 5e-05,
      "loss": 0.1281,
      "step": 22730
    },
    {
      "step": 22730,
      "train_flow_matching_loss": 0.146484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 22730,
      "train_flow_matching_loss": 0.09423828125,
      "train_physics_loss": 0.3232221305370331
    },
    {
      "step": 22730,
      "train_flow_matching_loss": 0.181640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 22730,
      "train_flow_matching_loss": 0.0281982421875,
      "train_physics_loss": 0.22770726680755615
    },
    {
      "grad_norm": 0.08241315931081772,
      "learning_rate": 5e-05,
      "loss": 0.1275,
      "step": 22740
    },
    {
      "step": 22740,
      "train_flow_matching_loss": 0.1513671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 22740,
      "train_flow_matching_loss": 0.0240478515625,
      "train_physics_loss": 0.25504592061042786
    },
    {
      "step": 22740,
      "train_flow_matching_loss": 0.11572265625,
      "train_physics_loss": 0.0
    },
    {
      "step": 22740,
      "train_flow_matching_loss": 0.03466796875,
      "train_physics_loss": 0.2414465993642807
    },
    {
      "grad_norm": 0.09724278002977371,
      "learning_rate": 5e-05,
      "loss": 0.127,
      "step": 22750
    },
    {
      "step": 22750,
      "train_flow_matching_loss": 0.11083984375,
      "train_physics_loss": 0.0
    },
    {
      "step": 22750,
      "train_flow_matching_loss": 0.0830078125,
      "train_physics_loss": 0.24031014740467072
    },
    {
      "step": 22750,
      "train_flow_matching_loss": 0.1572265625,
      "train_physics_loss": 0.0
    },
    {
      "step": 22750,
      "train_flow_matching_loss": 0.0263671875,
      "train_physics_loss": 0.262422651052475
    },
    {
      "grad_norm": 0.34268081188201904,
      "learning_rate": 5e-05,
      "loss": 0.1235,
      "step": 22760
    },
    {
      "step": 22760,
      "train_flow_matching_loss": 0.224609375,
      "train_physics_loss": 0.0
    },
    {
      "step": 22760,
      "train_flow_matching_loss": 0.0732421875,
      "train_physics_loss": 0.2900504171848297
    },
    {
      "step": 22760,
      "train_flow_matching_loss": 0.1767578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 22760,
      "train_flow_matching_loss": 0.042724609375,
      "train_physics_loss": 0.2721805274486542
    },
    {
      "grad_norm": 0.10480664670467377,
      "learning_rate": 5e-05,
      "loss": 0.1253,
      "step": 22770
    },
    {
      "step": 22770,
      "train_flow_matching_loss": 0.1328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 22770,
      "train_flow_matching_loss": 0.07666015625,
      "train_physics_loss": 0.26367586851119995
    },
    {
      "step": 22770,
      "train_flow_matching_loss": 0.1962890625,
      "train_physics_loss": 0.0
    },
    {
      "step": 22770,
      "train_flow_matching_loss": 0.0751953125,
      "train_physics_loss": 0.25127559900283813
    },
    {
      "grad_norm": 0.3041444420814514,
      "learning_rate": 5e-05,
      "loss": 0.1223,
      "step": 22780
    },
    {
      "step": 22780,
      "train_flow_matching_loss": 0.1455078125,
      "train_physics_loss": 0.0
    },
    {
      "step": 22780,
      "train_flow_matching_loss": 0.0849609375,
      "train_physics_loss": 0.23263348639011383
    },
    {
      "step": 22780,
      "train_flow_matching_loss": 0.15625,
      "train_physics_loss": 0.0
    },
    {
      "step": 22780,
      "train_flow_matching_loss": 0.03662109375,
      "train_physics_loss": 0.23659683763980865
    },
    {
      "grad_norm": 0.10588599741458893,
      "learning_rate": 5e-05,
      "loss": 0.1273,
      "step": 22790
    },
    {
      "step": 22790,
      "train_flow_matching_loss": 0.1240234375,
      "train_physics_loss": 0.0
    },
    {
      "step": 22790,
      "train_flow_matching_loss": 0.07958984375,
      "train_physics_loss": 0.22834163904190063
    },
    {
      "step": 22790,
      "train_flow_matching_loss": 0.11865234375,
      "train_physics_loss": 0.0
    },
    {
      "step": 22790,
      "train_flow_matching_loss": 0.0302734375,
      "train_physics_loss": 0.2530610263347626
    },
    {
      "grad_norm": 0.12954537570476532,
      "learning_rate": 5e-05,
      "loss": 0.1265,
      "step": 22800
    },
    {
      "step": 22800,
      "train_flow_matching_loss": 0.1328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 22800,
      "train_flow_matching_loss": 0.07373046875,
      "train_physics_loss": 0.25812602043151855
    },
    {
      "step": 22800,
      "train_flow_matching_loss": 0.09423828125,
      "train_physics_loss": 0.0
    },
    {
      "step": 22800,
      "train_flow_matching_loss": 0.064453125,
      "train_physics_loss": 0.21209457516670227
    },
    {
      "grad_norm": 0.07213099300861359,
      "learning_rate": 5e-05,
      "loss": 0.1269,
      "step": 22810
    },
    {
      "step": 22810,
      "train_flow_matching_loss": 0.10009765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 22810,
      "train_flow_matching_loss": 0.052490234375,
      "train_physics_loss": 0.22182227671146393
    },
    {
      "step": 22810,
      "train_flow_matching_loss": 0.1806640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 22810,
      "train_flow_matching_loss": 0.053955078125,
      "train_physics_loss": 0.31464123725891113
    },
    {
      "grad_norm": 0.09361850470304489,
      "learning_rate": 5e-05,
      "loss": 0.1244,
      "step": 22820
    },
    {
      "step": 22820,
      "train_flow_matching_loss": 0.09228515625,
      "train_physics_loss": 0.0
    },
    {
      "step": 22820,
      "train_flow_matching_loss": 0.042724609375,
      "train_physics_loss": 0.22696012258529663
    },
    {
      "step": 22820,
      "train_flow_matching_loss": 0.125,
      "train_physics_loss": 0.0
    },
    {
      "step": 22820,
      "train_flow_matching_loss": 0.08203125,
      "train_physics_loss": 0.2929286062717438
    },
    {
      "grad_norm": 0.1377972811460495,
      "learning_rate": 5e-05,
      "loss": 0.1166,
      "step": 22830
    },
    {
      "step": 22830,
      "train_flow_matching_loss": 0.1533203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 22830,
      "train_flow_matching_loss": 0.07080078125,
      "train_physics_loss": 0.19823305308818817
    },
    {
      "step": 22830,
      "train_flow_matching_loss": 0.154296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 22830,
      "train_flow_matching_loss": 0.044189453125,
      "train_physics_loss": 0.2357817143201828
    },
    {
      "grad_norm": 0.0765671655535698,
      "learning_rate": 5e-05,
      "loss": 0.1235,
      "step": 22840
    },
    {
      "step": 22840,
      "train_flow_matching_loss": 0.1123046875,
      "train_physics_loss": 0.0
    },
    {
      "step": 22840,
      "train_flow_matching_loss": 0.047119140625,
      "train_physics_loss": 0.278497576713562
    },
    {
      "step": 22840,
      "train_flow_matching_loss": 0.11328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 22840,
      "train_flow_matching_loss": 0.05810546875,
      "train_physics_loss": 0.32047587633132935
    },
    {
      "grad_norm": 0.08294522762298584,
      "learning_rate": 5e-05,
      "loss": 0.1231,
      "step": 22850
    },
    {
      "step": 22850,
      "train_flow_matching_loss": 0.138671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 22850,
      "train_flow_matching_loss": 0.03466796875,
      "train_physics_loss": 0.2454390972852707
    },
    {
      "step": 22850,
      "train_flow_matching_loss": 0.08349609375,
      "train_physics_loss": 0.0
    },
    {
      "step": 22850,
      "train_flow_matching_loss": 0.03564453125,
      "train_physics_loss": 0.2258628010749817
    },
    {
      "grad_norm": 0.10602440685033798,
      "learning_rate": 5e-05,
      "loss": 0.1258,
      "step": 22860
    },
    {
      "step": 22860,
      "train_flow_matching_loss": 0.12060546875,
      "train_physics_loss": 0.0
    },
    {
      "step": 22860,
      "train_flow_matching_loss": 0.09423828125,
      "train_physics_loss": 0.0
    },
    {
      "step": 22860,
      "train_flow_matching_loss": 0.09814453125,
      "train_physics_loss": 0.0
    },
    {
      "step": 22860,
      "train_flow_matching_loss": 0.056884765625,
      "train_physics_loss": 0.2532893419265747
    },
    {
      "grad_norm": 0.08734209835529327,
      "learning_rate": 5e-05,
      "loss": 0.1187,
      "step": 22870
    },
    {
      "step": 22870,
      "train_flow_matching_loss": 0.1630859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 22870,
      "train_flow_matching_loss": 0.1220703125,
      "train_physics_loss": 0.0
    },
    {
      "step": 22870,
      "train_flow_matching_loss": 0.1630859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 22870,
      "train_flow_matching_loss": 0.07177734375,
      "train_physics_loss": 0.2152515947818756
    },
    {
      "grad_norm": 0.09440590441226959,
      "learning_rate": 5e-05,
      "loss": 0.1246,
      "step": 22880
    },
    {
      "step": 22880,
      "train_flow_matching_loss": 0.12451171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 22880,
      "train_flow_matching_loss": 0.15625,
      "train_physics_loss": 0.0
    },
    {
      "step": 22880,
      "train_flow_matching_loss": 0.1201171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 22880,
      "train_flow_matching_loss": 0.044677734375,
      "train_physics_loss": 0.2771010100841522
    },
    {
      "grad_norm": 0.0993451401591301,
      "learning_rate": 5e-05,
      "loss": 0.1253,
      "step": 22890
    },
    {
      "step": 22890,
      "train_flow_matching_loss": 0.154296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 22890,
      "train_flow_matching_loss": 0.12255859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 22890,
      "train_flow_matching_loss": 0.1494140625,
      "train_physics_loss": 0.0
    },
    {
      "step": 22890,
      "train_flow_matching_loss": 0.08740234375,
      "train_physics_loss": 0.41819635033607483
    },
    {
      "grad_norm": 0.12835967540740967,
      "learning_rate": 5e-05,
      "loss": 0.121,
      "step": 22900
    },
    {
      "step": 22900,
      "train_flow_matching_loss": 0.1376953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 22900,
      "train_flow_matching_loss": 0.1484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 22900,
      "train_flow_matching_loss": 0.12060546875,
      "train_physics_loss": 0.0
    },
    {
      "step": 22900,
      "train_flow_matching_loss": 0.03271484375,
      "train_physics_loss": 0.2542698085308075
    },
    {
      "grad_norm": 0.08442186564207077,
      "learning_rate": 5e-05,
      "loss": 0.1237,
      "step": 22910
    },
    {
      "step": 22910,
      "train_flow_matching_loss": 0.10546875,
      "train_physics_loss": 0.0
    },
    {
      "step": 22910,
      "train_flow_matching_loss": 0.1708984375,
      "train_physics_loss": 0.0
    },
    {
      "step": 22910,
      "train_flow_matching_loss": 0.1552734375,
      "train_physics_loss": 0.0
    },
    {
      "step": 22910,
      "train_flow_matching_loss": 0.046630859375,
      "train_physics_loss": 0.23842573165893555
    },
    {
      "grad_norm": 0.09981361031532288,
      "learning_rate": 5e-05,
      "loss": 0.126,
      "step": 22920
    },
    {
      "step": 22920,
      "train_flow_matching_loss": 0.162109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 22920,
      "train_flow_matching_loss": 0.158203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 22920,
      "train_flow_matching_loss": 0.1083984375,
      "train_physics_loss": 0.0
    },
    {
      "step": 22920,
      "train_flow_matching_loss": 0.061767578125,
      "train_physics_loss": 0.27533963322639465
    },
    {
      "grad_norm": 0.103988878428936,
      "learning_rate": 5e-05,
      "loss": 0.1217,
      "step": 22930
    },
    {
      "step": 22930,
      "train_flow_matching_loss": 0.1376953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 22930,
      "train_flow_matching_loss": 0.1669921875,
      "train_physics_loss": 0.0
    },
    {
      "step": 22930,
      "train_flow_matching_loss": 0.181640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 22930,
      "train_flow_matching_loss": 0.07861328125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.1707683503627777,
      "learning_rate": 5e-05,
      "loss": 0.1175,
      "step": 22940
    },
    {
      "step": 22940,
      "train_flow_matching_loss": 0.10009765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 22940,
      "train_flow_matching_loss": 0.1455078125,
      "train_physics_loss": 0.0
    },
    {
      "step": 22940,
      "train_flow_matching_loss": 0.10498046875,
      "train_physics_loss": 0.0
    },
    {
      "step": 22940,
      "train_flow_matching_loss": 0.0830078125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.15734148025512695,
      "learning_rate": 5e-05,
      "loss": 0.126,
      "step": 22950
    },
    {
      "step": 22950,
      "train_flow_matching_loss": 0.07958984375,
      "train_physics_loss": 0.0
    },
    {
      "step": 22950,
      "train_flow_matching_loss": 0.08837890625,
      "train_physics_loss": 0.0
    },
    {
      "step": 22950,
      "train_flow_matching_loss": 0.11181640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 22950,
      "train_flow_matching_loss": 0.1220703125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09025968611240387,
      "learning_rate": 5e-05,
      "loss": 0.1214,
      "step": 22960
    },
    {
      "step": 22960,
      "train_flow_matching_loss": 0.10107421875,
      "train_physics_loss": 0.0
    },
    {
      "step": 22960,
      "train_flow_matching_loss": 0.1533203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 22960,
      "train_flow_matching_loss": 0.0986328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 22960,
      "train_flow_matching_loss": 0.1484375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11283539980649948,
      "learning_rate": 5e-05,
      "loss": 0.1168,
      "step": 22970
    },
    {
      "step": 22970,
      "train_flow_matching_loss": 0.11376953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 22970,
      "train_flow_matching_loss": 0.09423828125,
      "train_physics_loss": 0.0
    },
    {
      "step": 22970,
      "train_flow_matching_loss": 0.1357421875,
      "train_physics_loss": 0.0
    },
    {
      "step": 22970,
      "train_flow_matching_loss": 0.1474609375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09003663808107376,
      "learning_rate": 5e-05,
      "loss": 0.1263,
      "step": 22980
    },
    {
      "step": 22980,
      "train_flow_matching_loss": 0.1806640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 22980,
      "train_flow_matching_loss": 0.138671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 22980,
      "train_flow_matching_loss": 0.10498046875,
      "train_physics_loss": 0.0
    },
    {
      "step": 22980,
      "train_flow_matching_loss": 0.1416015625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10493927448987961,
      "learning_rate": 5e-05,
      "loss": 0.1193,
      "step": 22990
    },
    {
      "step": 22990,
      "train_flow_matching_loss": 0.1494140625,
      "train_physics_loss": 0.0
    },
    {
      "step": 22990,
      "train_flow_matching_loss": 0.1328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 22990,
      "train_flow_matching_loss": 0.1201171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 22990,
      "train_flow_matching_loss": 0.11865234375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09971526265144348,
      "learning_rate": 5e-05,
      "loss": 0.1212,
      "step": 23000
    },
    {
      "step": 23000,
      "train_flow_matching_loss": 0.1455078125,
      "train_physics_loss": 0.0
    },
    {
      "step": 23000,
      "train_flow_matching_loss": 0.08642578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 23000,
      "train_flow_matching_loss": 0.1181640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 23000,
      "train_flow_matching_loss": 0.08642578125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11392684280872345,
      "learning_rate": 5e-05,
      "loss": 0.1238,
      "step": 23010
    },
    {
      "step": 23010,
      "train_flow_matching_loss": 0.1123046875,
      "train_physics_loss": 0.0
    },
    {
      "step": 23010,
      "train_flow_matching_loss": 0.10498046875,
      "train_physics_loss": 0.0
    },
    {
      "step": 23010,
      "train_flow_matching_loss": 0.1455078125,
      "train_physics_loss": 0.0
    },
    {
      "step": 23010,
      "train_flow_matching_loss": 0.1142578125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09319261461496353,
      "learning_rate": 5e-05,
      "loss": 0.1222,
      "step": 23020
    },
    {
      "step": 23020,
      "train_flow_matching_loss": 0.1884765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 23020,
      "train_flow_matching_loss": 0.1982421875,
      "train_physics_loss": 0.0
    },
    {
      "step": 23020,
      "train_flow_matching_loss": 0.134765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 23020,
      "train_flow_matching_loss": 0.1376953125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.14411203563213348,
      "learning_rate": 5e-05,
      "loss": 0.1223,
      "step": 23030
    },
    {
      "step": 23030,
      "train_flow_matching_loss": 0.08349609375,
      "train_physics_loss": 0.0
    },
    {
      "step": 23030,
      "train_flow_matching_loss": 0.140625,
      "train_physics_loss": 0.0
    },
    {
      "step": 23030,
      "train_flow_matching_loss": 0.1064453125,
      "train_physics_loss": 0.0
    },
    {
      "step": 23030,
      "train_flow_matching_loss": 0.146484375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08298366516828537,
      "learning_rate": 5e-05,
      "loss": 0.1268,
      "step": 23040
    },
    {
      "step": 23040,
      "train_flow_matching_loss": 0.11279296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 23040,
      "train_flow_matching_loss": 0.0625,
      "train_physics_loss": 0.0
    },
    {
      "step": 23040,
      "train_flow_matching_loss": 0.1689453125,
      "train_physics_loss": 0.0
    },
    {
      "step": 23040,
      "train_flow_matching_loss": 0.1142578125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10104303061962128,
      "learning_rate": 5e-05,
      "loss": 0.1312,
      "step": 23050
    },
    {
      "step": 23050,
      "train_flow_matching_loss": 0.173828125,
      "train_physics_loss": 0.0
    },
    {
      "step": 23050,
      "train_flow_matching_loss": 0.09033203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 23050,
      "train_flow_matching_loss": 0.1474609375,
      "train_physics_loss": 0.0
    },
    {
      "step": 23050,
      "train_flow_matching_loss": 0.11767578125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.13764765858650208,
      "learning_rate": 5e-05,
      "loss": 0.1198,
      "step": 23060
    },
    {
      "step": 23060,
      "train_flow_matching_loss": 0.1533203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 23060,
      "train_flow_matching_loss": 0.0654296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 23060,
      "train_flow_matching_loss": 0.1357421875,
      "train_physics_loss": 0.0
    },
    {
      "step": 23060,
      "train_flow_matching_loss": 0.169921875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.18122872710227966,
      "learning_rate": 5e-05,
      "loss": 0.1264,
      "step": 23070
    },
    {
      "step": 23070,
      "train_flow_matching_loss": 0.1416015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 23070,
      "train_flow_matching_loss": 0.06884765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 23070,
      "train_flow_matching_loss": 0.11376953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 23070,
      "train_flow_matching_loss": 0.09033203125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11730631440877914,
      "learning_rate": 5e-05,
      "loss": 0.1286,
      "step": 23080
    },
    {
      "step": 23080,
      "train_flow_matching_loss": 0.201171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 23080,
      "train_flow_matching_loss": 0.08984375,
      "train_physics_loss": 0.0
    },
    {
      "step": 23080,
      "train_flow_matching_loss": 0.12158203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 23080,
      "train_flow_matching_loss": 0.12158203125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11578978598117828,
      "learning_rate": 5e-05,
      "loss": 0.1267,
      "step": 23090
    },
    {
      "step": 23090,
      "train_flow_matching_loss": 0.125,
      "train_physics_loss": 0.0
    },
    {
      "step": 23090,
      "train_flow_matching_loss": 0.10009765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 23090,
      "train_flow_matching_loss": 0.07177734375,
      "train_physics_loss": 0.0
    },
    {
      "step": 23090,
      "train_flow_matching_loss": 0.212890625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08208445459604263,
      "learning_rate": 5e-05,
      "loss": 0.1318,
      "step": 23100
    },
    {
      "step": 23100,
      "train_flow_matching_loss": 0.11083984375,
      "train_physics_loss": 0.0
    },
    {
      "step": 23100,
      "train_flow_matching_loss": 0.11962890625,
      "train_physics_loss": 0.0
    },
    {
      "step": 23100,
      "train_flow_matching_loss": 0.10791015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 23100,
      "train_flow_matching_loss": 0.2392578125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09375957399606705,
      "learning_rate": 5e-05,
      "loss": 0.1236,
      "step": 23110
    },
    {
      "step": 23110,
      "train_flow_matching_loss": 0.08740234375,
      "train_physics_loss": 0.0
    },
    {
      "step": 23110,
      "train_flow_matching_loss": 0.185546875,
      "train_physics_loss": 0.0
    },
    {
      "step": 23110,
      "train_flow_matching_loss": 0.1474609375,
      "train_physics_loss": 0.0
    },
    {
      "step": 23110,
      "train_flow_matching_loss": 0.142578125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.06167745962738991,
      "learning_rate": 5e-05,
      "loss": 0.1287,
      "step": 23120
    },
    {
      "step": 23120,
      "train_flow_matching_loss": 0.126953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 23120,
      "train_flow_matching_loss": 0.185546875,
      "train_physics_loss": 0.0
    },
    {
      "step": 23120,
      "train_flow_matching_loss": 0.078125,
      "train_physics_loss": 0.0
    },
    {
      "step": 23120,
      "train_flow_matching_loss": 0.1650390625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.07982843369245529,
      "learning_rate": 5e-05,
      "loss": 0.1245,
      "step": 23130
    },
    {
      "step": 23130,
      "train_flow_matching_loss": 0.1416015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 23130,
      "train_flow_matching_loss": 0.10791015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 23130,
      "train_flow_matching_loss": 0.11181640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 23130,
      "train_flow_matching_loss": 0.1591796875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.07416084408760071,
      "learning_rate": 5e-05,
      "loss": 0.1221,
      "step": 23140
    },
    {
      "step": 23140,
      "train_flow_matching_loss": 0.1240234375,
      "train_physics_loss": 0.0
    },
    {
      "step": 23140,
      "train_flow_matching_loss": 0.12451171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 23140,
      "train_flow_matching_loss": 0.09814453125,
      "train_physics_loss": 0.0
    },
    {
      "step": 23140,
      "train_flow_matching_loss": 0.203125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11931304633617401,
      "learning_rate": 5e-05,
      "loss": 0.1278,
      "step": 23150
    },
    {
      "step": 23150,
      "train_flow_matching_loss": 0.154296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 23150,
      "train_flow_matching_loss": 0.07763671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 23150,
      "train_flow_matching_loss": 0.09130859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 23150,
      "train_flow_matching_loss": 0.201171875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.13250496983528137,
      "learning_rate": 5e-05,
      "loss": 0.1288,
      "step": 23160
    },
    {
      "step": 23160,
      "train_flow_matching_loss": 0.1201171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 23160,
      "train_flow_matching_loss": 0.12060546875,
      "train_physics_loss": 0.0
    },
    {
      "step": 23160,
      "train_flow_matching_loss": 0.25,
      "train_physics_loss": 0.0
    },
    {
      "step": 23160,
      "train_flow_matching_loss": 0.1220703125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08774170279502869,
      "learning_rate": 5e-05,
      "loss": 0.1228,
      "step": 23170
    },
    {
      "step": 23170,
      "train_flow_matching_loss": 0.2177734375,
      "train_physics_loss": 0.0
    },
    {
      "step": 23170,
      "train_flow_matching_loss": 0.142578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 23170,
      "train_flow_matching_loss": 0.224609375,
      "train_physics_loss": 0.0
    },
    {
      "step": 23170,
      "train_flow_matching_loss": 0.185546875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10848604887723923,
      "learning_rate": 5e-05,
      "loss": 0.1268,
      "step": 23180
    },
    {
      "step": 23180,
      "train_flow_matching_loss": 0.1103515625,
      "train_physics_loss": 0.0
    },
    {
      "step": 23180,
      "train_flow_matching_loss": 0.10595703125,
      "train_physics_loss": 0.0
    },
    {
      "step": 23180,
      "train_flow_matching_loss": 0.11669921875,
      "train_physics_loss": 0.0
    },
    {
      "step": 23180,
      "train_flow_matching_loss": 0.208984375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08779916912317276,
      "learning_rate": 5e-05,
      "loss": 0.1199,
      "step": 23190
    },
    {
      "step": 23190,
      "train_flow_matching_loss": 0.173828125,
      "train_physics_loss": 0.0
    },
    {
      "step": 23190,
      "train_flow_matching_loss": 0.2080078125,
      "train_physics_loss": 0.0
    },
    {
      "step": 23190,
      "train_flow_matching_loss": 0.150390625,
      "train_physics_loss": 0.0
    },
    {
      "step": 23190,
      "train_flow_matching_loss": 0.11376953125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.12928852438926697,
      "learning_rate": 5e-05,
      "loss": 0.1254,
      "step": 23200
    },
    {
      "step": 23200,
      "train_flow_matching_loss": 0.1318359375,
      "train_physics_loss": 0.0
    },
    {
      "step": 23200,
      "train_flow_matching_loss": 0.1748046875,
      "train_physics_loss": 0.0
    },
    {
      "step": 23200,
      "train_flow_matching_loss": 0.0849609375,
      "train_physics_loss": 0.0
    },
    {
      "step": 23200,
      "train_flow_matching_loss": 0.197265625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.1147836372256279,
      "learning_rate": 5e-05,
      "loss": 0.1312,
      "step": 23210
    },
    {
      "step": 23210,
      "train_flow_matching_loss": 0.11962890625,
      "train_physics_loss": 0.0
    },
    {
      "step": 23210,
      "train_flow_matching_loss": 0.21875,
      "train_physics_loss": 0.0
    },
    {
      "step": 23210,
      "train_flow_matching_loss": 0.181640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 23210,
      "train_flow_matching_loss": 0.11279296875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.1024380698800087,
      "learning_rate": 5e-05,
      "loss": 0.1256,
      "step": 23220
    },
    {
      "step": 23220,
      "train_flow_matching_loss": 0.1435546875,
      "train_physics_loss": 0.0
    },
    {
      "step": 23220,
      "train_flow_matching_loss": 0.2080078125,
      "train_physics_loss": 0.0
    },
    {
      "step": 23220,
      "train_flow_matching_loss": 0.09912109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 23220,
      "train_flow_matching_loss": 0.0947265625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09465523809194565,
      "learning_rate": 5e-05,
      "loss": 0.1327,
      "step": 23230
    },
    {
      "step": 23230,
      "train_flow_matching_loss": 0.146484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 23230,
      "train_flow_matching_loss": 0.1533203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 23230,
      "train_flow_matching_loss": 0.2255859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 23230,
      "train_flow_matching_loss": 0.1259765625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.07062658667564392,
      "learning_rate": 5e-05,
      "loss": 0.1265,
      "step": 23240
    },
    {
      "step": 23240,
      "train_flow_matching_loss": 0.1416015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 23240,
      "train_flow_matching_loss": 0.0830078125,
      "train_physics_loss": 0.0
    },
    {
      "step": 23240,
      "train_flow_matching_loss": 0.1611328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 23240,
      "train_flow_matching_loss": 0.25390625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.0702725276350975,
      "learning_rate": 5e-05,
      "loss": 0.1322,
      "step": 23250
    },
    {
      "step": 23250,
      "train_flow_matching_loss": 0.11962890625,
      "train_physics_loss": 0.0
    },
    {
      "step": 23250,
      "train_flow_matching_loss": 0.1201171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 23250,
      "train_flow_matching_loss": 0.1650390625,
      "train_physics_loss": 0.0
    },
    {
      "step": 23250,
      "train_flow_matching_loss": 0.1259765625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09378783404827118,
      "learning_rate": 5e-05,
      "loss": 0.1295,
      "step": 23260
    },
    {
      "step": 23260,
      "train_flow_matching_loss": 0.08203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 23260,
      "train_flow_matching_loss": 0.1533203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 23260,
      "train_flow_matching_loss": 0.1396484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 23260,
      "train_flow_matching_loss": 0.07763671875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09010639041662216,
      "learning_rate": 5e-05,
      "loss": 0.1217,
      "step": 23270
    },
    {
      "step": 23270,
      "train_flow_matching_loss": 0.11474609375,
      "train_physics_loss": 0.0
    },
    {
      "step": 23270,
      "train_flow_matching_loss": 0.208984375,
      "train_physics_loss": 0.0
    },
    {
      "step": 23270,
      "train_flow_matching_loss": 0.09130859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 23270,
      "train_flow_matching_loss": 0.09521484375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10653682798147202,
      "learning_rate": 5e-05,
      "loss": 0.1263,
      "step": 23280
    },
    {
      "step": 23280,
      "train_flow_matching_loss": 0.15625,
      "train_physics_loss": 0.0
    },
    {
      "step": 23280,
      "train_flow_matching_loss": 0.314453125,
      "train_physics_loss": 0.0
    },
    {
      "step": 23280,
      "train_flow_matching_loss": 0.1142578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 23280,
      "train_flow_matching_loss": 0.11865234375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10487554967403412,
      "learning_rate": 5e-05,
      "loss": 0.1279,
      "step": 23290
    },
    {
      "step": 23290,
      "train_flow_matching_loss": 0.1611328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 23290,
      "train_flow_matching_loss": 0.248046875,
      "train_physics_loss": 0.0
    },
    {
      "step": 23290,
      "train_flow_matching_loss": 0.07275390625,
      "train_physics_loss": 0.0
    },
    {
      "step": 23290,
      "train_flow_matching_loss": 0.12353515625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.15104585886001587,
      "learning_rate": 5e-05,
      "loss": 0.1251,
      "step": 23300
    },
    {
      "step": 23300,
      "train_flow_matching_loss": 0.1298828125,
      "train_physics_loss": 0.0
    },
    {
      "step": 23300,
      "train_flow_matching_loss": 0.134765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 23300,
      "train_flow_matching_loss": 0.1015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 23300,
      "train_flow_matching_loss": 0.12255859375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09839212894439697,
      "learning_rate": 5e-05,
      "loss": 0.1243,
      "step": 23310
    },
    {
      "step": 23310,
      "train_flow_matching_loss": 0.06884765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 23310,
      "train_flow_matching_loss": 0.1357421875,
      "train_physics_loss": 0.0
    },
    {
      "step": 23310,
      "train_flow_matching_loss": 0.1748046875,
      "train_physics_loss": 0.0
    },
    {
      "step": 23310,
      "train_flow_matching_loss": 0.1494140625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09048082679510117,
      "learning_rate": 5e-05,
      "loss": 0.1276,
      "step": 23320
    },
    {
      "step": 23320,
      "train_flow_matching_loss": 0.138671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 23320,
      "train_flow_matching_loss": 0.263671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 23320,
      "train_flow_matching_loss": 0.1708984375,
      "train_physics_loss": 0.0
    },
    {
      "step": 23320,
      "train_flow_matching_loss": 0.09716796875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09110823273658752,
      "learning_rate": 5e-05,
      "loss": 0.1282,
      "step": 23330
    },
    {
      "step": 23330,
      "train_flow_matching_loss": 0.140625,
      "train_physics_loss": 0.0
    },
    {
      "step": 23330,
      "train_flow_matching_loss": 0.216796875,
      "train_physics_loss": 0.0
    },
    {
      "step": 23330,
      "train_flow_matching_loss": 0.1279296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 23330,
      "train_flow_matching_loss": 0.166015625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09939951449632645,
      "learning_rate": 5e-05,
      "loss": 0.1262,
      "step": 23340
    },
    {
      "step": 23340,
      "train_flow_matching_loss": 0.1650390625,
      "train_physics_loss": 0.0
    },
    {
      "step": 23340,
      "train_flow_matching_loss": 0.1728515625,
      "train_physics_loss": 0.0
    },
    {
      "step": 23340,
      "train_flow_matching_loss": 0.07373046875,
      "train_physics_loss": 0.0
    },
    {
      "step": 23340,
      "train_flow_matching_loss": 0.150390625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08070426434278488,
      "learning_rate": 5e-05,
      "loss": 0.1299,
      "step": 23350
    },
    {
      "step": 23350,
      "train_flow_matching_loss": 0.1240234375,
      "train_physics_loss": 0.0
    },
    {
      "step": 23350,
      "train_flow_matching_loss": 0.11279296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 23350,
      "train_flow_matching_loss": 0.1767578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 23350,
      "train_flow_matching_loss": 0.126953125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08713074773550034,
      "learning_rate": 5e-05,
      "loss": 0.1181,
      "step": 23360
    },
    {
      "step": 23360,
      "train_flow_matching_loss": 0.1669921875,
      "train_physics_loss": 0.0
    },
    {
      "step": 23360,
      "train_flow_matching_loss": 0.1259765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 23360,
      "train_flow_matching_loss": 0.126953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 23360,
      "train_flow_matching_loss": 0.10009765625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09238553047180176,
      "learning_rate": 5e-05,
      "loss": 0.1248,
      "step": 23370
    },
    {
      "step": 23370,
      "train_flow_matching_loss": 0.185546875,
      "train_physics_loss": 0.0
    },
    {
      "step": 23370,
      "train_flow_matching_loss": 0.07861328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 23370,
      "train_flow_matching_loss": 0.11181640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 23370,
      "train_flow_matching_loss": 0.10693359375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10854681581258774,
      "learning_rate": 5e-05,
      "loss": 0.1283,
      "step": 23380
    },
    {
      "step": 23380,
      "train_flow_matching_loss": 0.1826171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 23380,
      "train_flow_matching_loss": 0.078125,
      "train_physics_loss": 0.0
    },
    {
      "step": 23380,
      "train_flow_matching_loss": 0.10986328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 23380,
      "train_flow_matching_loss": 0.1259765625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10973220318555832,
      "learning_rate": 5e-05,
      "loss": 0.1259,
      "step": 23390
    },
    {
      "step": 23390,
      "train_flow_matching_loss": 0.1787109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 23390,
      "train_flow_matching_loss": 0.12109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 23390,
      "train_flow_matching_loss": 0.2109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 23390,
      "train_flow_matching_loss": 0.10498046875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.0658271461725235,
      "learning_rate": 5e-05,
      "loss": 0.1246,
      "step": 23400
    },
    {
      "step": 23400,
      "train_flow_matching_loss": 0.181640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 23400,
      "train_flow_matching_loss": 0.13671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 23400,
      "train_flow_matching_loss": 0.1396484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 23400,
      "train_flow_matching_loss": 0.10107421875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.07102502882480621,
      "learning_rate": 5e-05,
      "loss": 0.1262,
      "step": 23410
    },
    {
      "step": 23410,
      "train_flow_matching_loss": 0.173828125,
      "train_physics_loss": 0.0
    },
    {
      "step": 23410,
      "train_flow_matching_loss": 0.08349609375,
      "train_physics_loss": 0.0
    },
    {
      "step": 23410,
      "train_flow_matching_loss": 0.1435546875,
      "train_physics_loss": 0.0
    },
    {
      "step": 23410,
      "train_flow_matching_loss": 0.10888671875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.138605535030365,
      "learning_rate": 5e-05,
      "loss": 0.125,
      "step": 23420
    },
    {
      "step": 23420,
      "train_flow_matching_loss": 0.1884765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 23420,
      "train_flow_matching_loss": 0.109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 23420,
      "train_flow_matching_loss": 0.11669921875,
      "train_physics_loss": 0.0
    },
    {
      "step": 23420,
      "train_flow_matching_loss": 0.1220703125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08225906640291214,
      "learning_rate": 5e-05,
      "loss": 0.1291,
      "step": 23430
    },
    {
      "step": 23430,
      "train_flow_matching_loss": 0.1806640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 23430,
      "train_flow_matching_loss": 0.1005859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 23430,
      "train_flow_matching_loss": 0.169921875,
      "train_physics_loss": 0.0
    },
    {
      "step": 23430,
      "train_flow_matching_loss": 0.0927734375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.081422358751297,
      "learning_rate": 5e-05,
      "loss": 0.1232,
      "step": 23440
    },
    {
      "step": 23440,
      "train_flow_matching_loss": 0.1337890625,
      "train_physics_loss": 0.0
    },
    {
      "step": 23440,
      "train_flow_matching_loss": 0.15625,
      "train_physics_loss": 0.0
    },
    {
      "step": 23440,
      "train_flow_matching_loss": 0.1513671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 23440,
      "train_flow_matching_loss": 0.04638671875,
      "train_physics_loss": 0.2941991984844208
    },
    {
      "grad_norm": 0.0988847017288208,
      "learning_rate": 5e-05,
      "loss": 0.1242,
      "step": 23450
    },
    {
      "step": 23450,
      "train_flow_matching_loss": 0.1201171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 23450,
      "train_flow_matching_loss": 0.0986328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 23450,
      "train_flow_matching_loss": 0.10595703125,
      "train_physics_loss": 0.0
    },
    {
      "step": 23450,
      "train_flow_matching_loss": 0.07275390625,
      "train_physics_loss": 0.20911067724227905
    },
    {
      "grad_norm": 0.07382727414369583,
      "learning_rate": 5e-05,
      "loss": 0.1279,
      "step": 23460
    },
    {
      "step": 23460,
      "train_flow_matching_loss": 0.1728515625,
      "train_physics_loss": 0.0
    },
    {
      "step": 23460,
      "train_flow_matching_loss": 0.173828125,
      "train_physics_loss": 0.0
    },
    {
      "step": 23460,
      "train_flow_matching_loss": 0.10205078125,
      "train_physics_loss": 0.0
    },
    {
      "step": 23460,
      "train_flow_matching_loss": 0.04931640625,
      "train_physics_loss": 0.20311805605888367
    },
    {
      "grad_norm": 0.07811103761196136,
      "learning_rate": 5e-05,
      "loss": 0.126,
      "step": 23470
    },
    {
      "step": 23470,
      "train_flow_matching_loss": 0.177734375,
      "train_physics_loss": 0.0
    },
    {
      "step": 23470,
      "train_flow_matching_loss": 0.138671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 23470,
      "train_flow_matching_loss": 0.140625,
      "train_physics_loss": 0.0
    },
    {
      "step": 23470,
      "train_flow_matching_loss": 0.06298828125,
      "train_physics_loss": 0.2647345960140228
    },
    {
      "grad_norm": 0.11303863674402237,
      "learning_rate": 5e-05,
      "loss": 0.1288,
      "step": 23480
    },
    {
      "step": 23480,
      "train_flow_matching_loss": 0.1796875,
      "train_physics_loss": 0.0
    },
    {
      "step": 23480,
      "train_flow_matching_loss": 0.109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 23480,
      "train_flow_matching_loss": 0.1357421875,
      "train_physics_loss": 0.0
    },
    {
      "step": 23480,
      "train_flow_matching_loss": 0.03857421875,
      "train_physics_loss": 0.23524554073810577
    },
    {
      "grad_norm": 0.09110225737094879,
      "learning_rate": 5e-05,
      "loss": 0.1274,
      "step": 23490
    },
    {
      "step": 23490,
      "train_flow_matching_loss": 0.2158203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 23490,
      "train_flow_matching_loss": 0.09521484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 23490,
      "train_flow_matching_loss": 0.1162109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 23490,
      "train_flow_matching_loss": 0.04736328125,
      "train_physics_loss": 0.2099776417016983
    },
    {
      "grad_norm": 0.08675437420606613,
      "learning_rate": 5e-05,
      "loss": 0.1279,
      "step": 23500
    },
    {
      "step": 23500,
      "train_flow_matching_loss": 0.2373046875,
      "train_physics_loss": 0.0
    },
    {
      "step": 23500,
      "train_flow_matching_loss": 0.07763671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 23500,
      "train_flow_matching_loss": 0.11865234375,
      "train_physics_loss": 0.0
    },
    {
      "step": 23500,
      "train_flow_matching_loss": 0.029052734375,
      "train_physics_loss": 0.23590531945228577
    },
    {
      "grad_norm": 0.13244982063770294,
      "learning_rate": 5e-05,
      "loss": 0.1248,
      "step": 23510
    },
    {
      "step": 23510,
      "train_flow_matching_loss": 0.1943359375,
      "train_physics_loss": 0.0
    },
    {
      "step": 23510,
      "train_flow_matching_loss": 0.12109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 23510,
      "train_flow_matching_loss": 0.0830078125,
      "train_physics_loss": 0.0
    },
    {
      "step": 23510,
      "train_flow_matching_loss": 0.0162353515625,
      "train_physics_loss": 0.22913901507854462
    },
    {
      "grad_norm": 0.07715100795030594,
      "learning_rate": 5e-05,
      "loss": 0.1279,
      "step": 23520
    },
    {
      "step": 23520,
      "train_flow_matching_loss": 0.201171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 23520,
      "train_flow_matching_loss": 0.1416015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 23520,
      "train_flow_matching_loss": 0.146484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 23520,
      "train_flow_matching_loss": 0.05078125,
      "train_physics_loss": 0.23588700592517853
    },
    {
      "grad_norm": 0.10197543352842331,
      "learning_rate": 5e-05,
      "loss": 0.1223,
      "step": 23530
    },
    {
      "step": 23530,
      "train_flow_matching_loss": 0.1953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 23530,
      "train_flow_matching_loss": 0.1767578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 23530,
      "train_flow_matching_loss": 0.09375,
      "train_physics_loss": 0.0
    },
    {
      "step": 23530,
      "train_flow_matching_loss": 0.0172119140625,
      "train_physics_loss": 0.21705612540245056
    },
    {
      "grad_norm": 0.1319926530122757,
      "learning_rate": 5e-05,
      "loss": 0.1244,
      "step": 23540
    },
    {
      "step": 23540,
      "train_flow_matching_loss": 0.09521484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 23540,
      "train_flow_matching_loss": 0.1337890625,
      "train_physics_loss": 0.0
    },
    {
      "step": 23540,
      "train_flow_matching_loss": 0.109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 23540,
      "train_flow_matching_loss": 0.044189453125,
      "train_physics_loss": 0.22952811419963837
    },
    {
      "grad_norm": 0.08410383760929108,
      "learning_rate": 5e-05,
      "loss": 0.116,
      "step": 23550
    },
    {
      "step": 23550,
      "train_flow_matching_loss": 0.1044921875,
      "train_physics_loss": 0.0
    },
    {
      "step": 23550,
      "train_flow_matching_loss": 0.1201171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 23550,
      "train_flow_matching_loss": 0.1484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 23550,
      "train_flow_matching_loss": 0.10888671875,
      "train_physics_loss": 0.29094138741493225
    },
    {
      "grad_norm": 0.09829725325107574,
      "learning_rate": 5e-05,
      "loss": 0.1232,
      "step": 23560
    },
    {
      "step": 23560,
      "train_flow_matching_loss": 0.11572265625,
      "train_physics_loss": 0.0
    },
    {
      "step": 23560,
      "train_flow_matching_loss": 0.11474609375,
      "train_physics_loss": 0.0
    },
    {
      "step": 23560,
      "train_flow_matching_loss": 0.1357421875,
      "train_physics_loss": 0.0
    },
    {
      "step": 23560,
      "train_flow_matching_loss": 0.06396484375,
      "train_physics_loss": 0.1880580335855484
    },
    {
      "grad_norm": 0.7979317903518677,
      "learning_rate": 5e-05,
      "loss": 0.1249,
      "step": 23570
    },
    {
      "step": 23570,
      "train_flow_matching_loss": 0.154296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 23570,
      "train_flow_matching_loss": 0.08349609375,
      "train_physics_loss": 0.0
    },
    {
      "step": 23570,
      "train_flow_matching_loss": 0.1669921875,
      "train_physics_loss": 0.0
    },
    {
      "step": 23570,
      "train_flow_matching_loss": 0.0751953125,
      "train_physics_loss": 0.24007850885391235
    },
    {
      "grad_norm": 0.08566626161336899,
      "learning_rate": 5e-05,
      "loss": 0.1242,
      "step": 23580
    },
    {
      "step": 23580,
      "train_flow_matching_loss": 0.1962890625,
      "train_physics_loss": 0.0
    },
    {
      "step": 23580,
      "train_flow_matching_loss": 0.11767578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 23580,
      "train_flow_matching_loss": 0.09521484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 23580,
      "train_flow_matching_loss": 0.08154296875,
      "train_physics_loss": 0.24145610630512238
    },
    {
      "grad_norm": 0.10919869691133499,
      "learning_rate": 5e-05,
      "loss": 0.1254,
      "step": 23590
    },
    {
      "step": 23590,
      "train_flow_matching_loss": 0.126953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 23590,
      "train_flow_matching_loss": 0.1435546875,
      "train_physics_loss": 0.0
    },
    {
      "step": 23590,
      "train_flow_matching_loss": 0.15625,
      "train_physics_loss": 0.0
    },
    {
      "step": 23590,
      "train_flow_matching_loss": 0.052978515625,
      "train_physics_loss": 0.33550477027893066
    },
    {
      "grad_norm": 0.11616436392068863,
      "learning_rate": 5e-05,
      "loss": 0.1234,
      "step": 23600
    },
    {
      "step": 23600,
      "train_flow_matching_loss": 0.1494140625,
      "train_physics_loss": 0.0
    },
    {
      "step": 23600,
      "train_flow_matching_loss": 0.09814453125,
      "train_physics_loss": 0.0
    },
    {
      "step": 23600,
      "train_flow_matching_loss": 0.1474609375,
      "train_physics_loss": 0.0
    },
    {
      "step": 23600,
      "train_flow_matching_loss": 0.08154296875,
      "train_physics_loss": 0.2920510470867157
    },
    {
      "grad_norm": 0.0912313312292099,
      "learning_rate": 5e-05,
      "loss": 0.1242,
      "step": 23610
    },
    {
      "step": 23610,
      "train_flow_matching_loss": 0.0869140625,
      "train_physics_loss": 0.0
    },
    {
      "step": 23610,
      "train_flow_matching_loss": 0.11279296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 23610,
      "train_flow_matching_loss": 0.1640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 23610,
      "train_flow_matching_loss": 0.035400390625,
      "train_physics_loss": 0.21965721249580383
    },
    {
      "grad_norm": 0.0895819142460823,
      "learning_rate": 5e-05,
      "loss": 0.1235,
      "step": 23620
    },
    {
      "step": 23620,
      "train_flow_matching_loss": 0.0859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 23620,
      "train_flow_matching_loss": 0.07763671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 23620,
      "train_flow_matching_loss": 0.10205078125,
      "train_physics_loss": 0.0
    },
    {
      "step": 23620,
      "train_flow_matching_loss": 0.03955078125,
      "train_physics_loss": 0.2131677120923996
    },
    {
      "grad_norm": 0.1788177639245987,
      "learning_rate": 5e-05,
      "loss": 0.1238,
      "step": 23630
    },
    {
      "step": 23630,
      "train_flow_matching_loss": 0.140625,
      "train_physics_loss": 0.0
    },
    {
      "step": 23630,
      "train_flow_matching_loss": 0.1533203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 23630,
      "train_flow_matching_loss": 0.1259765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 23630,
      "train_flow_matching_loss": 0.024169921875,
      "train_physics_loss": 0.2110905796289444
    },
    {
      "grad_norm": 0.07335777580738068,
      "learning_rate": 5e-05,
      "loss": 0.1311,
      "step": 23640
    },
    {
      "step": 23640,
      "train_flow_matching_loss": 0.1025390625,
      "train_physics_loss": 0.0
    },
    {
      "step": 23640,
      "train_flow_matching_loss": 0.146484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 23640,
      "train_flow_matching_loss": 0.12451171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 23640,
      "train_flow_matching_loss": 0.04638671875,
      "train_physics_loss": 0.3185596466064453
    },
    {
      "grad_norm": 0.09895633906126022,
      "learning_rate": 5e-05,
      "loss": 0.1225,
      "step": 23650
    },
    {
      "step": 23650,
      "train_flow_matching_loss": 0.1181640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 23650,
      "train_flow_matching_loss": 0.1337890625,
      "train_physics_loss": 0.0
    },
    {
      "step": 23650,
      "train_flow_matching_loss": 0.1533203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 23650,
      "train_flow_matching_loss": 0.058837890625,
      "train_physics_loss": 0.2746032476425171
    },
    {
      "grad_norm": 0.11402463912963867,
      "learning_rate": 5e-05,
      "loss": 0.1202,
      "step": 23660
    },
    {
      "step": 23660,
      "train_flow_matching_loss": 0.212890625,
      "train_physics_loss": 0.0
    },
    {
      "step": 23660,
      "train_flow_matching_loss": 0.1123046875,
      "train_physics_loss": 0.0
    },
    {
      "step": 23660,
      "train_flow_matching_loss": 0.10400390625,
      "train_physics_loss": 0.0
    },
    {
      "step": 23660,
      "train_flow_matching_loss": 0.040771484375,
      "train_physics_loss": 0.2849857211112976
    },
    {
      "grad_norm": 0.10807251185178757,
      "learning_rate": 5e-05,
      "loss": 0.1281,
      "step": 23670
    },
    {
      "step": 23670,
      "train_flow_matching_loss": 0.0751953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 23670,
      "train_flow_matching_loss": 0.0908203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 23670,
      "train_flow_matching_loss": 0.09033203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 23670,
      "train_flow_matching_loss": 0.058349609375,
      "train_physics_loss": 0.2085835039615631
    },
    {
      "grad_norm": 0.10365642607212067,
      "learning_rate": 5e-05,
      "loss": 0.1242,
      "step": 23680
    },
    {
      "step": 23680,
      "train_flow_matching_loss": 0.0673828125,
      "train_physics_loss": 0.0
    },
    {
      "step": 23680,
      "train_flow_matching_loss": 0.154296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 23680,
      "train_flow_matching_loss": 0.09423828125,
      "train_physics_loss": 0.0
    },
    {
      "step": 23680,
      "train_flow_matching_loss": 0.039306640625,
      "train_physics_loss": 0.2719767391681671
    },
    {
      "grad_norm": 0.08756190538406372,
      "learning_rate": 5e-05,
      "loss": 0.1235,
      "step": 23690
    },
    {
      "step": 23690,
      "train_flow_matching_loss": 0.1142578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 23690,
      "train_flow_matching_loss": 0.1962890625,
      "train_physics_loss": 0.0
    },
    {
      "step": 23690,
      "train_flow_matching_loss": 0.10595703125,
      "train_physics_loss": 0.0
    },
    {
      "step": 23690,
      "train_flow_matching_loss": 0.08837890625,
      "train_physics_loss": 0.17027151584625244
    },
    {
      "grad_norm": 0.13529548048973083,
      "learning_rate": 5e-05,
      "loss": 0.1248,
      "step": 23700
    },
    {
      "step": 23700,
      "train_flow_matching_loss": 0.09912109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 23700,
      "train_flow_matching_loss": 0.1259765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 23700,
      "train_flow_matching_loss": 0.1494140625,
      "train_physics_loss": 0.0
    },
    {
      "step": 23700,
      "train_flow_matching_loss": 0.1494140625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08328273892402649,
      "learning_rate": 5e-05,
      "loss": 0.119,
      "step": 23710
    },
    {
      "step": 23710,
      "train_flow_matching_loss": 0.1298828125,
      "train_physics_loss": 0.0
    },
    {
      "step": 23710,
      "train_flow_matching_loss": 0.068359375,
      "train_physics_loss": 0.0
    },
    {
      "step": 23710,
      "train_flow_matching_loss": 0.08837890625,
      "train_physics_loss": 0.0
    },
    {
      "step": 23710,
      "train_flow_matching_loss": 0.1923828125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09099242836236954,
      "learning_rate": 5e-05,
      "loss": 0.1253,
      "step": 23720
    },
    {
      "step": 23720,
      "train_flow_matching_loss": 0.05224609375,
      "train_physics_loss": 0.0
    },
    {
      "step": 23720,
      "train_flow_matching_loss": 0.05615234375,
      "train_physics_loss": 0.0
    },
    {
      "step": 23720,
      "train_flow_matching_loss": 0.21875,
      "train_physics_loss": 0.0
    },
    {
      "step": 23720,
      "train_flow_matching_loss": 0.1416015625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.3207675516605377,
      "learning_rate": 5e-05,
      "loss": 0.1271,
      "step": 23730
    },
    {
      "step": 23730,
      "train_flow_matching_loss": 0.0859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 23730,
      "train_flow_matching_loss": 0.050537109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 23730,
      "train_flow_matching_loss": 0.1611328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 23730,
      "train_flow_matching_loss": 0.216796875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08443208783864975,
      "learning_rate": 5e-05,
      "loss": 0.1192,
      "step": 23740
    },
    {
      "step": 23740,
      "train_flow_matching_loss": 0.12353515625,
      "train_physics_loss": 0.0
    },
    {
      "step": 23740,
      "train_flow_matching_loss": 0.1572265625,
      "train_physics_loss": 0.0
    },
    {
      "step": 23740,
      "train_flow_matching_loss": 0.1728515625,
      "train_physics_loss": 0.0
    },
    {
      "step": 23740,
      "train_flow_matching_loss": 0.1533203125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10030732303857803,
      "learning_rate": 5e-05,
      "loss": 0.1272,
      "step": 23750
    },
    {
      "step": 23750,
      "train_flow_matching_loss": 0.078125,
      "train_physics_loss": 0.0
    },
    {
      "step": 23750,
      "train_flow_matching_loss": 0.134765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 23750,
      "train_flow_matching_loss": 0.12890625,
      "train_physics_loss": 0.0
    },
    {
      "step": 23750,
      "train_flow_matching_loss": 0.1796875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.1490173190832138,
      "learning_rate": 5e-05,
      "loss": 0.1244,
      "step": 23760
    },
    {
      "step": 23760,
      "train_flow_matching_loss": 0.06689453125,
      "train_physics_loss": 0.0
    },
    {
      "step": 23760,
      "train_flow_matching_loss": 0.1181640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 23760,
      "train_flow_matching_loss": 0.1083984375,
      "train_physics_loss": 0.0
    },
    {
      "step": 23760,
      "train_flow_matching_loss": 0.1796875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09125012904405594,
      "learning_rate": 5e-05,
      "loss": 0.1249,
      "step": 23770
    },
    {
      "step": 23770,
      "train_flow_matching_loss": 0.1279296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 23770,
      "train_flow_matching_loss": 0.185546875,
      "train_physics_loss": 0.0
    },
    {
      "step": 23770,
      "train_flow_matching_loss": 0.146484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 23770,
      "train_flow_matching_loss": 0.1337890625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.13015320897102356,
      "learning_rate": 5e-05,
      "loss": 0.1239,
      "step": 23780
    },
    {
      "step": 23780,
      "train_flow_matching_loss": 0.0927734375,
      "train_physics_loss": 0.0
    },
    {
      "step": 23780,
      "train_flow_matching_loss": 0.1640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 23780,
      "train_flow_matching_loss": 0.09814453125,
      "train_physics_loss": 0.0
    },
    {
      "step": 23780,
      "train_flow_matching_loss": 0.1337890625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.24991293251514435,
      "learning_rate": 5e-05,
      "loss": 0.1264,
      "step": 23790
    },
    {
      "step": 23790,
      "train_flow_matching_loss": 0.154296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 23790,
      "train_flow_matching_loss": 0.09765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 23790,
      "train_flow_matching_loss": 0.12060546875,
      "train_physics_loss": 0.0
    },
    {
      "step": 23790,
      "train_flow_matching_loss": 0.150390625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11001400649547577,
      "learning_rate": 5e-05,
      "loss": 0.1222,
      "step": 23800
    },
    {
      "step": 23800,
      "train_flow_matching_loss": 0.09033203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 23800,
      "train_flow_matching_loss": 0.1650390625,
      "train_physics_loss": 0.0
    },
    {
      "step": 23800,
      "train_flow_matching_loss": 0.09521484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 23800,
      "train_flow_matching_loss": 0.1953125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.29208478331565857,
      "learning_rate": 5e-05,
      "loss": 0.1245,
      "step": 23810
    },
    {
      "step": 23810,
      "train_flow_matching_loss": 0.140625,
      "train_physics_loss": 0.0
    },
    {
      "step": 23810,
      "train_flow_matching_loss": 0.08056640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 23810,
      "train_flow_matching_loss": 0.037841796875,
      "train_physics_loss": 0.0
    },
    {
      "step": 23810,
      "train_flow_matching_loss": 0.09033203125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.3600316643714905,
      "learning_rate": 5e-05,
      "loss": 0.1214,
      "step": 23820
    },
    {
      "step": 23820,
      "train_flow_matching_loss": 0.10986328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 23820,
      "train_flow_matching_loss": 0.12109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 23820,
      "train_flow_matching_loss": 0.07275390625,
      "train_physics_loss": 0.0
    },
    {
      "step": 23820,
      "train_flow_matching_loss": 0.171875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.2646816670894623,
      "learning_rate": 5e-05,
      "loss": 0.1305,
      "step": 23830
    },
    {
      "step": 23830,
      "train_flow_matching_loss": 0.2001953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 23830,
      "train_flow_matching_loss": 0.1767578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 23830,
      "train_flow_matching_loss": 0.06787109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 23830,
      "train_flow_matching_loss": 0.134765625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08686256408691406,
      "learning_rate": 5e-05,
      "loss": 0.1206,
      "step": 23840
    },
    {
      "step": 23840,
      "train_flow_matching_loss": 0.1533203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 23840,
      "train_flow_matching_loss": 0.11962890625,
      "train_physics_loss": 0.0
    },
    {
      "step": 23840,
      "train_flow_matching_loss": 0.06591796875,
      "train_physics_loss": 0.0
    },
    {
      "step": 23840,
      "train_flow_matching_loss": 0.1416015625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.0876956358551979,
      "learning_rate": 5e-05,
      "loss": 0.1283,
      "step": 23850
    },
    {
      "step": 23850,
      "train_flow_matching_loss": 0.1572265625,
      "train_physics_loss": 0.0
    },
    {
      "step": 23850,
      "train_flow_matching_loss": 0.1435546875,
      "train_physics_loss": 0.0
    },
    {
      "step": 23850,
      "train_flow_matching_loss": 0.08203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 23850,
      "train_flow_matching_loss": 0.189453125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10739786922931671,
      "learning_rate": 5e-05,
      "loss": 0.1278,
      "step": 23860
    },
    {
      "step": 23860,
      "train_flow_matching_loss": 0.125,
      "train_physics_loss": 0.0
    },
    {
      "step": 23860,
      "train_flow_matching_loss": 0.09228515625,
      "train_physics_loss": 0.0
    },
    {
      "step": 23860,
      "train_flow_matching_loss": 0.197265625,
      "train_physics_loss": 0.0
    },
    {
      "step": 23860,
      "train_flow_matching_loss": 0.1064453125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11335066705942154,
      "learning_rate": 5e-05,
      "loss": 0.1278,
      "step": 23870
    },
    {
      "step": 23870,
      "train_flow_matching_loss": 0.12109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 23870,
      "train_flow_matching_loss": 0.193359375,
      "train_physics_loss": 0.0
    },
    {
      "step": 23870,
      "train_flow_matching_loss": 0.0732421875,
      "train_physics_loss": 0.0
    },
    {
      "step": 23870,
      "train_flow_matching_loss": 0.134765625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.1092081293463707,
      "learning_rate": 5e-05,
      "loss": 0.1319,
      "step": 23880
    },
    {
      "step": 23880,
      "train_flow_matching_loss": 0.1455078125,
      "train_physics_loss": 0.0
    },
    {
      "step": 23880,
      "train_flow_matching_loss": 0.154296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 23880,
      "train_flow_matching_loss": 0.181640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 23880,
      "train_flow_matching_loss": 0.1484375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08692030608654022,
      "learning_rate": 5e-05,
      "loss": 0.124,
      "step": 23890
    },
    {
      "step": 23890,
      "train_flow_matching_loss": 0.1357421875,
      "train_physics_loss": 0.0
    },
    {
      "step": 23890,
      "train_flow_matching_loss": 0.08984375,
      "train_physics_loss": 0.0
    },
    {
      "step": 23890,
      "train_flow_matching_loss": 0.10595703125,
      "train_physics_loss": 0.0
    },
    {
      "step": 23890,
      "train_flow_matching_loss": 0.2236328125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.07035956531763077,
      "learning_rate": 5e-05,
      "loss": 0.1239,
      "step": 23900
    },
    {
      "step": 23900,
      "train_flow_matching_loss": 0.1484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 23900,
      "train_flow_matching_loss": 0.1689453125,
      "train_physics_loss": 0.0
    },
    {
      "step": 23900,
      "train_flow_matching_loss": 0.1513671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 23900,
      "train_flow_matching_loss": 0.1748046875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.1352807581424713,
      "learning_rate": 5e-05,
      "loss": 0.1266,
      "step": 23910
    },
    {
      "step": 23910,
      "train_flow_matching_loss": 0.1826171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 23910,
      "train_flow_matching_loss": 0.130859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 23910,
      "train_flow_matching_loss": 0.11474609375,
      "train_physics_loss": 0.0
    },
    {
      "step": 23910,
      "train_flow_matching_loss": 0.1474609375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10930582135915756,
      "learning_rate": 5e-05,
      "loss": 0.1324,
      "step": 23920
    },
    {
      "step": 23920,
      "train_flow_matching_loss": 0.1064453125,
      "train_physics_loss": 0.0
    },
    {
      "step": 23920,
      "train_flow_matching_loss": 0.09423828125,
      "train_physics_loss": 0.0
    },
    {
      "step": 23920,
      "train_flow_matching_loss": 0.10986328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 23920,
      "train_flow_matching_loss": 0.15625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.2241266518831253,
      "learning_rate": 5e-05,
      "loss": 0.127,
      "step": 23930
    },
    {
      "step": 23930,
      "train_flow_matching_loss": 0.177734375,
      "train_physics_loss": 0.0
    },
    {
      "step": 23930,
      "train_flow_matching_loss": 0.142578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 23930,
      "train_flow_matching_loss": 0.1962890625,
      "train_physics_loss": 0.0
    },
    {
      "step": 23930,
      "train_flow_matching_loss": 0.11962890625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08428886532783508,
      "learning_rate": 5e-05,
      "loss": 0.1223,
      "step": 23940
    },
    {
      "step": 23940,
      "train_flow_matching_loss": 0.0859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 23940,
      "train_flow_matching_loss": 0.09130859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 23940,
      "train_flow_matching_loss": 0.10888671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 23940,
      "train_flow_matching_loss": 0.1416015625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08810973912477493,
      "learning_rate": 5e-05,
      "loss": 0.1263,
      "step": 23950
    },
    {
      "step": 23950,
      "train_flow_matching_loss": 0.1826171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 23950,
      "train_flow_matching_loss": 0.1787109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 23950,
      "train_flow_matching_loss": 0.08447265625,
      "train_physics_loss": 0.0
    },
    {
      "step": 23950,
      "train_flow_matching_loss": 0.1025390625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.1979113519191742,
      "learning_rate": 5e-05,
      "loss": 0.1255,
      "step": 23960
    },
    {
      "step": 23960,
      "train_flow_matching_loss": 0.1748046875,
      "train_physics_loss": 0.0
    },
    {
      "step": 23960,
      "train_flow_matching_loss": 0.07958984375,
      "train_physics_loss": 0.0
    },
    {
      "step": 23960,
      "train_flow_matching_loss": 0.0947265625,
      "train_physics_loss": 0.0
    },
    {
      "step": 23960,
      "train_flow_matching_loss": 0.1484375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10052542388439178,
      "learning_rate": 5e-05,
      "loss": 0.1198,
      "step": 23970
    },
    {
      "step": 23970,
      "train_flow_matching_loss": 0.142578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 23970,
      "train_flow_matching_loss": 0.08984375,
      "train_physics_loss": 0.0
    },
    {
      "step": 23970,
      "train_flow_matching_loss": 0.12451171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 23970,
      "train_flow_matching_loss": 0.146484375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.1485314816236496,
      "learning_rate": 5e-05,
      "loss": 0.124,
      "step": 23980
    },
    {
      "step": 23980,
      "train_flow_matching_loss": 0.1630859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 23980,
      "train_flow_matching_loss": 0.08447265625,
      "train_physics_loss": 0.0
    },
    {
      "step": 23980,
      "train_flow_matching_loss": 0.10498046875,
      "train_physics_loss": 0.0
    },
    {
      "step": 23980,
      "train_flow_matching_loss": 0.1318359375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.13203226029872894,
      "learning_rate": 5e-05,
      "loss": 0.1253,
      "step": 23990
    },
    {
      "step": 23990,
      "train_flow_matching_loss": 0.10400390625,
      "train_physics_loss": 0.0
    },
    {
      "step": 23990,
      "train_flow_matching_loss": 0.1455078125,
      "train_physics_loss": 0.0
    },
    {
      "step": 23990,
      "train_flow_matching_loss": 0.0947265625,
      "train_physics_loss": 0.0
    },
    {
      "step": 23990,
      "train_flow_matching_loss": 0.185546875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09416943043470383,
      "learning_rate": 5e-05,
      "loss": 0.1262,
      "step": 24000
    },
    {
      "step": 24000,
      "train_flow_matching_loss": 0.0927734375,
      "train_physics_loss": 0.0
    },
    {
      "step": 24000,
      "train_flow_matching_loss": 0.1943359375,
      "train_physics_loss": 0.0
    },
    {
      "step": 24000,
      "train_flow_matching_loss": 0.1826171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 24000,
      "train_flow_matching_loss": 0.08544921875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.06764804571866989,
      "learning_rate": 5e-05,
      "loss": 0.1248,
      "step": 24010
    },
    {
      "step": 24010,
      "train_flow_matching_loss": 0.1318359375,
      "train_physics_loss": 0.0
    },
    {
      "step": 24010,
      "train_flow_matching_loss": 0.181640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 24010,
      "train_flow_matching_loss": 0.1455078125,
      "train_physics_loss": 0.0
    },
    {
      "step": 24010,
      "train_flow_matching_loss": 0.1640625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.077050119638443,
      "learning_rate": 5e-05,
      "loss": 0.1299,
      "step": 24020
    },
    {
      "step": 24020,
      "train_flow_matching_loss": 0.1455078125,
      "train_physics_loss": 0.0
    },
    {
      "step": 24020,
      "train_flow_matching_loss": 0.0859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 24020,
      "train_flow_matching_loss": 0.06787109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 24020,
      "train_flow_matching_loss": 0.08837890625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.07516958564519882,
      "learning_rate": 5e-05,
      "loss": 0.1225,
      "step": 24030
    },
    {
      "step": 24030,
      "train_flow_matching_loss": 0.11572265625,
      "train_physics_loss": 0.0
    },
    {
      "step": 24030,
      "train_flow_matching_loss": 0.07177734375,
      "train_physics_loss": 0.0
    },
    {
      "step": 24030,
      "train_flow_matching_loss": 0.1337890625,
      "train_physics_loss": 0.0
    },
    {
      "step": 24030,
      "train_flow_matching_loss": 0.08935546875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.1722668707370758,
      "learning_rate": 5e-05,
      "loss": 0.1226,
      "step": 24040
    },
    {
      "step": 24040,
      "train_flow_matching_loss": 0.1279296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 24040,
      "train_flow_matching_loss": 0.1142578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 24040,
      "train_flow_matching_loss": 0.0986328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 24040,
      "train_flow_matching_loss": 0.0966796875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09468359500169754,
      "learning_rate": 5e-05,
      "loss": 0.1218,
      "step": 24050
    },
    {
      "step": 24050,
      "train_flow_matching_loss": 0.032958984375,
      "train_physics_loss": 0.19738823175430298
    },
    {
      "step": 24050,
      "train_flow_matching_loss": 0.0908203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 24050,
      "train_flow_matching_loss": 0.134765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 24050,
      "train_flow_matching_loss": 0.1416015625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11120546609163284,
      "learning_rate": 5e-05,
      "loss": 0.1226,
      "step": 24060
    },
    {
      "step": 24060,
      "train_flow_matching_loss": 0.07080078125,
      "train_physics_loss": 0.184570774435997
    },
    {
      "step": 24060,
      "train_flow_matching_loss": 0.193359375,
      "train_physics_loss": 0.0
    },
    {
      "step": 24060,
      "train_flow_matching_loss": 0.11669921875,
      "train_physics_loss": 0.0
    },
    {
      "step": 24060,
      "train_flow_matching_loss": 0.12060546875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.18918251991271973,
      "learning_rate": 5e-05,
      "loss": 0.1261,
      "step": 24070
    },
    {
      "step": 24070,
      "train_flow_matching_loss": 0.03369140625,
      "train_physics_loss": 0.2216649204492569
    },
    {
      "step": 24070,
      "train_flow_matching_loss": 0.2158203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 24070,
      "train_flow_matching_loss": 0.158203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 24070,
      "train_flow_matching_loss": 0.228515625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.07526208460330963,
      "learning_rate": 5e-05,
      "loss": 0.1248,
      "step": 24080
    },
    {
      "step": 24080,
      "train_flow_matching_loss": 0.055419921875,
      "train_physics_loss": 0.19232897460460663
    },
    {
      "step": 24080,
      "train_flow_matching_loss": 0.09228515625,
      "train_physics_loss": 0.0
    },
    {
      "step": 24080,
      "train_flow_matching_loss": 0.1064453125,
      "train_physics_loss": 0.0
    },
    {
      "step": 24080,
      "train_flow_matching_loss": 0.0703125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.13784846663475037,
      "learning_rate": 5e-05,
      "loss": 0.1283,
      "step": 24090
    },
    {
      "step": 24090,
      "train_flow_matching_loss": 0.0390625,
      "train_physics_loss": 0.21599535644054413
    },
    {
      "step": 24090,
      "train_flow_matching_loss": 0.1015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 24090,
      "train_flow_matching_loss": 0.134765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 24090,
      "train_flow_matching_loss": 0.1416015625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11986684054136276,
      "learning_rate": 5e-05,
      "loss": 0.119,
      "step": 24100
    },
    {
      "step": 24100,
      "train_flow_matching_loss": 0.06787109375,
      "train_physics_loss": 0.2891782224178314
    },
    {
      "step": 24100,
      "train_flow_matching_loss": 0.09375,
      "train_physics_loss": 0.0
    },
    {
      "step": 24100,
      "train_flow_matching_loss": 0.11669921875,
      "train_physics_loss": 0.0
    },
    {
      "step": 24100,
      "train_flow_matching_loss": 0.2412109375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11374739557504654,
      "learning_rate": 5e-05,
      "loss": 0.1271,
      "step": 24110
    },
    {
      "step": 24110,
      "train_flow_matching_loss": 0.04296875,
      "train_physics_loss": 0.20415489375591278
    },
    {
      "step": 24110,
      "train_flow_matching_loss": 0.061767578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 24110,
      "train_flow_matching_loss": 0.2158203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 24110,
      "train_flow_matching_loss": 0.09033203125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.24397669732570648,
      "learning_rate": 5e-05,
      "loss": 0.1281,
      "step": 24120
    },
    {
      "step": 24120,
      "train_flow_matching_loss": 0.0771484375,
      "train_physics_loss": 0.32470130920410156
    },
    {
      "step": 24120,
      "train_flow_matching_loss": 0.09912109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 24120,
      "train_flow_matching_loss": 0.150390625,
      "train_physics_loss": 0.0
    },
    {
      "step": 24120,
      "train_flow_matching_loss": 0.177734375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09796439856290817,
      "learning_rate": 5e-05,
      "loss": 0.1234,
      "step": 24130
    },
    {
      "step": 24130,
      "train_flow_matching_loss": 0.0693359375,
      "train_physics_loss": 0.21112439036369324
    },
    {
      "step": 24130,
      "train_flow_matching_loss": 0.1552734375,
      "train_physics_loss": 0.0
    },
    {
      "step": 24130,
      "train_flow_matching_loss": 0.189453125,
      "train_physics_loss": 0.0
    },
    {
      "step": 24130,
      "train_flow_matching_loss": 0.10791015625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.0871945396065712,
      "learning_rate": 5e-05,
      "loss": 0.123,
      "step": 24140
    },
    {
      "step": 24140,
      "train_flow_matching_loss": 0.026123046875,
      "train_physics_loss": 0.20442607998847961
    },
    {
      "step": 24140,
      "train_flow_matching_loss": 0.150390625,
      "train_physics_loss": 0.0
    },
    {
      "step": 24140,
      "train_flow_matching_loss": 0.21875,
      "train_physics_loss": 0.0
    },
    {
      "step": 24140,
      "train_flow_matching_loss": 0.150390625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09183186292648315,
      "learning_rate": 5e-05,
      "loss": 0.1233,
      "step": 24150
    },
    {
      "step": 24150,
      "train_flow_matching_loss": 0.125,
      "train_physics_loss": 0.22133533656597137
    },
    {
      "step": 24150,
      "train_flow_matching_loss": 0.1708984375,
      "train_physics_loss": 0.0
    },
    {
      "step": 24150,
      "train_flow_matching_loss": 0.162109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 24150,
      "train_flow_matching_loss": 0.056396484375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.12181390821933746,
      "learning_rate": 5e-05,
      "loss": 0.1223,
      "step": 24160
    },
    {
      "step": 24160,
      "train_flow_matching_loss": 0.03466796875,
      "train_physics_loss": 0.19762781262397766
    },
    {
      "step": 24160,
      "train_flow_matching_loss": 0.09326171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 24160,
      "train_flow_matching_loss": 0.1240234375,
      "train_physics_loss": 0.0
    },
    {
      "step": 24160,
      "train_flow_matching_loss": 0.12451171875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08727218210697174,
      "learning_rate": 5e-05,
      "loss": 0.1231,
      "step": 24170
    },
    {
      "step": 24170,
      "train_flow_matching_loss": 0.06494140625,
      "train_physics_loss": 0.25947657227516174
    },
    {
      "step": 24170,
      "train_flow_matching_loss": 0.1650390625,
      "train_physics_loss": 0.0
    },
    {
      "step": 24170,
      "train_flow_matching_loss": 0.0986328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 24170,
      "train_flow_matching_loss": 0.1943359375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08725469559431076,
      "learning_rate": 5e-05,
      "loss": 0.1229,
      "step": 24180
    },
    {
      "step": 24180,
      "train_flow_matching_loss": 0.03759765625,
      "train_physics_loss": 0.20273949205875397
    },
    {
      "step": 24180,
      "train_flow_matching_loss": 0.146484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 24180,
      "train_flow_matching_loss": 0.0732421875,
      "train_physics_loss": 0.0
    },
    {
      "step": 24180,
      "train_flow_matching_loss": 0.07958984375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08800971508026123,
      "learning_rate": 5e-05,
      "loss": 0.127,
      "step": 24190
    },
    {
      "step": 24190,
      "train_flow_matching_loss": 0.0849609375,
      "train_physics_loss": 0.2496911883354187
    },
    {
      "step": 24190,
      "train_flow_matching_loss": 0.193359375,
      "train_physics_loss": 0.0
    },
    {
      "step": 24190,
      "train_flow_matching_loss": 0.1884765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 24190,
      "train_flow_matching_loss": 0.162109375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10334265232086182,
      "learning_rate": 5e-05,
      "loss": 0.122,
      "step": 24200
    },
    {
      "step": 24200,
      "train_flow_matching_loss": 0.09765625,
      "train_physics_loss": 0.35451287031173706
    },
    {
      "step": 24200,
      "train_flow_matching_loss": 0.1376953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 24200,
      "train_flow_matching_loss": 0.046142578125,
      "train_physics_loss": 0.2005849927663803
    },
    {
      "step": 24200,
      "train_flow_matching_loss": 0.158203125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08541910350322723,
      "learning_rate": 5e-05,
      "loss": 0.1254,
      "step": 24210
    },
    {
      "step": 24210,
      "train_flow_matching_loss": 0.041259765625,
      "train_physics_loss": 0.22168517112731934
    },
    {
      "step": 24210,
      "train_flow_matching_loss": 0.0859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 24210,
      "train_flow_matching_loss": 0.06884765625,
      "train_physics_loss": 0.3194471299648285
    },
    {
      "step": 24210,
      "train_flow_matching_loss": 0.11328125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11288553476333618,
      "learning_rate": 5e-05,
      "loss": 0.125,
      "step": 24220
    },
    {
      "step": 24220,
      "train_flow_matching_loss": 0.039306640625,
      "train_physics_loss": 0.21298176050186157
    },
    {
      "step": 24220,
      "train_flow_matching_loss": 0.11181640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 24220,
      "train_flow_matching_loss": 0.039794921875,
      "train_physics_loss": 0.2109149694442749
    },
    {
      "step": 24220,
      "train_flow_matching_loss": 0.12890625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10097158700227737,
      "learning_rate": 5e-05,
      "loss": 0.1216,
      "step": 24230
    },
    {
      "step": 24230,
      "train_flow_matching_loss": 0.051513671875,
      "train_physics_loss": 0.26882532238960266
    },
    {
      "step": 24230,
      "train_flow_matching_loss": 0.158203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 24230,
      "train_flow_matching_loss": 0.08984375,
      "train_physics_loss": 0.3086676299571991
    },
    {
      "step": 24230,
      "train_flow_matching_loss": 0.134765625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.0970781147480011,
      "learning_rate": 5e-05,
      "loss": 0.1272,
      "step": 24240
    },
    {
      "step": 24240,
      "train_flow_matching_loss": 0.052490234375,
      "train_physics_loss": 0.20552217960357666
    },
    {
      "step": 24240,
      "train_flow_matching_loss": 0.205078125,
      "train_physics_loss": 0.0
    },
    {
      "step": 24240,
      "train_flow_matching_loss": 0.058837890625,
      "train_physics_loss": 0.22854889929294586
    },
    {
      "step": 24240,
      "train_flow_matching_loss": 0.09423828125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.13149656355381012,
      "learning_rate": 5e-05,
      "loss": 0.1201,
      "step": 24250
    },
    {
      "step": 24250,
      "train_flow_matching_loss": 0.08154296875,
      "train_physics_loss": 0.30727091431617737
    },
    {
      "step": 24250,
      "train_flow_matching_loss": 0.1708984375,
      "train_physics_loss": 0.0
    },
    {
      "step": 24250,
      "train_flow_matching_loss": 0.0234375,
      "train_physics_loss": 0.22886134684085846
    },
    {
      "step": 24250,
      "train_flow_matching_loss": 0.10888671875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.22215582430362701,
      "learning_rate": 5e-05,
      "loss": 0.1275,
      "step": 24260
    },
    {
      "step": 24260,
      "train_flow_matching_loss": 0.03955078125,
      "train_physics_loss": 0.2050980031490326
    },
    {
      "step": 24260,
      "train_flow_matching_loss": 0.1552734375,
      "train_physics_loss": 0.0
    },
    {
      "step": 24260,
      "train_flow_matching_loss": 0.064453125,
      "train_physics_loss": 0.20670408010482788
    },
    {
      "step": 24260,
      "train_flow_matching_loss": 0.1103515625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.0878826454281807,
      "learning_rate": 5e-05,
      "loss": 0.1207,
      "step": 24270
    },
    {
      "step": 24270,
      "train_flow_matching_loss": 0.035400390625,
      "train_physics_loss": 0.2690197825431824
    },
    {
      "step": 24270,
      "train_flow_matching_loss": 0.154296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 24270,
      "train_flow_matching_loss": 0.03857421875,
      "train_physics_loss": 0.3004905879497528
    },
    {
      "step": 24270,
      "train_flow_matching_loss": 0.166015625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11247863620519638,
      "learning_rate": 5e-05,
      "loss": 0.1232,
      "step": 24280
    },
    {
      "step": 24280,
      "train_flow_matching_loss": 0.038330078125,
      "train_physics_loss": 0.20927347242832184
    },
    {
      "step": 24280,
      "train_flow_matching_loss": 0.208984375,
      "train_physics_loss": 0.0
    },
    {
      "step": 24280,
      "train_flow_matching_loss": 0.05322265625,
      "train_physics_loss": 0.2049267590045929
    },
    {
      "step": 24280,
      "train_flow_matching_loss": 0.2021484375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10681216418743134,
      "learning_rate": 5e-05,
      "loss": 0.1211,
      "step": 24290
    },
    {
      "step": 24290,
      "train_flow_matching_loss": 0.032470703125,
      "train_physics_loss": 0.19767633080482483
    },
    {
      "step": 24290,
      "train_flow_matching_loss": 0.1298828125,
      "train_physics_loss": 0.0
    },
    {
      "step": 24290,
      "train_flow_matching_loss": 0.03564453125,
      "train_physics_loss": 0.21480049192905426
    },
    {
      "step": 24290,
      "train_flow_matching_loss": 0.2099609375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10571921616792679,
      "learning_rate": 5e-05,
      "loss": 0.1253,
      "step": 24300
    },
    {
      "step": 24300,
      "train_flow_matching_loss": 0.2353515625,
      "train_physics_loss": 0.0
    },
    {
      "step": 24300,
      "train_flow_matching_loss": 0.1845703125,
      "train_physics_loss": 0.0
    },
    {
      "step": 24300,
      "train_flow_matching_loss": 0.034423828125,
      "train_physics_loss": 0.22940561175346375
    },
    {
      "step": 24300,
      "train_flow_matching_loss": 0.08447265625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.16481083631515503,
      "learning_rate": 5e-05,
      "loss": 0.1222,
      "step": 24310
    },
    {
      "step": 24310,
      "train_flow_matching_loss": 0.1162109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 24310,
      "train_flow_matching_loss": 0.21875,
      "train_physics_loss": 0.0
    },
    {
      "step": 24310,
      "train_flow_matching_loss": 0.0673828125,
      "train_physics_loss": 0.2988581657409668
    },
    {
      "step": 24310,
      "train_flow_matching_loss": 0.173828125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10173725336790085,
      "learning_rate": 5e-05,
      "loss": 0.1273,
      "step": 24320
    },
    {
      "step": 24320,
      "train_flow_matching_loss": 0.134765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 24320,
      "train_flow_matching_loss": 0.1513671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 24320,
      "train_flow_matching_loss": 0.04052734375,
      "train_physics_loss": 0.22939282655715942
    },
    {
      "step": 24320,
      "train_flow_matching_loss": 0.0927734375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09567797183990479,
      "learning_rate": 5e-05,
      "loss": 0.1259,
      "step": 24330
    },
    {
      "step": 24330,
      "train_flow_matching_loss": 0.189453125,
      "train_physics_loss": 0.0
    },
    {
      "step": 24330,
      "train_flow_matching_loss": 0.1767578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 24330,
      "train_flow_matching_loss": 0.04931640625,
      "train_physics_loss": 0.2027140110731125
    },
    {
      "step": 24330,
      "train_flow_matching_loss": 0.1337890625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10302209854125977,
      "learning_rate": 5e-05,
      "loss": 0.124,
      "step": 24340
    },
    {
      "step": 24340,
      "train_flow_matching_loss": 0.181640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 24340,
      "train_flow_matching_loss": 0.1376953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 24340,
      "train_flow_matching_loss": 0.020751953125,
      "train_physics_loss": 0.1946595311164856
    },
    {
      "step": 24340,
      "train_flow_matching_loss": 0.1591796875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.12729382514953613,
      "learning_rate": 5e-05,
      "loss": 0.1207,
      "step": 24350
    },
    {
      "step": 24350,
      "train_flow_matching_loss": 0.1669921875,
      "train_physics_loss": 0.0
    },
    {
      "step": 24350,
      "train_flow_matching_loss": 0.1181640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 24350,
      "train_flow_matching_loss": 0.06640625,
      "train_physics_loss": 0.22897137701511383
    },
    {
      "step": 24350,
      "train_flow_matching_loss": 0.11181640625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.1860753744840622,
      "learning_rate": 5e-05,
      "loss": 0.1257,
      "step": 24360
    },
    {
      "step": 24360,
      "train_flow_matching_loss": 0.1630859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 24360,
      "train_flow_matching_loss": 0.055419921875,
      "train_physics_loss": 0.0
    },
    {
      "step": 24360,
      "train_flow_matching_loss": 0.040771484375,
      "train_physics_loss": 0.21313978731632233
    },
    {
      "step": 24360,
      "train_flow_matching_loss": 0.1220703125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.4960009753704071,
      "learning_rate": 5e-05,
      "loss": 0.1245,
      "step": 24370
    },
    {
      "step": 24370,
      "train_flow_matching_loss": 0.2216796875,
      "train_physics_loss": 0.0
    },
    {
      "step": 24370,
      "train_flow_matching_loss": 0.07666015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 24370,
      "train_flow_matching_loss": 0.07861328125,
      "train_physics_loss": 0.21785849332809448
    },
    {
      "step": 24370,
      "train_flow_matching_loss": 0.140625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.12296899408102036,
      "learning_rate": 5e-05,
      "loss": 0.1224,
      "step": 24380
    },
    {
      "step": 24380,
      "train_flow_matching_loss": 0.2265625,
      "train_physics_loss": 0.0
    },
    {
      "step": 24380,
      "train_flow_matching_loss": 0.134765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 24380,
      "train_flow_matching_loss": 0.06640625,
      "train_physics_loss": 0.2621949315071106
    },
    {
      "step": 24380,
      "train_flow_matching_loss": 0.11376953125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.15696023404598236,
      "learning_rate": 5e-05,
      "loss": 0.1295,
      "step": 24390
    },
    {
      "step": 24390,
      "train_flow_matching_loss": 0.1220703125,
      "train_physics_loss": 0.0
    },
    {
      "step": 24390,
      "train_flow_matching_loss": 0.09814453125,
      "train_physics_loss": 0.0
    },
    {
      "step": 24390,
      "train_flow_matching_loss": 0.036865234375,
      "train_physics_loss": 0.2557491064071655
    },
    {
      "step": 24390,
      "train_flow_matching_loss": 0.1103515625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.210398867726326,
      "learning_rate": 5e-05,
      "loss": 0.1236,
      "step": 24400
    },
    {
      "step": 24400,
      "train_flow_matching_loss": 0.1904296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 24400,
      "train_flow_matching_loss": 0.1337890625,
      "train_physics_loss": 0.0
    },
    {
      "step": 24400,
      "train_flow_matching_loss": 0.0849609375,
      "train_physics_loss": 0.21986238658428192
    },
    {
      "step": 24400,
      "train_flow_matching_loss": 0.1279296875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.1637757271528244,
      "learning_rate": 5e-05,
      "loss": 0.1255,
      "step": 24410
    },
    {
      "step": 24410,
      "train_flow_matching_loss": 0.1484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 24410,
      "train_flow_matching_loss": 0.169921875,
      "train_physics_loss": 0.0
    },
    {
      "step": 24410,
      "train_flow_matching_loss": 0.06005859375,
      "train_physics_loss": 0.24709539115428925
    },
    {
      "step": 24410,
      "train_flow_matching_loss": 0.10400390625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.13010327517986298,
      "learning_rate": 5e-05,
      "loss": 0.1306,
      "step": 24420
    },
    {
      "step": 24420,
      "train_flow_matching_loss": 0.12109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 24420,
      "train_flow_matching_loss": 0.1533203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 24420,
      "train_flow_matching_loss": 0.02734375,
      "train_physics_loss": 0.1995704472064972
    },
    {
      "step": 24420,
      "train_flow_matching_loss": 0.1494140625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.13899409770965576,
      "learning_rate": 5e-05,
      "loss": 0.1194,
      "step": 24430
    },
    {
      "step": 24430,
      "train_flow_matching_loss": 0.1943359375,
      "train_physics_loss": 0.0
    },
    {
      "step": 24430,
      "train_flow_matching_loss": 0.1630859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 24430,
      "train_flow_matching_loss": 0.057861328125,
      "train_physics_loss": 0.20705364644527435
    },
    {
      "step": 24430,
      "train_flow_matching_loss": 0.126953125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.1356523633003235,
      "learning_rate": 5e-05,
      "loss": 0.1227,
      "step": 24440
    },
    {
      "step": 24440,
      "train_flow_matching_loss": 0.1865234375,
      "train_physics_loss": 0.0
    },
    {
      "step": 24440,
      "train_flow_matching_loss": 0.11328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 24440,
      "train_flow_matching_loss": 0.047119140625,
      "train_physics_loss": 0.21105770766735077
    },
    {
      "step": 24440,
      "train_flow_matching_loss": 0.1474609375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10446921736001968,
      "learning_rate": 5e-05,
      "loss": 0.1245,
      "step": 24450
    },
    {
      "step": 24450,
      "train_flow_matching_loss": 0.11474609375,
      "train_physics_loss": 0.0
    },
    {
      "step": 24450,
      "train_flow_matching_loss": 0.12158203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 24450,
      "train_flow_matching_loss": 0.1943359375,
      "train_physics_loss": 0.0
    },
    {
      "step": 24450,
      "train_flow_matching_loss": 0.11962890625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.1130424365401268,
      "learning_rate": 5e-05,
      "loss": 0.1208,
      "step": 24460
    },
    {
      "step": 24460,
      "train_flow_matching_loss": 0.173828125,
      "train_physics_loss": 0.0
    },
    {
      "step": 24460,
      "train_flow_matching_loss": 0.11865234375,
      "train_physics_loss": 0.0
    },
    {
      "step": 24460,
      "train_flow_matching_loss": 0.2216796875,
      "train_physics_loss": 0.0
    },
    {
      "step": 24460,
      "train_flow_matching_loss": 0.1044921875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09202572703361511,
      "learning_rate": 5e-05,
      "loss": 0.1276,
      "step": 24470
    },
    {
      "step": 24470,
      "train_flow_matching_loss": 0.1201171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 24470,
      "train_flow_matching_loss": 0.166015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 24470,
      "train_flow_matching_loss": 0.1357421875,
      "train_physics_loss": 0.0
    },
    {
      "step": 24470,
      "train_flow_matching_loss": 0.1494140625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11101333051919937,
      "learning_rate": 5e-05,
      "loss": 0.1255,
      "step": 24480
    },
    {
      "step": 24480,
      "train_flow_matching_loss": 0.1513671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 24480,
      "train_flow_matching_loss": 0.12890625,
      "train_physics_loss": 0.0
    },
    {
      "step": 24480,
      "train_flow_matching_loss": 0.171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 24480,
      "train_flow_matching_loss": 0.1826171875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10262257605791092,
      "learning_rate": 5e-05,
      "loss": 0.1249,
      "step": 24490
    },
    {
      "step": 24490,
      "train_flow_matching_loss": 0.2353515625,
      "train_physics_loss": 0.0
    },
    {
      "step": 24490,
      "train_flow_matching_loss": 0.1162109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 24490,
      "train_flow_matching_loss": 0.1513671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 24490,
      "train_flow_matching_loss": 0.1669921875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08521492034196854,
      "learning_rate": 5e-05,
      "loss": 0.1233,
      "step": 24500
    },
    {
      "step": 24500,
      "train_flow_matching_loss": 0.09375,
      "train_physics_loss": 0.0
    },
    {
      "step": 24500,
      "train_flow_matching_loss": 0.142578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 24500,
      "train_flow_matching_loss": 0.140625,
      "train_physics_loss": 0.0
    },
    {
      "step": 24500,
      "train_flow_matching_loss": 0.12353515625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09222529083490372,
      "learning_rate": 5e-05,
      "loss": 0.1229,
      "step": 24510
    },
    {
      "step": 24510,
      "train_flow_matching_loss": 0.0849609375,
      "train_physics_loss": 0.0
    },
    {
      "step": 24510,
      "train_flow_matching_loss": 0.2119140625,
      "train_physics_loss": 0.0
    },
    {
      "step": 24510,
      "train_flow_matching_loss": 0.134765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 24510,
      "train_flow_matching_loss": 0.140625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11637543886899948,
      "learning_rate": 5e-05,
      "loss": 0.1232,
      "step": 24520
    },
    {
      "step": 24520,
      "train_flow_matching_loss": 0.1845703125,
      "train_physics_loss": 0.0
    },
    {
      "step": 24520,
      "train_flow_matching_loss": 0.12109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 24520,
      "train_flow_matching_loss": 0.1484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 24520,
      "train_flow_matching_loss": 0.150390625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.07998579740524292,
      "learning_rate": 5e-05,
      "loss": 0.1278,
      "step": 24530
    },
    {
      "step": 24530,
      "train_flow_matching_loss": 0.193359375,
      "train_physics_loss": 0.0
    },
    {
      "step": 24530,
      "train_flow_matching_loss": 0.203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 24530,
      "train_flow_matching_loss": 0.1552734375,
      "train_physics_loss": 0.0
    },
    {
      "step": 24530,
      "train_flow_matching_loss": 0.10205078125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.104271300137043,
      "learning_rate": 5e-05,
      "loss": 0.1267,
      "step": 24540
    },
    {
      "step": 24540,
      "train_flow_matching_loss": 0.0966796875,
      "train_physics_loss": 0.0
    },
    {
      "step": 24540,
      "train_flow_matching_loss": 0.26171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 24540,
      "train_flow_matching_loss": 0.13671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 24540,
      "train_flow_matching_loss": 0.109375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.0676940456032753,
      "learning_rate": 5e-05,
      "loss": 0.1246,
      "step": 24550
    },
    {
      "step": 24550,
      "train_flow_matching_loss": 0.2158203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 24550,
      "train_flow_matching_loss": 0.1689453125,
      "train_physics_loss": 0.0
    },
    {
      "step": 24550,
      "train_flow_matching_loss": 0.06494140625,
      "train_physics_loss": 0.0
    },
    {
      "step": 24550,
      "train_flow_matching_loss": 0.11962890625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09832961857318878,
      "learning_rate": 5e-05,
      "loss": 0.1189,
      "step": 24560
    },
    {
      "step": 24560,
      "train_flow_matching_loss": 0.1298828125,
      "train_physics_loss": 0.0
    },
    {
      "step": 24560,
      "train_flow_matching_loss": 0.1572265625,
      "train_physics_loss": 0.0
    },
    {
      "step": 24560,
      "train_flow_matching_loss": 0.09521484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 24560,
      "train_flow_matching_loss": 0.07861328125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09705985337495804,
      "learning_rate": 5e-05,
      "loss": 0.1189,
      "step": 24570
    },
    {
      "step": 24570,
      "train_flow_matching_loss": 0.1494140625,
      "train_physics_loss": 0.0
    },
    {
      "step": 24570,
      "train_flow_matching_loss": 0.10986328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 24570,
      "train_flow_matching_loss": 0.203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 24570,
      "train_flow_matching_loss": 0.150390625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.07258299738168716,
      "learning_rate": 5e-05,
      "loss": 0.1182,
      "step": 24580
    },
    {
      "step": 24580,
      "train_flow_matching_loss": 0.1416015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 24580,
      "train_flow_matching_loss": 0.1416015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 24580,
      "train_flow_matching_loss": 0.0966796875,
      "train_physics_loss": 0.0
    },
    {
      "step": 24580,
      "train_flow_matching_loss": 0.1171875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10606775432825089,
      "learning_rate": 5e-05,
      "loss": 0.1282,
      "step": 24590
    },
    {
      "step": 24590,
      "train_flow_matching_loss": 0.10546875,
      "train_physics_loss": 0.0
    },
    {
      "step": 24590,
      "train_flow_matching_loss": 0.1728515625,
      "train_physics_loss": 0.0
    },
    {
      "step": 24590,
      "train_flow_matching_loss": 0.15625,
      "train_physics_loss": 0.0
    },
    {
      "step": 24590,
      "train_flow_matching_loss": 0.1552734375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.07590071111917496,
      "learning_rate": 5e-05,
      "loss": 0.1251,
      "step": 24600
    },
    {
      "step": 24600,
      "train_flow_matching_loss": 0.1494140625,
      "train_physics_loss": 0.0
    },
    {
      "step": 24600,
      "train_flow_matching_loss": 0.15625,
      "train_physics_loss": 0.0
    },
    {
      "step": 24600,
      "train_flow_matching_loss": 0.1474609375,
      "train_physics_loss": 0.0
    },
    {
      "step": 24600,
      "train_flow_matching_loss": 0.08740234375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.0942627415060997,
      "learning_rate": 5e-05,
      "loss": 0.1202,
      "step": 24610
    },
    {
      "step": 24610,
      "train_flow_matching_loss": 0.09765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 24610,
      "train_flow_matching_loss": 0.1611328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 24610,
      "train_flow_matching_loss": 0.09765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 24610,
      "train_flow_matching_loss": 0.11376953125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08516070246696472,
      "learning_rate": 5e-05,
      "loss": 0.1253,
      "step": 24620
    },
    {
      "step": 24620,
      "train_flow_matching_loss": 0.1787109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 24620,
      "train_flow_matching_loss": 0.07763671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 24620,
      "train_flow_matching_loss": 0.1630859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 24620,
      "train_flow_matching_loss": 0.0986328125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09285292774438858,
      "learning_rate": 5e-05,
      "loss": 0.1214,
      "step": 24630
    },
    {
      "step": 24630,
      "train_flow_matching_loss": 0.1474609375,
      "train_physics_loss": 0.0
    },
    {
      "step": 24630,
      "train_flow_matching_loss": 0.166015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 24630,
      "train_flow_matching_loss": 0.1416015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 24630,
      "train_flow_matching_loss": 0.08740234375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11954494565725327,
      "learning_rate": 5e-05,
      "loss": 0.1197,
      "step": 24640
    },
    {
      "step": 24640,
      "train_flow_matching_loss": 0.154296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 24640,
      "train_flow_matching_loss": 0.1669921875,
      "train_physics_loss": 0.0
    },
    {
      "step": 24640,
      "train_flow_matching_loss": 0.1279296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 24640,
      "train_flow_matching_loss": 0.1416015625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11184214055538177,
      "learning_rate": 5e-05,
      "loss": 0.1198,
      "step": 24650
    },
    {
      "step": 24650,
      "train_flow_matching_loss": 0.1962890625,
      "train_physics_loss": 0.0
    },
    {
      "step": 24650,
      "train_flow_matching_loss": 0.109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 24650,
      "train_flow_matching_loss": 0.10107421875,
      "train_physics_loss": 0.0
    },
    {
      "step": 24650,
      "train_flow_matching_loss": 0.1533203125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10131662338972092,
      "learning_rate": 5e-05,
      "loss": 0.1185,
      "step": 24660
    },
    {
      "step": 24660,
      "train_flow_matching_loss": 0.11328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 24660,
      "train_flow_matching_loss": 0.1083984375,
      "train_physics_loss": 0.0
    },
    {
      "step": 24660,
      "train_flow_matching_loss": 0.1533203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 24660,
      "train_flow_matching_loss": 0.07958984375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10071223229169846,
      "learning_rate": 5e-05,
      "loss": 0.1237,
      "step": 24670
    },
    {
      "step": 24670,
      "train_flow_matching_loss": 0.193359375,
      "train_physics_loss": 0.0
    },
    {
      "step": 24670,
      "train_flow_matching_loss": 0.1064453125,
      "train_physics_loss": 0.0
    },
    {
      "step": 24670,
      "train_flow_matching_loss": 0.15625,
      "train_physics_loss": 0.0
    },
    {
      "step": 24670,
      "train_flow_matching_loss": 0.10595703125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08749455958604813,
      "learning_rate": 5e-05,
      "loss": 0.1208,
      "step": 24680
    },
    {
      "step": 24680,
      "train_flow_matching_loss": 0.201171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 24680,
      "train_flow_matching_loss": 0.1826171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 24680,
      "train_flow_matching_loss": 0.1171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 24680,
      "train_flow_matching_loss": 0.134765625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.0946325957775116,
      "learning_rate": 5e-05,
      "loss": 0.1272,
      "step": 24690
    },
    {
      "step": 24690,
      "train_flow_matching_loss": 0.1748046875,
      "train_physics_loss": 0.0
    },
    {
      "step": 24690,
      "train_flow_matching_loss": 0.134765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 24690,
      "train_flow_matching_loss": 0.1337890625,
      "train_physics_loss": 0.0
    },
    {
      "step": 24690,
      "train_flow_matching_loss": 0.048095703125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.0683617889881134,
      "learning_rate": 5e-05,
      "loss": 0.1232,
      "step": 24700
    },
    {
      "step": 24700,
      "train_flow_matching_loss": 0.1884765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 24700,
      "train_flow_matching_loss": 0.1591796875,
      "train_physics_loss": 0.0
    },
    {
      "step": 24700,
      "train_flow_matching_loss": 0.09130859375,
      "train_physics_loss": 0.19090768694877625
    },
    {
      "step": 24700,
      "train_flow_matching_loss": 0.10302734375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.13476622104644775,
      "learning_rate": 5e-05,
      "loss": 0.1182,
      "step": 24710
    },
    {
      "step": 24710,
      "train_flow_matching_loss": 0.1494140625,
      "train_physics_loss": 0.0
    },
    {
      "step": 24710,
      "train_flow_matching_loss": 0.09423828125,
      "train_physics_loss": 0.0
    },
    {
      "step": 24710,
      "train_flow_matching_loss": 0.10791015625,
      "train_physics_loss": 0.3226257562637329
    },
    {
      "step": 24710,
      "train_flow_matching_loss": 0.1376953125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08455189317464828,
      "learning_rate": 5e-05,
      "loss": 0.1212,
      "step": 24720
    },
    {
      "step": 24720,
      "train_flow_matching_loss": 0.10400390625,
      "train_physics_loss": 0.0
    },
    {
      "step": 24720,
      "train_flow_matching_loss": 0.1552734375,
      "train_physics_loss": 0.0
    },
    {
      "step": 24720,
      "train_flow_matching_loss": 0.038818359375,
      "train_physics_loss": 0.1761065125465393
    },
    {
      "step": 24720,
      "train_flow_matching_loss": 0.0927734375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.07085958123207092,
      "learning_rate": 5e-05,
      "loss": 0.1213,
      "step": 24730
    },
    {
      "step": 24730,
      "train_flow_matching_loss": 0.173828125,
      "train_physics_loss": 0.0
    },
    {
      "step": 24730,
      "train_flow_matching_loss": 0.1220703125,
      "train_physics_loss": 0.0
    },
    {
      "step": 24730,
      "train_flow_matching_loss": 0.07275390625,
      "train_physics_loss": 0.3148713707923889
    },
    {
      "step": 24730,
      "train_flow_matching_loss": 0.07861328125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.07339376956224442,
      "learning_rate": 5e-05,
      "loss": 0.1155,
      "step": 24740
    },
    {
      "step": 24740,
      "train_flow_matching_loss": 0.1201171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 24740,
      "train_flow_matching_loss": 0.12158203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 24740,
      "train_flow_matching_loss": 0.09814453125,
      "train_physics_loss": 0.18795520067214966
    },
    {
      "step": 24740,
      "train_flow_matching_loss": 0.1533203125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08571138978004456,
      "learning_rate": 5e-05,
      "loss": 0.1236,
      "step": 24750
    },
    {
      "step": 24750,
      "train_flow_matching_loss": 0.1142578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 24750,
      "train_flow_matching_loss": 0.193359375,
      "train_physics_loss": 0.0
    },
    {
      "step": 24750,
      "train_flow_matching_loss": 0.048828125,
      "train_physics_loss": 0.2049352377653122
    },
    {
      "step": 24750,
      "train_flow_matching_loss": 0.10595703125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.07119401544332504,
      "learning_rate": 5e-05,
      "loss": 0.1152,
      "step": 24760
    },
    {
      "step": 24760,
      "train_flow_matching_loss": 0.130859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 24760,
      "train_flow_matching_loss": 0.1279296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 24760,
      "train_flow_matching_loss": 0.055419921875,
      "train_physics_loss": 0.25773951411247253
    },
    {
      "step": 24760,
      "train_flow_matching_loss": 0.1201171875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10945002734661102,
      "learning_rate": 5e-05,
      "loss": 0.1181,
      "step": 24770
    },
    {
      "step": 24770,
      "train_flow_matching_loss": 0.1318359375,
      "train_physics_loss": 0.0
    },
    {
      "step": 24770,
      "train_flow_matching_loss": 0.09033203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 24770,
      "train_flow_matching_loss": 0.037109375,
      "train_physics_loss": 0.21934731304645538
    },
    {
      "step": 24770,
      "train_flow_matching_loss": 0.12158203125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.07769057899713516,
      "learning_rate": 5e-05,
      "loss": 0.1181,
      "step": 24780
    },
    {
      "step": 24780,
      "train_flow_matching_loss": 0.11865234375,
      "train_physics_loss": 0.0
    },
    {
      "step": 24780,
      "train_flow_matching_loss": 0.203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 24780,
      "train_flow_matching_loss": 0.08740234375,
      "train_physics_loss": 0.25105103850364685
    },
    {
      "step": 24780,
      "train_flow_matching_loss": 0.08447265625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.0877608060836792,
      "learning_rate": 5e-05,
      "loss": 0.1224,
      "step": 24790
    },
    {
      "step": 24790,
      "train_flow_matching_loss": 0.10205078125,
      "train_physics_loss": 0.0
    },
    {
      "step": 24790,
      "train_flow_matching_loss": 0.1630859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 24790,
      "train_flow_matching_loss": 0.046630859375,
      "train_physics_loss": 0.2214297503232956
    },
    {
      "step": 24790,
      "train_flow_matching_loss": 0.1669921875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.18364396691322327,
      "learning_rate": 5e-05,
      "loss": 0.1195,
      "step": 24800
    },
    {
      "step": 24800,
      "train_flow_matching_loss": 0.10546875,
      "train_physics_loss": 0.0
    },
    {
      "step": 24800,
      "train_flow_matching_loss": 0.1416015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 24800,
      "train_flow_matching_loss": 0.04541015625,
      "train_physics_loss": 0.19150696694850922
    },
    {
      "step": 24800,
      "train_flow_matching_loss": 0.125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08537941426038742,
      "learning_rate": 5e-05,
      "loss": 0.1188,
      "step": 24810
    },
    {
      "step": 24810,
      "train_flow_matching_loss": 0.052734375,
      "train_physics_loss": 0.20394785702228546
    },
    {
      "step": 24810,
      "train_flow_matching_loss": 0.11328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 24810,
      "train_flow_matching_loss": 0.0986328125,
      "train_physics_loss": 0.2499590963125229
    },
    {
      "step": 24810,
      "train_flow_matching_loss": 0.11962890625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.0963796079158783,
      "learning_rate": 5e-05,
      "loss": 0.1155,
      "step": 24820
    },
    {
      "step": 24820,
      "train_flow_matching_loss": 0.037109375,
      "train_physics_loss": 0.19167882204055786
    },
    {
      "step": 24820,
      "train_flow_matching_loss": 0.1484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 24820,
      "train_flow_matching_loss": 0.06787109375,
      "train_physics_loss": 0.20636709034442902
    },
    {
      "step": 24820,
      "train_flow_matching_loss": 0.11376953125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.107231505215168,
      "learning_rate": 5e-05,
      "loss": 0.1197,
      "step": 24830
    },
    {
      "step": 24830,
      "train_flow_matching_loss": 0.036865234375,
      "train_physics_loss": 0.2069859355688095
    },
    {
      "step": 24830,
      "train_flow_matching_loss": 0.09765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 24830,
      "train_flow_matching_loss": 0.06884765625,
      "train_physics_loss": 0.21608634293079376
    },
    {
      "step": 24830,
      "train_flow_matching_loss": 0.171875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10666745901107788,
      "learning_rate": 5e-05,
      "loss": 0.1221,
      "step": 24840
    },
    {
      "step": 24840,
      "train_flow_matching_loss": 0.07373046875,
      "train_physics_loss": 0.26810818910598755
    },
    {
      "step": 24840,
      "train_flow_matching_loss": 0.1884765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 24840,
      "train_flow_matching_loss": 0.034912109375,
      "train_physics_loss": 0.18720893561840057
    },
    {
      "step": 24840,
      "train_flow_matching_loss": 0.146484375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.074546679854393,
      "learning_rate": 5e-05,
      "loss": 0.1278,
      "step": 24850
    },
    {
      "step": 24850,
      "train_flow_matching_loss": 0.0634765625,
      "train_physics_loss": 0.219436913728714
    },
    {
      "step": 24850,
      "train_flow_matching_loss": 0.12158203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 24850,
      "train_flow_matching_loss": 0.04638671875,
      "train_physics_loss": 0.23392266035079956
    },
    {
      "step": 24850,
      "train_flow_matching_loss": 0.0966796875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11792159080505371,
      "learning_rate": 5e-05,
      "loss": 0.1256,
      "step": 24860
    },
    {
      "step": 24860,
      "train_flow_matching_loss": 0.02734375,
      "train_physics_loss": 0.1985674351453781
    },
    {
      "step": 24860,
      "train_flow_matching_loss": 0.09765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 24860,
      "train_flow_matching_loss": 0.0274658203125,
      "train_physics_loss": 0.20218370854854584
    },
    {
      "step": 24860,
      "train_flow_matching_loss": 0.08642578125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11334838718175888,
      "learning_rate": 5e-05,
      "loss": 0.1131,
      "step": 24870
    },
    {
      "step": 24870,
      "train_flow_matching_loss": 0.07421875,
      "train_physics_loss": 0.0
    },
    {
      "step": 24870,
      "train_flow_matching_loss": 0.058349609375,
      "train_physics_loss": 0.0
    },
    {
      "step": 24870,
      "train_flow_matching_loss": 0.0712890625,
      "train_physics_loss": 0.1681741178035736
    },
    {
      "step": 24870,
      "train_flow_matching_loss": 0.10986328125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09176071733236313,
      "learning_rate": 5e-05,
      "loss": 0.1237,
      "step": 24880
    },
    {
      "step": 24880,
      "train_flow_matching_loss": 0.038330078125,
      "train_physics_loss": 0.21109150350093842
    },
    {
      "step": 24880,
      "train_flow_matching_loss": 0.212890625,
      "train_physics_loss": 0.0
    },
    {
      "step": 24880,
      "train_flow_matching_loss": 0.046630859375,
      "train_physics_loss": 0.1906287670135498
    },
    {
      "step": 24880,
      "train_flow_matching_loss": 0.09814453125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09986605495214462,
      "learning_rate": 5e-05,
      "loss": 0.1217,
      "step": 24890
    },
    {
      "step": 24890,
      "train_flow_matching_loss": 0.0849609375,
      "train_physics_loss": 0.2522573173046112
    },
    {
      "step": 24890,
      "train_flow_matching_loss": 0.1708984375,
      "train_physics_loss": 0.0
    },
    {
      "step": 24890,
      "train_flow_matching_loss": 0.041259765625,
      "train_physics_loss": 0.21673527359962463
    },
    {
      "step": 24890,
      "train_flow_matching_loss": 0.060302734375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08436087518930435,
      "learning_rate": 5e-05,
      "loss": 0.1219,
      "step": 24900
    },
    {
      "step": 24900,
      "train_flow_matching_loss": 0.03564453125,
      "train_physics_loss": 0.17939493060112
    },
    {
      "step": 24900,
      "train_flow_matching_loss": 0.09716796875,
      "train_physics_loss": 0.0
    },
    {
      "step": 24900,
      "train_flow_matching_loss": 0.04931640625,
      "train_physics_loss": 0.21869046986103058
    },
    {
      "step": 24900,
      "train_flow_matching_loss": 0.1630859375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09750203788280487,
      "learning_rate": 5e-05,
      "loss": 0.1169,
      "step": 24910
    },
    {
      "step": 24910,
      "train_flow_matching_loss": 0.060302734375,
      "train_physics_loss": 0.1946716010570526
    },
    {
      "step": 24910,
      "train_flow_matching_loss": 0.09326171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 24910,
      "train_flow_matching_loss": 0.06884765625,
      "train_physics_loss": 0.2028137594461441
    },
    {
      "step": 24910,
      "train_flow_matching_loss": 0.12109375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09032470732927322,
      "learning_rate": 5e-05,
      "loss": 0.1164,
      "step": 24920
    },
    {
      "step": 24920,
      "train_flow_matching_loss": 0.049072265625,
      "train_physics_loss": 0.0
    },
    {
      "step": 24920,
      "train_flow_matching_loss": 0.0771484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 24920,
      "train_flow_matching_loss": 0.04296875,
      "train_physics_loss": 0.22570112347602844
    },
    {
      "step": 24920,
      "train_flow_matching_loss": 0.07080078125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.1005881279706955,
      "learning_rate": 5e-05,
      "loss": 0.1199,
      "step": 24930
    },
    {
      "step": 24930,
      "train_flow_matching_loss": 0.1044921875,
      "train_physics_loss": 0.1589096188545227
    },
    {
      "step": 24930,
      "train_flow_matching_loss": 0.1064453125,
      "train_physics_loss": 0.0
    },
    {
      "step": 24930,
      "train_flow_matching_loss": 0.0390625,
      "train_physics_loss": 0.20575104653835297
    },
    {
      "step": 24930,
      "train_flow_matching_loss": 0.0966796875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.13560861349105835,
      "learning_rate": 5e-05,
      "loss": 0.1249,
      "step": 24940
    },
    {
      "step": 24940,
      "train_flow_matching_loss": 0.04931640625,
      "train_physics_loss": 0.2089623212814331
    },
    {
      "step": 24940,
      "train_flow_matching_loss": 0.11865234375,
      "train_physics_loss": 0.0
    },
    {
      "step": 24940,
      "train_flow_matching_loss": 0.042724609375,
      "train_physics_loss": 0.21965382993221283
    },
    {
      "step": 24940,
      "train_flow_matching_loss": 0.07470703125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.07789713889360428,
      "learning_rate": 5e-05,
      "loss": 0.1179,
      "step": 24950
    },
    {
      "step": 24950,
      "train_flow_matching_loss": 0.043212890625,
      "train_physics_loss": 0.15939554572105408
    },
    {
      "step": 24950,
      "train_flow_matching_loss": 0.1982421875,
      "train_physics_loss": 0.0
    },
    {
      "step": 24950,
      "train_flow_matching_loss": 0.09619140625,
      "train_physics_loss": 0.292381227016449
    },
    {
      "step": 24950,
      "train_flow_matching_loss": 0.1865234375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.07382730394601822,
      "learning_rate": 5e-05,
      "loss": 0.1179,
      "step": 24960
    },
    {
      "step": 24960,
      "train_flow_matching_loss": 0.09326171875,
      "train_physics_loss": 0.19713842868804932
    },
    {
      "step": 24960,
      "train_flow_matching_loss": 0.12451171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 24960,
      "train_flow_matching_loss": 0.1640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 24960,
      "train_flow_matching_loss": 0.1337890625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.07879842817783356,
      "learning_rate": 5e-05,
      "loss": 0.1255,
      "step": 24970
    },
    {
      "step": 24970,
      "train_flow_matching_loss": 0.033447265625,
      "train_physics_loss": 0.15384860336780548
    },
    {
      "step": 24970,
      "train_flow_matching_loss": 0.140625,
      "train_physics_loss": 0.0
    },
    {
      "step": 24970,
      "train_flow_matching_loss": 0.12890625,
      "train_physics_loss": 0.0
    },
    {
      "step": 24970,
      "train_flow_matching_loss": 0.09521484375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08936160057783127,
      "learning_rate": 5e-05,
      "loss": 0.1179,
      "step": 24980
    },
    {
      "step": 24980,
      "train_flow_matching_loss": 0.056396484375,
      "train_physics_loss": 0.23418131470680237
    },
    {
      "step": 24980,
      "train_flow_matching_loss": 0.1767578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 24980,
      "train_flow_matching_loss": 0.1220703125,
      "train_physics_loss": 0.0
    },
    {
      "step": 24980,
      "train_flow_matching_loss": 0.04150390625,
      "train_physics_loss": 0.16833911836147308
    },
    {
      "grad_norm": 0.06410595029592514,
      "learning_rate": 5e-05,
      "loss": 0.1213,
      "step": 24990
    },
    {
      "step": 24990,
      "train_flow_matching_loss": 0.053955078125,
      "train_physics_loss": 0.2296494096517563
    },
    {
      "step": 24990,
      "train_flow_matching_loss": 0.12451171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 24990,
      "train_flow_matching_loss": 0.125,
      "train_physics_loss": 0.0
    },
    {
      "step": 24990,
      "train_flow_matching_loss": 0.07568359375,
      "train_physics_loss": 0.1968727856874466
    },
    {
      "grad_norm": 0.10864505171775818,
      "learning_rate": 5e-05,
      "loss": 0.12,
      "step": 25000
    },
    {
      "step": 25000,
      "train_flow_matching_loss": 0.1787109375,
      "train_physics_loss": 0.20005059242248535
    },
    {
      "step": 25000,
      "train_flow_matching_loss": 0.11083984375,
      "train_physics_loss": 0.0
    },
    {
      "step": 25000,
      "train_flow_matching_loss": 0.130859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 25000,
      "train_flow_matching_loss": 0.09130859375,
      "train_physics_loss": 0.22867617011070251
    },
    {
      "grad_norm": 0.09555885195732117,
      "learning_rate": 5e-05,
      "loss": 0.124,
      "step": 25010
    },
    {
      "step": 25010,
      "train_flow_matching_loss": 0.0673828125,
      "train_physics_loss": 0.25803086161613464
    },
    {
      "step": 25010,
      "train_flow_matching_loss": 0.10791015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 25010,
      "train_flow_matching_loss": 0.197265625,
      "train_physics_loss": 0.0
    },
    {
      "step": 25010,
      "train_flow_matching_loss": 0.07666015625,
      "train_physics_loss": 0.18818798661231995
    },
    {
      "grad_norm": 0.15669037401676178,
      "learning_rate": 5e-05,
      "loss": 0.1228,
      "step": 25020
    },
    {
      "step": 25020,
      "train_flow_matching_loss": 0.04296875,
      "train_physics_loss": 0.1990402787923813
    },
    {
      "step": 25020,
      "train_flow_matching_loss": 0.10498046875,
      "train_physics_loss": 0.0
    },
    {
      "step": 25020,
      "train_flow_matching_loss": 0.2099609375,
      "train_physics_loss": 0.0
    },
    {
      "step": 25020,
      "train_flow_matching_loss": 0.035400390625,
      "train_physics_loss": 0.17104099690914154
    },
    {
      "grad_norm": 0.09952245652675629,
      "learning_rate": 5e-05,
      "loss": 0.1161,
      "step": 25030
    },
    {
      "step": 25030,
      "train_flow_matching_loss": 0.10498046875,
      "train_physics_loss": 0.19112980365753174
    },
    {
      "step": 25030,
      "train_flow_matching_loss": 0.08642578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 25030,
      "train_flow_matching_loss": 0.08544921875,
      "train_physics_loss": 0.0
    },
    {
      "step": 25030,
      "train_flow_matching_loss": 0.09814453125,
      "train_physics_loss": 0.19915400445461273
    },
    {
      "grad_norm": 0.35452109575271606,
      "learning_rate": 5e-05,
      "loss": 0.1152,
      "step": 25040
    },
    {
      "step": 25040,
      "train_flow_matching_loss": 0.033447265625,
      "train_physics_loss": 0.1718916893005371
    },
    {
      "step": 25040,
      "train_flow_matching_loss": 0.138671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 25040,
      "train_flow_matching_loss": 0.1142578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 25040,
      "train_flow_matching_loss": 0.038330078125,
      "train_physics_loss": 0.17823253571987152
    },
    {
      "grad_norm": 0.36020979285240173,
      "learning_rate": 5e-05,
      "loss": 0.119,
      "step": 25050
    },
    {
      "step": 25050,
      "train_flow_matching_loss": 0.06689453125,
      "train_physics_loss": 0.2288510650396347
    },
    {
      "step": 25050,
      "train_flow_matching_loss": 0.1396484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 25050,
      "train_flow_matching_loss": 0.11572265625,
      "train_physics_loss": 0.0
    },
    {
      "step": 25050,
      "train_flow_matching_loss": 0.044921875,
      "train_physics_loss": 0.18539147078990936
    },
    {
      "grad_norm": 0.0956154465675354,
      "learning_rate": 5e-05,
      "loss": 0.1206,
      "step": 25060
    },
    {
      "step": 25060,
      "train_flow_matching_loss": 0.06298828125,
      "train_physics_loss": 0.1985420286655426
    },
    {
      "step": 25060,
      "train_flow_matching_loss": 0.1298828125,
      "train_physics_loss": 0.0
    },
    {
      "step": 25060,
      "train_flow_matching_loss": 0.0986328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 25060,
      "train_flow_matching_loss": 0.0267333984375,
      "train_physics_loss": 0.1961604207754135
    },
    {
      "grad_norm": 0.10964027792215347,
      "learning_rate": 5e-05,
      "loss": 0.1172,
      "step": 25070
    },
    {
      "step": 25070,
      "train_flow_matching_loss": 0.1650390625,
      "train_physics_loss": 0.0
    },
    {
      "step": 25070,
      "train_flow_matching_loss": 0.1416015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 25070,
      "train_flow_matching_loss": 0.1484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 25070,
      "train_flow_matching_loss": 0.0947265625,
      "train_physics_loss": 0.2819494307041168
    },
    {
      "grad_norm": 0.10419673472642899,
      "learning_rate": 5e-05,
      "loss": 0.1182,
      "step": 25080
    },
    {
      "step": 25080,
      "train_flow_matching_loss": 0.1552734375,
      "train_physics_loss": 0.0
    },
    {
      "step": 25080,
      "train_flow_matching_loss": 0.205078125,
      "train_physics_loss": 0.0
    },
    {
      "step": 25080,
      "train_flow_matching_loss": 0.302734375,
      "train_physics_loss": 0.0
    },
    {
      "step": 25080,
      "train_flow_matching_loss": 0.040283203125,
      "train_physics_loss": 0.1848539412021637
    },
    {
      "grad_norm": 0.10454247146844864,
      "learning_rate": 5e-05,
      "loss": 0.1179,
      "step": 25090
    },
    {
      "step": 25090,
      "train_flow_matching_loss": 0.10107421875,
      "train_physics_loss": 0.0
    },
    {
      "step": 25090,
      "train_flow_matching_loss": 0.171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 25090,
      "train_flow_matching_loss": 0.173828125,
      "train_physics_loss": 0.0
    },
    {
      "step": 25090,
      "train_flow_matching_loss": 0.044677734375,
      "train_physics_loss": 0.22509536147117615
    },
    {
      "grad_norm": 0.12033101916313171,
      "learning_rate": 5e-05,
      "loss": 0.1224,
      "step": 25100
    },
    {
      "step": 25100,
      "train_flow_matching_loss": 0.10302734375,
      "train_physics_loss": 0.0
    },
    {
      "step": 25100,
      "train_flow_matching_loss": 0.1552734375,
      "train_physics_loss": 0.0
    },
    {
      "step": 25100,
      "train_flow_matching_loss": 0.10302734375,
      "train_physics_loss": 0.0
    },
    {
      "step": 25100,
      "train_flow_matching_loss": 0.038330078125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09428933262825012,
      "learning_rate": 5e-05,
      "loss": 0.1256,
      "step": 25110
    },
    {
      "step": 25110,
      "train_flow_matching_loss": 0.08447265625,
      "train_physics_loss": 0.0
    },
    {
      "step": 25110,
      "train_flow_matching_loss": 0.11865234375,
      "train_physics_loss": 0.0
    },
    {
      "step": 25110,
      "train_flow_matching_loss": 0.248046875,
      "train_physics_loss": 0.0
    },
    {
      "step": 25110,
      "train_flow_matching_loss": 0.0810546875,
      "train_physics_loss": 0.27412015199661255
    },
    {
      "grad_norm": 0.10801492631435394,
      "learning_rate": 5e-05,
      "loss": 0.1219,
      "step": 25120
    },
    {
      "step": 25120,
      "train_flow_matching_loss": 0.09521484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 25120,
      "train_flow_matching_loss": 0.1298828125,
      "train_physics_loss": 0.0
    },
    {
      "step": 25120,
      "train_flow_matching_loss": 0.09814453125,
      "train_physics_loss": 0.0
    },
    {
      "step": 25120,
      "train_flow_matching_loss": 0.035400390625,
      "train_physics_loss": 0.16227024793624878
    },
    {
      "grad_norm": 0.07708173245191574,
      "learning_rate": 5e-05,
      "loss": 0.1168,
      "step": 25130
    },
    {
      "step": 25130,
      "train_flow_matching_loss": 0.146484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 25130,
      "train_flow_matching_loss": 0.189453125,
      "train_physics_loss": 0.0
    },
    {
      "step": 25130,
      "train_flow_matching_loss": 0.10888671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 25130,
      "train_flow_matching_loss": 0.03173828125,
      "train_physics_loss": 0.1688195914030075
    },
    {
      "grad_norm": 0.06958217173814774,
      "learning_rate": 5e-05,
      "loss": 0.1175,
      "step": 25140
    },
    {
      "step": 25140,
      "train_flow_matching_loss": 0.11328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 25140,
      "train_flow_matching_loss": 0.1328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 25140,
      "train_flow_matching_loss": 0.08251953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 25140,
      "train_flow_matching_loss": 0.05029296875,
      "train_physics_loss": 0.18972276151180267
    },
    {
      "grad_norm": 0.1054825633764267,
      "learning_rate": 5e-05,
      "loss": 0.1245,
      "step": 25150
    },
    {
      "step": 25150,
      "train_flow_matching_loss": 0.11962890625,
      "train_physics_loss": 0.0
    },
    {
      "step": 25150,
      "train_flow_matching_loss": 0.1630859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 25150,
      "train_flow_matching_loss": 0.1572265625,
      "train_physics_loss": 0.0
    },
    {
      "step": 25150,
      "train_flow_matching_loss": 0.036865234375,
      "train_physics_loss": 0.22929337620735168
    },
    {
      "grad_norm": 0.08093603700399399,
      "learning_rate": 5e-05,
      "loss": 0.12,
      "step": 25160
    },
    {
      "step": 25160,
      "train_flow_matching_loss": 0.1689453125,
      "train_physics_loss": 0.0
    },
    {
      "step": 25160,
      "train_flow_matching_loss": 0.1357421875,
      "train_physics_loss": 0.0
    },
    {
      "step": 25160,
      "train_flow_matching_loss": 0.1494140625,
      "train_physics_loss": 0.0
    },
    {
      "step": 25160,
      "train_flow_matching_loss": 0.041259765625,
      "train_physics_loss": 0.19470728933811188
    },
    {
      "grad_norm": 0.09998655319213867,
      "learning_rate": 5e-05,
      "loss": 0.1212,
      "step": 25170
    },
    {
      "step": 25170,
      "train_flow_matching_loss": 0.1123046875,
      "train_physics_loss": 0.0
    },
    {
      "step": 25170,
      "train_flow_matching_loss": 0.08837890625,
      "train_physics_loss": 0.0
    },
    {
      "step": 25170,
      "train_flow_matching_loss": 0.146484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 25170,
      "train_flow_matching_loss": 0.05810546875,
      "train_physics_loss": 0.20088428258895874
    },
    {
      "grad_norm": 0.07871191203594208,
      "learning_rate": 5e-05,
      "loss": 0.1206,
      "step": 25180
    },
    {
      "step": 25180,
      "train_flow_matching_loss": 0.138671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 25180,
      "train_flow_matching_loss": 0.1767578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 25180,
      "train_flow_matching_loss": 0.1708984375,
      "train_physics_loss": 0.0
    },
    {
      "step": 25180,
      "train_flow_matching_loss": 0.0224609375,
      "train_physics_loss": 0.23111632466316223
    },
    {
      "grad_norm": 0.0833851620554924,
      "learning_rate": 5e-05,
      "loss": 0.117,
      "step": 25190
    },
    {
      "step": 25190,
      "train_flow_matching_loss": 0.10595703125,
      "train_physics_loss": 0.0
    },
    {
      "step": 25190,
      "train_flow_matching_loss": 0.0830078125,
      "train_physics_loss": 0.0
    },
    {
      "step": 25190,
      "train_flow_matching_loss": 0.2001953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 25190,
      "train_flow_matching_loss": 0.0211181640625,
      "train_physics_loss": 0.16208505630493164
    },
    {
      "grad_norm": 0.07574760168790817,
      "learning_rate": 5e-05,
      "loss": 0.12,
      "step": 25200
    },
    {
      "step": 25200,
      "train_flow_matching_loss": 0.1279296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 25200,
      "train_flow_matching_loss": 0.09375,
      "train_physics_loss": 0.0
    },
    {
      "step": 25200,
      "train_flow_matching_loss": 0.13671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 25200,
      "train_flow_matching_loss": 0.048095703125,
      "train_physics_loss": 0.19770774245262146
    },
    {
      "grad_norm": 0.08906219154596329,
      "learning_rate": 5e-05,
      "loss": 0.1154,
      "step": 25210
    },
    {
      "step": 25210,
      "train_flow_matching_loss": 0.08203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 25210,
      "train_flow_matching_loss": 0.09619140625,
      "train_physics_loss": 0.0
    },
    {
      "step": 25210,
      "train_flow_matching_loss": 0.1298828125,
      "train_physics_loss": 0.0
    },
    {
      "step": 25210,
      "train_flow_matching_loss": 0.06201171875,
      "train_physics_loss": 0.2375108301639557
    },
    {
      "grad_norm": 0.10115779936313629,
      "learning_rate": 5e-05,
      "loss": 0.1205,
      "step": 25220
    },
    {
      "step": 25220,
      "train_flow_matching_loss": 0.1494140625,
      "train_physics_loss": 0.0
    },
    {
      "step": 25220,
      "train_flow_matching_loss": 0.1103515625,
      "train_physics_loss": 0.0
    },
    {
      "step": 25220,
      "train_flow_matching_loss": 0.197265625,
      "train_physics_loss": 0.0
    },
    {
      "step": 25220,
      "train_flow_matching_loss": 0.03369140625,
      "train_physics_loss": 0.20001326501369476
    },
    {
      "grad_norm": 0.09188804030418396,
      "learning_rate": 5e-05,
      "loss": 0.1222,
      "step": 25230
    },
    {
      "step": 25230,
      "train_flow_matching_loss": 0.1416015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 25230,
      "train_flow_matching_loss": 0.169921875,
      "train_physics_loss": 0.0
    },
    {
      "step": 25230,
      "train_flow_matching_loss": 0.1787109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 25230,
      "train_flow_matching_loss": 0.11376953125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09923204779624939,
      "learning_rate": 5e-05,
      "loss": 0.1167,
      "step": 25240
    },
    {
      "step": 25240,
      "train_flow_matching_loss": 0.138671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 25240,
      "train_flow_matching_loss": 0.2021484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 25240,
      "train_flow_matching_loss": 0.0869140625,
      "train_physics_loss": 0.0
    },
    {
      "step": 25240,
      "train_flow_matching_loss": 0.162109375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.18787094950675964,
      "learning_rate": 5e-05,
      "loss": 0.124,
      "step": 25250
    },
    {
      "step": 25250,
      "train_flow_matching_loss": 0.07275390625,
      "train_physics_loss": 0.0
    },
    {
      "step": 25250,
      "train_flow_matching_loss": 0.146484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 25250,
      "train_flow_matching_loss": 0.181640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 25250,
      "train_flow_matching_loss": 0.1591796875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09809219837188721,
      "learning_rate": 5e-05,
      "loss": 0.1197,
      "step": 25260
    },
    {
      "step": 25260,
      "train_flow_matching_loss": 0.1103515625,
      "train_physics_loss": 0.0
    },
    {
      "step": 25260,
      "train_flow_matching_loss": 0.181640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 25260,
      "train_flow_matching_loss": 0.11669921875,
      "train_physics_loss": 0.0
    },
    {
      "step": 25260,
      "train_flow_matching_loss": 0.09375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11526166647672653,
      "learning_rate": 5e-05,
      "loss": 0.1251,
      "step": 25270
    },
    {
      "step": 25270,
      "train_flow_matching_loss": 0.08642578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 25270,
      "train_flow_matching_loss": 0.10546875,
      "train_physics_loss": 0.0
    },
    {
      "step": 25270,
      "train_flow_matching_loss": 0.1103515625,
      "train_physics_loss": 0.0
    },
    {
      "step": 25270,
      "train_flow_matching_loss": 0.1005859375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.07720720022916794,
      "learning_rate": 5e-05,
      "loss": 0.119,
      "step": 25280
    },
    {
      "step": 25280,
      "train_flow_matching_loss": 0.11767578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 25280,
      "train_flow_matching_loss": 0.06494140625,
      "train_physics_loss": 0.0
    },
    {
      "step": 25280,
      "train_flow_matching_loss": 0.11767578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 25280,
      "train_flow_matching_loss": 0.1298828125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.1225542202591896,
      "learning_rate": 5e-05,
      "loss": 0.1179,
      "step": 25290
    },
    {
      "step": 25290,
      "train_flow_matching_loss": 0.10009765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 25290,
      "train_flow_matching_loss": 0.11572265625,
      "train_physics_loss": 0.0
    },
    {
      "step": 25290,
      "train_flow_matching_loss": 0.1591796875,
      "train_physics_loss": 0.0
    },
    {
      "step": 25290,
      "train_flow_matching_loss": 0.06103515625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10257872939109802,
      "learning_rate": 5e-05,
      "loss": 0.1194,
      "step": 25300
    },
    {
      "step": 25300,
      "train_flow_matching_loss": 0.1376953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 25300,
      "train_flow_matching_loss": 0.173828125,
      "train_physics_loss": 0.0
    },
    {
      "step": 25300,
      "train_flow_matching_loss": 0.1552734375,
      "train_physics_loss": 0.0
    },
    {
      "step": 25300,
      "train_flow_matching_loss": 0.146484375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10781602561473846,
      "learning_rate": 5e-05,
      "loss": 0.121,
      "step": 25310
    },
    {
      "step": 25310,
      "train_flow_matching_loss": 0.150390625,
      "train_physics_loss": 0.0
    },
    {
      "step": 25310,
      "train_flow_matching_loss": 0.1318359375,
      "train_physics_loss": 0.0
    },
    {
      "step": 25310,
      "train_flow_matching_loss": 0.1591796875,
      "train_physics_loss": 0.0
    },
    {
      "step": 25310,
      "train_flow_matching_loss": 0.10986328125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.0980265736579895,
      "learning_rate": 5e-05,
      "loss": 0.1204,
      "step": 25320
    },
    {
      "step": 25320,
      "train_flow_matching_loss": 0.1416015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 25320,
      "train_flow_matching_loss": 0.1318359375,
      "train_physics_loss": 0.0
    },
    {
      "step": 25320,
      "train_flow_matching_loss": 0.0986328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 25320,
      "train_flow_matching_loss": 0.0947265625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.1145964041352272,
      "learning_rate": 5e-05,
      "loss": 0.1222,
      "step": 25330
    },
    {
      "step": 25330,
      "train_flow_matching_loss": 0.12353515625,
      "train_physics_loss": 0.0
    },
    {
      "step": 25330,
      "train_flow_matching_loss": 0.1708984375,
      "train_physics_loss": 0.0
    },
    {
      "step": 25330,
      "train_flow_matching_loss": 0.1357421875,
      "train_physics_loss": 0.0
    },
    {
      "step": 25330,
      "train_flow_matching_loss": 0.1279296875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10524912923574448,
      "learning_rate": 5e-05,
      "loss": 0.1177,
      "step": 25340
    },
    {
      "step": 25340,
      "train_flow_matching_loss": 0.150390625,
      "train_physics_loss": 0.0
    },
    {
      "step": 25340,
      "train_flow_matching_loss": 0.1455078125,
      "train_physics_loss": 0.0
    },
    {
      "step": 25340,
      "train_flow_matching_loss": 0.10986328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 25340,
      "train_flow_matching_loss": 0.1279296875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09386015683412552,
      "learning_rate": 5e-05,
      "loss": 0.1211,
      "step": 25350
    },
    {
      "step": 25350,
      "train_flow_matching_loss": 0.154296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 25350,
      "train_flow_matching_loss": 0.1474609375,
      "train_physics_loss": 0.0
    },
    {
      "step": 25350,
      "train_flow_matching_loss": 0.0869140625,
      "train_physics_loss": 0.0
    },
    {
      "step": 25350,
      "train_flow_matching_loss": 0.0771484375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09770605713129044,
      "learning_rate": 5e-05,
      "loss": 0.12,
      "step": 25360
    },
    {
      "step": 25360,
      "train_flow_matching_loss": 0.169921875,
      "train_physics_loss": 0.0
    },
    {
      "step": 25360,
      "train_flow_matching_loss": 0.1572265625,
      "train_physics_loss": 0.0
    },
    {
      "step": 25360,
      "train_flow_matching_loss": 0.10986328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 25360,
      "train_flow_matching_loss": 0.1376953125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.07513302564620972,
      "learning_rate": 5e-05,
      "loss": 0.1284,
      "step": 25370
    },
    {
      "step": 25370,
      "train_flow_matching_loss": 0.10791015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 25370,
      "train_flow_matching_loss": 0.10009765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 25370,
      "train_flow_matching_loss": 0.1416015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 25370,
      "train_flow_matching_loss": 0.051025390625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.1092667505145073,
      "learning_rate": 5e-05,
      "loss": 0.1215,
      "step": 25380
    },
    {
      "step": 25380,
      "train_flow_matching_loss": 0.11669921875,
      "train_physics_loss": 0.0
    },
    {
      "step": 25380,
      "train_flow_matching_loss": 0.126953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 25380,
      "train_flow_matching_loss": 0.146484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 25380,
      "train_flow_matching_loss": 0.048095703125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09505924582481384,
      "learning_rate": 5e-05,
      "loss": 0.1229,
      "step": 25390
    },
    {
      "step": 25390,
      "train_flow_matching_loss": 0.130859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 25390,
      "train_flow_matching_loss": 0.1298828125,
      "train_physics_loss": 0.0
    },
    {
      "step": 25390,
      "train_flow_matching_loss": 0.1572265625,
      "train_physics_loss": 0.0
    },
    {
      "step": 25390,
      "train_flow_matching_loss": 0.09912109375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.3861309885978699,
      "learning_rate": 5e-05,
      "loss": 0.1192,
      "step": 25400
    },
    {
      "step": 25400,
      "train_flow_matching_loss": 0.1650390625,
      "train_physics_loss": 0.0
    },
    {
      "step": 25400,
      "train_flow_matching_loss": 0.11328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 25400,
      "train_flow_matching_loss": 0.1181640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 25400,
      "train_flow_matching_loss": 0.10498046875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08026393502950668,
      "learning_rate": 5e-05,
      "loss": 0.1221,
      "step": 25410
    },
    {
      "step": 25410,
      "train_flow_matching_loss": 0.1220703125,
      "train_physics_loss": 0.0
    },
    {
      "step": 25410,
      "train_flow_matching_loss": 0.06884765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 25410,
      "train_flow_matching_loss": 0.1416015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 25410,
      "train_flow_matching_loss": 0.09912109375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08197057992219925,
      "learning_rate": 5e-05,
      "loss": 0.126,
      "step": 25420
    },
    {
      "step": 25420,
      "train_flow_matching_loss": 0.1572265625,
      "train_physics_loss": 0.0
    },
    {
      "step": 25420,
      "train_flow_matching_loss": 0.1728515625,
      "train_physics_loss": 0.0
    },
    {
      "step": 25420,
      "train_flow_matching_loss": 0.1015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 25420,
      "train_flow_matching_loss": 0.10986328125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.1120668351650238,
      "learning_rate": 5e-05,
      "loss": 0.1233,
      "step": 25430
    },
    {
      "step": 25430,
      "train_flow_matching_loss": 0.212890625,
      "train_physics_loss": 0.0
    },
    {
      "step": 25430,
      "train_flow_matching_loss": 0.109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 25430,
      "train_flow_matching_loss": 0.0927734375,
      "train_physics_loss": 0.0
    },
    {
      "step": 25430,
      "train_flow_matching_loss": 0.138671875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.1226634830236435,
      "learning_rate": 5e-05,
      "loss": 0.1292,
      "step": 25440
    },
    {
      "step": 25440,
      "train_flow_matching_loss": 0.150390625,
      "train_physics_loss": 0.0
    },
    {
      "step": 25440,
      "train_flow_matching_loss": 0.11474609375,
      "train_physics_loss": 0.0
    },
    {
      "step": 25440,
      "train_flow_matching_loss": 0.126953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 25440,
      "train_flow_matching_loss": 0.1689453125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.07229195535182953,
      "learning_rate": 5e-05,
      "loss": 0.1238,
      "step": 25450
    },
    {
      "step": 25450,
      "train_flow_matching_loss": 0.173828125,
      "train_physics_loss": 0.0
    },
    {
      "step": 25450,
      "train_flow_matching_loss": 0.09765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 25450,
      "train_flow_matching_loss": 0.1220703125,
      "train_physics_loss": 0.0
    },
    {
      "step": 25450,
      "train_flow_matching_loss": 0.181640625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.0824616402387619,
      "learning_rate": 5e-05,
      "loss": 0.1282,
      "step": 25460
    },
    {
      "step": 25460,
      "train_flow_matching_loss": 0.1865234375,
      "train_physics_loss": 0.0
    },
    {
      "step": 25460,
      "train_flow_matching_loss": 0.1416015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 25460,
      "train_flow_matching_loss": 0.138671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 25460,
      "train_flow_matching_loss": 0.06982421875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10537533462047577,
      "learning_rate": 5e-05,
      "loss": 0.1213,
      "step": 25470
    },
    {
      "step": 25470,
      "train_flow_matching_loss": 0.1787109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 25470,
      "train_flow_matching_loss": 0.125,
      "train_physics_loss": 0.0
    },
    {
      "step": 25470,
      "train_flow_matching_loss": 0.126953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 25470,
      "train_flow_matching_loss": 0.1396484375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.12273062765598297,
      "learning_rate": 5e-05,
      "loss": 0.1206,
      "step": 25480
    },
    {
      "step": 25480,
      "train_flow_matching_loss": 0.1767578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 25480,
      "train_flow_matching_loss": 0.1416015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 25480,
      "train_flow_matching_loss": 0.154296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 25480,
      "train_flow_matching_loss": 0.171875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.13306021690368652,
      "learning_rate": 5e-05,
      "loss": 0.1219,
      "step": 25490
    },
    {
      "step": 25490,
      "train_flow_matching_loss": 0.11572265625,
      "train_physics_loss": 0.0
    },
    {
      "step": 25490,
      "train_flow_matching_loss": 0.12255859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 25490,
      "train_flow_matching_loss": 0.0751953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 25490,
      "train_flow_matching_loss": 0.2080078125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.1827205866575241,
      "learning_rate": 5e-05,
      "loss": 0.1216,
      "step": 25500
    },
    {
      "step": 25500,
      "train_flow_matching_loss": 0.189453125,
      "train_physics_loss": 0.0
    },
    {
      "step": 25500,
      "train_flow_matching_loss": 0.08251953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 25500,
      "train_flow_matching_loss": 0.2314453125,
      "train_physics_loss": 0.0
    },
    {
      "step": 25500,
      "train_flow_matching_loss": 0.1826171875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10302425175905228,
      "learning_rate": 5e-05,
      "loss": 0.126,
      "step": 25510
    },
    {
      "step": 25510,
      "train_flow_matching_loss": 0.15625,
      "train_physics_loss": 0.0
    },
    {
      "step": 25510,
      "train_flow_matching_loss": 0.1162109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 25510,
      "train_flow_matching_loss": 0.1494140625,
      "train_physics_loss": 0.0
    },
    {
      "step": 25510,
      "train_flow_matching_loss": 0.1435546875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10189338773488998,
      "learning_rate": 5e-05,
      "loss": 0.1245,
      "step": 25520
    },
    {
      "step": 25520,
      "train_flow_matching_loss": 0.1259765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 25520,
      "train_flow_matching_loss": 0.169921875,
      "train_physics_loss": 0.0
    },
    {
      "step": 25520,
      "train_flow_matching_loss": 0.1591796875,
      "train_physics_loss": 0.0
    },
    {
      "step": 25520,
      "train_flow_matching_loss": 0.1025390625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.0876142829656601,
      "learning_rate": 5e-05,
      "loss": 0.1311,
      "step": 25530
    },
    {
      "step": 25530,
      "train_flow_matching_loss": 0.1015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 25530,
      "train_flow_matching_loss": 0.1357421875,
      "train_physics_loss": 0.0
    },
    {
      "step": 25530,
      "train_flow_matching_loss": 0.1474609375,
      "train_physics_loss": 0.0
    },
    {
      "step": 25530,
      "train_flow_matching_loss": 0.1806640625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.07597032189369202,
      "learning_rate": 5e-05,
      "loss": 0.1259,
      "step": 25540
    },
    {
      "step": 25540,
      "train_flow_matching_loss": 0.158203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 25540,
      "train_flow_matching_loss": 0.142578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 25540,
      "train_flow_matching_loss": 0.1787109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 25540,
      "train_flow_matching_loss": 0.1416015625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09291023761034012,
      "learning_rate": 5e-05,
      "loss": 0.1213,
      "step": 25550
    },
    {
      "step": 25550,
      "train_flow_matching_loss": 0.2216796875,
      "train_physics_loss": 0.0
    },
    {
      "step": 25550,
      "train_flow_matching_loss": 0.10205078125,
      "train_physics_loss": 0.0
    },
    {
      "step": 25550,
      "train_flow_matching_loss": 0.169921875,
      "train_physics_loss": 0.0
    },
    {
      "step": 25550,
      "train_flow_matching_loss": 0.1640625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.1117812916636467,
      "learning_rate": 5e-05,
      "loss": 0.1242,
      "step": 25560
    },
    {
      "step": 25560,
      "train_flow_matching_loss": 0.09326171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 25560,
      "train_flow_matching_loss": 0.125,
      "train_physics_loss": 0.0
    },
    {
      "step": 25560,
      "train_flow_matching_loss": 0.1435546875,
      "train_physics_loss": 0.0
    },
    {
      "step": 25560,
      "train_flow_matching_loss": 0.1298828125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.1073426753282547,
      "learning_rate": 5e-05,
      "loss": 0.1249,
      "step": 25570
    },
    {
      "step": 25570,
      "train_flow_matching_loss": 0.1015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 25570,
      "train_flow_matching_loss": 0.09716796875,
      "train_physics_loss": 0.0
    },
    {
      "step": 25570,
      "train_flow_matching_loss": 0.1025390625,
      "train_physics_loss": 0.0
    },
    {
      "step": 25570,
      "train_flow_matching_loss": 0.1396484375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08690439909696579,
      "learning_rate": 5e-05,
      "loss": 0.1211,
      "step": 25580
    },
    {
      "step": 25580,
      "train_flow_matching_loss": 0.1728515625,
      "train_physics_loss": 0.0
    },
    {
      "step": 25580,
      "train_flow_matching_loss": 0.1396484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 25580,
      "train_flow_matching_loss": 0.11865234375,
      "train_physics_loss": 0.0
    },
    {
      "step": 25580,
      "train_flow_matching_loss": 0.10546875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.15928596258163452,
      "learning_rate": 5e-05,
      "loss": 0.1212,
      "step": 25590
    },
    {
      "step": 25590,
      "train_flow_matching_loss": 0.078125,
      "train_physics_loss": 0.0
    },
    {
      "step": 25590,
      "train_flow_matching_loss": 0.10791015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 25590,
      "train_flow_matching_loss": 0.1611328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 25590,
      "train_flow_matching_loss": 0.0732421875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.1175299733877182,
      "learning_rate": 5e-05,
      "loss": 0.1195,
      "step": 25600
    },
    {
      "step": 25600,
      "train_flow_matching_loss": 0.1572265625,
      "train_physics_loss": 0.0
    },
    {
      "step": 25600,
      "train_flow_matching_loss": 0.0859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 25600,
      "train_flow_matching_loss": 0.08349609375,
      "train_physics_loss": 0.0
    },
    {
      "step": 25600,
      "train_flow_matching_loss": 0.1337890625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.07032414525747299,
      "learning_rate": 5e-05,
      "loss": 0.1219,
      "step": 25610
    },
    {
      "step": 25610,
      "train_flow_matching_loss": 0.1259765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 25610,
      "train_flow_matching_loss": 0.1689453125,
      "train_physics_loss": 0.0
    },
    {
      "step": 25610,
      "train_flow_matching_loss": 0.0625,
      "train_physics_loss": 0.0
    },
    {
      "step": 25610,
      "train_flow_matching_loss": 0.181640625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10349921882152557,
      "learning_rate": 5e-05,
      "loss": 0.1194,
      "step": 25620
    },
    {
      "step": 25620,
      "train_flow_matching_loss": 0.1064453125,
      "train_physics_loss": 0.0
    },
    {
      "step": 25620,
      "train_flow_matching_loss": 0.0947265625,
      "train_physics_loss": 0.0
    },
    {
      "step": 25620,
      "train_flow_matching_loss": 0.2275390625,
      "train_physics_loss": 0.0
    },
    {
      "step": 25620,
      "train_flow_matching_loss": 0.11474609375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.14011691510677338,
      "learning_rate": 5e-05,
      "loss": 0.1224,
      "step": 25630
    },
    {
      "step": 25630,
      "train_flow_matching_loss": 0.1064453125,
      "train_physics_loss": 0.0
    },
    {
      "step": 25630,
      "train_flow_matching_loss": 0.1298828125,
      "train_physics_loss": 0.0
    },
    {
      "step": 25630,
      "train_flow_matching_loss": 0.1650390625,
      "train_physics_loss": 0.0
    },
    {
      "step": 25630,
      "train_flow_matching_loss": 0.1689453125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09048271924257278,
      "learning_rate": 5e-05,
      "loss": 0.1196,
      "step": 25640
    },
    {
      "step": 25640,
      "train_flow_matching_loss": 0.1357421875,
      "train_physics_loss": 0.0
    },
    {
      "step": 25640,
      "train_flow_matching_loss": 0.140625,
      "train_physics_loss": 0.0
    },
    {
      "step": 25640,
      "train_flow_matching_loss": 0.10693359375,
      "train_physics_loss": 0.0
    },
    {
      "step": 25640,
      "train_flow_matching_loss": 0.173828125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09173958748579025,
      "learning_rate": 5e-05,
      "loss": 0.1231,
      "step": 25650
    },
    {
      "step": 25650,
      "train_flow_matching_loss": 0.1279296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 25650,
      "train_flow_matching_loss": 0.06005859375,
      "train_physics_loss": 0.23430173099040985
    },
    {
      "step": 25650,
      "train_flow_matching_loss": 0.08544921875,
      "train_physics_loss": 0.0
    },
    {
      "step": 25650,
      "train_flow_matching_loss": 0.1416015625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09822151809930801,
      "learning_rate": 5e-05,
      "loss": 0.1221,
      "step": 25660
    },
    {
      "step": 25660,
      "train_flow_matching_loss": 0.1455078125,
      "train_physics_loss": 0.0
    },
    {
      "step": 25660,
      "train_flow_matching_loss": 0.05615234375,
      "train_physics_loss": 0.23508033156394958
    },
    {
      "step": 25660,
      "train_flow_matching_loss": 0.1484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 25660,
      "train_flow_matching_loss": 0.08154296875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10079916566610336,
      "learning_rate": 5e-05,
      "loss": 0.1174,
      "step": 25670
    },
    {
      "step": 25670,
      "train_flow_matching_loss": 0.09375,
      "train_physics_loss": 0.0
    },
    {
      "step": 25670,
      "train_flow_matching_loss": 0.02734375,
      "train_physics_loss": 0.1886637657880783
    },
    {
      "step": 25670,
      "train_flow_matching_loss": 0.08203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 25670,
      "train_flow_matching_loss": 0.15625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.13163809478282928,
      "learning_rate": 5e-05,
      "loss": 0.1205,
      "step": 25680
    },
    {
      "step": 25680,
      "train_flow_matching_loss": 0.1123046875,
      "train_physics_loss": 0.0
    },
    {
      "step": 25680,
      "train_flow_matching_loss": 0.060546875,
      "train_physics_loss": 0.2738690972328186
    },
    {
      "step": 25680,
      "train_flow_matching_loss": 0.08251953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 25680,
      "train_flow_matching_loss": 0.10009765625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.13312752544879913,
      "learning_rate": 5e-05,
      "loss": 0.1218,
      "step": 25690
    },
    {
      "step": 25690,
      "train_flow_matching_loss": 0.1982421875,
      "train_physics_loss": 0.0
    },
    {
      "step": 25690,
      "train_flow_matching_loss": 0.05810546875,
      "train_physics_loss": 0.2483612447977066
    },
    {
      "step": 25690,
      "train_flow_matching_loss": 0.154296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 25690,
      "train_flow_matching_loss": 0.1337890625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11307680606842041,
      "learning_rate": 5e-05,
      "loss": 0.1249,
      "step": 25700
    },
    {
      "step": 25700,
      "train_flow_matching_loss": 0.07568359375,
      "train_physics_loss": 0.0
    },
    {
      "step": 25700,
      "train_flow_matching_loss": 0.052978515625,
      "train_physics_loss": 0.23344635963439941
    },
    {
      "step": 25700,
      "train_flow_matching_loss": 0.150390625,
      "train_physics_loss": 0.0
    },
    {
      "step": 25700,
      "train_flow_matching_loss": 0.0751953125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09903482347726822,
      "learning_rate": 5e-05,
      "loss": 0.1177,
      "step": 25710
    },
    {
      "step": 25710,
      "train_flow_matching_loss": 0.1484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 25710,
      "train_flow_matching_loss": 0.042236328125,
      "train_physics_loss": 0.2501181662082672
    },
    {
      "step": 25710,
      "train_flow_matching_loss": 0.11865234375,
      "train_physics_loss": 0.0
    },
    {
      "step": 25710,
      "train_flow_matching_loss": 0.146484375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.0982675775885582,
      "learning_rate": 5e-05,
      "loss": 0.1231,
      "step": 25720
    },
    {
      "step": 25720,
      "train_flow_matching_loss": 0.09619140625,
      "train_physics_loss": 0.0
    },
    {
      "step": 25720,
      "train_flow_matching_loss": 0.054443359375,
      "train_physics_loss": 0.18633407354354858
    },
    {
      "step": 25720,
      "train_flow_matching_loss": 0.1650390625,
      "train_physics_loss": 0.0
    },
    {
      "step": 25720,
      "train_flow_matching_loss": 0.09326171875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09965161979198456,
      "learning_rate": 5e-05,
      "loss": 0.1225,
      "step": 25730
    },
    {
      "step": 25730,
      "train_flow_matching_loss": 0.171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 25730,
      "train_flow_matching_loss": 0.0546875,
      "train_physics_loss": 0.20271363854408264
    },
    {
      "step": 25730,
      "train_flow_matching_loss": 0.2294921875,
      "train_physics_loss": 0.0
    },
    {
      "step": 25730,
      "train_flow_matching_loss": 0.08251953125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.07576847821474075,
      "learning_rate": 5e-05,
      "loss": 0.1173,
      "step": 25740
    },
    {
      "step": 25740,
      "train_flow_matching_loss": 0.1416015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 25740,
      "train_flow_matching_loss": 0.045654296875,
      "train_physics_loss": 0.232966810464859
    },
    {
      "step": 25740,
      "train_flow_matching_loss": 0.1513671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 25740,
      "train_flow_matching_loss": 0.08349609375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10498001426458359,
      "learning_rate": 5e-05,
      "loss": 0.1133,
      "step": 25750
    },
    {
      "step": 25750,
      "train_flow_matching_loss": 0.11767578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 25750,
      "train_flow_matching_loss": 0.021240234375,
      "train_physics_loss": 0.15092059969902039
    },
    {
      "step": 25750,
      "train_flow_matching_loss": 0.1650390625,
      "train_physics_loss": 0.0
    },
    {
      "step": 25750,
      "train_flow_matching_loss": 0.07958984375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.06894851475954056,
      "learning_rate": 5e-05,
      "loss": 0.1189,
      "step": 25760
    },
    {
      "step": 25760,
      "train_flow_matching_loss": 0.1337890625,
      "train_physics_loss": 0.0
    },
    {
      "step": 25760,
      "train_flow_matching_loss": 0.044921875,
      "train_physics_loss": 0.16592739522457123
    },
    {
      "step": 25760,
      "train_flow_matching_loss": 0.09326171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 25760,
      "train_flow_matching_loss": 0.12109375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11229605972766876,
      "learning_rate": 5e-05,
      "loss": 0.122,
      "step": 25770
    },
    {
      "step": 25770,
      "train_flow_matching_loss": 0.0947265625,
      "train_physics_loss": 0.0
    },
    {
      "step": 25770,
      "train_flow_matching_loss": 0.07373046875,
      "train_physics_loss": 0.16505372524261475
    },
    {
      "step": 25770,
      "train_flow_matching_loss": 0.1318359375,
      "train_physics_loss": 0.0
    },
    {
      "step": 25770,
      "train_flow_matching_loss": 0.11767578125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09745851904153824,
      "learning_rate": 5e-05,
      "loss": 0.1241,
      "step": 25780
    },
    {
      "step": 25780,
      "train_flow_matching_loss": 0.142578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 25780,
      "train_flow_matching_loss": 0.051025390625,
      "train_physics_loss": 0.1832994818687439
    },
    {
      "step": 25780,
      "train_flow_matching_loss": 0.1943359375,
      "train_physics_loss": 0.0
    },
    {
      "step": 25780,
      "train_flow_matching_loss": 0.061767578125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10765022039413452,
      "learning_rate": 5e-05,
      "loss": 0.1202,
      "step": 25790
    },
    {
      "step": 25790,
      "train_flow_matching_loss": 0.1337890625,
      "train_physics_loss": 0.0
    },
    {
      "step": 25790,
      "train_flow_matching_loss": 0.04931640625,
      "train_physics_loss": 0.2306109219789505
    },
    {
      "step": 25790,
      "train_flow_matching_loss": 0.1787109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 25790,
      "train_flow_matching_loss": 0.06884765625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.21228697896003723,
      "learning_rate": 5e-05,
      "loss": 0.1185,
      "step": 25800
    },
    {
      "step": 25800,
      "train_flow_matching_loss": 0.1708984375,
      "train_physics_loss": 0.0
    },
    {
      "step": 25800,
      "train_flow_matching_loss": 0.039794921875,
      "train_physics_loss": 0.21829965710639954
    },
    {
      "step": 25800,
      "train_flow_matching_loss": 0.1552734375,
      "train_physics_loss": 0.0
    },
    {
      "step": 25800,
      "train_flow_matching_loss": 0.10498046875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.07639029622077942,
      "learning_rate": 5e-05,
      "loss": 0.1186,
      "step": 25810
    },
    {
      "step": 25810,
      "train_flow_matching_loss": 0.1328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 25810,
      "train_flow_matching_loss": 0.09033203125,
      "train_physics_loss": 0.18150335550308228
    },
    {
      "step": 25810,
      "train_flow_matching_loss": 0.1982421875,
      "train_physics_loss": 0.0
    },
    {
      "step": 25810,
      "train_flow_matching_loss": 0.11572265625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.2950279712677002,
      "learning_rate": 5e-05,
      "loss": 0.1198,
      "step": 25820
    },
    {
      "step": 25820,
      "train_flow_matching_loss": 0.2099609375,
      "train_physics_loss": 0.0
    },
    {
      "step": 25820,
      "train_flow_matching_loss": 0.030029296875,
      "train_physics_loss": 0.19546669721603394
    },
    {
      "step": 25820,
      "train_flow_matching_loss": 0.158203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 25820,
      "train_flow_matching_loss": 0.09228515625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.1422051042318344,
      "learning_rate": 5e-05,
      "loss": 0.1185,
      "step": 25830
    },
    {
      "step": 25830,
      "train_flow_matching_loss": 0.1298828125,
      "train_physics_loss": 0.0
    },
    {
      "step": 25830,
      "train_flow_matching_loss": 0.10400390625,
      "train_physics_loss": 0.2521511912345886
    },
    {
      "step": 25830,
      "train_flow_matching_loss": 0.1875,
      "train_physics_loss": 0.0
    },
    {
      "step": 25830,
      "train_flow_matching_loss": 0.130859375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08799561113119125,
      "learning_rate": 5e-05,
      "loss": 0.1212,
      "step": 25840
    },
    {
      "step": 25840,
      "train_flow_matching_loss": 0.177734375,
      "train_physics_loss": 0.0
    },
    {
      "step": 25840,
      "train_flow_matching_loss": 0.062255859375,
      "train_physics_loss": 0.21445249021053314
    },
    {
      "step": 25840,
      "train_flow_matching_loss": 0.1630859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 25840,
      "train_flow_matching_loss": 0.1376953125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11319659650325775,
      "learning_rate": 5e-05,
      "loss": 0.1246,
      "step": 25850
    },
    {
      "step": 25850,
      "train_flow_matching_loss": 0.173828125,
      "train_physics_loss": 0.0
    },
    {
      "step": 25850,
      "train_flow_matching_loss": 0.07421875,
      "train_physics_loss": 0.16725049912929535
    },
    {
      "step": 25850,
      "train_flow_matching_loss": 0.0859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 25850,
      "train_flow_matching_loss": 0.1201171875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.0907357707619667,
      "learning_rate": 5e-05,
      "loss": 0.126,
      "step": 25860
    },
    {
      "step": 25860,
      "train_flow_matching_loss": 0.11474609375,
      "train_physics_loss": 0.0
    },
    {
      "step": 25860,
      "train_flow_matching_loss": 0.046142578125,
      "train_physics_loss": 0.19441695511341095
    },
    {
      "step": 25860,
      "train_flow_matching_loss": 0.11181640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 25860,
      "train_flow_matching_loss": 0.09375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.38655805587768555,
      "learning_rate": 5e-05,
      "loss": 0.1204,
      "step": 25870
    },
    {
      "step": 25870,
      "train_flow_matching_loss": 0.1474609375,
      "train_physics_loss": 0.0
    },
    {
      "step": 25870,
      "train_flow_matching_loss": 0.0306396484375,
      "train_physics_loss": 0.16731972992420197
    },
    {
      "step": 25870,
      "train_flow_matching_loss": 0.1865234375,
      "train_physics_loss": 0.0
    },
    {
      "step": 25870,
      "train_flow_matching_loss": 0.09619140625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11223917454481125,
      "learning_rate": 5e-05,
      "loss": 0.1185,
      "step": 25880
    },
    {
      "step": 25880,
      "train_flow_matching_loss": 0.126953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 25880,
      "train_flow_matching_loss": 0.03759765625,
      "train_physics_loss": 0.20766660571098328
    },
    {
      "step": 25880,
      "train_flow_matching_loss": 0.154296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 25880,
      "train_flow_matching_loss": 0.050537109375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09493470191955566,
      "learning_rate": 5e-05,
      "loss": 0.1196,
      "step": 25890
    },
    {
      "step": 25890,
      "train_flow_matching_loss": 0.1728515625,
      "train_physics_loss": 0.0
    },
    {
      "step": 25890,
      "train_flow_matching_loss": 0.03857421875,
      "train_physics_loss": 0.21382172405719757
    },
    {
      "step": 25890,
      "train_flow_matching_loss": 0.1884765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 25890,
      "train_flow_matching_loss": 0.07958984375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.07260682433843613,
      "learning_rate": 5e-05,
      "loss": 0.1196,
      "step": 25900
    },
    {
      "step": 25900,
      "train_flow_matching_loss": 0.1455078125,
      "train_physics_loss": 0.0
    },
    {
      "step": 25900,
      "train_flow_matching_loss": 0.037109375,
      "train_physics_loss": 0.1929575502872467
    },
    {
      "step": 25900,
      "train_flow_matching_loss": 0.1416015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 25900,
      "train_flow_matching_loss": 0.09033203125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.06134914606809616,
      "learning_rate": 5e-05,
      "loss": 0.1237,
      "step": 25910
    },
    {
      "step": 25910,
      "train_flow_matching_loss": 0.1396484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 25910,
      "train_flow_matching_loss": 0.041748046875,
      "train_physics_loss": 0.1874352991580963
    },
    {
      "step": 25910,
      "train_flow_matching_loss": 0.146484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 25910,
      "train_flow_matching_loss": 0.1298828125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10502099245786667,
      "learning_rate": 5e-05,
      "loss": 0.1152,
      "step": 25920
    },
    {
      "step": 25920,
      "train_flow_matching_loss": 0.12158203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 25920,
      "train_flow_matching_loss": 0.043701171875,
      "train_physics_loss": 0.17725859582424164
    },
    {
      "step": 25920,
      "train_flow_matching_loss": 0.125,
      "train_physics_loss": 0.0
    },
    {
      "step": 25920,
      "train_flow_matching_loss": 0.09912109375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10253304243087769,
      "learning_rate": 5e-05,
      "loss": 0.1166,
      "step": 25930
    },
    {
      "step": 25930,
      "train_flow_matching_loss": 0.07373046875,
      "train_physics_loss": 0.0
    },
    {
      "step": 25930,
      "train_flow_matching_loss": 0.037841796875,
      "train_physics_loss": 0.1772996485233307
    },
    {
      "step": 25930,
      "train_flow_matching_loss": 0.181640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 25930,
      "train_flow_matching_loss": 0.07080078125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08626465499401093,
      "learning_rate": 5e-05,
      "loss": 0.1193,
      "step": 25940
    },
    {
      "step": 25940,
      "train_flow_matching_loss": 0.1533203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 25940,
      "train_flow_matching_loss": 0.05810546875,
      "train_physics_loss": 0.18349722027778625
    },
    {
      "step": 25940,
      "train_flow_matching_loss": 0.109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 25940,
      "train_flow_matching_loss": 0.10107421875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.07614577561616898,
      "learning_rate": 5e-05,
      "loss": 0.1198,
      "step": 25950
    },
    {
      "step": 25950,
      "train_flow_matching_loss": 0.1591796875,
      "train_physics_loss": 0.0
    },
    {
      "step": 25950,
      "train_flow_matching_loss": 0.0634765625,
      "train_physics_loss": 0.2550126612186432
    },
    {
      "step": 25950,
      "train_flow_matching_loss": 0.1796875,
      "train_physics_loss": 0.0
    },
    {
      "step": 25950,
      "train_flow_matching_loss": 0.06298828125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.12039270997047424,
      "learning_rate": 5e-05,
      "loss": 0.1177,
      "step": 25960
    },
    {
      "step": 25960,
      "train_flow_matching_loss": 0.1904296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 25960,
      "train_flow_matching_loss": 0.0830078125,
      "train_physics_loss": 0.237859308719635
    },
    {
      "step": 25960,
      "train_flow_matching_loss": 0.140625,
      "train_physics_loss": 0.0
    },
    {
      "step": 25960,
      "train_flow_matching_loss": 0.08984375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.1043476089835167,
      "learning_rate": 5e-05,
      "loss": 0.1196,
      "step": 25970
    },
    {
      "step": 25970,
      "train_flow_matching_loss": 0.158203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 25970,
      "train_flow_matching_loss": 0.048583984375,
      "train_physics_loss": 0.19817690551280975
    },
    {
      "step": 25970,
      "train_flow_matching_loss": 0.1259765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 25970,
      "train_flow_matching_loss": 0.1669921875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.07752349227666855,
      "learning_rate": 5e-05,
      "loss": 0.1177,
      "step": 25980
    },
    {
      "step": 25980,
      "train_flow_matching_loss": 0.1259765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 25980,
      "train_flow_matching_loss": 0.057861328125,
      "train_physics_loss": 0.11770658195018768
    },
    {
      "step": 25980,
      "train_flow_matching_loss": 0.1494140625,
      "train_physics_loss": 0.0
    },
    {
      "step": 25980,
      "train_flow_matching_loss": 0.0771484375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11913543939590454,
      "learning_rate": 5e-05,
      "loss": 0.1227,
      "step": 25990
    },
    {
      "step": 25990,
      "train_flow_matching_loss": 0.1103515625,
      "train_physics_loss": 0.0
    },
    {
      "step": 25990,
      "train_flow_matching_loss": 0.0179443359375,
      "train_physics_loss": 0.15520913898944855
    },
    {
      "step": 25990,
      "train_flow_matching_loss": 0.12109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 25990,
      "train_flow_matching_loss": 0.181640625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09966602176427841,
      "learning_rate": 5e-05,
      "loss": 0.1228,
      "step": 26000
    },
    {
      "step": 26000,
      "train_flow_matching_loss": 0.1005859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 26000,
      "train_flow_matching_loss": 0.12109375,
      "train_physics_loss": 0.16214823722839355
    },
    {
      "step": 26000,
      "train_flow_matching_loss": 0.1220703125,
      "train_physics_loss": 0.0
    },
    {
      "step": 26000,
      "train_flow_matching_loss": 0.181640625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.1562533676624298,
      "learning_rate": 5e-05,
      "loss": 0.1194,
      "step": 26010
    },
    {
      "step": 26010,
      "train_flow_matching_loss": 0.1650390625,
      "train_physics_loss": 0.0
    },
    {
      "step": 26010,
      "train_flow_matching_loss": 0.035888671875,
      "train_physics_loss": 0.2222876399755478
    },
    {
      "step": 26010,
      "train_flow_matching_loss": 0.07421875,
      "train_physics_loss": 0.0
    },
    {
      "step": 26010,
      "train_flow_matching_loss": 0.1689453125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11215832084417343,
      "learning_rate": 5e-05,
      "loss": 0.1206,
      "step": 26020
    },
    {
      "step": 26020,
      "train_flow_matching_loss": 0.1513671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 26020,
      "train_flow_matching_loss": 0.06396484375,
      "train_physics_loss": 0.26167723536491394
    },
    {
      "step": 26020,
      "train_flow_matching_loss": 0.1259765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 26020,
      "train_flow_matching_loss": 0.134765625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11143795400857925,
      "learning_rate": 5e-05,
      "loss": 0.1219,
      "step": 26030
    },
    {
      "step": 26030,
      "train_flow_matching_loss": 0.0966796875,
      "train_physics_loss": 0.0
    },
    {
      "step": 26030,
      "train_flow_matching_loss": 0.04150390625,
      "train_physics_loss": 0.2679729759693146
    },
    {
      "step": 26030,
      "train_flow_matching_loss": 0.1572265625,
      "train_physics_loss": 0.0
    },
    {
      "step": 26030,
      "train_flow_matching_loss": 0.08837890625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09905584901571274,
      "learning_rate": 5e-05,
      "loss": 0.1198,
      "step": 26040
    },
    {
      "step": 26040,
      "train_flow_matching_loss": 0.11865234375,
      "train_physics_loss": 0.0
    },
    {
      "step": 26040,
      "train_flow_matching_loss": 0.07470703125,
      "train_physics_loss": 0.18521171808242798
    },
    {
      "step": 26040,
      "train_flow_matching_loss": 0.061767578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 26040,
      "train_flow_matching_loss": 0.07177734375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.13268788158893585,
      "learning_rate": 5e-05,
      "loss": 0.117,
      "step": 26050
    },
    {
      "step": 26050,
      "train_flow_matching_loss": 0.1669921875,
      "train_physics_loss": 0.0
    },
    {
      "step": 26050,
      "train_flow_matching_loss": 0.051513671875,
      "train_physics_loss": 0.20310141146183014
    },
    {
      "step": 26050,
      "train_flow_matching_loss": 0.150390625,
      "train_physics_loss": 0.0
    },
    {
      "step": 26050,
      "train_flow_matching_loss": 0.1845703125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11195748299360275,
      "learning_rate": 5e-05,
      "loss": 0.1212,
      "step": 26060
    },
    {
      "step": 26060,
      "train_flow_matching_loss": 0.1806640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 26060,
      "train_flow_matching_loss": 0.0322265625,
      "train_physics_loss": 0.15504190325737
    },
    {
      "step": 26060,
      "train_flow_matching_loss": 0.1298828125,
      "train_physics_loss": 0.0
    },
    {
      "step": 26060,
      "train_flow_matching_loss": 0.08203125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09482553601264954,
      "learning_rate": 5e-05,
      "loss": 0.1167,
      "step": 26070
    },
    {
      "step": 26070,
      "train_flow_matching_loss": 0.10400390625,
      "train_physics_loss": 0.0
    },
    {
      "step": 26070,
      "train_flow_matching_loss": 0.032958984375,
      "train_physics_loss": 0.23256061971187592
    },
    {
      "step": 26070,
      "train_flow_matching_loss": 0.13671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 26070,
      "train_flow_matching_loss": 0.1083984375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11295755207538605,
      "learning_rate": 5e-05,
      "loss": 0.1226,
      "step": 26080
    },
    {
      "step": 26080,
      "train_flow_matching_loss": 0.12353515625,
      "train_physics_loss": 0.0
    },
    {
      "step": 26080,
      "train_flow_matching_loss": 0.060546875,
      "train_physics_loss": 0.20348994433879852
    },
    {
      "step": 26080,
      "train_flow_matching_loss": 0.10791015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 26080,
      "train_flow_matching_loss": 0.08447265625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09657066315412521,
      "learning_rate": 5e-05,
      "loss": 0.1208,
      "step": 26090
    },
    {
      "step": 26090,
      "train_flow_matching_loss": 0.1259765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 26090,
      "train_flow_matching_loss": 0.04541015625,
      "train_physics_loss": 0.20418423414230347
    },
    {
      "step": 26090,
      "train_flow_matching_loss": 0.07958984375,
      "train_physics_loss": 0.0
    },
    {
      "step": 26090,
      "train_flow_matching_loss": 0.1318359375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.12730494141578674,
      "learning_rate": 5e-05,
      "loss": 0.1225,
      "step": 26100
    },
    {
      "step": 26100,
      "train_flow_matching_loss": 0.1728515625,
      "train_physics_loss": 0.0
    },
    {
      "step": 26100,
      "train_flow_matching_loss": 0.07568359375,
      "train_physics_loss": 0.1815173327922821
    },
    {
      "step": 26100,
      "train_flow_matching_loss": 0.134765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 26100,
      "train_flow_matching_loss": 0.10986328125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.12387190759181976,
      "learning_rate": 5e-05,
      "loss": 0.1224,
      "step": 26110
    },
    {
      "step": 26110,
      "train_flow_matching_loss": 0.06201171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 26110,
      "train_flow_matching_loss": 0.05322265625,
      "train_physics_loss": 0.17828042805194855
    },
    {
      "step": 26110,
      "train_flow_matching_loss": 0.10498046875,
      "train_physics_loss": 0.0
    },
    {
      "step": 26110,
      "train_flow_matching_loss": 0.1328125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10925086587667465,
      "learning_rate": 5e-05,
      "loss": 0.1213,
      "step": 26120
    },
    {
      "step": 26120,
      "train_flow_matching_loss": 0.09619140625,
      "train_physics_loss": 0.0
    },
    {
      "step": 26120,
      "train_flow_matching_loss": 0.053466796875,
      "train_physics_loss": 0.1624288260936737
    },
    {
      "step": 26120,
      "train_flow_matching_loss": 0.11767578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 26120,
      "train_flow_matching_loss": 0.1484375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.07122480124235153,
      "learning_rate": 5e-05,
      "loss": 0.1211,
      "step": 26130
    },
    {
      "step": 26130,
      "train_flow_matching_loss": 0.07177734375,
      "train_physics_loss": 0.0
    },
    {
      "step": 26130,
      "train_flow_matching_loss": 0.03955078125,
      "train_physics_loss": 0.22922943532466888
    },
    {
      "step": 26130,
      "train_flow_matching_loss": 0.1162109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 26130,
      "train_flow_matching_loss": 0.10205078125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.07184579968452454,
      "learning_rate": 5e-05,
      "loss": 0.1141,
      "step": 26140
    },
    {
      "step": 26140,
      "train_flow_matching_loss": 0.0654296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 26140,
      "train_flow_matching_loss": 0.06103515625,
      "train_physics_loss": 0.19713130593299866
    },
    {
      "step": 26140,
      "train_flow_matching_loss": 0.10791015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 26140,
      "train_flow_matching_loss": 0.193359375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.1006510853767395,
      "learning_rate": 5e-05,
      "loss": 0.1268,
      "step": 26150
    },
    {
      "step": 26150,
      "train_flow_matching_loss": 0.11328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 26150,
      "train_flow_matching_loss": 0.06884765625,
      "train_physics_loss": 0.2393244206905365
    },
    {
      "step": 26150,
      "train_flow_matching_loss": 0.1630859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 26150,
      "train_flow_matching_loss": 0.1728515625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.07619854062795639,
      "learning_rate": 5e-05,
      "loss": 0.1213,
      "step": 26160
    },
    {
      "step": 26160,
      "train_flow_matching_loss": 0.1357421875,
      "train_physics_loss": 0.0
    },
    {
      "step": 26160,
      "train_flow_matching_loss": 0.0546875,
      "train_physics_loss": 0.0
    },
    {
      "step": 26160,
      "train_flow_matching_loss": 0.08837890625,
      "train_physics_loss": 0.0
    },
    {
      "step": 26160,
      "train_flow_matching_loss": 0.193359375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08116094022989273,
      "learning_rate": 5e-05,
      "loss": 0.121,
      "step": 26170
    },
    {
      "step": 26170,
      "train_flow_matching_loss": 0.1064453125,
      "train_physics_loss": 0.0
    },
    {
      "step": 26170,
      "train_flow_matching_loss": 0.045166015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 26170,
      "train_flow_matching_loss": 0.11083984375,
      "train_physics_loss": 0.0
    },
    {
      "step": 26170,
      "train_flow_matching_loss": 0.205078125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09032357484102249,
      "learning_rate": 5e-05,
      "loss": 0.1277,
      "step": 26180
    },
    {
      "step": 26180,
      "train_flow_matching_loss": 0.1240234375,
      "train_physics_loss": 0.0
    },
    {
      "step": 26180,
      "train_flow_matching_loss": 0.09228515625,
      "train_physics_loss": 0.0
    },
    {
      "step": 26180,
      "train_flow_matching_loss": 0.142578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 26180,
      "train_flow_matching_loss": 0.08642578125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11178769916296005,
      "learning_rate": 5e-05,
      "loss": 0.119,
      "step": 26190
    },
    {
      "step": 26190,
      "train_flow_matching_loss": 0.07275390625,
      "train_physics_loss": 0.0
    },
    {
      "step": 26190,
      "train_flow_matching_loss": 0.07666015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 26190,
      "train_flow_matching_loss": 0.1240234375,
      "train_physics_loss": 0.0
    },
    {
      "step": 26190,
      "train_flow_matching_loss": 0.1103515625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10414329916238785,
      "learning_rate": 5e-05,
      "loss": 0.1224,
      "step": 26200
    },
    {
      "step": 26200,
      "train_flow_matching_loss": 0.1171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 26200,
      "train_flow_matching_loss": 0.166015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 26200,
      "train_flow_matching_loss": 0.12060546875,
      "train_physics_loss": 0.0
    },
    {
      "step": 26200,
      "train_flow_matching_loss": 0.1181640625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.13774031400680542,
      "learning_rate": 5e-05,
      "loss": 0.1198,
      "step": 26210
    },
    {
      "step": 26210,
      "train_flow_matching_loss": 0.06689453125,
      "train_physics_loss": 0.0
    },
    {
      "step": 26210,
      "train_flow_matching_loss": 0.1005859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 26210,
      "train_flow_matching_loss": 0.1435546875,
      "train_physics_loss": 0.0
    },
    {
      "step": 26210,
      "train_flow_matching_loss": 0.10888671875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.15070809423923492,
      "learning_rate": 5e-05,
      "loss": 0.1235,
      "step": 26220
    },
    {
      "step": 26220,
      "train_flow_matching_loss": 0.09619140625,
      "train_physics_loss": 0.0
    },
    {
      "step": 26220,
      "train_flow_matching_loss": 0.10986328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 26220,
      "train_flow_matching_loss": 0.197265625,
      "train_physics_loss": 0.0
    },
    {
      "step": 26220,
      "train_flow_matching_loss": 0.09423828125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.07848137617111206,
      "learning_rate": 5e-05,
      "loss": 0.1245,
      "step": 26230
    },
    {
      "step": 26230,
      "train_flow_matching_loss": 0.12890625,
      "train_physics_loss": 0.0
    },
    {
      "step": 26230,
      "train_flow_matching_loss": 0.0869140625,
      "train_physics_loss": 0.0
    },
    {
      "step": 26230,
      "train_flow_matching_loss": 0.12158203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 26230,
      "train_flow_matching_loss": 0.11474609375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08660656958818436,
      "learning_rate": 5e-05,
      "loss": 0.1195,
      "step": 26240
    },
    {
      "step": 26240,
      "train_flow_matching_loss": 0.08203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 26240,
      "train_flow_matching_loss": 0.0869140625,
      "train_physics_loss": 0.0
    },
    {
      "step": 26240,
      "train_flow_matching_loss": 0.1337890625,
      "train_physics_loss": 0.0
    },
    {
      "step": 26240,
      "train_flow_matching_loss": 0.154296875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11542962491512299,
      "learning_rate": 5e-05,
      "loss": 0.1218,
      "step": 26250
    },
    {
      "step": 26250,
      "train_flow_matching_loss": 0.07421875,
      "train_physics_loss": 0.0
    },
    {
      "step": 26250,
      "train_flow_matching_loss": 0.1455078125,
      "train_physics_loss": 0.0
    },
    {
      "step": 26250,
      "train_flow_matching_loss": 0.1982421875,
      "train_physics_loss": 0.0
    },
    {
      "step": 26250,
      "train_flow_matching_loss": 0.1025390625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10044504702091217,
      "learning_rate": 5e-05,
      "loss": 0.126,
      "step": 26260
    },
    {
      "step": 26260,
      "train_flow_matching_loss": 0.09716796875,
      "train_physics_loss": 0.0
    },
    {
      "step": 26260,
      "train_flow_matching_loss": 0.09375,
      "train_physics_loss": 0.0
    },
    {
      "step": 26260,
      "train_flow_matching_loss": 0.12890625,
      "train_physics_loss": 0.0
    },
    {
      "step": 26260,
      "train_flow_matching_loss": 0.1787109375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.21012909710407257,
      "learning_rate": 5e-05,
      "loss": 0.121,
      "step": 26270
    },
    {
      "step": 26270,
      "train_flow_matching_loss": 0.1220703125,
      "train_physics_loss": 0.0
    },
    {
      "step": 26270,
      "train_flow_matching_loss": 0.1767578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 26270,
      "train_flow_matching_loss": 0.1884765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 26270,
      "train_flow_matching_loss": 0.1171875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10239928215742111,
      "learning_rate": 5e-05,
      "loss": 0.1202,
      "step": 26280
    },
    {
      "step": 26280,
      "train_flow_matching_loss": 0.0634765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 26280,
      "train_flow_matching_loss": 0.1357421875,
      "train_physics_loss": 0.0
    },
    {
      "step": 26280,
      "train_flow_matching_loss": 0.1396484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 26280,
      "train_flow_matching_loss": 0.08251953125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.13286606967449188,
      "learning_rate": 5e-05,
      "loss": 0.1257,
      "step": 26290
    },
    {
      "step": 26290,
      "train_flow_matching_loss": 0.08935546875,
      "train_physics_loss": 0.0
    },
    {
      "step": 26290,
      "train_flow_matching_loss": 0.06640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 26290,
      "train_flow_matching_loss": 0.1259765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 26290,
      "train_flow_matching_loss": 0.140625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10578396916389465,
      "learning_rate": 5e-05,
      "loss": 0.1223,
      "step": 26300
    },
    {
      "step": 26300,
      "train_flow_matching_loss": 0.1376953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 26300,
      "train_flow_matching_loss": 0.1484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 26300,
      "train_flow_matching_loss": 0.10986328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 26300,
      "train_flow_matching_loss": 0.1669921875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08467564731836319,
      "learning_rate": 5e-05,
      "loss": 0.1231,
      "step": 26310
    },
    {
      "step": 26310,
      "train_flow_matching_loss": 0.12353515625,
      "train_physics_loss": 0.0
    },
    {
      "step": 26310,
      "train_flow_matching_loss": 0.125,
      "train_physics_loss": 0.0
    },
    {
      "step": 26310,
      "train_flow_matching_loss": 0.09765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 26310,
      "train_flow_matching_loss": 0.1943359375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08548440784215927,
      "learning_rate": 5e-05,
      "loss": 0.1191,
      "step": 26320
    },
    {
      "step": 26320,
      "train_flow_matching_loss": 0.12060546875,
      "train_physics_loss": 0.0
    },
    {
      "step": 26320,
      "train_flow_matching_loss": 0.154296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 26320,
      "train_flow_matching_loss": 0.1455078125,
      "train_physics_loss": 0.0
    },
    {
      "step": 26320,
      "train_flow_matching_loss": 0.1201171875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09410054981708527,
      "learning_rate": 5e-05,
      "loss": 0.1272,
      "step": 26330
    },
    {
      "step": 26330,
      "train_flow_matching_loss": 0.1796875,
      "train_physics_loss": 0.0
    },
    {
      "step": 26330,
      "train_flow_matching_loss": 0.1650390625,
      "train_physics_loss": 0.0
    },
    {
      "step": 26330,
      "train_flow_matching_loss": 0.11181640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 26330,
      "train_flow_matching_loss": 0.1474609375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08828092366456985,
      "learning_rate": 5e-05,
      "loss": 0.1213,
      "step": 26340
    },
    {
      "step": 26340,
      "train_flow_matching_loss": 0.09619140625,
      "train_physics_loss": 0.0
    },
    {
      "step": 26340,
      "train_flow_matching_loss": 0.2041015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 26340,
      "train_flow_matching_loss": 0.1689453125,
      "train_physics_loss": 0.0
    },
    {
      "step": 26340,
      "train_flow_matching_loss": 0.125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.0975295677781105,
      "learning_rate": 5e-05,
      "loss": 0.1255,
      "step": 26350
    },
    {
      "step": 26350,
      "train_flow_matching_loss": 0.051025390625,
      "train_physics_loss": 0.0
    },
    {
      "step": 26350,
      "train_flow_matching_loss": 0.1513671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 26350,
      "train_flow_matching_loss": 0.12353515625,
      "train_physics_loss": 0.0
    },
    {
      "step": 26350,
      "train_flow_matching_loss": 0.177734375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.16266068816184998,
      "learning_rate": 5e-05,
      "loss": 0.1196,
      "step": 26360
    },
    {
      "step": 26360,
      "train_flow_matching_loss": 0.046875,
      "train_physics_loss": 0.0
    },
    {
      "step": 26360,
      "train_flow_matching_loss": 0.08544921875,
      "train_physics_loss": 0.0
    },
    {
      "step": 26360,
      "train_flow_matching_loss": 0.1318359375,
      "train_physics_loss": 0.0
    },
    {
      "step": 26360,
      "train_flow_matching_loss": 0.1474609375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08230669051408768,
      "learning_rate": 5e-05,
      "loss": 0.1239,
      "step": 26370
    },
    {
      "step": 26370,
      "train_flow_matching_loss": 0.150390625,
      "train_physics_loss": 0.0
    },
    {
      "step": 26370,
      "train_flow_matching_loss": 0.1591796875,
      "train_physics_loss": 0.0
    },
    {
      "step": 26370,
      "train_flow_matching_loss": 0.11865234375,
      "train_physics_loss": 0.0
    },
    {
      "step": 26370,
      "train_flow_matching_loss": 0.1396484375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09428369998931885,
      "learning_rate": 5e-05,
      "loss": 0.1253,
      "step": 26380
    },
    {
      "step": 26380,
      "train_flow_matching_loss": 0.10107421875,
      "train_physics_loss": 0.0
    },
    {
      "step": 26380,
      "train_flow_matching_loss": 0.05712890625,
      "train_physics_loss": 0.0
    },
    {
      "step": 26380,
      "train_flow_matching_loss": 0.0888671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 26380,
      "train_flow_matching_loss": 0.2041015625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.133854940533638,
      "learning_rate": 5e-05,
      "loss": 0.1158,
      "step": 26390
    },
    {
      "step": 26390,
      "train_flow_matching_loss": 0.154296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 26390,
      "train_flow_matching_loss": 0.0869140625,
      "train_physics_loss": 0.0
    },
    {
      "step": 26390,
      "train_flow_matching_loss": 0.1298828125,
      "train_physics_loss": 0.0
    },
    {
      "step": 26390,
      "train_flow_matching_loss": 0.10107421875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08846879750490189,
      "learning_rate": 5e-05,
      "loss": 0.1201,
      "step": 26400
    },
    {
      "step": 26400,
      "train_flow_matching_loss": 0.1318359375,
      "train_physics_loss": 0.0
    },
    {
      "step": 26400,
      "train_flow_matching_loss": 0.1025390625,
      "train_physics_loss": 0.0
    },
    {
      "step": 26400,
      "train_flow_matching_loss": 0.138671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 26400,
      "train_flow_matching_loss": 0.1640625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09327234327793121,
      "learning_rate": 5e-05,
      "loss": 0.1241,
      "step": 26410
    },
    {
      "step": 26410,
      "train_flow_matching_loss": 0.1171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 26410,
      "train_flow_matching_loss": 0.078125,
      "train_physics_loss": 0.0
    },
    {
      "step": 26410,
      "train_flow_matching_loss": 0.10498046875,
      "train_physics_loss": 0.0
    },
    {
      "step": 26410,
      "train_flow_matching_loss": 0.173828125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.07942179590463638,
      "learning_rate": 5e-05,
      "loss": 0.1198,
      "step": 26420
    },
    {
      "step": 26420,
      "train_flow_matching_loss": 0.134765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 26420,
      "train_flow_matching_loss": 0.11962890625,
      "train_physics_loss": 0.0
    },
    {
      "step": 26420,
      "train_flow_matching_loss": 0.1474609375,
      "train_physics_loss": 0.0
    },
    {
      "step": 26420,
      "train_flow_matching_loss": 0.1923828125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.12614871561527252,
      "learning_rate": 5e-05,
      "loss": 0.1206,
      "step": 26430
    },
    {
      "step": 26430,
      "train_flow_matching_loss": 0.13671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 26430,
      "train_flow_matching_loss": 0.1025390625,
      "train_physics_loss": 0.0
    },
    {
      "step": 26430,
      "train_flow_matching_loss": 0.1494140625,
      "train_physics_loss": 0.0
    },
    {
      "step": 26430,
      "train_flow_matching_loss": 0.1708984375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10563156753778458,
      "learning_rate": 5e-05,
      "loss": 0.1213,
      "step": 26440
    },
    {
      "step": 26440,
      "train_flow_matching_loss": 0.1279296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 26440,
      "train_flow_matching_loss": 0.1640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 26440,
      "train_flow_matching_loss": 0.162109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 26440,
      "train_flow_matching_loss": 0.1474609375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09922877699136734,
      "learning_rate": 5e-05,
      "loss": 0.126,
      "step": 26450
    },
    {
      "step": 26450,
      "train_flow_matching_loss": 0.126953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 26450,
      "train_flow_matching_loss": 0.2099609375,
      "train_physics_loss": 0.0
    },
    {
      "step": 26450,
      "train_flow_matching_loss": 0.1591796875,
      "train_physics_loss": 0.0
    },
    {
      "step": 26450,
      "train_flow_matching_loss": 0.173828125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09696795791387558,
      "learning_rate": 5e-05,
      "loss": 0.1228,
      "step": 26460
    },
    {
      "step": 26460,
      "train_flow_matching_loss": 0.045166015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 26460,
      "train_flow_matching_loss": 0.1484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 26460,
      "train_flow_matching_loss": 0.09912109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 26460,
      "train_flow_matching_loss": 0.11083984375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.06752415001392365,
      "learning_rate": 5e-05,
      "loss": 0.1189,
      "step": 26470
    },
    {
      "step": 26470,
      "train_flow_matching_loss": 0.134765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 26470,
      "train_flow_matching_loss": 0.0712890625,
      "train_physics_loss": 0.0
    },
    {
      "step": 26470,
      "train_flow_matching_loss": 0.140625,
      "train_physics_loss": 0.0
    },
    {
      "step": 26470,
      "train_flow_matching_loss": 0.134765625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10913768410682678,
      "learning_rate": 5e-05,
      "loss": 0.1225,
      "step": 26480
    },
    {
      "step": 26480,
      "train_flow_matching_loss": 0.2080078125,
      "train_physics_loss": 0.0
    },
    {
      "step": 26480,
      "train_flow_matching_loss": 0.1962890625,
      "train_physics_loss": 0.0
    },
    {
      "step": 26480,
      "train_flow_matching_loss": 0.10693359375,
      "train_physics_loss": 0.0
    },
    {
      "step": 26480,
      "train_flow_matching_loss": 0.158203125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08349139243364334,
      "learning_rate": 5e-05,
      "loss": 0.1157,
      "step": 26490
    },
    {
      "step": 26490,
      "train_flow_matching_loss": 0.10986328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 26490,
      "train_flow_matching_loss": 0.169921875,
      "train_physics_loss": 0.0
    },
    {
      "step": 26490,
      "train_flow_matching_loss": 0.09814453125,
      "train_physics_loss": 0.0
    },
    {
      "step": 26490,
      "train_flow_matching_loss": 0.039306640625,
      "train_physics_loss": 0.1798250824213028
    },
    {
      "grad_norm": 0.12086963653564453,
      "learning_rate": 5e-05,
      "loss": 0.1255,
      "step": 26500
    },
    {
      "step": 26500,
      "train_flow_matching_loss": 0.0771484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 26500,
      "train_flow_matching_loss": 0.1484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 26500,
      "train_flow_matching_loss": 0.1064453125,
      "train_physics_loss": 0.0
    },
    {
      "step": 26500,
      "train_flow_matching_loss": 0.03125,
      "train_physics_loss": 0.1950521022081375
    },
    {
      "grad_norm": 0.29864954948425293,
      "learning_rate": 5e-05,
      "loss": 0.1194,
      "step": 26510
    },
    {
      "step": 26510,
      "train_flow_matching_loss": 0.12255859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 26510,
      "train_flow_matching_loss": 0.10400390625,
      "train_physics_loss": 0.0
    },
    {
      "step": 26510,
      "train_flow_matching_loss": 0.08056640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 26510,
      "train_flow_matching_loss": 0.04833984375,
      "train_physics_loss": 0.14570237696170807
    },
    {
      "grad_norm": 0.07488079369068146,
      "learning_rate": 5e-05,
      "loss": 0.1208,
      "step": 26520
    },
    {
      "step": 26520,
      "train_flow_matching_loss": 0.169921875,
      "train_physics_loss": 0.0
    },
    {
      "step": 26520,
      "train_flow_matching_loss": 0.09912109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 26520,
      "train_flow_matching_loss": 0.1474609375,
      "train_physics_loss": 0.0
    },
    {
      "step": 26520,
      "train_flow_matching_loss": 0.03125,
      "train_physics_loss": 0.1475946307182312
    },
    {
      "grad_norm": 0.11876799911260605,
      "learning_rate": 5e-05,
      "loss": 0.1137,
      "step": 26530
    },
    {
      "step": 26530,
      "train_flow_matching_loss": 0.11279296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 26530,
      "train_flow_matching_loss": 0.11474609375,
      "train_physics_loss": 0.0
    },
    {
      "step": 26530,
      "train_flow_matching_loss": 0.07666015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 26530,
      "train_flow_matching_loss": 0.07177734375,
      "train_physics_loss": 0.14846675097942352
    },
    {
      "grad_norm": 0.07164324074983597,
      "learning_rate": 5e-05,
      "loss": 0.125,
      "step": 26540
    },
    {
      "step": 26540,
      "train_flow_matching_loss": 0.07666015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 26540,
      "train_flow_matching_loss": 0.09130859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 26540,
      "train_flow_matching_loss": 0.1103515625,
      "train_physics_loss": 0.0
    },
    {
      "step": 26540,
      "train_flow_matching_loss": 0.049072265625,
      "train_physics_loss": 0.14192230999469757
    },
    {
      "grad_norm": 0.08402585238218307,
      "learning_rate": 5e-05,
      "loss": 0.1237,
      "step": 26550
    },
    {
      "step": 26550,
      "train_flow_matching_loss": 0.1044921875,
      "train_physics_loss": 0.0
    },
    {
      "step": 26550,
      "train_flow_matching_loss": 0.1640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 26550,
      "train_flow_matching_loss": 0.08935546875,
      "train_physics_loss": 0.0
    },
    {
      "step": 26550,
      "train_flow_matching_loss": 0.044189453125,
      "train_physics_loss": 0.16201664507389069
    },
    {
      "grad_norm": 0.09500603377819061,
      "learning_rate": 5e-05,
      "loss": 0.118,
      "step": 26560
    },
    {
      "step": 26560,
      "train_flow_matching_loss": 0.1689453125,
      "train_physics_loss": 0.0
    },
    {
      "step": 26560,
      "train_flow_matching_loss": 0.12890625,
      "train_physics_loss": 0.0
    },
    {
      "step": 26560,
      "train_flow_matching_loss": 0.08984375,
      "train_physics_loss": 0.0
    },
    {
      "step": 26560,
      "train_flow_matching_loss": 0.061279296875,
      "train_physics_loss": 0.12955665588378906
    },
    {
      "grad_norm": 0.0796678364276886,
      "learning_rate": 5e-05,
      "loss": 0.1199,
      "step": 26570
    },
    {
      "step": 26570,
      "train_flow_matching_loss": 0.12890625,
      "train_physics_loss": 0.16122712194919586
    },
    {
      "step": 26570,
      "train_flow_matching_loss": 0.054931640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 26570,
      "train_flow_matching_loss": 0.1494140625,
      "train_physics_loss": 0.0
    },
    {
      "step": 26570,
      "train_flow_matching_loss": 0.0673828125,
      "train_physics_loss": 0.17189399898052216
    },
    {
      "grad_norm": 0.13217829167842865,
      "learning_rate": 5e-05,
      "loss": 0.1237,
      "step": 26580
    },
    {
      "step": 26580,
      "train_flow_matching_loss": 0.045654296875,
      "train_physics_loss": 0.18034546077251434
    },
    {
      "step": 26580,
      "train_flow_matching_loss": 0.11962890625,
      "train_physics_loss": 0.0
    },
    {
      "step": 26580,
      "train_flow_matching_loss": 0.1455078125,
      "train_physics_loss": 0.0
    },
    {
      "step": 26580,
      "train_flow_matching_loss": 0.06689453125,
      "train_physics_loss": 0.22833481431007385
    },
    {
      "grad_norm": 0.07209676504135132,
      "learning_rate": 5e-05,
      "loss": 0.1194,
      "step": 26590
    },
    {
      "step": 26590,
      "train_flow_matching_loss": 0.07666015625,
      "train_physics_loss": 0.20433536171913147
    },
    {
      "step": 26590,
      "train_flow_matching_loss": 0.169921875,
      "train_physics_loss": 0.0
    },
    {
      "step": 26590,
      "train_flow_matching_loss": 0.07568359375,
      "train_physics_loss": 0.0
    },
    {
      "step": 26590,
      "train_flow_matching_loss": 0.059814453125,
      "train_physics_loss": 0.12013901770114899
    },
    {
      "grad_norm": 0.102706179022789,
      "learning_rate": 5e-05,
      "loss": 0.1207,
      "step": 26600
    },
    {
      "step": 26600,
      "train_flow_matching_loss": 0.054931640625,
      "train_physics_loss": 0.16925173997879028
    },
    {
      "step": 26600,
      "train_flow_matching_loss": 0.173828125,
      "train_physics_loss": 0.0
    },
    {
      "step": 26600,
      "train_flow_matching_loss": 0.1181640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 26600,
      "train_flow_matching_loss": 0.041748046875,
      "train_physics_loss": 0.13809660077095032
    },
    {
      "grad_norm": 0.08548688888549805,
      "learning_rate": 5e-05,
      "loss": 0.1203,
      "step": 26610
    },
    {
      "step": 26610,
      "train_flow_matching_loss": 0.021728515625,
      "train_physics_loss": 0.14568978548049927
    },
    {
      "step": 26610,
      "train_flow_matching_loss": 0.10205078125,
      "train_physics_loss": 0.0
    },
    {
      "step": 26610,
      "train_flow_matching_loss": 0.08154296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 26610,
      "train_flow_matching_loss": 0.0693359375,
      "train_physics_loss": 0.20751625299453735
    },
    {
      "grad_norm": 0.09990048408508301,
      "learning_rate": 5e-05,
      "loss": 0.1194,
      "step": 26620
    },
    {
      "step": 26620,
      "train_flow_matching_loss": 0.05810546875,
      "train_physics_loss": 0.17616836726665497
    },
    {
      "step": 26620,
      "train_flow_matching_loss": 0.07080078125,
      "train_physics_loss": 0.0
    },
    {
      "step": 26620,
      "train_flow_matching_loss": 0.13671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 26620,
      "train_flow_matching_loss": 0.060546875,
      "train_physics_loss": 0.1315295249223709
    },
    {
      "grad_norm": 0.10472753643989563,
      "learning_rate": 5e-05,
      "loss": 0.1248,
      "step": 26630
    },
    {
      "step": 26630,
      "train_flow_matching_loss": 0.056884765625,
      "train_physics_loss": 0.19851653277873993
    },
    {
      "step": 26630,
      "train_flow_matching_loss": 0.1142578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 26630,
      "train_flow_matching_loss": 0.1376953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 26630,
      "train_flow_matching_loss": 0.05224609375,
      "train_physics_loss": 0.1739223599433899
    },
    {
      "grad_norm": 0.09558746963739395,
      "learning_rate": 5e-05,
      "loss": 0.1237,
      "step": 26640
    },
    {
      "step": 26640,
      "train_flow_matching_loss": 0.07958984375,
      "train_physics_loss": 0.2336881458759308
    },
    {
      "step": 26640,
      "train_flow_matching_loss": 0.10400390625,
      "train_physics_loss": 0.0
    },
    {
      "step": 26640,
      "train_flow_matching_loss": 0.126953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 26640,
      "train_flow_matching_loss": 0.05859375,
      "train_physics_loss": 0.2854846119880676
    },
    {
      "grad_norm": 0.08689246326684952,
      "learning_rate": 5e-05,
      "loss": 0.1209,
      "step": 26650
    },
    {
      "step": 26650,
      "train_flow_matching_loss": 0.04541015625,
      "train_physics_loss": 0.15703463554382324
    },
    {
      "step": 26650,
      "train_flow_matching_loss": 0.11962890625,
      "train_physics_loss": 0.0
    },
    {
      "step": 26650,
      "train_flow_matching_loss": 0.1611328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 26650,
      "train_flow_matching_loss": 0.051513671875,
      "train_physics_loss": 0.19234997034072876
    },
    {
      "grad_norm": 0.07982627302408218,
      "learning_rate": 5e-05,
      "loss": 0.1168,
      "step": 26660
    },
    {
      "step": 26660,
      "train_flow_matching_loss": 0.03955078125,
      "train_physics_loss": 0.1650928407907486
    },
    {
      "step": 26660,
      "train_flow_matching_loss": 0.1806640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 26660,
      "train_flow_matching_loss": 0.07373046875,
      "train_physics_loss": 0.0
    },
    {
      "step": 26660,
      "train_flow_matching_loss": 0.10546875,
      "train_physics_loss": 0.17954660952091217
    },
    {
      "grad_norm": 0.14977890253067017,
      "learning_rate": 5e-05,
      "loss": 0.1183,
      "step": 26670
    },
    {
      "step": 26670,
      "train_flow_matching_loss": 0.08447265625,
      "train_physics_loss": 0.17261002957820892
    },
    {
      "step": 26670,
      "train_flow_matching_loss": 0.10888671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 26670,
      "train_flow_matching_loss": 0.08642578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 26670,
      "train_flow_matching_loss": 0.0250244140625,
      "train_physics_loss": 0.1528114527463913
    },
    {
      "grad_norm": 0.08578520268201828,
      "learning_rate": 5e-05,
      "loss": 0.1204,
      "step": 26680
    },
    {
      "step": 26680,
      "train_flow_matching_loss": 0.053466796875,
      "train_physics_loss": 0.1827128678560257
    },
    {
      "step": 26680,
      "train_flow_matching_loss": 0.1787109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 26680,
      "train_flow_matching_loss": 0.05810546875,
      "train_physics_loss": 0.0
    },
    {
      "step": 26680,
      "train_flow_matching_loss": 0.1572265625,
      "train_physics_loss": 0.4942685663700104
    },
    {
      "grad_norm": 0.19568520784378052,
      "learning_rate": 5e-05,
      "loss": 0.1183,
      "step": 26690
    },
    {
      "step": 26690,
      "train_flow_matching_loss": 0.038818359375,
      "train_physics_loss": 0.15835179388523102
    },
    {
      "step": 26690,
      "train_flow_matching_loss": 0.2041015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 26690,
      "train_flow_matching_loss": 0.1630859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 26690,
      "train_flow_matching_loss": 0.03662109375,
      "train_physics_loss": 0.21830065548419952
    },
    {
      "grad_norm": 0.08430016040802002,
      "learning_rate": 5e-05,
      "loss": 0.1206,
      "step": 26700
    },
    {
      "step": 26700,
      "train_flow_matching_loss": 0.052978515625,
      "train_physics_loss": 0.19088315963745117
    },
    {
      "step": 26700,
      "train_flow_matching_loss": 0.1552734375,
      "train_physics_loss": 0.0
    },
    {
      "step": 26700,
      "train_flow_matching_loss": 0.12109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 26700,
      "train_flow_matching_loss": 0.039794921875,
      "train_physics_loss": 0.14904417097568512
    },
    {
      "grad_norm": 0.09625343233346939,
      "learning_rate": 5e-05,
      "loss": 0.1221,
      "step": 26710
    },
    {
      "step": 26710,
      "train_flow_matching_loss": 0.044677734375,
      "train_physics_loss": 0.14048489928245544
    },
    {
      "step": 26710,
      "train_flow_matching_loss": 0.150390625,
      "train_physics_loss": 0.0
    },
    {
      "step": 26710,
      "train_flow_matching_loss": 0.06005859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 26710,
      "train_flow_matching_loss": 0.09716796875,
      "train_physics_loss": 0.20431561768054962
    },
    {
      "grad_norm": 0.12069901078939438,
      "learning_rate": 5e-05,
      "loss": 0.1238,
      "step": 26720
    },
    {
      "step": 26720,
      "train_flow_matching_loss": 0.033447265625,
      "train_physics_loss": 0.1515350192785263
    },
    {
      "step": 26720,
      "train_flow_matching_loss": 0.06982421875,
      "train_physics_loss": 0.0
    },
    {
      "step": 26720,
      "train_flow_matching_loss": 0.12255859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 26720,
      "train_flow_matching_loss": 0.017333984375,
      "train_physics_loss": 0.12799213826656342
    },
    {
      "grad_norm": 0.0721869021654129,
      "learning_rate": 5e-05,
      "loss": 0.1137,
      "step": 26730
    },
    {
      "step": 26730,
      "train_flow_matching_loss": 0.048828125,
      "train_physics_loss": 0.2242223173379898
    },
    {
      "step": 26730,
      "train_flow_matching_loss": 0.140625,
      "train_physics_loss": 0.0
    },
    {
      "step": 26730,
      "train_flow_matching_loss": 0.15625,
      "train_physics_loss": 0.0
    },
    {
      "step": 26730,
      "train_flow_matching_loss": 0.07373046875,
      "train_physics_loss": 0.21398644149303436
    },
    {
      "grad_norm": 0.10364880412817001,
      "learning_rate": 5e-05,
      "loss": 0.1155,
      "step": 26740
    },
    {
      "step": 26740,
      "train_flow_matching_loss": 0.033203125,
      "train_physics_loss": 0.1300385594367981
    },
    {
      "step": 26740,
      "train_flow_matching_loss": 0.1474609375,
      "train_physics_loss": 0.0
    },
    {
      "step": 26740,
      "train_flow_matching_loss": 0.11865234375,
      "train_physics_loss": 0.0
    },
    {
      "step": 26740,
      "train_flow_matching_loss": 0.057861328125,
      "train_physics_loss": 0.3580559492111206
    },
    {
      "grad_norm": 0.10292547941207886,
      "learning_rate": 5e-05,
      "loss": 0.1182,
      "step": 26750
    },
    {
      "step": 26750,
      "train_flow_matching_loss": 0.055419921875,
      "train_physics_loss": 0.2426108568906784
    },
    {
      "step": 26750,
      "train_flow_matching_loss": 0.1796875,
      "train_physics_loss": 0.0
    },
    {
      "step": 26750,
      "train_flow_matching_loss": 0.0986328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 26750,
      "train_flow_matching_loss": 0.0283203125,
      "train_physics_loss": 0.12850157916545868
    },
    {
      "grad_norm": 0.14011211693286896,
      "learning_rate": 5e-05,
      "loss": 0.1195,
      "step": 26760
    },
    {
      "step": 26760,
      "train_flow_matching_loss": 0.042724609375,
      "train_physics_loss": 0.1623150110244751
    },
    {
      "step": 26760,
      "train_flow_matching_loss": 0.1630859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 26760,
      "train_flow_matching_loss": 0.0751953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 26760,
      "train_flow_matching_loss": 0.10546875,
      "train_physics_loss": 0.1858064979314804
    },
    {
      "grad_norm": 0.10752060264348984,
      "learning_rate": 5e-05,
      "loss": 0.1203,
      "step": 26770
    },
    {
      "step": 26770,
      "train_flow_matching_loss": 0.038330078125,
      "train_physics_loss": 0.17918464541435242
    },
    {
      "step": 26770,
      "train_flow_matching_loss": 0.1953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 26770,
      "train_flow_matching_loss": 0.11279296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 26770,
      "train_flow_matching_loss": 0.048583984375,
      "train_physics_loss": 0.17206493020057678
    },
    {
      "grad_norm": 0.10765362530946732,
      "learning_rate": 5e-05,
      "loss": 0.1201,
      "step": 26780
    },
    {
      "step": 26780,
      "train_flow_matching_loss": 0.1181640625,
      "train_physics_loss": 0.14656862616539001
    },
    {
      "step": 26780,
      "train_flow_matching_loss": 0.1669921875,
      "train_physics_loss": 0.0
    },
    {
      "step": 26780,
      "train_flow_matching_loss": 0.08984375,
      "train_physics_loss": 0.0
    },
    {
      "step": 26780,
      "train_flow_matching_loss": 0.033935546875,
      "train_physics_loss": 0.1360200196504593
    },
    {
      "grad_norm": 0.08047866821289062,
      "learning_rate": 5e-05,
      "loss": 0.1231,
      "step": 26790
    },
    {
      "step": 26790,
      "train_flow_matching_loss": 0.036376953125,
      "train_physics_loss": 0.15516182780265808
    },
    {
      "step": 26790,
      "train_flow_matching_loss": 0.291015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 26790,
      "train_flow_matching_loss": 0.1787109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 26790,
      "train_flow_matching_loss": 0.041015625,
      "train_physics_loss": 0.15308928489685059
    },
    {
      "grad_norm": 0.15973852574825287,
      "learning_rate": 5e-05,
      "loss": 0.1257,
      "step": 26800
    },
    {
      "step": 26800,
      "train_flow_matching_loss": 0.07861328125,
      "train_physics_loss": 0.41992589831352234
    },
    {
      "step": 26800,
      "train_flow_matching_loss": 0.1494140625,
      "train_physics_loss": 0.0
    },
    {
      "step": 26800,
      "train_flow_matching_loss": 0.1630859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 26800,
      "train_flow_matching_loss": 0.0859375,
      "train_physics_loss": 0.1229613646864891
    },
    {
      "grad_norm": 0.123814158141613,
      "learning_rate": 5e-05,
      "loss": 0.122,
      "step": 26810
    },
    {
      "step": 26810,
      "train_flow_matching_loss": 0.03759765625,
      "train_physics_loss": 0.2487003207206726
    },
    {
      "step": 26810,
      "train_flow_matching_loss": 0.2021484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 26810,
      "train_flow_matching_loss": 0.189453125,
      "train_physics_loss": 0.0
    },
    {
      "step": 26810,
      "train_flow_matching_loss": 0.05224609375,
      "train_physics_loss": 0.16560585796833038
    },
    {
      "grad_norm": 0.0816858783364296,
      "learning_rate": 5e-05,
      "loss": 0.1222,
      "step": 26820
    },
    {
      "step": 26820,
      "train_flow_matching_loss": 0.05322265625,
      "train_physics_loss": 0.19067160785198212
    },
    {
      "step": 26820,
      "train_flow_matching_loss": 0.1494140625,
      "train_physics_loss": 0.0
    },
    {
      "step": 26820,
      "train_flow_matching_loss": 0.12158203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 26820,
      "train_flow_matching_loss": 0.031494140625,
      "train_physics_loss": 0.24997501075267792
    },
    {
      "grad_norm": 0.06500141322612762,
      "learning_rate": 5e-05,
      "loss": 0.1215,
      "step": 26830
    },
    {
      "step": 26830,
      "train_flow_matching_loss": 0.04736328125,
      "train_physics_loss": 0.20493613183498383
    },
    {
      "step": 26830,
      "train_flow_matching_loss": 0.146484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 26830,
      "train_flow_matching_loss": 0.1416015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 26830,
      "train_flow_matching_loss": 0.0277099609375,
      "train_physics_loss": 0.12225983291864395
    },
    {
      "grad_norm": 0.07754265516996384,
      "learning_rate": 5e-05,
      "loss": 0.1282,
      "step": 26840
    },
    {
      "step": 26840,
      "train_flow_matching_loss": 0.03857421875,
      "train_physics_loss": 0.17369574308395386
    },
    {
      "step": 26840,
      "train_flow_matching_loss": 0.171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 26840,
      "train_flow_matching_loss": 0.09228515625,
      "train_physics_loss": 0.0
    },
    {
      "step": 26840,
      "train_flow_matching_loss": 0.060546875,
      "train_physics_loss": 0.10443925857543945
    },
    {
      "grad_norm": 0.10229641199111938,
      "learning_rate": 5e-05,
      "loss": 0.1258,
      "step": 26850
    },
    {
      "step": 26850,
      "train_flow_matching_loss": 0.08349609375,
      "train_physics_loss": 0.24017110466957092
    },
    {
      "step": 26850,
      "train_flow_matching_loss": 0.185546875,
      "train_physics_loss": 0.0
    },
    {
      "step": 26850,
      "train_flow_matching_loss": 0.11376953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 26850,
      "train_flow_matching_loss": 0.045166015625,
      "train_physics_loss": 0.13124649226665497
    },
    {
      "grad_norm": 0.122431181371212,
      "learning_rate": 5e-05,
      "loss": 0.1261,
      "step": 26860
    },
    {
      "step": 26860,
      "train_flow_matching_loss": 0.0283203125,
      "train_physics_loss": 0.126786470413208
    },
    {
      "step": 26860,
      "train_flow_matching_loss": 0.173828125,
      "train_physics_loss": 0.0
    },
    {
      "step": 26860,
      "train_flow_matching_loss": 0.193359375,
      "train_physics_loss": 0.0
    },
    {
      "step": 26860,
      "train_flow_matching_loss": 0.04931640625,
      "train_physics_loss": 0.1629249006509781
    },
    {
      "grad_norm": 0.09392596781253815,
      "learning_rate": 5e-05,
      "loss": 0.124,
      "step": 26870
    },
    {
      "step": 26870,
      "train_flow_matching_loss": 0.03466796875,
      "train_physics_loss": 0.1403358429670334
    },
    {
      "step": 26870,
      "train_flow_matching_loss": 0.1708984375,
      "train_physics_loss": 0.0
    },
    {
      "step": 26870,
      "train_flow_matching_loss": 0.1435546875,
      "train_physics_loss": 0.0
    },
    {
      "step": 26870,
      "train_flow_matching_loss": 0.047119140625,
      "train_physics_loss": 0.16205187141895294
    },
    {
      "grad_norm": 0.09481934458017349,
      "learning_rate": 5e-05,
      "loss": 0.1271,
      "step": 26880
    },
    {
      "step": 26880,
      "train_flow_matching_loss": 0.041259765625,
      "train_physics_loss": 0.15397188067436218
    },
    {
      "step": 26880,
      "train_flow_matching_loss": 0.1650390625,
      "train_physics_loss": 0.0
    },
    {
      "step": 26880,
      "train_flow_matching_loss": 0.166015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 26880,
      "train_flow_matching_loss": 0.05712890625,
      "train_physics_loss": 0.19366927444934845
    },
    {
      "grad_norm": 0.08321713656187057,
      "learning_rate": 5e-05,
      "loss": 0.1199,
      "step": 26890
    },
    {
      "step": 26890,
      "train_flow_matching_loss": 0.07275390625,
      "train_physics_loss": 0.1291649341583252
    },
    {
      "step": 26890,
      "train_flow_matching_loss": 0.1201171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 26890,
      "train_flow_matching_loss": 0.130859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 26890,
      "train_flow_matching_loss": 0.029296875,
      "train_physics_loss": 0.14770419895648956
    },
    {
      "grad_norm": 0.11163809895515442,
      "learning_rate": 5e-05,
      "loss": 0.1234,
      "step": 26900
    },
    {
      "step": 26900,
      "train_flow_matching_loss": 0.054931640625,
      "train_physics_loss": 0.12510575354099274
    },
    {
      "step": 26900,
      "train_flow_matching_loss": 0.2119140625,
      "train_physics_loss": 0.0
    },
    {
      "step": 26900,
      "train_flow_matching_loss": 0.1064453125,
      "train_physics_loss": 0.0
    },
    {
      "step": 26900,
      "train_flow_matching_loss": 0.046630859375,
      "train_physics_loss": 0.1415366679430008
    },
    {
      "grad_norm": 0.10355188697576523,
      "learning_rate": 5e-05,
      "loss": 0.1218,
      "step": 26910
    },
    {
      "step": 26910,
      "train_flow_matching_loss": 0.049560546875,
      "train_physics_loss": 0.1658451408147812
    },
    {
      "step": 26910,
      "train_flow_matching_loss": 0.11376953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 26910,
      "train_flow_matching_loss": 0.177734375,
      "train_physics_loss": 0.0
    },
    {
      "step": 26910,
      "train_flow_matching_loss": 0.0712890625,
      "train_physics_loss": 0.14114072918891907
    },
    {
      "grad_norm": 0.18663977086544037,
      "learning_rate": 5e-05,
      "loss": 0.1187,
      "step": 26920
    },
    {
      "step": 26920,
      "train_flow_matching_loss": 0.0634765625,
      "train_physics_loss": 0.17794740200042725
    },
    {
      "step": 26920,
      "train_flow_matching_loss": 0.1923828125,
      "train_physics_loss": 0.0
    },
    {
      "step": 26920,
      "train_flow_matching_loss": 0.078125,
      "train_physics_loss": 0.0
    },
    {
      "step": 26920,
      "train_flow_matching_loss": 0.0274658203125,
      "train_physics_loss": 0.10798380523920059
    },
    {
      "grad_norm": 0.09687449038028717,
      "learning_rate": 5e-05,
      "loss": 0.1181,
      "step": 26930
    },
    {
      "step": 26930,
      "train_flow_matching_loss": 0.041748046875,
      "train_physics_loss": 0.1581435650587082
    },
    {
      "step": 26930,
      "train_flow_matching_loss": 0.33203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 26930,
      "train_flow_matching_loss": 0.1259765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 26930,
      "train_flow_matching_loss": 0.031494140625,
      "train_physics_loss": 0.15262585878372192
    },
    {
      "grad_norm": 0.18430085480213165,
      "learning_rate": 5e-05,
      "loss": 0.1291,
      "step": 26940
    },
    {
      "step": 26940,
      "train_flow_matching_loss": 0.051025390625,
      "train_physics_loss": 0.17686402797698975
    },
    {
      "step": 26940,
      "train_flow_matching_loss": 0.1484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 26940,
      "train_flow_matching_loss": 0.12060546875,
      "train_physics_loss": 0.0
    },
    {
      "step": 26940,
      "train_flow_matching_loss": 0.033935546875,
      "train_physics_loss": 0.1344153732061386
    },
    {
      "grad_norm": 0.10653308779001236,
      "learning_rate": 5e-05,
      "loss": 0.1206,
      "step": 26950
    },
    {
      "step": 26950,
      "train_flow_matching_loss": 0.03955078125,
      "train_physics_loss": 0.11784739047288895
    },
    {
      "step": 26950,
      "train_flow_matching_loss": 0.109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 26950,
      "train_flow_matching_loss": 0.173828125,
      "train_physics_loss": 0.0
    },
    {
      "step": 26950,
      "train_flow_matching_loss": 0.0546875,
      "train_physics_loss": 0.16191792488098145
    },
    {
      "grad_norm": 0.09535860270261765,
      "learning_rate": 5e-05,
      "loss": 0.1198,
      "step": 26960
    },
    {
      "step": 26960,
      "train_flow_matching_loss": 0.039306640625,
      "train_physics_loss": 0.12788531184196472
    },
    {
      "step": 26960,
      "train_flow_matching_loss": 0.1357421875,
      "train_physics_loss": 0.0
    },
    {
      "step": 26960,
      "train_flow_matching_loss": 0.1025390625,
      "train_physics_loss": 0.0
    },
    {
      "step": 26960,
      "train_flow_matching_loss": 0.07177734375,
      "train_physics_loss": 0.14722687005996704
    },
    {
      "grad_norm": 0.11691097915172577,
      "learning_rate": 5e-05,
      "loss": 0.1204,
      "step": 26970
    },
    {
      "step": 26970,
      "train_flow_matching_loss": 0.0306396484375,
      "train_physics_loss": 0.13082323968410492
    },
    {
      "step": 26970,
      "train_flow_matching_loss": 0.1298828125,
      "train_physics_loss": 0.0
    },
    {
      "step": 26970,
      "train_flow_matching_loss": 0.1337890625,
      "train_physics_loss": 0.0
    },
    {
      "step": 26970,
      "train_flow_matching_loss": 0.048583984375,
      "train_physics_loss": 0.14523987472057343
    },
    {
      "grad_norm": 0.27876731753349304,
      "learning_rate": 5e-05,
      "loss": 0.1179,
      "step": 26980
    },
    {
      "step": 26980,
      "train_flow_matching_loss": 0.030517578125,
      "train_physics_loss": 0.11455965787172318
    },
    {
      "step": 26980,
      "train_flow_matching_loss": 0.1083984375,
      "train_physics_loss": 0.0
    },
    {
      "step": 26980,
      "train_flow_matching_loss": 0.1826171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 26980,
      "train_flow_matching_loss": 0.06689453125,
      "train_physics_loss": 0.14455382525920868
    },
    {
      "grad_norm": 0.11956176906824112,
      "learning_rate": 5e-05,
      "loss": 0.1257,
      "step": 26990
    },
    {
      "step": 26990,
      "train_flow_matching_loss": 0.043212890625,
      "train_physics_loss": 0.13404138386249542
    },
    {
      "step": 26990,
      "train_flow_matching_loss": 0.142578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 26990,
      "train_flow_matching_loss": 0.1767578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 26990,
      "train_flow_matching_loss": 0.07177734375,
      "train_physics_loss": 0.1854846477508545
    },
    {
      "grad_norm": 0.05802999809384346,
      "learning_rate": 5e-05,
      "loss": 0.1209,
      "step": 27000
    },
    {
      "step": 27000,
      "train_flow_matching_loss": 0.0252685546875,
      "train_physics_loss": 0.1341009885072708
    },
    {
      "step": 27000,
      "train_flow_matching_loss": 0.1982421875,
      "train_physics_loss": 0.0
    },
    {
      "step": 27000,
      "train_flow_matching_loss": 0.12353515625,
      "train_physics_loss": 0.0
    },
    {
      "step": 27000,
      "train_flow_matching_loss": 0.06982421875,
      "train_physics_loss": 0.15936441719532013
    },
    {
      "grad_norm": 0.10498743504285812,
      "learning_rate": 5e-05,
      "loss": 0.1259,
      "step": 27010
    },
    {
      "step": 27010,
      "train_flow_matching_loss": 0.0277099609375,
      "train_physics_loss": 0.14800839126110077
    },
    {
      "step": 27010,
      "train_flow_matching_loss": 0.1123046875,
      "train_physics_loss": 0.0
    },
    {
      "step": 27010,
      "train_flow_matching_loss": 0.1279296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 27010,
      "train_flow_matching_loss": 0.1484375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11936069279909134,
      "learning_rate": 5e-05,
      "loss": 0.1203,
      "step": 27020
    },
    {
      "step": 27020,
      "train_flow_matching_loss": 0.031982421875,
      "train_physics_loss": 0.1439812183380127
    },
    {
      "step": 27020,
      "train_flow_matching_loss": 0.0947265625,
      "train_physics_loss": 0.0
    },
    {
      "step": 27020,
      "train_flow_matching_loss": 0.10107421875,
      "train_physics_loss": 0.0
    },
    {
      "step": 27020,
      "train_flow_matching_loss": 0.0927734375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10528150200843811,
      "learning_rate": 5e-05,
      "loss": 0.1199,
      "step": 27030
    },
    {
      "step": 27030,
      "train_flow_matching_loss": 0.033447265625,
      "train_physics_loss": 0.14374490082263947
    },
    {
      "step": 27030,
      "train_flow_matching_loss": 0.1591796875,
      "train_physics_loss": 0.0
    },
    {
      "step": 27030,
      "train_flow_matching_loss": 0.1337890625,
      "train_physics_loss": 0.0
    },
    {
      "step": 27030,
      "train_flow_matching_loss": 0.07275390625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10319763422012329,
      "learning_rate": 5e-05,
      "loss": 0.1194,
      "step": 27040
    },
    {
      "step": 27040,
      "train_flow_matching_loss": 0.05908203125,
      "train_physics_loss": 0.16463235020637512
    },
    {
      "step": 27040,
      "train_flow_matching_loss": 0.0849609375,
      "train_physics_loss": 0.0
    },
    {
      "step": 27040,
      "train_flow_matching_loss": 0.126953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 27040,
      "train_flow_matching_loss": 0.1279296875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08039037883281708,
      "learning_rate": 5e-05,
      "loss": 0.1225,
      "step": 27050
    },
    {
      "step": 27050,
      "train_flow_matching_loss": 0.0283203125,
      "train_physics_loss": 0.09396496415138245
    },
    {
      "step": 27050,
      "train_flow_matching_loss": 0.1640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 27050,
      "train_flow_matching_loss": 0.21875,
      "train_physics_loss": 0.0
    },
    {
      "step": 27050,
      "train_flow_matching_loss": 0.09716796875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.713290810585022,
      "learning_rate": 5e-05,
      "loss": 0.1178,
      "step": 27060
    },
    {
      "step": 27060,
      "train_flow_matching_loss": 0.050048828125,
      "train_physics_loss": 0.13640223443508148
    },
    {
      "step": 27060,
      "train_flow_matching_loss": 0.216796875,
      "train_physics_loss": 0.0
    },
    {
      "step": 27060,
      "train_flow_matching_loss": 0.10498046875,
      "train_physics_loss": 0.0
    },
    {
      "step": 27060,
      "train_flow_matching_loss": 0.109375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11772450804710388,
      "learning_rate": 5e-05,
      "loss": 0.1206,
      "step": 27070
    },
    {
      "step": 27070,
      "train_flow_matching_loss": 0.041015625,
      "train_physics_loss": 0.1989639401435852
    },
    {
      "step": 27070,
      "train_flow_matching_loss": 0.2412109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 27070,
      "train_flow_matching_loss": 0.2373046875,
      "train_physics_loss": 0.0
    },
    {
      "step": 27070,
      "train_flow_matching_loss": 0.109375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10479573905467987,
      "learning_rate": 5e-05,
      "loss": 0.1166,
      "step": 27080
    },
    {
      "step": 27080,
      "train_flow_matching_loss": 0.033935546875,
      "train_physics_loss": 0.10448110103607178
    },
    {
      "step": 27080,
      "train_flow_matching_loss": 0.1767578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 27080,
      "train_flow_matching_loss": 0.10693359375,
      "train_physics_loss": 0.0
    },
    {
      "step": 27080,
      "train_flow_matching_loss": 0.1376953125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.07753001898527145,
      "learning_rate": 5e-05,
      "loss": 0.1225,
      "step": 27090
    },
    {
      "step": 27090,
      "train_flow_matching_loss": 0.1181640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 27090,
      "train_flow_matching_loss": 0.0830078125,
      "train_physics_loss": 0.0
    },
    {
      "step": 27090,
      "train_flow_matching_loss": 0.1201171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 27090,
      "train_flow_matching_loss": 0.1708984375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11919461190700531,
      "learning_rate": 5e-05,
      "loss": 0.1205,
      "step": 27100
    },
    {
      "step": 27100,
      "train_flow_matching_loss": 0.12890625,
      "train_physics_loss": 0.0
    },
    {
      "step": 27100,
      "train_flow_matching_loss": 0.1484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 27100,
      "train_flow_matching_loss": 0.09716796875,
      "train_physics_loss": 0.0
    },
    {
      "step": 27100,
      "train_flow_matching_loss": 0.08837890625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11779075860977173,
      "learning_rate": 5e-05,
      "loss": 0.121,
      "step": 27110
    },
    {
      "step": 27110,
      "train_flow_matching_loss": 0.126953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 27110,
      "train_flow_matching_loss": 0.0888671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 27110,
      "train_flow_matching_loss": 0.134765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 27110,
      "train_flow_matching_loss": 0.09521484375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09556315094232559,
      "learning_rate": 5e-05,
      "loss": 0.1172,
      "step": 27120
    },
    {
      "step": 27120,
      "train_flow_matching_loss": 0.138671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 27120,
      "train_flow_matching_loss": 0.1123046875,
      "train_physics_loss": 0.0
    },
    {
      "step": 27120,
      "train_flow_matching_loss": 0.1279296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 27120,
      "train_flow_matching_loss": 0.13671875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10476851463317871,
      "learning_rate": 5e-05,
      "loss": 0.12,
      "step": 27130
    },
    {
      "step": 27130,
      "train_flow_matching_loss": 0.109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 27130,
      "train_flow_matching_loss": 0.08349609375,
      "train_physics_loss": 0.0
    },
    {
      "step": 27130,
      "train_flow_matching_loss": 0.142578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 27130,
      "train_flow_matching_loss": 0.11083984375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10474436730146408,
      "learning_rate": 5e-05,
      "loss": 0.1168,
      "step": 27140
    },
    {
      "step": 27140,
      "train_flow_matching_loss": 0.1181640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 27140,
      "train_flow_matching_loss": 0.177734375,
      "train_physics_loss": 0.0
    },
    {
      "step": 27140,
      "train_flow_matching_loss": 0.1279296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 27140,
      "train_flow_matching_loss": 0.1455078125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09684763848781586,
      "learning_rate": 5e-05,
      "loss": 0.1223,
      "step": 27150
    },
    {
      "step": 27150,
      "train_flow_matching_loss": 0.11865234375,
      "train_physics_loss": 0.0
    },
    {
      "step": 27150,
      "train_flow_matching_loss": 0.12890625,
      "train_physics_loss": 0.0
    },
    {
      "step": 27150,
      "train_flow_matching_loss": 0.142578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 27150,
      "train_flow_matching_loss": 0.134765625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11985542625188828,
      "learning_rate": 5e-05,
      "loss": 0.119,
      "step": 27160
    },
    {
      "step": 27160,
      "train_flow_matching_loss": 0.126953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 27160,
      "train_flow_matching_loss": 0.166015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 27160,
      "train_flow_matching_loss": 0.08154296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 27160,
      "train_flow_matching_loss": 0.12353515625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.12566359341144562,
      "learning_rate": 5e-05,
      "loss": 0.1159,
      "step": 27170
    },
    {
      "step": 27170,
      "train_flow_matching_loss": 0.150390625,
      "train_physics_loss": 0.0
    },
    {
      "step": 27170,
      "train_flow_matching_loss": 0.11474609375,
      "train_physics_loss": 0.0
    },
    {
      "step": 27170,
      "train_flow_matching_loss": 0.1328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 27170,
      "train_flow_matching_loss": 0.11669921875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.087064728140831,
      "learning_rate": 5e-05,
      "loss": 0.1273,
      "step": 27180
    },
    {
      "step": 27180,
      "train_flow_matching_loss": 0.09814453125,
      "train_physics_loss": 0.0
    },
    {
      "step": 27180,
      "train_flow_matching_loss": 0.12451171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 27180,
      "train_flow_matching_loss": 0.208984375,
      "train_physics_loss": 0.0
    },
    {
      "step": 27180,
      "train_flow_matching_loss": 0.1181640625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.13373510539531708,
      "learning_rate": 5e-05,
      "loss": 0.1215,
      "step": 27190
    },
    {
      "step": 27190,
      "train_flow_matching_loss": 0.11572265625,
      "train_physics_loss": 0.0
    },
    {
      "step": 27190,
      "train_flow_matching_loss": 0.109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 27190,
      "train_flow_matching_loss": 0.1376953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 27190,
      "train_flow_matching_loss": 0.11572265625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09812599420547485,
      "learning_rate": 5e-05,
      "loss": 0.1203,
      "step": 27200
    },
    {
      "step": 27200,
      "train_flow_matching_loss": 0.09375,
      "train_physics_loss": 0.0
    },
    {
      "step": 27200,
      "train_flow_matching_loss": 0.2421875,
      "train_physics_loss": 0.0
    },
    {
      "step": 27200,
      "train_flow_matching_loss": 0.11328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 27200,
      "train_flow_matching_loss": 0.0947265625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.06299819052219391,
      "learning_rate": 5e-05,
      "loss": 0.1159,
      "step": 27210
    },
    {
      "step": 27210,
      "train_flow_matching_loss": 0.1572265625,
      "train_physics_loss": 0.0
    },
    {
      "step": 27210,
      "train_flow_matching_loss": 0.10205078125,
      "train_physics_loss": 0.0
    },
    {
      "step": 27210,
      "train_flow_matching_loss": 0.1435546875,
      "train_physics_loss": 0.0
    },
    {
      "step": 27210,
      "train_flow_matching_loss": 0.15625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10218831896781921,
      "learning_rate": 5e-05,
      "loss": 0.1135,
      "step": 27220
    },
    {
      "step": 27220,
      "train_flow_matching_loss": 0.12060546875,
      "train_physics_loss": 0.0
    },
    {
      "step": 27220,
      "train_flow_matching_loss": 0.1728515625,
      "train_physics_loss": 0.0
    },
    {
      "step": 27220,
      "train_flow_matching_loss": 0.07421875,
      "train_physics_loss": 0.0
    },
    {
      "step": 27220,
      "train_flow_matching_loss": 0.134765625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08675751090049744,
      "learning_rate": 5e-05,
      "loss": 0.1172,
      "step": 27230
    },
    {
      "step": 27230,
      "train_flow_matching_loss": 0.142578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 27230,
      "train_flow_matching_loss": 0.15625,
      "train_physics_loss": 0.0
    },
    {
      "step": 27230,
      "train_flow_matching_loss": 0.12451171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 27230,
      "train_flow_matching_loss": 0.10302734375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.1299685388803482,
      "learning_rate": 5e-05,
      "loss": 0.1202,
      "step": 27240
    },
    {
      "step": 27240,
      "train_flow_matching_loss": 0.1171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 27240,
      "train_flow_matching_loss": 0.1494140625,
      "train_physics_loss": 0.0
    },
    {
      "step": 27240,
      "train_flow_matching_loss": 0.10107421875,
      "train_physics_loss": 0.0
    },
    {
      "step": 27240,
      "train_flow_matching_loss": 0.1787109375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.1298971176147461,
      "learning_rate": 5e-05,
      "loss": 0.1168,
      "step": 27250
    },
    {
      "step": 27250,
      "train_flow_matching_loss": 0.1455078125,
      "train_physics_loss": 0.0
    },
    {
      "step": 27250,
      "train_flow_matching_loss": 0.064453125,
      "train_physics_loss": 0.0
    },
    {
      "step": 27250,
      "train_flow_matching_loss": 0.09619140625,
      "train_physics_loss": 0.0
    },
    {
      "step": 27250,
      "train_flow_matching_loss": 0.1875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10867169499397278,
      "learning_rate": 5e-05,
      "loss": 0.1195,
      "step": 27260
    },
    {
      "step": 27260,
      "train_flow_matching_loss": 0.146484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 27260,
      "train_flow_matching_loss": 0.1298828125,
      "train_physics_loss": 0.0
    },
    {
      "step": 27260,
      "train_flow_matching_loss": 0.07080078125,
      "train_physics_loss": 0.0
    },
    {
      "step": 27260,
      "train_flow_matching_loss": 0.017822265625,
      "train_physics_loss": 0.11458732932806015
    },
    {
      "grad_norm": 0.10355269908905029,
      "learning_rate": 5e-05,
      "loss": 0.1173,
      "step": 27270
    },
    {
      "step": 27270,
      "train_flow_matching_loss": 0.1455078125,
      "train_physics_loss": 0.0
    },
    {
      "step": 27270,
      "train_flow_matching_loss": 0.1689453125,
      "train_physics_loss": 0.0
    },
    {
      "step": 27270,
      "train_flow_matching_loss": 0.2412109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 27270,
      "train_flow_matching_loss": 0.0272216796875,
      "train_physics_loss": 0.13148151338100433
    },
    {
      "grad_norm": 0.10765101760625839,
      "learning_rate": 5e-05,
      "loss": 0.123,
      "step": 27280
    },
    {
      "step": 27280,
      "train_flow_matching_loss": 0.08740234375,
      "train_physics_loss": 0.0
    },
    {
      "step": 27280,
      "train_flow_matching_loss": 0.1220703125,
      "train_physics_loss": 0.0
    },
    {
      "step": 27280,
      "train_flow_matching_loss": 0.2021484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 27280,
      "train_flow_matching_loss": 0.1171875,
      "train_physics_loss": 0.18226048350334167
    },
    {
      "grad_norm": 0.14215336740016937,
      "learning_rate": 5e-05,
      "loss": 0.1218,
      "step": 27290
    },
    {
      "step": 27290,
      "train_flow_matching_loss": 0.12158203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 27290,
      "train_flow_matching_loss": 0.134765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 27290,
      "train_flow_matching_loss": 0.1904296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 27290,
      "train_flow_matching_loss": 0.1044921875,
      "train_physics_loss": 0.11156371980905533
    },
    {
      "grad_norm": 0.1142609566450119,
      "learning_rate": 5e-05,
      "loss": 0.1203,
      "step": 27300
    },
    {
      "step": 27300,
      "train_flow_matching_loss": 0.1162109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 27300,
      "train_flow_matching_loss": 0.181640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 27300,
      "train_flow_matching_loss": 0.1669921875,
      "train_physics_loss": 0.0
    },
    {
      "step": 27300,
      "train_flow_matching_loss": 0.058837890625,
      "train_physics_loss": 0.14072547852993011
    },
    {
      "grad_norm": 0.11711039394140244,
      "learning_rate": 5e-05,
      "loss": 0.1156,
      "step": 27310
    },
    {
      "step": 27310,
      "train_flow_matching_loss": 0.134765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 27310,
      "train_flow_matching_loss": 0.1142578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 27310,
      "train_flow_matching_loss": 0.0986328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 27310,
      "train_flow_matching_loss": 0.07568359375,
      "train_physics_loss": 0.2087673842906952
    },
    {
      "grad_norm": 0.09480424225330353,
      "learning_rate": 5e-05,
      "loss": 0.121,
      "step": 27320
    },
    {
      "step": 27320,
      "train_flow_matching_loss": 0.09765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 27320,
      "train_flow_matching_loss": 0.166015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 27320,
      "train_flow_matching_loss": 0.166015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 27320,
      "train_flow_matching_loss": 0.037353515625,
      "train_physics_loss": 0.16666026413440704
    },
    {
      "grad_norm": 0.10822032392024994,
      "learning_rate": 5e-05,
      "loss": 0.1166,
      "step": 27330
    },
    {
      "step": 27330,
      "train_flow_matching_loss": 0.166015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 27330,
      "train_flow_matching_loss": 0.2470703125,
      "train_physics_loss": 0.0
    },
    {
      "step": 27330,
      "train_flow_matching_loss": 0.1279296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 27330,
      "train_flow_matching_loss": 0.06494140625,
      "train_physics_loss": 0.17801779508590698
    },
    {
      "grad_norm": 0.09020990133285522,
      "learning_rate": 5e-05,
      "loss": 0.1221,
      "step": 27340
    },
    {
      "step": 27340,
      "train_flow_matching_loss": 0.15625,
      "train_physics_loss": 0.0
    },
    {
      "step": 27340,
      "train_flow_matching_loss": 0.193359375,
      "train_physics_loss": 0.0
    },
    {
      "step": 27340,
      "train_flow_matching_loss": 0.1591796875,
      "train_physics_loss": 0.0
    },
    {
      "step": 27340,
      "train_flow_matching_loss": 0.03857421875,
      "train_physics_loss": 0.15886953473091125
    },
    {
      "grad_norm": 0.10858635604381561,
      "learning_rate": 5e-05,
      "loss": 0.1256,
      "step": 27350
    },
    {
      "step": 27350,
      "train_flow_matching_loss": 0.216796875,
      "train_physics_loss": 0.0
    },
    {
      "step": 27350,
      "train_flow_matching_loss": 0.10595703125,
      "train_physics_loss": 0.0
    },
    {
      "step": 27350,
      "train_flow_matching_loss": 0.10107421875,
      "train_physics_loss": 0.0
    },
    {
      "step": 27350,
      "train_flow_matching_loss": 0.032470703125,
      "train_physics_loss": 0.14066816866397858
    },
    {
      "grad_norm": 0.0787900760769844,
      "learning_rate": 5e-05,
      "loss": 0.1237,
      "step": 27360
    },
    {
      "step": 27360,
      "train_flow_matching_loss": 0.154296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 27360,
      "train_flow_matching_loss": 0.1337890625,
      "train_physics_loss": 0.0
    },
    {
      "step": 27360,
      "train_flow_matching_loss": 0.08203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 27360,
      "train_flow_matching_loss": 0.0216064453125,
      "train_physics_loss": 0.1298658549785614
    },
    {
      "grad_norm": 0.09362740814685822,
      "learning_rate": 5e-05,
      "loss": 0.1241,
      "step": 27370
    },
    {
      "step": 27370,
      "train_flow_matching_loss": 0.09375,
      "train_physics_loss": 0.0
    },
    {
      "step": 27370,
      "train_flow_matching_loss": 0.1611328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 27370,
      "train_flow_matching_loss": 0.10009765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 27370,
      "train_flow_matching_loss": 0.040771484375,
      "train_physics_loss": 0.13575007021427155
    },
    {
      "grad_norm": 0.14473991096019745,
      "learning_rate": 5e-05,
      "loss": 0.1153,
      "step": 27380
    },
    {
      "step": 27380,
      "train_flow_matching_loss": 0.1796875,
      "train_physics_loss": 0.0
    },
    {
      "step": 27380,
      "train_flow_matching_loss": 0.1865234375,
      "train_physics_loss": 0.0
    },
    {
      "step": 27380,
      "train_flow_matching_loss": 0.1884765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 27380,
      "train_flow_matching_loss": 0.029541015625,
      "train_physics_loss": 0.15023311972618103
    },
    {
      "grad_norm": 0.10484857112169266,
      "learning_rate": 5e-05,
      "loss": 0.1225,
      "step": 27390
    },
    {
      "step": 27390,
      "train_flow_matching_loss": 0.09765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 27390,
      "train_flow_matching_loss": 0.1142578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 27390,
      "train_flow_matching_loss": 0.0986328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 27390,
      "train_flow_matching_loss": 0.050537109375,
      "train_physics_loss": 0.20012304186820984
    },
    {
      "grad_norm": 0.29585084319114685,
      "learning_rate": 5e-05,
      "loss": 0.1212,
      "step": 27400
    },
    {
      "step": 27400,
      "train_flow_matching_loss": 0.10986328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 27400,
      "train_flow_matching_loss": 0.1328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 27400,
      "train_flow_matching_loss": 0.185546875,
      "train_physics_loss": 0.0
    },
    {
      "step": 27400,
      "train_flow_matching_loss": 0.06494140625,
      "train_physics_loss": 0.12130194157361984
    },
    {
      "grad_norm": 0.08687400072813034,
      "learning_rate": 5e-05,
      "loss": 0.1208,
      "step": 27410
    },
    {
      "step": 27410,
      "train_flow_matching_loss": 0.1435546875,
      "train_physics_loss": 0.0
    },
    {
      "step": 27410,
      "train_flow_matching_loss": 0.154296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 27410,
      "train_flow_matching_loss": 0.1435546875,
      "train_physics_loss": 0.0
    },
    {
      "step": 27410,
      "train_flow_matching_loss": 0.041748046875,
      "train_physics_loss": 0.10887071490287781
    },
    {
      "grad_norm": 0.0801505446434021,
      "learning_rate": 5e-05,
      "loss": 0.1234,
      "step": 27420
    },
    {
      "step": 27420,
      "train_flow_matching_loss": 0.1875,
      "train_physics_loss": 0.0
    },
    {
      "step": 27420,
      "train_flow_matching_loss": 0.134765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 27420,
      "train_flow_matching_loss": 0.11376953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 27420,
      "train_flow_matching_loss": 0.06298828125,
      "train_physics_loss": 0.15260908007621765
    },
    {
      "grad_norm": 0.20322294533252716,
      "learning_rate": 5e-05,
      "loss": 0.1181,
      "step": 27430
    },
    {
      "step": 27430,
      "train_flow_matching_loss": 0.1064453125,
      "train_physics_loss": 0.0
    },
    {
      "step": 27430,
      "train_flow_matching_loss": 0.1416015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 27430,
      "train_flow_matching_loss": 0.228515625,
      "train_physics_loss": 0.0
    },
    {
      "step": 27430,
      "train_flow_matching_loss": 0.04833984375,
      "train_physics_loss": 0.14079786837100983
    },
    {
      "grad_norm": 0.07628343999385834,
      "learning_rate": 5e-05,
      "loss": 0.1187,
      "step": 27440
    },
    {
      "step": 27440,
      "train_flow_matching_loss": 0.1357421875,
      "train_physics_loss": 0.0
    },
    {
      "step": 27440,
      "train_flow_matching_loss": 0.1416015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 27440,
      "train_flow_matching_loss": 0.1220703125,
      "train_physics_loss": 0.0
    },
    {
      "step": 27440,
      "train_flow_matching_loss": 0.06787109375,
      "train_physics_loss": 0.1501360982656479
    },
    {
      "grad_norm": 0.08411255478858948,
      "learning_rate": 5e-05,
      "loss": 0.1167,
      "step": 27450
    },
    {
      "step": 27450,
      "train_flow_matching_loss": 0.07177734375,
      "train_physics_loss": 0.0
    },
    {
      "step": 27450,
      "train_flow_matching_loss": 0.1396484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 27450,
      "train_flow_matching_loss": 0.11669921875,
      "train_physics_loss": 0.0
    },
    {
      "step": 27450,
      "train_flow_matching_loss": 0.04345703125,
      "train_physics_loss": 0.13820192217826843
    },
    {
      "grad_norm": 0.08931617438793182,
      "learning_rate": 5e-05,
      "loss": 0.1248,
      "step": 27460
    },
    {
      "step": 27460,
      "train_flow_matching_loss": 0.142578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 27460,
      "train_flow_matching_loss": 0.09912109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 27460,
      "train_flow_matching_loss": 0.1201171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 27460,
      "train_flow_matching_loss": 0.047607421875,
      "train_physics_loss": 0.18519242107868195
    },
    {
      "grad_norm": 0.13307072222232819,
      "learning_rate": 5e-05,
      "loss": 0.122,
      "step": 27470
    },
    {
      "step": 27470,
      "train_flow_matching_loss": 0.1015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 27470,
      "train_flow_matching_loss": 0.142578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 27470,
      "train_flow_matching_loss": 0.1533203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 27470,
      "train_flow_matching_loss": 0.0296630859375,
      "train_physics_loss": 0.16460567712783813
    },
    {
      "grad_norm": 0.11053770780563354,
      "learning_rate": 5e-05,
      "loss": 0.1183,
      "step": 27480
    },
    {
      "step": 27480,
      "train_flow_matching_loss": 0.0966796875,
      "train_physics_loss": 0.0
    },
    {
      "step": 27480,
      "train_flow_matching_loss": 0.1982421875,
      "train_physics_loss": 0.0
    },
    {
      "step": 27480,
      "train_flow_matching_loss": 0.0986328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 27480,
      "train_flow_matching_loss": 0.0205078125,
      "train_physics_loss": 0.1162770688533783
    },
    {
      "grad_norm": 0.12975601851940155,
      "learning_rate": 5e-05,
      "loss": 0.1251,
      "step": 27490
    },
    {
      "step": 27490,
      "train_flow_matching_loss": 0.125,
      "train_physics_loss": 0.0
    },
    {
      "step": 27490,
      "train_flow_matching_loss": 0.1572265625,
      "train_physics_loss": 0.0
    },
    {
      "step": 27490,
      "train_flow_matching_loss": 0.07373046875,
      "train_physics_loss": 0.0
    },
    {
      "step": 27490,
      "train_flow_matching_loss": 0.050537109375,
      "train_physics_loss": 0.105378158390522
    },
    {
      "grad_norm": 0.09642042964696884,
      "learning_rate": 5e-05,
      "loss": 0.1258,
      "step": 27500
    },
    {
      "step": 27500,
      "train_flow_matching_loss": 0.1005859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 27500,
      "train_flow_matching_loss": 0.09423828125,
      "train_physics_loss": 0.0
    },
    {
      "step": 27500,
      "train_flow_matching_loss": 0.109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 27500,
      "train_flow_matching_loss": 0.06591796875,
      "train_physics_loss": 0.16941829025745392
    },
    {
      "grad_norm": 0.09792458266019821,
      "learning_rate": 5e-05,
      "loss": 0.1199,
      "step": 27510
    },
    {
      "step": 27510,
      "train_flow_matching_loss": 0.07421875,
      "train_physics_loss": 0.0
    },
    {
      "step": 27510,
      "train_flow_matching_loss": 0.1279296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 27510,
      "train_flow_matching_loss": 0.12060546875,
      "train_physics_loss": 0.0
    },
    {
      "step": 27510,
      "train_flow_matching_loss": 0.03271484375,
      "train_physics_loss": 0.20036505162715912
    },
    {
      "grad_norm": 0.09494519233703613,
      "learning_rate": 5e-05,
      "loss": 0.1221,
      "step": 27520
    },
    {
      "step": 27520,
      "train_flow_matching_loss": 0.1171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 27520,
      "train_flow_matching_loss": 0.130859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 27520,
      "train_flow_matching_loss": 0.15625,
      "train_physics_loss": 0.0
    },
    {
      "step": 27520,
      "train_flow_matching_loss": 0.15625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11828254163265228,
      "learning_rate": 5e-05,
      "loss": 0.1151,
      "step": 27530
    },
    {
      "step": 27530,
      "train_flow_matching_loss": 0.1630859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 27530,
      "train_flow_matching_loss": 0.1328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 27530,
      "train_flow_matching_loss": 0.13671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 27530,
      "train_flow_matching_loss": 0.1005859375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10973738878965378,
      "learning_rate": 5e-05,
      "loss": 0.1197,
      "step": 27540
    },
    {
      "step": 27540,
      "train_flow_matching_loss": 0.0732421875,
      "train_physics_loss": 0.0
    },
    {
      "step": 27540,
      "train_flow_matching_loss": 0.11376953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 27540,
      "train_flow_matching_loss": 0.125,
      "train_physics_loss": 0.0
    },
    {
      "step": 27540,
      "train_flow_matching_loss": 0.130859375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.0770164504647255,
      "learning_rate": 5e-05,
      "loss": 0.1196,
      "step": 27550
    },
    {
      "step": 27550,
      "train_flow_matching_loss": 0.166015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 27550,
      "train_flow_matching_loss": 0.11083984375,
      "train_physics_loss": 0.0
    },
    {
      "step": 27550,
      "train_flow_matching_loss": 0.193359375,
      "train_physics_loss": 0.0
    },
    {
      "step": 27550,
      "train_flow_matching_loss": 0.1552734375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10275239497423172,
      "learning_rate": 5e-05,
      "loss": 0.1201,
      "step": 27560
    },
    {
      "step": 27560,
      "train_flow_matching_loss": 0.150390625,
      "train_physics_loss": 0.0
    },
    {
      "step": 27560,
      "train_flow_matching_loss": 0.12109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 27560,
      "train_flow_matching_loss": 0.1181640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 27560,
      "train_flow_matching_loss": 0.05712890625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09288587421178818,
      "learning_rate": 5e-05,
      "loss": 0.1201,
      "step": 27570
    },
    {
      "step": 27570,
      "train_flow_matching_loss": 0.1611328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 27570,
      "train_flow_matching_loss": 0.150390625,
      "train_physics_loss": 0.0
    },
    {
      "step": 27570,
      "train_flow_matching_loss": 0.11962890625,
      "train_physics_loss": 0.0
    },
    {
      "step": 27570,
      "train_flow_matching_loss": 0.1689453125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.07837997376918793,
      "learning_rate": 5e-05,
      "loss": 0.1164,
      "step": 27580
    },
    {
      "step": 27580,
      "train_flow_matching_loss": 0.166015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 27580,
      "train_flow_matching_loss": 0.1435546875,
      "train_physics_loss": 0.0
    },
    {
      "step": 27580,
      "train_flow_matching_loss": 0.10791015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 27580,
      "train_flow_matching_loss": 0.07666015625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.1628345549106598,
      "learning_rate": 5e-05,
      "loss": 0.1152,
      "step": 27590
    },
    {
      "step": 27590,
      "train_flow_matching_loss": 0.1982421875,
      "train_physics_loss": 0.0
    },
    {
      "step": 27590,
      "train_flow_matching_loss": 0.10986328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 27590,
      "train_flow_matching_loss": 0.1650390625,
      "train_physics_loss": 0.0
    },
    {
      "step": 27590,
      "train_flow_matching_loss": 0.1611328125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.12840542197227478,
      "learning_rate": 5e-05,
      "loss": 0.1211,
      "step": 27600
    },
    {
      "step": 27600,
      "train_flow_matching_loss": 0.1240234375,
      "train_physics_loss": 0.0
    },
    {
      "step": 27600,
      "train_flow_matching_loss": 0.130859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 27600,
      "train_flow_matching_loss": 0.1416015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 27600,
      "train_flow_matching_loss": 0.1533203125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.12645699083805084,
      "learning_rate": 5e-05,
      "loss": 0.1171,
      "step": 27610
    },
    {
      "step": 27610,
      "train_flow_matching_loss": 0.12255859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 27610,
      "train_flow_matching_loss": 0.1064453125,
      "train_physics_loss": 0.0
    },
    {
      "step": 27610,
      "train_flow_matching_loss": 0.10302734375,
      "train_physics_loss": 0.0
    },
    {
      "step": 27610,
      "train_flow_matching_loss": 0.1064453125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.12204720824956894,
      "learning_rate": 5e-05,
      "loss": 0.117,
      "step": 27620
    },
    {
      "step": 27620,
      "train_flow_matching_loss": 0.1591796875,
      "train_physics_loss": 0.0
    },
    {
      "step": 27620,
      "train_flow_matching_loss": 0.1474609375,
      "train_physics_loss": 0.0
    },
    {
      "step": 27620,
      "train_flow_matching_loss": 0.1376953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 27620,
      "train_flow_matching_loss": 0.12451171875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.06210966780781746,
      "learning_rate": 5e-05,
      "loss": 0.1195,
      "step": 27630
    },
    {
      "step": 27630,
      "train_flow_matching_loss": 0.12890625,
      "train_physics_loss": 0.0
    },
    {
      "step": 27630,
      "train_flow_matching_loss": 0.0830078125,
      "train_physics_loss": 0.0
    },
    {
      "step": 27630,
      "train_flow_matching_loss": 0.1279296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 27630,
      "train_flow_matching_loss": 0.06005859375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.1183054968714714,
      "learning_rate": 5e-05,
      "loss": 0.117,
      "step": 27640
    },
    {
      "step": 27640,
      "train_flow_matching_loss": 0.158203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 27640,
      "train_flow_matching_loss": 0.154296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 27640,
      "train_flow_matching_loss": 0.1416015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 27640,
      "train_flow_matching_loss": 0.1240234375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09295482933521271,
      "learning_rate": 5e-05,
      "loss": 0.1216,
      "step": 27650
    },
    {
      "step": 27650,
      "train_flow_matching_loss": 0.12451171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 27650,
      "train_flow_matching_loss": 0.2158203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 27650,
      "train_flow_matching_loss": 0.1103515625,
      "train_physics_loss": 0.0
    },
    {
      "step": 27650,
      "train_flow_matching_loss": 0.0712890625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.13270770013332367,
      "learning_rate": 5e-05,
      "loss": 0.1195,
      "step": 27660
    },
    {
      "step": 27660,
      "train_flow_matching_loss": 0.109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 27660,
      "train_flow_matching_loss": 0.1015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 27660,
      "train_flow_matching_loss": 0.11328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 27660,
      "train_flow_matching_loss": 0.1259765625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11066563427448273,
      "learning_rate": 5e-05,
      "loss": 0.1209,
      "step": 27670
    },
    {
      "step": 27670,
      "train_flow_matching_loss": 0.15625,
      "train_physics_loss": 0.0
    },
    {
      "step": 27670,
      "train_flow_matching_loss": 0.138671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 27670,
      "train_flow_matching_loss": 0.1962890625,
      "train_physics_loss": 0.0
    },
    {
      "step": 27670,
      "train_flow_matching_loss": 0.1630859375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.1643727868795395,
      "learning_rate": 5e-05,
      "loss": 0.119,
      "step": 27680
    },
    {
      "step": 27680,
      "train_flow_matching_loss": 0.1630859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 27680,
      "train_flow_matching_loss": 0.0830078125,
      "train_physics_loss": 0.0
    },
    {
      "step": 27680,
      "train_flow_matching_loss": 0.07763671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 27680,
      "train_flow_matching_loss": 0.1259765625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11076430231332779,
      "learning_rate": 5e-05,
      "loss": 0.1161,
      "step": 27690
    },
    {
      "step": 27690,
      "train_flow_matching_loss": 0.1484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 27690,
      "train_flow_matching_loss": 0.169921875,
      "train_physics_loss": 0.0
    },
    {
      "step": 27690,
      "train_flow_matching_loss": 0.11669921875,
      "train_physics_loss": 0.0
    },
    {
      "step": 27690,
      "train_flow_matching_loss": 0.1591796875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.12523053586483002,
      "learning_rate": 5e-05,
      "loss": 0.1158,
      "step": 27700
    },
    {
      "step": 27700,
      "train_flow_matching_loss": 0.12060546875,
      "train_physics_loss": 0.0
    },
    {
      "step": 27700,
      "train_flow_matching_loss": 0.1318359375,
      "train_physics_loss": 0.0
    },
    {
      "step": 27700,
      "train_flow_matching_loss": 0.11767578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 27700,
      "train_flow_matching_loss": 0.134765625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.0815395787358284,
      "learning_rate": 5e-05,
      "loss": 0.1131,
      "step": 27710
    },
    {
      "step": 27710,
      "train_flow_matching_loss": 0.1455078125,
      "train_physics_loss": 0.0
    },
    {
      "step": 27710,
      "train_flow_matching_loss": 0.1435546875,
      "train_physics_loss": 0.0
    },
    {
      "step": 27710,
      "train_flow_matching_loss": 0.11865234375,
      "train_physics_loss": 0.0
    },
    {
      "step": 27710,
      "train_flow_matching_loss": 0.1640625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09937728941440582,
      "learning_rate": 5e-05,
      "loss": 0.1138,
      "step": 27720
    },
    {
      "step": 27720,
      "train_flow_matching_loss": 0.1376953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 27720,
      "train_flow_matching_loss": 0.130859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 27720,
      "train_flow_matching_loss": 0.162109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 27720,
      "train_flow_matching_loss": 0.109375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10092145204544067,
      "learning_rate": 5e-05,
      "loss": 0.119,
      "step": 27730
    },
    {
      "step": 27730,
      "train_flow_matching_loss": 0.126953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 27730,
      "train_flow_matching_loss": 0.06982421875,
      "train_physics_loss": 0.0
    },
    {
      "step": 27730,
      "train_flow_matching_loss": 0.08935546875,
      "train_physics_loss": 0.0
    },
    {
      "step": 27730,
      "train_flow_matching_loss": 0.130859375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.1034892275929451,
      "learning_rate": 5e-05,
      "loss": 0.1184,
      "step": 27740
    },
    {
      "step": 27740,
      "train_flow_matching_loss": 0.177734375,
      "train_physics_loss": 0.0
    },
    {
      "step": 27740,
      "train_flow_matching_loss": 0.1162109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 27740,
      "train_flow_matching_loss": 0.154296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 27740,
      "train_flow_matching_loss": 0.050048828125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.12446238845586777,
      "learning_rate": 5e-05,
      "loss": 0.1181,
      "step": 27750
    },
    {
      "step": 27750,
      "train_flow_matching_loss": 0.2041015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 27750,
      "train_flow_matching_loss": 0.142578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 27750,
      "train_flow_matching_loss": 0.130859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 27750,
      "train_flow_matching_loss": 0.12890625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.14566153287887573,
      "learning_rate": 5e-05,
      "loss": 0.1197,
      "step": 27760
    },
    {
      "step": 27760,
      "train_flow_matching_loss": 0.1728515625,
      "train_physics_loss": 0.0
    },
    {
      "step": 27760,
      "train_flow_matching_loss": 0.189453125,
      "train_physics_loss": 0.0
    },
    {
      "step": 27760,
      "train_flow_matching_loss": 0.1396484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 27760,
      "train_flow_matching_loss": 0.1279296875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09992223978042603,
      "learning_rate": 5e-05,
      "loss": 0.119,
      "step": 27770
    },
    {
      "step": 27770,
      "train_flow_matching_loss": 0.1806640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 27770,
      "train_flow_matching_loss": 0.1494140625,
      "train_physics_loss": 0.0
    },
    {
      "step": 27770,
      "train_flow_matching_loss": 0.1044921875,
      "train_physics_loss": 0.0
    },
    {
      "step": 27770,
      "train_flow_matching_loss": 0.13671875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10203251987695694,
      "learning_rate": 5e-05,
      "loss": 0.1184,
      "step": 27780
    },
    {
      "step": 27780,
      "train_flow_matching_loss": 0.10205078125,
      "train_physics_loss": 0.0
    },
    {
      "step": 27780,
      "train_flow_matching_loss": 0.1767578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 27780,
      "train_flow_matching_loss": 0.1416015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 27780,
      "train_flow_matching_loss": 0.1318359375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.15094022452831268,
      "learning_rate": 5e-05,
      "loss": 0.119,
      "step": 27790
    },
    {
      "step": 27790,
      "train_flow_matching_loss": 0.11865234375,
      "train_physics_loss": 0.0
    },
    {
      "step": 27790,
      "train_flow_matching_loss": 0.1396484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 27790,
      "train_flow_matching_loss": 0.0927734375,
      "train_physics_loss": 0.152363583445549
    },
    {
      "step": 27790,
      "train_flow_matching_loss": 0.236328125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.12799552083015442,
      "learning_rate": 5e-05,
      "loss": 0.114,
      "step": 27800
    },
    {
      "step": 27800,
      "train_flow_matching_loss": 0.08544921875,
      "train_physics_loss": 0.0
    },
    {
      "step": 27800,
      "train_flow_matching_loss": 0.1865234375,
      "train_physics_loss": 0.0
    },
    {
      "step": 27800,
      "train_flow_matching_loss": 0.0390625,
      "train_physics_loss": 0.14289125800132751
    },
    {
      "step": 27800,
      "train_flow_matching_loss": 0.1513671875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.1169164776802063,
      "learning_rate": 5e-05,
      "loss": 0.1211,
      "step": 27810
    },
    {
      "step": 27810,
      "train_flow_matching_loss": 0.11474609375,
      "train_physics_loss": 0.0
    },
    {
      "step": 27810,
      "train_flow_matching_loss": 0.181640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 27810,
      "train_flow_matching_loss": 0.043701171875,
      "train_physics_loss": 0.14309003949165344
    },
    {
      "step": 27810,
      "train_flow_matching_loss": 0.1806640625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09681425243616104,
      "learning_rate": 5e-05,
      "loss": 0.1195,
      "step": 27820
    },
    {
      "step": 27820,
      "train_flow_matching_loss": 0.10205078125,
      "train_physics_loss": 0.0
    },
    {
      "step": 27820,
      "train_flow_matching_loss": 0.2265625,
      "train_physics_loss": 0.0
    },
    {
      "step": 27820,
      "train_flow_matching_loss": 0.052734375,
      "train_physics_loss": 0.1214766800403595
    },
    {
      "step": 27820,
      "train_flow_matching_loss": 0.130859375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10052533447742462,
      "learning_rate": 5e-05,
      "loss": 0.1164,
      "step": 27830
    },
    {
      "step": 27830,
      "train_flow_matching_loss": 0.09326171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 27830,
      "train_flow_matching_loss": 0.1103515625,
      "train_physics_loss": 0.0
    },
    {
      "step": 27830,
      "train_flow_matching_loss": 0.06494140625,
      "train_physics_loss": 0.15216483175754547
    },
    {
      "step": 27830,
      "train_flow_matching_loss": 0.1875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.12908807396888733,
      "learning_rate": 5e-05,
      "loss": 0.1195,
      "step": 27840
    },
    {
      "step": 27840,
      "train_flow_matching_loss": 0.08935546875,
      "train_physics_loss": 0.0
    },
    {
      "step": 27840,
      "train_flow_matching_loss": 0.142578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 27840,
      "train_flow_matching_loss": 0.07373046875,
      "train_physics_loss": 0.13259357213974
    },
    {
      "step": 27840,
      "train_flow_matching_loss": 0.09716796875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10260359197854996,
      "learning_rate": 5e-05,
      "loss": 0.1169,
      "step": 27850
    },
    {
      "step": 27850,
      "train_flow_matching_loss": 0.07763671875,
      "train_physics_loss": 0.320684015750885
    },
    {
      "step": 27850,
      "train_flow_matching_loss": 0.1279296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 27850,
      "train_flow_matching_loss": 0.040283203125,
      "train_physics_loss": 0.15907050669193268
    },
    {
      "step": 27850,
      "train_flow_matching_loss": 0.1123046875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11338122934103012,
      "learning_rate": 5e-05,
      "loss": 0.1172,
      "step": 27860
    },
    {
      "step": 27860,
      "train_flow_matching_loss": 0.07861328125,
      "train_physics_loss": 0.18906505405902863
    },
    {
      "step": 27860,
      "train_flow_matching_loss": 0.2177734375,
      "train_physics_loss": 0.0
    },
    {
      "step": 27860,
      "train_flow_matching_loss": 0.047607421875,
      "train_physics_loss": 0.13006167113780975
    },
    {
      "step": 27860,
      "train_flow_matching_loss": 0.1552734375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08978500962257385,
      "learning_rate": 5e-05,
      "loss": 0.1215,
      "step": 27870
    },
    {
      "step": 27870,
      "train_flow_matching_loss": 0.10986328125,
      "train_physics_loss": 0.2517739236354828
    },
    {
      "step": 27870,
      "train_flow_matching_loss": 0.11181640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 27870,
      "train_flow_matching_loss": 0.037841796875,
      "train_physics_loss": 0.11554420739412308
    },
    {
      "step": 27870,
      "train_flow_matching_loss": 0.09423828125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08018870651721954,
      "learning_rate": 5e-05,
      "loss": 0.1169,
      "step": 27880
    },
    {
      "step": 27880,
      "train_flow_matching_loss": 0.05029296875,
      "train_physics_loss": 0.15135827660560608
    },
    {
      "step": 27880,
      "train_flow_matching_loss": 0.197265625,
      "train_physics_loss": 0.0
    },
    {
      "step": 27880,
      "train_flow_matching_loss": 0.10546875,
      "train_physics_loss": 0.1417291909456253
    },
    {
      "step": 27880,
      "train_flow_matching_loss": 0.1962890625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.0878964439034462,
      "learning_rate": 5e-05,
      "loss": 0.122,
      "step": 27890
    },
    {
      "step": 27890,
      "train_flow_matching_loss": 0.0289306640625,
      "train_physics_loss": 0.07619139552116394
    },
    {
      "step": 27890,
      "train_flow_matching_loss": 0.21875,
      "train_physics_loss": 0.0
    },
    {
      "step": 27890,
      "train_flow_matching_loss": 0.05908203125,
      "train_physics_loss": 0.11857908964157104
    },
    {
      "step": 27890,
      "train_flow_matching_loss": 0.1845703125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.13088229298591614,
      "learning_rate": 5e-05,
      "loss": 0.1185,
      "step": 27900
    },
    {
      "step": 27900,
      "train_flow_matching_loss": 0.0810546875,
      "train_physics_loss": 0.21793213486671448
    },
    {
      "step": 27900,
      "train_flow_matching_loss": 0.11328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 27900,
      "train_flow_matching_loss": 0.044921875,
      "train_physics_loss": 0.12452934682369232
    },
    {
      "step": 27900,
      "train_flow_matching_loss": 0.150390625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11782210320234299,
      "learning_rate": 5e-05,
      "loss": 0.1208,
      "step": 27910
    },
    {
      "step": 27910,
      "train_flow_matching_loss": 0.10205078125,
      "train_physics_loss": 0.10524322837591171
    },
    {
      "step": 27910,
      "train_flow_matching_loss": 0.12353515625,
      "train_physics_loss": 0.0
    },
    {
      "step": 27910,
      "train_flow_matching_loss": 0.09765625,
      "train_physics_loss": 0.13921409845352173
    },
    {
      "step": 27910,
      "train_flow_matching_loss": 0.109375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09173274040222168,
      "learning_rate": 5e-05,
      "loss": 0.1226,
      "step": 27920
    },
    {
      "step": 27920,
      "train_flow_matching_loss": 0.04052734375,
      "train_physics_loss": 0.10804308205842972
    },
    {
      "step": 27920,
      "train_flow_matching_loss": 0.138671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 27920,
      "train_flow_matching_loss": 0.042236328125,
      "train_physics_loss": 0.08515579998493195
    },
    {
      "step": 27920,
      "train_flow_matching_loss": 0.158203125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08331822603940964,
      "learning_rate": 5e-05,
      "loss": 0.1221,
      "step": 27930
    },
    {
      "step": 27930,
      "train_flow_matching_loss": 0.05615234375,
      "train_physics_loss": 0.10655443370342255
    },
    {
      "step": 27930,
      "train_flow_matching_loss": 0.1259765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 27930,
      "train_flow_matching_loss": 0.062255859375,
      "train_physics_loss": 0.14787115156650543
    },
    {
      "step": 27930,
      "train_flow_matching_loss": 0.1220703125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10195967555046082,
      "learning_rate": 5e-05,
      "loss": 0.1218,
      "step": 27940
    },
    {
      "step": 27940,
      "train_flow_matching_loss": 0.08984375,
      "train_physics_loss": 0.24714989960193634
    },
    {
      "step": 27940,
      "train_flow_matching_loss": 0.162109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 27940,
      "train_flow_matching_loss": 0.0306396484375,
      "train_physics_loss": 0.1594846099615097
    },
    {
      "step": 27940,
      "train_flow_matching_loss": 0.1044921875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10533392429351807,
      "learning_rate": 5e-05,
      "loss": 0.1174,
      "step": 27950
    },
    {
      "step": 27950,
      "train_flow_matching_loss": 0.034423828125,
      "train_physics_loss": 0.1374061554670334
    },
    {
      "step": 27950,
      "train_flow_matching_loss": 0.06396484375,
      "train_physics_loss": 0.1795577108860016
    },
    {
      "step": 27950,
      "train_flow_matching_loss": 0.031982421875,
      "train_physics_loss": 0.12067559361457825
    },
    {
      "step": 27950,
      "train_flow_matching_loss": 0.12451171875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09652707725763321,
      "learning_rate": 5e-05,
      "loss": 0.1177,
      "step": 27960
    },
    {
      "step": 27960,
      "train_flow_matching_loss": 0.035888671875,
      "train_physics_loss": 0.13387855887413025
    },
    {
      "step": 27960,
      "train_flow_matching_loss": 0.04443359375,
      "train_physics_loss": 0.15010371804237366
    },
    {
      "step": 27960,
      "train_flow_matching_loss": 0.07470703125,
      "train_physics_loss": 0.14094851911067963
    },
    {
      "step": 27960,
      "train_flow_matching_loss": 0.154296875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.07763365656137466,
      "learning_rate": 5e-05,
      "loss": 0.1195,
      "step": 27970
    },
    {
      "step": 27970,
      "train_flow_matching_loss": 0.04052734375,
      "train_physics_loss": 0.10143232345581055
    },
    {
      "step": 27970,
      "train_flow_matching_loss": 0.0289306640625,
      "train_physics_loss": 0.134042888879776
    },
    {
      "step": 27970,
      "train_flow_matching_loss": 0.051513671875,
      "train_physics_loss": 0.1337745189666748
    },
    {
      "step": 27970,
      "train_flow_matching_loss": 0.138671875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10474427044391632,
      "learning_rate": 5e-05,
      "loss": 0.1186,
      "step": 27980
    },
    {
      "step": 27980,
      "train_flow_matching_loss": 0.032470703125,
      "train_physics_loss": 0.157284215092659
    },
    {
      "step": 27980,
      "train_flow_matching_loss": 0.08837890625,
      "train_physics_loss": 0.2349974513053894
    },
    {
      "step": 27980,
      "train_flow_matching_loss": 0.03662109375,
      "train_physics_loss": 0.1793174296617508
    },
    {
      "step": 27980,
      "train_flow_matching_loss": 0.1201171875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.13709452748298645,
      "learning_rate": 5e-05,
      "loss": 0.1144,
      "step": 27990
    },
    {
      "step": 27990,
      "train_flow_matching_loss": 0.044677734375,
      "train_physics_loss": 0.13650472462177277
    },
    {
      "step": 27990,
      "train_flow_matching_loss": 0.029296875,
      "train_physics_loss": 0.1436215192079544
    },
    {
      "step": 27990,
      "train_flow_matching_loss": 0.036865234375,
      "train_physics_loss": 0.1459917575120926
    },
    {
      "step": 27990,
      "train_flow_matching_loss": 0.1298828125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08864488452672958,
      "learning_rate": 5e-05,
      "loss": 0.1164,
      "step": 28000
    },
    {
      "step": 28000,
      "train_flow_matching_loss": 0.03466796875,
      "train_physics_loss": 0.1323496401309967
    },
    {
      "step": 28000,
      "train_flow_matching_loss": 0.0208740234375,
      "train_physics_loss": 0.12343508750200272
    },
    {
      "step": 28000,
      "train_flow_matching_loss": 0.056640625,
      "train_physics_loss": 0.16987772285938263
    },
    {
      "step": 28000,
      "train_flow_matching_loss": 0.05322265625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10494557023048401,
      "learning_rate": 5e-05,
      "loss": 0.1182,
      "step": 28010
    },
    {
      "step": 28010,
      "train_flow_matching_loss": 0.056640625,
      "train_physics_loss": 0.14356932044029236
    },
    {
      "step": 28010,
      "train_flow_matching_loss": 0.09228515625,
      "train_physics_loss": 0.16129624843597412
    },
    {
      "step": 28010,
      "train_flow_matching_loss": 0.03759765625,
      "train_physics_loss": 0.13254012167453766
    },
    {
      "step": 28010,
      "train_flow_matching_loss": 0.0810546875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09056280553340912,
      "learning_rate": 5e-05,
      "loss": 0.1233,
      "step": 28020
    },
    {
      "step": 28020,
      "train_flow_matching_loss": 0.0247802734375,
      "train_physics_loss": 0.12680134177207947
    },
    {
      "step": 28020,
      "train_flow_matching_loss": 0.041259765625,
      "train_physics_loss": 0.13765747845172882
    },
    {
      "step": 28020,
      "train_flow_matching_loss": 0.0279541015625,
      "train_physics_loss": 0.121088907122612
    },
    {
      "step": 28020,
      "train_flow_matching_loss": 0.134765625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11235819011926651,
      "learning_rate": 5e-05,
      "loss": 0.1156,
      "step": 28030
    },
    {
      "step": 28030,
      "train_flow_matching_loss": 0.050048828125,
      "train_physics_loss": 0.1187298521399498
    },
    {
      "step": 28030,
      "train_flow_matching_loss": 0.052490234375,
      "train_physics_loss": 0.18933796882629395
    },
    {
      "step": 28030,
      "train_flow_matching_loss": 0.041015625,
      "train_physics_loss": 0.20027600228786469
    },
    {
      "step": 28030,
      "train_flow_matching_loss": 0.10546875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10631302744150162,
      "learning_rate": 5e-05,
      "loss": 0.1237,
      "step": 28040
    },
    {
      "step": 28040,
      "train_flow_matching_loss": 0.037841796875,
      "train_physics_loss": 0.13204319775104523
    },
    {
      "step": 28040,
      "train_flow_matching_loss": 0.032958984375,
      "train_physics_loss": 0.0884726271033287
    },
    {
      "step": 28040,
      "train_flow_matching_loss": 0.0654296875,
      "train_physics_loss": 0.1409658044576645
    },
    {
      "step": 28040,
      "train_flow_matching_loss": 0.0830078125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08629219233989716,
      "learning_rate": 5e-05,
      "loss": 0.1163,
      "step": 28050
    },
    {
      "step": 28050,
      "train_flow_matching_loss": 0.09912109375,
      "train_physics_loss": 0.28582680225372314
    },
    {
      "step": 28050,
      "train_flow_matching_loss": 0.05078125,
      "train_physics_loss": 0.12302097678184509
    },
    {
      "step": 28050,
      "train_flow_matching_loss": 0.1240234375,
      "train_physics_loss": 0.0
    },
    {
      "step": 28050,
      "train_flow_matching_loss": 0.10791015625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10443174093961716,
      "learning_rate": 5e-05,
      "loss": 0.1175,
      "step": 28060
    },
    {
      "step": 28060,
      "train_flow_matching_loss": 0.06396484375,
      "train_physics_loss": 0.1457904875278473
    },
    {
      "step": 28060,
      "train_flow_matching_loss": 0.052001953125,
      "train_physics_loss": 0.1681147962808609
    },
    {
      "step": 28060,
      "train_flow_matching_loss": 0.12890625,
      "train_physics_loss": 0.0
    },
    {
      "step": 28060,
      "train_flow_matching_loss": 0.08837890625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10327212512493134,
      "learning_rate": 5e-05,
      "loss": 0.114,
      "step": 28070
    },
    {
      "step": 28070,
      "train_flow_matching_loss": 0.057373046875,
      "train_physics_loss": 0.12577055394649506
    },
    {
      "step": 28070,
      "train_flow_matching_loss": 0.032958984375,
      "train_physics_loss": 0.13074585795402527
    },
    {
      "step": 28070,
      "train_flow_matching_loss": 0.130859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 28070,
      "train_flow_matching_loss": 0.1767578125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.07614554464817047,
      "learning_rate": 5e-05,
      "loss": 0.1171,
      "step": 28080
    },
    {
      "step": 28080,
      "train_flow_matching_loss": 0.07275390625,
      "train_physics_loss": 0.14989124238491058
    },
    {
      "step": 28080,
      "train_flow_matching_loss": 0.052490234375,
      "train_physics_loss": 0.1578955501317978
    },
    {
      "step": 28080,
      "train_flow_matching_loss": 0.1337890625,
      "train_physics_loss": 0.0
    },
    {
      "step": 28080,
      "train_flow_matching_loss": 0.0771484375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.15190733969211578,
      "learning_rate": 5e-05,
      "loss": 0.1209,
      "step": 28090
    },
    {
      "step": 28090,
      "train_flow_matching_loss": 0.06494140625,
      "train_physics_loss": 0.1328783482313156
    },
    {
      "step": 28090,
      "train_flow_matching_loss": 0.049072265625,
      "train_physics_loss": 0.11704456806182861
    },
    {
      "step": 28090,
      "train_flow_matching_loss": 0.1376953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 28090,
      "train_flow_matching_loss": 0.10107421875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11567281186580658,
      "learning_rate": 5e-05,
      "loss": 0.1253,
      "step": 28100
    },
    {
      "step": 28100,
      "train_flow_matching_loss": 0.058837890625,
      "train_physics_loss": 0.09468439966440201
    },
    {
      "step": 28100,
      "train_flow_matching_loss": 0.0634765625,
      "train_physics_loss": 0.21364572644233704
    },
    {
      "step": 28100,
      "train_flow_matching_loss": 0.150390625,
      "train_physics_loss": 0.0
    },
    {
      "step": 28100,
      "train_flow_matching_loss": 0.1298828125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.1297483742237091,
      "learning_rate": 5e-05,
      "loss": 0.1165,
      "step": 28110
    },
    {
      "step": 28110,
      "train_flow_matching_loss": 0.146484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 28110,
      "train_flow_matching_loss": 0.03125,
      "train_physics_loss": 0.13291116058826447
    },
    {
      "step": 28110,
      "train_flow_matching_loss": 0.10107421875,
      "train_physics_loss": 0.0
    },
    {
      "step": 28110,
      "train_flow_matching_loss": 0.10009765625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10175083577632904,
      "learning_rate": 5e-05,
      "loss": 0.1213,
      "step": 28120
    },
    {
      "step": 28120,
      "train_flow_matching_loss": 0.09912109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 28120,
      "train_flow_matching_loss": 0.03955078125,
      "train_physics_loss": 0.19213858246803284
    },
    {
      "step": 28120,
      "train_flow_matching_loss": 0.138671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 28120,
      "train_flow_matching_loss": 0.2255859375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08471988141536713,
      "learning_rate": 5e-05,
      "loss": 0.1169,
      "step": 28130
    },
    {
      "step": 28130,
      "train_flow_matching_loss": 0.126953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 28130,
      "train_flow_matching_loss": 0.029052734375,
      "train_physics_loss": 0.0
    },
    {
      "step": 28130,
      "train_flow_matching_loss": 0.1884765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 28130,
      "train_flow_matching_loss": 0.1083984375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.12235070019960403,
      "learning_rate": 5e-05,
      "loss": 0.112,
      "step": 28140
    },
    {
      "step": 28140,
      "train_flow_matching_loss": 0.1357421875,
      "train_physics_loss": 0.0
    },
    {
      "step": 28140,
      "train_flow_matching_loss": 0.064453125,
      "train_physics_loss": 0.15771861374378204
    },
    {
      "step": 28140,
      "train_flow_matching_loss": 0.09228515625,
      "train_physics_loss": 0.0
    },
    {
      "step": 28140,
      "train_flow_matching_loss": 0.1044921875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08694256842136383,
      "learning_rate": 5e-05,
      "loss": 0.1136,
      "step": 28150
    },
    {
      "step": 28150,
      "train_flow_matching_loss": 0.06884765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 28150,
      "train_flow_matching_loss": 0.07275390625,
      "train_physics_loss": 0.12628783285617828
    },
    {
      "step": 28150,
      "train_flow_matching_loss": 0.1552734375,
      "train_physics_loss": 0.0
    },
    {
      "step": 28150,
      "train_flow_matching_loss": 0.07470703125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.12353956699371338,
      "learning_rate": 5e-05,
      "loss": 0.12,
      "step": 28160
    },
    {
      "step": 28160,
      "train_flow_matching_loss": 0.1396484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 28160,
      "train_flow_matching_loss": 0.04248046875,
      "train_physics_loss": 0.11622707545757294
    },
    {
      "step": 28160,
      "train_flow_matching_loss": 0.08154296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 28160,
      "train_flow_matching_loss": 0.0859375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.1468089073896408,
      "learning_rate": 5e-05,
      "loss": 0.1184,
      "step": 28170
    },
    {
      "step": 28170,
      "train_flow_matching_loss": 0.1220703125,
      "train_physics_loss": 0.0
    },
    {
      "step": 28170,
      "train_flow_matching_loss": 0.04443359375,
      "train_physics_loss": 0.16852064430713654
    },
    {
      "step": 28170,
      "train_flow_matching_loss": 0.2236328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 28170,
      "train_flow_matching_loss": 0.1298828125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.07468272000551224,
      "learning_rate": 5e-05,
      "loss": 0.1183,
      "step": 28180
    },
    {
      "step": 28180,
      "train_flow_matching_loss": 0.1298828125,
      "train_physics_loss": 0.0
    },
    {
      "step": 28180,
      "train_flow_matching_loss": 0.048095703125,
      "train_physics_loss": 0.17884285748004913
    },
    {
      "step": 28180,
      "train_flow_matching_loss": 0.173828125,
      "train_physics_loss": 0.0
    },
    {
      "step": 28180,
      "train_flow_matching_loss": 0.043212890625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.07573188096284866,
      "learning_rate": 5e-05,
      "loss": 0.1151,
      "step": 28190
    },
    {
      "step": 28190,
      "train_flow_matching_loss": 0.10595703125,
      "train_physics_loss": 0.0
    },
    {
      "step": 28190,
      "train_flow_matching_loss": 0.04443359375,
      "train_physics_loss": 0.15669040381908417
    },
    {
      "step": 28190,
      "train_flow_matching_loss": 0.150390625,
      "train_physics_loss": 0.0
    },
    {
      "step": 28190,
      "train_flow_matching_loss": 0.1513671875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08636467158794403,
      "learning_rate": 5e-05,
      "loss": 0.1223,
      "step": 28200
    },
    {
      "step": 28200,
      "train_flow_matching_loss": 0.1708984375,
      "train_physics_loss": 0.0
    },
    {
      "step": 28200,
      "train_flow_matching_loss": 0.150390625,
      "train_physics_loss": 0.29812803864479065
    },
    {
      "step": 28200,
      "train_flow_matching_loss": 0.162109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 28200,
      "train_flow_matching_loss": 0.1806640625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08301368355751038,
      "learning_rate": 5e-05,
      "loss": 0.1211,
      "step": 28210
    },
    {
      "step": 28210,
      "train_flow_matching_loss": 0.1455078125,
      "train_physics_loss": 0.0
    },
    {
      "step": 28210,
      "train_flow_matching_loss": 0.12890625,
      "train_physics_loss": 0.0
    },
    {
      "step": 28210,
      "train_flow_matching_loss": 0.201171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 28210,
      "train_flow_matching_loss": 0.0947265625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.12348692864179611,
      "learning_rate": 5e-05,
      "loss": 0.1211,
      "step": 28220
    },
    {
      "step": 28220,
      "train_flow_matching_loss": 0.10888671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 28220,
      "train_flow_matching_loss": 0.150390625,
      "train_physics_loss": 0.0
    },
    {
      "step": 28220,
      "train_flow_matching_loss": 0.1064453125,
      "train_physics_loss": 0.0
    },
    {
      "step": 28220,
      "train_flow_matching_loss": 0.10986328125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10992780327796936,
      "learning_rate": 5e-05,
      "loss": 0.1217,
      "step": 28230
    },
    {
      "step": 28230,
      "train_flow_matching_loss": 0.1181640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 28230,
      "train_flow_matching_loss": 0.08740234375,
      "train_physics_loss": 0.0
    },
    {
      "step": 28230,
      "train_flow_matching_loss": 0.1025390625,
      "train_physics_loss": 0.0
    },
    {
      "step": 28230,
      "train_flow_matching_loss": 0.12451171875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.0761227235198021,
      "learning_rate": 5e-05,
      "loss": 0.1206,
      "step": 28240
    },
    {
      "step": 28240,
      "train_flow_matching_loss": 0.2138671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 28240,
      "train_flow_matching_loss": 0.134765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 28240,
      "train_flow_matching_loss": 0.15625,
      "train_physics_loss": 0.0
    },
    {
      "step": 28240,
      "train_flow_matching_loss": 0.060791015625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10014299303293228,
      "learning_rate": 5e-05,
      "loss": 0.1186,
      "step": 28250
    },
    {
      "step": 28250,
      "train_flow_matching_loss": 0.10986328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 28250,
      "train_flow_matching_loss": 0.134765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 28250,
      "train_flow_matching_loss": 0.13671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 28250,
      "train_flow_matching_loss": 0.1484375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.16903607547283173,
      "learning_rate": 5e-05,
      "loss": 0.1212,
      "step": 28260
    },
    {
      "step": 28260,
      "train_flow_matching_loss": 0.1650390625,
      "train_physics_loss": 0.0
    },
    {
      "step": 28260,
      "train_flow_matching_loss": 0.1025390625,
      "train_physics_loss": 0.0
    },
    {
      "step": 28260,
      "train_flow_matching_loss": 0.1123046875,
      "train_physics_loss": 0.0
    },
    {
      "step": 28260,
      "train_flow_matching_loss": 0.1279296875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.0919145941734314,
      "learning_rate": 5e-05,
      "loss": 0.1178,
      "step": 28270
    },
    {
      "step": 28270,
      "train_flow_matching_loss": 0.1474609375,
      "train_physics_loss": 0.0
    },
    {
      "step": 28270,
      "train_flow_matching_loss": 0.10888671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 28270,
      "train_flow_matching_loss": 0.09228515625,
      "train_physics_loss": 0.0
    },
    {
      "step": 28270,
      "train_flow_matching_loss": 0.06591796875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08964044600725174,
      "learning_rate": 5e-05,
      "loss": 0.1155,
      "step": 28280
    },
    {
      "step": 28280,
      "train_flow_matching_loss": 0.078125,
      "train_physics_loss": 0.0
    },
    {
      "step": 28280,
      "train_flow_matching_loss": 0.10595703125,
      "train_physics_loss": 0.0
    },
    {
      "step": 28280,
      "train_flow_matching_loss": 0.1650390625,
      "train_physics_loss": 0.0
    },
    {
      "step": 28280,
      "train_flow_matching_loss": 0.154296875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10366921126842499,
      "learning_rate": 5e-05,
      "loss": 0.1162,
      "step": 28290
    },
    {
      "step": 28290,
      "train_flow_matching_loss": 0.1767578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 28290,
      "train_flow_matching_loss": 0.0673828125,
      "train_physics_loss": 0.0
    },
    {
      "step": 28290,
      "train_flow_matching_loss": 0.1943359375,
      "train_physics_loss": 0.0
    },
    {
      "step": 28290,
      "train_flow_matching_loss": 0.11962890625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10033810138702393,
      "learning_rate": 5e-05,
      "loss": 0.1159,
      "step": 28300
    },
    {
      "step": 28300,
      "train_flow_matching_loss": 0.1455078125,
      "train_physics_loss": 0.0
    },
    {
      "step": 28300,
      "train_flow_matching_loss": 0.09423828125,
      "train_physics_loss": 0.0
    },
    {
      "step": 28300,
      "train_flow_matching_loss": 0.1767578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 28300,
      "train_flow_matching_loss": 0.1181640625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08895985037088394,
      "learning_rate": 5e-05,
      "loss": 0.1127,
      "step": 28310
    },
    {
      "step": 28310,
      "train_flow_matching_loss": 0.1787109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 28310,
      "train_flow_matching_loss": 0.09765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 28310,
      "train_flow_matching_loss": 0.12353515625,
      "train_physics_loss": 0.0
    },
    {
      "step": 28310,
      "train_flow_matching_loss": 0.150390625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09906337410211563,
      "learning_rate": 5e-05,
      "loss": 0.1176,
      "step": 28320
    },
    {
      "step": 28320,
      "train_flow_matching_loss": 0.0625,
      "train_physics_loss": 0.0
    },
    {
      "step": 28320,
      "train_flow_matching_loss": 0.0654296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 28320,
      "train_flow_matching_loss": 0.1044921875,
      "train_physics_loss": 0.0
    },
    {
      "step": 28320,
      "train_flow_matching_loss": 0.05419921875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.06577824056148529,
      "learning_rate": 5e-05,
      "loss": 0.121,
      "step": 28330
    },
    {
      "step": 28330,
      "train_flow_matching_loss": 0.11572265625,
      "train_physics_loss": 0.0
    },
    {
      "step": 28330,
      "train_flow_matching_loss": 0.0947265625,
      "train_physics_loss": 0.0
    },
    {
      "step": 28330,
      "train_flow_matching_loss": 0.1767578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 28330,
      "train_flow_matching_loss": 0.11083984375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11510001868009567,
      "learning_rate": 5e-05,
      "loss": 0.1184,
      "step": 28340
    },
    {
      "step": 28340,
      "train_flow_matching_loss": 0.11962890625,
      "train_physics_loss": 0.0
    },
    {
      "step": 28340,
      "train_flow_matching_loss": 0.1484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 28340,
      "train_flow_matching_loss": 0.134765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 28340,
      "train_flow_matching_loss": 0.185546875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.12667632102966309,
      "learning_rate": 5e-05,
      "loss": 0.1213,
      "step": 28350
    },
    {
      "step": 28350,
      "train_flow_matching_loss": 0.0732421875,
      "train_physics_loss": 0.0
    },
    {
      "step": 28350,
      "train_flow_matching_loss": 0.10498046875,
      "train_physics_loss": 0.0
    },
    {
      "step": 28350,
      "train_flow_matching_loss": 0.1318359375,
      "train_physics_loss": 0.0
    },
    {
      "step": 28350,
      "train_flow_matching_loss": 0.138671875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09593580663204193,
      "learning_rate": 5e-05,
      "loss": 0.1175,
      "step": 28360
    },
    {
      "step": 28360,
      "train_flow_matching_loss": 0.11474609375,
      "train_physics_loss": 0.0
    },
    {
      "step": 28360,
      "train_flow_matching_loss": 0.0712890625,
      "train_physics_loss": 0.0
    },
    {
      "step": 28360,
      "train_flow_matching_loss": 0.1708984375,
      "train_physics_loss": 0.0
    },
    {
      "step": 28360,
      "train_flow_matching_loss": 0.1298828125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.07262039929628372,
      "learning_rate": 5e-05,
      "loss": 0.1165,
      "step": 28370
    },
    {
      "step": 28370,
      "train_flow_matching_loss": 0.06689453125,
      "train_physics_loss": 0.0
    },
    {
      "step": 28370,
      "train_flow_matching_loss": 0.10791015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 28370,
      "train_flow_matching_loss": 0.1201171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 28370,
      "train_flow_matching_loss": 0.25,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.0928858071565628,
      "learning_rate": 5e-05,
      "loss": 0.1208,
      "step": 28380
    },
    {
      "step": 28380,
      "train_flow_matching_loss": 0.1005859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 28380,
      "train_flow_matching_loss": 0.11572265625,
      "train_physics_loss": 0.0
    },
    {
      "step": 28380,
      "train_flow_matching_loss": 0.1298828125,
      "train_physics_loss": 0.0
    },
    {
      "step": 28380,
      "train_flow_matching_loss": 0.10009765625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10839726030826569,
      "learning_rate": 5e-05,
      "loss": 0.1166,
      "step": 28390
    },
    {
      "step": 28390,
      "train_flow_matching_loss": 0.11865234375,
      "train_physics_loss": 0.0
    },
    {
      "step": 28390,
      "train_flow_matching_loss": 0.09130859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 28390,
      "train_flow_matching_loss": 0.2021484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 28390,
      "train_flow_matching_loss": 0.06982421875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.0845341831445694,
      "learning_rate": 5e-05,
      "loss": 0.1153,
      "step": 28400
    },
    {
      "step": 28400,
      "train_flow_matching_loss": 0.1318359375,
      "train_physics_loss": 0.0
    },
    {
      "step": 28400,
      "train_flow_matching_loss": 0.10888671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 28400,
      "train_flow_matching_loss": 0.1728515625,
      "train_physics_loss": 0.0
    },
    {
      "step": 28400,
      "train_flow_matching_loss": 0.07568359375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08812949806451797,
      "learning_rate": 5e-05,
      "loss": 0.1225,
      "step": 28410
    },
    {
      "step": 28410,
      "train_flow_matching_loss": 0.1630859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 28410,
      "train_flow_matching_loss": 0.056396484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 28410,
      "train_flow_matching_loss": 0.07177734375,
      "train_physics_loss": 0.0
    },
    {
      "step": 28410,
      "train_flow_matching_loss": 0.1416015625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10762255638837814,
      "learning_rate": 5e-05,
      "loss": 0.1198,
      "step": 28420
    },
    {
      "step": 28420,
      "train_flow_matching_loss": 0.1064453125,
      "train_physics_loss": 0.0
    },
    {
      "step": 28420,
      "train_flow_matching_loss": 0.1298828125,
      "train_physics_loss": 0.0
    },
    {
      "step": 28420,
      "train_flow_matching_loss": 0.259765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 28420,
      "train_flow_matching_loss": 0.11376953125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09426674991846085,
      "learning_rate": 5e-05,
      "loss": 0.1168,
      "step": 28430
    },
    {
      "step": 28430,
      "train_flow_matching_loss": 0.130859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 28430,
      "train_flow_matching_loss": 0.1875,
      "train_physics_loss": 0.0
    },
    {
      "step": 28430,
      "train_flow_matching_loss": 0.154296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 28430,
      "train_flow_matching_loss": 0.2373046875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.1098853349685669,
      "learning_rate": 5e-05,
      "loss": 0.1234,
      "step": 28440
    },
    {
      "step": 28440,
      "train_flow_matching_loss": 0.1259765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 28440,
      "train_flow_matching_loss": 0.07080078125,
      "train_physics_loss": 0.0
    },
    {
      "step": 28440,
      "train_flow_matching_loss": 0.0986328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 28440,
      "train_flow_matching_loss": 0.109375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.1064753532409668,
      "learning_rate": 5e-05,
      "loss": 0.1188,
      "step": 28450
    },
    {
      "step": 28450,
      "train_flow_matching_loss": 0.1396484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 28450,
      "train_flow_matching_loss": 0.1474609375,
      "train_physics_loss": 0.0
    },
    {
      "step": 28450,
      "train_flow_matching_loss": 0.09326171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 28450,
      "train_flow_matching_loss": 0.130859375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09906245768070221,
      "learning_rate": 5e-05,
      "loss": 0.1232,
      "step": 28460
    },
    {
      "step": 28460,
      "train_flow_matching_loss": 0.12255859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 28460,
      "train_flow_matching_loss": 0.1982421875,
      "train_physics_loss": 0.0
    },
    {
      "step": 28460,
      "train_flow_matching_loss": 0.0908203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 28460,
      "train_flow_matching_loss": 0.1240234375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08524290472269058,
      "learning_rate": 5e-05,
      "loss": 0.1185,
      "step": 28470
    },
    {
      "step": 28470,
      "train_flow_matching_loss": 0.146484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 28470,
      "train_flow_matching_loss": 0.12255859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 28470,
      "train_flow_matching_loss": 0.2197265625,
      "train_physics_loss": 0.0
    },
    {
      "step": 28470,
      "train_flow_matching_loss": 0.10986328125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.19082383811473846,
      "learning_rate": 5e-05,
      "loss": 0.1226,
      "step": 28480
    },
    {
      "step": 28480,
      "train_flow_matching_loss": 0.142578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 28480,
      "train_flow_matching_loss": 0.11083984375,
      "train_physics_loss": 0.0
    },
    {
      "step": 28480,
      "train_flow_matching_loss": 0.1640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 28480,
      "train_flow_matching_loss": 0.146484375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.8094887137413025,
      "learning_rate": 5e-05,
      "loss": 0.1268,
      "step": 28490
    },
    {
      "step": 28490,
      "train_flow_matching_loss": 0.1298828125,
      "train_physics_loss": 0.0
    },
    {
      "step": 28490,
      "train_flow_matching_loss": 0.11474609375,
      "train_physics_loss": 0.0
    },
    {
      "step": 28490,
      "train_flow_matching_loss": 0.1904296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 28490,
      "train_flow_matching_loss": 0.12353515625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11407855153083801,
      "learning_rate": 5e-05,
      "loss": 0.1201,
      "step": 28500
    },
    {
      "step": 28500,
      "train_flow_matching_loss": 0.181640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 28500,
      "train_flow_matching_loss": 0.142578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 28500,
      "train_flow_matching_loss": 0.1806640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 28500,
      "train_flow_matching_loss": 0.109375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09929490834474564,
      "learning_rate": 5e-05,
      "loss": 0.1192,
      "step": 28510
    },
    {
      "step": 28510,
      "train_flow_matching_loss": 0.09423828125,
      "train_physics_loss": 0.0
    },
    {
      "step": 28510,
      "train_flow_matching_loss": 0.0986328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 28510,
      "train_flow_matching_loss": 0.12060546875,
      "train_physics_loss": 0.0
    },
    {
      "step": 28510,
      "train_flow_matching_loss": 0.0859375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.0941091999411583,
      "learning_rate": 5e-05,
      "loss": 0.1194,
      "step": 28520
    },
    {
      "step": 28520,
      "train_flow_matching_loss": 0.12060546875,
      "train_physics_loss": 0.0
    },
    {
      "step": 28520,
      "train_flow_matching_loss": 0.142578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 28520,
      "train_flow_matching_loss": 0.1455078125,
      "train_physics_loss": 0.0
    },
    {
      "step": 28520,
      "train_flow_matching_loss": 0.07666015625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.12867538630962372,
      "learning_rate": 5e-05,
      "loss": 0.1165,
      "step": 28530
    },
    {
      "step": 28530,
      "train_flow_matching_loss": 0.09765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 28530,
      "train_flow_matching_loss": 0.12890625,
      "train_physics_loss": 0.0
    },
    {
      "step": 28530,
      "train_flow_matching_loss": 0.1884765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 28530,
      "train_flow_matching_loss": 0.1279296875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09049176424741745,
      "learning_rate": 5e-05,
      "loss": 0.1231,
      "step": 28540
    },
    {
      "step": 28540,
      "train_flow_matching_loss": 0.1591796875,
      "train_physics_loss": 0.0
    },
    {
      "step": 28540,
      "train_flow_matching_loss": 0.1240234375,
      "train_physics_loss": 0.0
    },
    {
      "step": 28540,
      "train_flow_matching_loss": 0.140625,
      "train_physics_loss": 0.0
    },
    {
      "step": 28540,
      "train_flow_matching_loss": 0.1484375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.0897572785615921,
      "learning_rate": 5e-05,
      "loss": 0.1194,
      "step": 28550
    },
    {
      "step": 28550,
      "train_flow_matching_loss": 0.177734375,
      "train_physics_loss": 0.0
    },
    {
      "step": 28550,
      "train_flow_matching_loss": 0.0751953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 28550,
      "train_flow_matching_loss": 0.1123046875,
      "train_physics_loss": 0.0
    },
    {
      "step": 28550,
      "train_flow_matching_loss": 0.12109375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11176637560129166,
      "learning_rate": 5e-05,
      "loss": 0.1142,
      "step": 28560
    },
    {
      "step": 28560,
      "train_flow_matching_loss": 0.1298828125,
      "train_physics_loss": 0.0
    },
    {
      "step": 28560,
      "train_flow_matching_loss": 0.130859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 28560,
      "train_flow_matching_loss": 0.166015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 28560,
      "train_flow_matching_loss": 0.1494140625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10286516696214676,
      "learning_rate": 5e-05,
      "loss": 0.1152,
      "step": 28570
    },
    {
      "step": 28570,
      "train_flow_matching_loss": 0.2451171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 28570,
      "train_flow_matching_loss": 0.1748046875,
      "train_physics_loss": 0.0
    },
    {
      "step": 28570,
      "train_flow_matching_loss": 0.11962890625,
      "train_physics_loss": 0.0
    },
    {
      "step": 28570,
      "train_flow_matching_loss": 0.2109375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.0780075415968895,
      "learning_rate": 5e-05,
      "loss": 0.1228,
      "step": 28580
    },
    {
      "step": 28580,
      "train_flow_matching_loss": 0.1318359375,
      "train_physics_loss": 0.0
    },
    {
      "step": 28580,
      "train_flow_matching_loss": 0.1875,
      "train_physics_loss": 0.0
    },
    {
      "step": 28580,
      "train_flow_matching_loss": 0.10205078125,
      "train_physics_loss": 0.0
    },
    {
      "step": 28580,
      "train_flow_matching_loss": 0.12451171875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11123745888471603,
      "learning_rate": 5e-05,
      "loss": 0.1186,
      "step": 28590
    },
    {
      "step": 28590,
      "train_flow_matching_loss": 0.09716796875,
      "train_physics_loss": 0.0
    },
    {
      "step": 28590,
      "train_flow_matching_loss": 0.1181640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 28590,
      "train_flow_matching_loss": 0.185546875,
      "train_physics_loss": 0.0
    },
    {
      "step": 28590,
      "train_flow_matching_loss": 0.1875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.15972818434238434,
      "learning_rate": 5e-05,
      "loss": 0.1195,
      "step": 28600
    },
    {
      "step": 28600,
      "train_flow_matching_loss": 0.10546875,
      "train_physics_loss": 0.0
    },
    {
      "step": 28600,
      "train_flow_matching_loss": 0.09423828125,
      "train_physics_loss": 0.0
    },
    {
      "step": 28600,
      "train_flow_matching_loss": 0.10595703125,
      "train_physics_loss": 0.0
    },
    {
      "step": 28600,
      "train_flow_matching_loss": 0.10546875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.07370695471763611,
      "learning_rate": 5e-05,
      "loss": 0.1162,
      "step": 28610
    },
    {
      "step": 28610,
      "train_flow_matching_loss": 0.08837890625,
      "train_physics_loss": 0.0
    },
    {
      "step": 28610,
      "train_flow_matching_loss": 0.1181640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 28610,
      "train_flow_matching_loss": 0.1474609375,
      "train_physics_loss": 0.0
    },
    {
      "step": 28610,
      "train_flow_matching_loss": 0.1435546875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10099723935127258,
      "learning_rate": 5e-05,
      "loss": 0.127,
      "step": 28620
    },
    {
      "step": 28620,
      "train_flow_matching_loss": 0.06591796875,
      "train_physics_loss": 0.1037917509675026
    },
    {
      "step": 28620,
      "train_flow_matching_loss": 0.0986328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 28620,
      "train_flow_matching_loss": 0.130859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 28620,
      "train_flow_matching_loss": 0.11474609375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09756772965192795,
      "learning_rate": 5e-05,
      "loss": 0.1214,
      "step": 28630
    },
    {
      "step": 28630,
      "train_flow_matching_loss": 0.037841796875,
      "train_physics_loss": 0.1267009675502777
    },
    {
      "step": 28630,
      "train_flow_matching_loss": 0.203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 28630,
      "train_flow_matching_loss": 0.09619140625,
      "train_physics_loss": 0.0
    },
    {
      "step": 28630,
      "train_flow_matching_loss": 0.169921875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08149731159210205,
      "learning_rate": 5e-05,
      "loss": 0.12,
      "step": 28640
    },
    {
      "step": 28640,
      "train_flow_matching_loss": 0.058349609375,
      "train_physics_loss": 0.18617185950279236
    },
    {
      "step": 28640,
      "train_flow_matching_loss": 0.1435546875,
      "train_physics_loss": 0.0
    },
    {
      "step": 28640,
      "train_flow_matching_loss": 0.1298828125,
      "train_physics_loss": 0.0
    },
    {
      "step": 28640,
      "train_flow_matching_loss": 0.1728515625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.0974273607134819,
      "learning_rate": 5e-05,
      "loss": 0.1158,
      "step": 28650
    },
    {
      "step": 28650,
      "train_flow_matching_loss": 0.0240478515625,
      "train_physics_loss": 0.1133737564086914
    },
    {
      "step": 28650,
      "train_flow_matching_loss": 0.1513671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 28650,
      "train_flow_matching_loss": 0.1630859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 28650,
      "train_flow_matching_loss": 0.11767578125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09908438473939896,
      "learning_rate": 5e-05,
      "loss": 0.117,
      "step": 28660
    },
    {
      "step": 28660,
      "train_flow_matching_loss": 0.0306396484375,
      "train_physics_loss": 0.133797287940979
    },
    {
      "step": 28660,
      "train_flow_matching_loss": 0.11181640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 28660,
      "train_flow_matching_loss": 0.0927734375,
      "train_physics_loss": 0.0
    },
    {
      "step": 28660,
      "train_flow_matching_loss": 0.10546875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09533406794071198,
      "learning_rate": 5e-05,
      "loss": 0.1231,
      "step": 28670
    },
    {
      "step": 28670,
      "train_flow_matching_loss": 0.0732421875,
      "train_physics_loss": 0.14949418604373932
    },
    {
      "step": 28670,
      "train_flow_matching_loss": 0.1689453125,
      "train_physics_loss": 0.0
    },
    {
      "step": 28670,
      "train_flow_matching_loss": 0.1474609375,
      "train_physics_loss": 0.0
    },
    {
      "step": 28670,
      "train_flow_matching_loss": 0.1474609375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09128222614526749,
      "learning_rate": 5e-05,
      "loss": 0.1222,
      "step": 28680
    },
    {
      "step": 28680,
      "train_flow_matching_loss": 0.0673828125,
      "train_physics_loss": 0.16901956498622894
    },
    {
      "step": 28680,
      "train_flow_matching_loss": 0.0810546875,
      "train_physics_loss": 0.0
    },
    {
      "step": 28680,
      "train_flow_matching_loss": 0.1513671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 28680,
      "train_flow_matching_loss": 0.09521484375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.15064269304275513,
      "learning_rate": 5e-05,
      "loss": 0.1174,
      "step": 28690
    },
    {
      "step": 28690,
      "train_flow_matching_loss": 0.035400390625,
      "train_physics_loss": 0.14151346683502197
    },
    {
      "step": 28690,
      "train_flow_matching_loss": 0.1298828125,
      "train_physics_loss": 0.0
    },
    {
      "step": 28690,
      "train_flow_matching_loss": 0.150390625,
      "train_physics_loss": 0.0
    },
    {
      "step": 28690,
      "train_flow_matching_loss": 0.138671875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08725026249885559,
      "learning_rate": 5e-05,
      "loss": 0.1179,
      "step": 28700
    },
    {
      "step": 28700,
      "train_flow_matching_loss": 0.040283203125,
      "train_physics_loss": 0.13524602353572845
    },
    {
      "step": 28700,
      "train_flow_matching_loss": 0.1630859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 28700,
      "train_flow_matching_loss": 0.07275390625,
      "train_physics_loss": 0.0
    },
    {
      "step": 28700,
      "train_flow_matching_loss": 0.07421875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09063425660133362,
      "learning_rate": 5e-05,
      "loss": 0.1177,
      "step": 28710
    },
    {
      "step": 28710,
      "train_flow_matching_loss": 0.050537109375,
      "train_physics_loss": 0.1267087459564209
    },
    {
      "step": 28710,
      "train_flow_matching_loss": 0.142578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 28710,
      "train_flow_matching_loss": 0.1376953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 28710,
      "train_flow_matching_loss": 0.15625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.1334678828716278,
      "learning_rate": 5e-05,
      "loss": 0.1183,
      "step": 28720
    },
    {
      "step": 28720,
      "train_flow_matching_loss": 0.060546875,
      "train_physics_loss": 0.13355912268161774
    },
    {
      "step": 28720,
      "train_flow_matching_loss": 0.125,
      "train_physics_loss": 0.0
    },
    {
      "step": 28720,
      "train_flow_matching_loss": 0.10888671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 28720,
      "train_flow_matching_loss": 0.146484375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10364717245101929,
      "learning_rate": 5e-05,
      "loss": 0.1184,
      "step": 28730
    },
    {
      "step": 28730,
      "train_flow_matching_loss": 0.032470703125,
      "train_physics_loss": 0.11644473671913147
    },
    {
      "step": 28730,
      "train_flow_matching_loss": 0.1513671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 28730,
      "train_flow_matching_loss": 0.1416015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 28730,
      "train_flow_matching_loss": 0.10205078125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.1385839879512787,
      "learning_rate": 5e-05,
      "loss": 0.1144,
      "step": 28740
    },
    {
      "step": 28740,
      "train_flow_matching_loss": 0.0625,
      "train_physics_loss": 0.11299513280391693
    },
    {
      "step": 28740,
      "train_flow_matching_loss": 0.13671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 28740,
      "train_flow_matching_loss": 0.11376953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 28740,
      "train_flow_matching_loss": 0.0908203125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.13723178207874298,
      "learning_rate": 5e-05,
      "loss": 0.1164,
      "step": 28750
    },
    {
      "step": 28750,
      "train_flow_matching_loss": 0.0230712890625,
      "train_physics_loss": 0.10667914152145386
    },
    {
      "step": 28750,
      "train_flow_matching_loss": 0.173828125,
      "train_physics_loss": 0.0
    },
    {
      "step": 28750,
      "train_flow_matching_loss": 0.1015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 28750,
      "train_flow_matching_loss": 0.1337890625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08114110678434372,
      "learning_rate": 5e-05,
      "loss": 0.1205,
      "step": 28760
    },
    {
      "step": 28760,
      "train_flow_matching_loss": 0.05029296875,
      "train_physics_loss": 0.08839928358793259
    },
    {
      "step": 28760,
      "train_flow_matching_loss": 0.1357421875,
      "train_physics_loss": 0.0
    },
    {
      "step": 28760,
      "train_flow_matching_loss": 0.134765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 28760,
      "train_flow_matching_loss": 0.130859375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08780211210250854,
      "learning_rate": 5e-05,
      "loss": 0.1176,
      "step": 28770
    },
    {
      "step": 28770,
      "train_flow_matching_loss": 0.031494140625,
      "train_physics_loss": 0.10968339443206787
    },
    {
      "step": 28770,
      "train_flow_matching_loss": 0.171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 28770,
      "train_flow_matching_loss": 0.1845703125,
      "train_physics_loss": 0.0
    },
    {
      "step": 28770,
      "train_flow_matching_loss": 0.109375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10008876770734787,
      "learning_rate": 5e-05,
      "loss": 0.1215,
      "step": 28780
    },
    {
      "step": 28780,
      "train_flow_matching_loss": 0.0693359375,
      "train_physics_loss": 0.1007927656173706
    },
    {
      "step": 28780,
      "train_flow_matching_loss": 0.1748046875,
      "train_physics_loss": 0.0
    },
    {
      "step": 28780,
      "train_flow_matching_loss": 0.1533203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 28780,
      "train_flow_matching_loss": 0.216796875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10528595000505447,
      "learning_rate": 5e-05,
      "loss": 0.1192,
      "step": 28790
    },
    {
      "step": 28790,
      "train_flow_matching_loss": 0.04345703125,
      "train_physics_loss": 0.13976359367370605
    },
    {
      "step": 28790,
      "train_flow_matching_loss": 0.11962890625,
      "train_physics_loss": 0.0
    },
    {
      "step": 28790,
      "train_flow_matching_loss": 0.1923828125,
      "train_physics_loss": 0.0
    },
    {
      "step": 28790,
      "train_flow_matching_loss": 0.150390625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11899267137050629,
      "learning_rate": 5e-05,
      "loss": 0.118,
      "step": 28800
    },
    {
      "step": 28800,
      "train_flow_matching_loss": 0.055419921875,
      "train_physics_loss": 0.14695383608341217
    },
    {
      "step": 28800,
      "train_flow_matching_loss": 0.1455078125,
      "train_physics_loss": 0.0
    },
    {
      "step": 28800,
      "train_flow_matching_loss": 0.125,
      "train_physics_loss": 0.0
    },
    {
      "step": 28800,
      "train_flow_matching_loss": 0.2021484375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.13820898532867432,
      "learning_rate": 5e-05,
      "loss": 0.1192,
      "step": 28810
    },
    {
      "step": 28810,
      "train_flow_matching_loss": 0.056884765625,
      "train_physics_loss": 0.11030305176973343
    },
    {
      "step": 28810,
      "train_flow_matching_loss": 0.1494140625,
      "train_physics_loss": 0.0
    },
    {
      "step": 28810,
      "train_flow_matching_loss": 0.1357421875,
      "train_physics_loss": 0.0
    },
    {
      "step": 28810,
      "train_flow_matching_loss": 0.1650390625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08078283816576004,
      "learning_rate": 5e-05,
      "loss": 0.1191,
      "step": 28820
    },
    {
      "step": 28820,
      "train_flow_matching_loss": 0.04931640625,
      "train_physics_loss": 0.07394048571586609
    },
    {
      "step": 28820,
      "train_flow_matching_loss": 0.2314453125,
      "train_physics_loss": 0.0
    },
    {
      "step": 28820,
      "train_flow_matching_loss": 0.08349609375,
      "train_physics_loss": 0.0
    },
    {
      "step": 28820,
      "train_flow_matching_loss": 0.1591796875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09143350273370743,
      "learning_rate": 5e-05,
      "loss": 0.1162,
      "step": 28830
    },
    {
      "step": 28830,
      "train_flow_matching_loss": 0.05322265625,
      "train_physics_loss": 0.1236656978726387
    },
    {
      "step": 28830,
      "train_flow_matching_loss": 0.166015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 28830,
      "train_flow_matching_loss": 0.07080078125,
      "train_physics_loss": 0.0
    },
    {
      "step": 28830,
      "train_flow_matching_loss": 0.1630859375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.0867709144949913,
      "learning_rate": 5e-05,
      "loss": 0.1152,
      "step": 28840
    },
    {
      "step": 28840,
      "train_flow_matching_loss": 0.03271484375,
      "train_physics_loss": 0.1250198781490326
    },
    {
      "step": 28840,
      "train_flow_matching_loss": 0.2314453125,
      "train_physics_loss": 0.0
    },
    {
      "step": 28840,
      "train_flow_matching_loss": 0.158203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 28840,
      "train_flow_matching_loss": 0.158203125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.1284402310848236,
      "learning_rate": 5e-05,
      "loss": 0.1228,
      "step": 28850
    },
    {
      "step": 28850,
      "train_flow_matching_loss": 0.0732421875,
      "train_physics_loss": 0.10847447067499161
    },
    {
      "step": 28850,
      "train_flow_matching_loss": 0.1171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 28850,
      "train_flow_matching_loss": 0.1181640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 28850,
      "train_flow_matching_loss": 0.1796875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.1114196628332138,
      "learning_rate": 5e-05,
      "loss": 0.1216,
      "step": 28860
    },
    {
      "step": 28860,
      "train_flow_matching_loss": 0.0693359375,
      "train_physics_loss": 0.11032484471797943
    },
    {
      "step": 28860,
      "train_flow_matching_loss": 0.2216796875,
      "train_physics_loss": 0.0
    },
    {
      "step": 28860,
      "train_flow_matching_loss": 0.173828125,
      "train_physics_loss": 0.0
    },
    {
      "step": 28860,
      "train_flow_matching_loss": 0.1279296875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08537470549345016,
      "learning_rate": 5e-05,
      "loss": 0.1244,
      "step": 28870
    },
    {
      "step": 28870,
      "train_flow_matching_loss": 0.0751953125,
      "train_physics_loss": 0.16047650575637817
    },
    {
      "step": 28870,
      "train_flow_matching_loss": 0.1240234375,
      "train_physics_loss": 0.0
    },
    {
      "step": 28870,
      "train_flow_matching_loss": 0.125,
      "train_physics_loss": 0.0
    },
    {
      "step": 28870,
      "train_flow_matching_loss": 0.185546875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10052318871021271,
      "learning_rate": 5e-05,
      "loss": 0.1179,
      "step": 28880
    },
    {
      "step": 28880,
      "train_flow_matching_loss": 0.0308837890625,
      "train_physics_loss": 0.11779989302158356
    },
    {
      "step": 28880,
      "train_flow_matching_loss": 0.1611328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 28880,
      "train_flow_matching_loss": 0.1650390625,
      "train_physics_loss": 0.0
    },
    {
      "step": 28880,
      "train_flow_matching_loss": 0.1435546875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08718036860227585,
      "learning_rate": 5e-05,
      "loss": 0.1173,
      "step": 28890
    },
    {
      "step": 28890,
      "train_flow_matching_loss": 0.08203125,
      "train_physics_loss": 0.13811835646629333
    },
    {
      "step": 28890,
      "train_flow_matching_loss": 0.150390625,
      "train_physics_loss": 0.0
    },
    {
      "step": 28890,
      "train_flow_matching_loss": 0.109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 28890,
      "train_flow_matching_loss": 0.12451171875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.121559277176857,
      "learning_rate": 5e-05,
      "loss": 0.1226,
      "step": 28900
    },
    {
      "step": 28900,
      "train_flow_matching_loss": 0.0712890625,
      "train_physics_loss": 0.10402223467826843
    },
    {
      "step": 28900,
      "train_flow_matching_loss": 0.1533203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 28900,
      "train_flow_matching_loss": 0.146484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 28900,
      "train_flow_matching_loss": 0.1357421875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10202550888061523,
      "learning_rate": 5e-05,
      "loss": 0.1222,
      "step": 28910
    },
    {
      "step": 28910,
      "train_flow_matching_loss": 0.031005859375,
      "train_physics_loss": 0.09750166535377502
    },
    {
      "step": 28910,
      "train_flow_matching_loss": 0.1416015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 28910,
      "train_flow_matching_loss": 0.1474609375,
      "train_physics_loss": 0.0
    },
    {
      "step": 28910,
      "train_flow_matching_loss": 0.09814453125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11025633662939072,
      "learning_rate": 5e-05,
      "loss": 0.1204,
      "step": 28920
    },
    {
      "step": 28920,
      "train_flow_matching_loss": 0.019775390625,
      "train_physics_loss": 0.10446274280548096
    },
    {
      "step": 28920,
      "train_flow_matching_loss": 0.1611328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 28920,
      "train_flow_matching_loss": 0.109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 28920,
      "train_flow_matching_loss": 0.11767578125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09793683141469955,
      "learning_rate": 5e-05,
      "loss": 0.1271,
      "step": 28930
    },
    {
      "step": 28930,
      "train_flow_matching_loss": 0.1201171875,
      "train_physics_loss": 0.17944568395614624
    },
    {
      "step": 28930,
      "train_flow_matching_loss": 0.1005859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 28930,
      "train_flow_matching_loss": 0.11962890625,
      "train_physics_loss": 0.0
    },
    {
      "step": 28930,
      "train_flow_matching_loss": 0.1572265625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.07252851873636246,
      "learning_rate": 5e-05,
      "loss": 0.1124,
      "step": 28940
    },
    {
      "step": 28940,
      "train_flow_matching_loss": 0.030029296875,
      "train_physics_loss": 0.10408046096563339
    },
    {
      "step": 28940,
      "train_flow_matching_loss": 0.08154296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 28940,
      "train_flow_matching_loss": 0.11962890625,
      "train_physics_loss": 0.0
    },
    {
      "step": 28940,
      "train_flow_matching_loss": 0.1533203125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.2279512584209442,
      "learning_rate": 5e-05,
      "loss": 0.1149,
      "step": 28950
    },
    {
      "step": 28950,
      "train_flow_matching_loss": 0.03369140625,
      "train_physics_loss": 0.09147827327251434
    },
    {
      "step": 28950,
      "train_flow_matching_loss": 0.1630859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 28950,
      "train_flow_matching_loss": 0.1533203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 28950,
      "train_flow_matching_loss": 0.12890625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10440265387296677,
      "learning_rate": 5e-05,
      "loss": 0.1194,
      "step": 28960
    },
    {
      "step": 28960,
      "train_flow_matching_loss": 0.10498046875,
      "train_physics_loss": 0.10152123123407364
    },
    {
      "step": 28960,
      "train_flow_matching_loss": 0.07373046875,
      "train_physics_loss": 0.0
    },
    {
      "step": 28960,
      "train_flow_matching_loss": 0.11083984375,
      "train_physics_loss": 0.0
    },
    {
      "step": 28960,
      "train_flow_matching_loss": 0.1767578125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08869234472513199,
      "learning_rate": 5e-05,
      "loss": 0.1213,
      "step": 28970
    },
    {
      "step": 28970,
      "train_flow_matching_loss": 0.0283203125,
      "train_physics_loss": 0.12140398472547531
    },
    {
      "step": 28970,
      "train_flow_matching_loss": 0.0771484375,
      "train_physics_loss": 0.12104852497577667
    },
    {
      "step": 28970,
      "train_flow_matching_loss": 0.11328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 28970,
      "train_flow_matching_loss": 0.12060546875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.0959106907248497,
      "learning_rate": 5e-05,
      "loss": 0.1194,
      "step": 28980
    },
    {
      "step": 28980,
      "train_flow_matching_loss": 0.126953125,
      "train_physics_loss": 0.12658624351024628
    },
    {
      "step": 28980,
      "train_flow_matching_loss": 0.048828125,
      "train_physics_loss": 0.09912750869989395
    },
    {
      "step": 28980,
      "train_flow_matching_loss": 0.2265625,
      "train_physics_loss": 0.0
    },
    {
      "step": 28980,
      "train_flow_matching_loss": 0.11279296875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09303025156259537,
      "learning_rate": 5e-05,
      "loss": 0.1263,
      "step": 28990
    },
    {
      "step": 28990,
      "train_flow_matching_loss": 0.0712890625,
      "train_physics_loss": 0.15240007638931274
    },
    {
      "step": 28990,
      "train_flow_matching_loss": 0.0712890625,
      "train_physics_loss": 0.12797948718070984
    },
    {
      "step": 28990,
      "train_flow_matching_loss": 0.1044921875,
      "train_physics_loss": 0.0
    },
    {
      "step": 28990,
      "train_flow_matching_loss": 0.1650390625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.07775590568780899,
      "learning_rate": 5e-05,
      "loss": 0.1179,
      "step": 29000
    },
    {
      "step": 29000,
      "train_flow_matching_loss": 0.049560546875,
      "train_physics_loss": 0.08937422931194305
    },
    {
      "step": 29000,
      "train_flow_matching_loss": 0.061279296875,
      "train_physics_loss": 0.1468777060508728
    },
    {
      "step": 29000,
      "train_flow_matching_loss": 0.1630859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 29000,
      "train_flow_matching_loss": 0.1708984375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08859433233737946,
      "learning_rate": 5e-05,
      "loss": 0.1231,
      "step": 29010
    },
    {
      "step": 29010,
      "train_flow_matching_loss": 0.0281982421875,
      "train_physics_loss": 0.12159397453069687
    },
    {
      "step": 29010,
      "train_flow_matching_loss": 0.08349609375,
      "train_physics_loss": 0.10677178204059601
    },
    {
      "step": 29010,
      "train_flow_matching_loss": 0.1591796875,
      "train_physics_loss": 0.0
    },
    {
      "step": 29010,
      "train_flow_matching_loss": 0.1572265625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.0881834626197815,
      "learning_rate": 5e-05,
      "loss": 0.1194,
      "step": 29020
    },
    {
      "step": 29020,
      "train_flow_matching_loss": 0.04248046875,
      "train_physics_loss": 0.15338674187660217
    },
    {
      "step": 29020,
      "train_flow_matching_loss": 0.039794921875,
      "train_physics_loss": 0.09859932214021683
    },
    {
      "step": 29020,
      "train_flow_matching_loss": 0.1611328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 29020,
      "train_flow_matching_loss": 0.1298828125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09538273513317108,
      "learning_rate": 5e-05,
      "loss": 0.1191,
      "step": 29030
    },
    {
      "step": 29030,
      "train_flow_matching_loss": 0.0311279296875,
      "train_physics_loss": 0.11640945822000504
    },
    {
      "step": 29030,
      "train_flow_matching_loss": 0.045166015625,
      "train_physics_loss": 0.10866999626159668
    },
    {
      "step": 29030,
      "train_flow_matching_loss": 0.125,
      "train_physics_loss": 0.0
    },
    {
      "step": 29030,
      "train_flow_matching_loss": 0.06787109375,
      "train_physics_loss": 0.28887635469436646
    },
    {
      "grad_norm": 0.10839303582906723,
      "learning_rate": 5e-05,
      "loss": 0.1244,
      "step": 29040
    },
    {
      "step": 29040,
      "train_flow_matching_loss": 0.06396484375,
      "train_physics_loss": 0.12458033114671707
    },
    {
      "step": 29040,
      "train_flow_matching_loss": 0.023193359375,
      "train_physics_loss": 0.11618063598871231
    },
    {
      "step": 29040,
      "train_flow_matching_loss": 0.10498046875,
      "train_physics_loss": 0.0
    },
    {
      "step": 29040,
      "train_flow_matching_loss": 0.04345703125,
      "train_physics_loss": 0.1470707356929779
    },
    {
      "grad_norm": 0.10911644250154495,
      "learning_rate": 5e-05,
      "loss": 0.1172,
      "step": 29050
    },
    {
      "step": 29050,
      "train_flow_matching_loss": 0.04833984375,
      "train_physics_loss": 0.07634492963552475
    },
    {
      "step": 29050,
      "train_flow_matching_loss": 0.08544921875,
      "train_physics_loss": 0.10744807869195938
    },
    {
      "step": 29050,
      "train_flow_matching_loss": 0.232421875,
      "train_physics_loss": 0.0
    },
    {
      "step": 29050,
      "train_flow_matching_loss": 0.0225830078125,
      "train_physics_loss": 0.10328764468431473
    },
    {
      "grad_norm": 0.12671644985675812,
      "learning_rate": 5e-05,
      "loss": 0.1222,
      "step": 29060
    },
    {
      "step": 29060,
      "train_flow_matching_loss": 0.06396484375,
      "train_physics_loss": 0.11882209777832031
    },
    {
      "step": 29060,
      "train_flow_matching_loss": 0.045166015625,
      "train_physics_loss": 0.13855841755867004
    },
    {
      "step": 29060,
      "train_flow_matching_loss": 0.07958984375,
      "train_physics_loss": 0.0
    },
    {
      "step": 29060,
      "train_flow_matching_loss": 0.1064453125,
      "train_physics_loss": 0.1357363760471344
    },
    {
      "grad_norm": 0.08064426481723785,
      "learning_rate": 5e-05,
      "loss": 0.1187,
      "step": 29070
    },
    {
      "step": 29070,
      "train_flow_matching_loss": 0.029052734375,
      "train_physics_loss": 0.12014096975326538
    },
    {
      "step": 29070,
      "train_flow_matching_loss": 0.035400390625,
      "train_physics_loss": 0.0942435935139656
    },
    {
      "step": 29070,
      "train_flow_matching_loss": 0.037353515625,
      "train_physics_loss": 0.15834049880504608
    },
    {
      "step": 29070,
      "train_flow_matching_loss": 0.03515625,
      "train_physics_loss": 0.15279558300971985
    },
    {
      "grad_norm": 0.1189350038766861,
      "learning_rate": 5e-05,
      "loss": 0.1204,
      "step": 29080
    },
    {
      "step": 29080,
      "train_flow_matching_loss": 0.050537109375,
      "train_physics_loss": 0.09701734781265259
    },
    {
      "step": 29080,
      "train_flow_matching_loss": 0.045166015625,
      "train_physics_loss": 0.11396358907222748
    },
    {
      "step": 29080,
      "train_flow_matching_loss": 0.07080078125,
      "train_physics_loss": 0.30621466040611267
    },
    {
      "step": 29080,
      "train_flow_matching_loss": 0.052978515625,
      "train_physics_loss": 0.18420468270778656
    },
    {
      "grad_norm": 0.06380618363618851,
      "learning_rate": 5e-05,
      "loss": 0.1235,
      "step": 29090
    },
    {
      "step": 29090,
      "train_flow_matching_loss": 0.056640625,
      "train_physics_loss": 0.11957927793264389
    },
    {
      "step": 29090,
      "train_flow_matching_loss": 0.0791015625,
      "train_physics_loss": 0.1514342874288559
    },
    {
      "step": 29090,
      "train_flow_matching_loss": 0.041259765625,
      "train_physics_loss": 0.15799564123153687
    },
    {
      "step": 29090,
      "train_flow_matching_loss": 0.059814453125,
      "train_physics_loss": 0.09348428249359131
    },
    {
      "grad_norm": 0.10043800622224808,
      "learning_rate": 5e-05,
      "loss": 0.1176,
      "step": 29100
    },
    {
      "step": 29100,
      "train_flow_matching_loss": 0.03466796875,
      "train_physics_loss": 0.14593513309955597
    },
    {
      "step": 29100,
      "train_flow_matching_loss": 0.046875,
      "train_physics_loss": 0.10539529472589493
    },
    {
      "step": 29100,
      "train_flow_matching_loss": 0.04736328125,
      "train_physics_loss": 0.10507258027791977
    },
    {
      "step": 29100,
      "train_flow_matching_loss": 0.04931640625,
      "train_physics_loss": 0.10088632255792618
    },
    {
      "grad_norm": 0.11699943244457245,
      "learning_rate": 5e-05,
      "loss": 0.1243,
      "step": 29110
    },
    {
      "step": 29110,
      "train_flow_matching_loss": 0.0751953125,
      "train_physics_loss": 0.09867900609970093
    },
    {
      "step": 29110,
      "train_flow_matching_loss": 0.0240478515625,
      "train_physics_loss": 0.1164608895778656
    },
    {
      "step": 29110,
      "train_flow_matching_loss": 0.06982421875,
      "train_physics_loss": 0.1464841514825821
    },
    {
      "step": 29110,
      "train_flow_matching_loss": 0.033447265625,
      "train_physics_loss": 0.09288699179887772
    },
    {
      "grad_norm": 0.0646224096417427,
      "learning_rate": 5e-05,
      "loss": 0.1173,
      "step": 29120
    },
    {
      "step": 29120,
      "train_flow_matching_loss": 0.08447265625,
      "train_physics_loss": 0.08591328561306
    },
    {
      "step": 29120,
      "train_flow_matching_loss": 0.060791015625,
      "train_physics_loss": 0.15279927849769592
    },
    {
      "step": 29120,
      "train_flow_matching_loss": 0.05810546875,
      "train_physics_loss": 0.1086013913154602
    },
    {
      "step": 29120,
      "train_flow_matching_loss": 0.045654296875,
      "train_physics_loss": 0.14445175230503082
    },
    {
      "grad_norm": 0.0925908014178276,
      "learning_rate": 5e-05,
      "loss": 0.1175,
      "step": 29130
    },
    {
      "step": 29130,
      "train_flow_matching_loss": 0.046875,
      "train_physics_loss": 0.13589948415756226
    },
    {
      "step": 29130,
      "train_flow_matching_loss": 0.031494140625,
      "train_physics_loss": 0.13476812839508057
    },
    {
      "step": 29130,
      "train_flow_matching_loss": 0.11572265625,
      "train_physics_loss": 0.15403297543525696
    },
    {
      "step": 29130,
      "train_flow_matching_loss": 0.03369140625,
      "train_physics_loss": 0.10241833329200745
    },
    {
      "grad_norm": 0.11616130918264389,
      "learning_rate": 5e-05,
      "loss": 0.1203,
      "step": 29140
    },
    {
      "step": 29140,
      "train_flow_matching_loss": 0.1337890625,
      "train_physics_loss": 0.0
    },
    {
      "step": 29140,
      "train_flow_matching_loss": 0.04833984375,
      "train_physics_loss": 0.10149019211530685
    },
    {
      "step": 29140,
      "train_flow_matching_loss": 0.041748046875,
      "train_physics_loss": 0.10599387437105179
    },
    {
      "step": 29140,
      "train_flow_matching_loss": 0.048828125,
      "train_physics_loss": 0.09809782356023788
    },
    {
      "grad_norm": 0.09102144092321396,
      "learning_rate": 5e-05,
      "loss": 0.1158,
      "step": 29150
    },
    {
      "step": 29150,
      "train_flow_matching_loss": 0.1015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 29150,
      "train_flow_matching_loss": 0.05126953125,
      "train_physics_loss": 0.10005820542573929
    },
    {
      "step": 29150,
      "train_flow_matching_loss": 0.033935546875,
      "train_physics_loss": 0.10378038883209229
    },
    {
      "step": 29150,
      "train_flow_matching_loss": 0.0164794921875,
      "train_physics_loss": 0.07342492043972015
    },
    {
      "grad_norm": 0.06069101020693779,
      "learning_rate": 5e-05,
      "loss": 0.1146,
      "step": 29160
    },
    {
      "step": 29160,
      "train_flow_matching_loss": 0.08056640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 29160,
      "train_flow_matching_loss": 0.052001953125,
      "train_physics_loss": 0.11092590540647507
    },
    {
      "step": 29160,
      "train_flow_matching_loss": 0.0262451171875,
      "train_physics_loss": 0.10250555723905563
    },
    {
      "step": 29160,
      "train_flow_matching_loss": 0.025146484375,
      "train_physics_loss": 0.09065353125333786
    },
    {
      "grad_norm": 0.17086347937583923,
      "learning_rate": 5e-05,
      "loss": 0.1183,
      "step": 29170
    },
    {
      "step": 29170,
      "train_flow_matching_loss": 0.1435546875,
      "train_physics_loss": 0.0
    },
    {
      "step": 29170,
      "train_flow_matching_loss": 0.040283203125,
      "train_physics_loss": 0.1435241848230362
    },
    {
      "step": 29170,
      "train_flow_matching_loss": 0.06982421875,
      "train_physics_loss": 0.12154347449541092
    },
    {
      "step": 29170,
      "train_flow_matching_loss": 0.031494140625,
      "train_physics_loss": 0.16001367568969727
    },
    {
      "grad_norm": 0.08988968282938004,
      "learning_rate": 5e-05,
      "loss": 0.1197,
      "step": 29180
    },
    {
      "step": 29180,
      "train_flow_matching_loss": 0.0849609375,
      "train_physics_loss": 0.0
    },
    {
      "step": 29180,
      "train_flow_matching_loss": 0.0654296875,
      "train_physics_loss": 0.09734179824590683
    },
    {
      "step": 29180,
      "train_flow_matching_loss": 0.0267333984375,
      "train_physics_loss": 0.15544025599956512
    },
    {
      "step": 29180,
      "train_flow_matching_loss": 0.048095703125,
      "train_physics_loss": 0.1396302729845047
    },
    {
      "grad_norm": 0.09511175006628036,
      "learning_rate": 5e-05,
      "loss": 0.1171,
      "step": 29190
    },
    {
      "step": 29190,
      "train_flow_matching_loss": 0.1982421875,
      "train_physics_loss": 0.0
    },
    {
      "step": 29190,
      "train_flow_matching_loss": 0.05029296875,
      "train_physics_loss": 0.08626015484333038
    },
    {
      "step": 29190,
      "train_flow_matching_loss": 0.056884765625,
      "train_physics_loss": 0.147512748837471
    },
    {
      "step": 29190,
      "train_flow_matching_loss": 0.05224609375,
      "train_physics_loss": 0.21307377517223358
    },
    {
      "grad_norm": 0.0841279849410057,
      "learning_rate": 5e-05,
      "loss": 0.1174,
      "step": 29200
    },
    {
      "step": 29200,
      "train_flow_matching_loss": 0.0654296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 29200,
      "train_flow_matching_loss": 0.08447265625,
      "train_physics_loss": 0.14557935297489166
    },
    {
      "step": 29200,
      "train_flow_matching_loss": 0.058837890625,
      "train_physics_loss": 0.12559251487255096
    },
    {
      "step": 29200,
      "train_flow_matching_loss": 0.0751953125,
      "train_physics_loss": 0.16898201406002045
    },
    {
      "grad_norm": 0.0701741874217987,
      "learning_rate": 5e-05,
      "loss": 0.1209,
      "step": 29210
    },
    {
      "step": 29210,
      "train_flow_matching_loss": 0.09130859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 29210,
      "train_flow_matching_loss": 0.04443359375,
      "train_physics_loss": 0.13504786789417267
    },
    {
      "step": 29210,
      "train_flow_matching_loss": 0.10009765625,
      "train_physics_loss": 0.1441473364830017
    },
    {
      "step": 29210,
      "train_flow_matching_loss": 0.04345703125,
      "train_physics_loss": 0.1298561692237854
    },
    {
      "grad_norm": 0.10418157279491425,
      "learning_rate": 5e-05,
      "loss": 0.1197,
      "step": 29220
    },
    {
      "step": 29220,
      "train_flow_matching_loss": 0.0732421875,
      "train_physics_loss": 0.0
    },
    {
      "step": 29220,
      "train_flow_matching_loss": 0.07373046875,
      "train_physics_loss": 0.08924489468336105
    },
    {
      "step": 29220,
      "train_flow_matching_loss": 0.04345703125,
      "train_physics_loss": 0.15685012936592102
    },
    {
      "step": 29220,
      "train_flow_matching_loss": 0.038818359375,
      "train_physics_loss": 0.08985108137130737
    },
    {
      "grad_norm": 0.09148305654525757,
      "learning_rate": 5e-05,
      "loss": 0.1163,
      "step": 29230
    },
    {
      "step": 29230,
      "train_flow_matching_loss": 0.07958984375,
      "train_physics_loss": 0.0
    },
    {
      "step": 29230,
      "train_flow_matching_loss": 0.173828125,
      "train_physics_loss": 0.0
    },
    {
      "step": 29230,
      "train_flow_matching_loss": 0.06640625,
      "train_physics_loss": 0.18927796185016632
    },
    {
      "step": 29230,
      "train_flow_matching_loss": 0.03271484375,
      "train_physics_loss": 0.06410858035087585
    },
    {
      "grad_norm": 0.07669293880462646,
      "learning_rate": 5e-05,
      "loss": 0.1173,
      "step": 29240
    },
    {
      "step": 29240,
      "train_flow_matching_loss": 0.1328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 29240,
      "train_flow_matching_loss": 0.236328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 29240,
      "train_flow_matching_loss": 0.040771484375,
      "train_physics_loss": 0.12310484796762466
    },
    {
      "step": 29240,
      "train_flow_matching_loss": 0.034912109375,
      "train_physics_loss": 0.1111770048737526
    },
    {
      "grad_norm": 0.09430154412984848,
      "learning_rate": 5e-05,
      "loss": 0.1121,
      "step": 29250
    },
    {
      "step": 29250,
      "train_flow_matching_loss": 0.1181640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 29250,
      "train_flow_matching_loss": 0.205078125,
      "train_physics_loss": 0.0
    },
    {
      "step": 29250,
      "train_flow_matching_loss": 0.044921875,
      "train_physics_loss": 0.1471543312072754
    },
    {
      "step": 29250,
      "train_flow_matching_loss": 0.03466796875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.14719507098197937,
      "learning_rate": 5e-05,
      "loss": 0.1194,
      "step": 29260
    },
    {
      "step": 29260,
      "train_flow_matching_loss": 0.08984375,
      "train_physics_loss": 0.0
    },
    {
      "step": 29260,
      "train_flow_matching_loss": 0.1611328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 29260,
      "train_flow_matching_loss": 0.06396484375,
      "train_physics_loss": 0.15844282507896423
    },
    {
      "step": 29260,
      "train_flow_matching_loss": 0.0625,
      "train_physics_loss": 0.10614578425884247
    },
    {
      "grad_norm": 0.08057865500450134,
      "learning_rate": 5e-05,
      "loss": 0.1221,
      "step": 29270
    },
    {
      "step": 29270,
      "train_flow_matching_loss": 0.1484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 29270,
      "train_flow_matching_loss": 0.1103515625,
      "train_physics_loss": 0.0
    },
    {
      "step": 29270,
      "train_flow_matching_loss": 0.0264892578125,
      "train_physics_loss": 0.14993758499622345
    },
    {
      "step": 29270,
      "train_flow_matching_loss": 0.030029296875,
      "train_physics_loss": 0.09487534314393997
    },
    {
      "grad_norm": 0.09311511367559433,
      "learning_rate": 5e-05,
      "loss": 0.1165,
      "step": 29280
    },
    {
      "step": 29280,
      "train_flow_matching_loss": 0.169921875,
      "train_physics_loss": 0.0
    },
    {
      "step": 29280,
      "train_flow_matching_loss": 0.150390625,
      "train_physics_loss": 0.0
    },
    {
      "step": 29280,
      "train_flow_matching_loss": 0.0810546875,
      "train_physics_loss": 0.14509063959121704
    },
    {
      "step": 29280,
      "train_flow_matching_loss": 0.04150390625,
      "train_physics_loss": 0.13173432648181915
    },
    {
      "grad_norm": 0.0952124297618866,
      "learning_rate": 5e-05,
      "loss": 0.1194,
      "step": 29290
    },
    {
      "step": 29290,
      "train_flow_matching_loss": 0.1162109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 29290,
      "train_flow_matching_loss": 0.1533203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 29290,
      "train_flow_matching_loss": 0.07763671875,
      "train_physics_loss": 0.1291094869375229
    },
    {
      "step": 29290,
      "train_flow_matching_loss": 0.125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.14389409124851227,
      "learning_rate": 5e-05,
      "loss": 0.112,
      "step": 29300
    },
    {
      "step": 29300,
      "train_flow_matching_loss": 0.142578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 29300,
      "train_flow_matching_loss": 0.166015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 29300,
      "train_flow_matching_loss": 0.037841796875,
      "train_physics_loss": 0.06621803343296051
    },
    {
      "step": 29300,
      "train_flow_matching_loss": 0.208984375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11064550280570984,
      "learning_rate": 5e-05,
      "loss": 0.1218,
      "step": 29310
    },
    {
      "step": 29310,
      "train_flow_matching_loss": 0.1396484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 29310,
      "train_flow_matching_loss": 0.1123046875,
      "train_physics_loss": 0.0
    },
    {
      "step": 29310,
      "train_flow_matching_loss": 0.057373046875,
      "train_physics_loss": 0.15282265841960907
    },
    {
      "step": 29310,
      "train_flow_matching_loss": 0.0771484375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10727288573980331,
      "learning_rate": 5e-05,
      "loss": 0.1175,
      "step": 29320
    },
    {
      "step": 29320,
      "train_flow_matching_loss": 0.1318359375,
      "train_physics_loss": 0.0
    },
    {
      "step": 29320,
      "train_flow_matching_loss": 0.1728515625,
      "train_physics_loss": 0.0
    },
    {
      "step": 29320,
      "train_flow_matching_loss": 0.1357421875,
      "train_physics_loss": 0.18802234530448914
    },
    {
      "step": 29320,
      "train_flow_matching_loss": 0.15625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.07598307728767395,
      "learning_rate": 5e-05,
      "loss": 0.1243,
      "step": 29330
    },
    {
      "step": 29330,
      "train_flow_matching_loss": 0.09521484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 29330,
      "train_flow_matching_loss": 0.1708984375,
      "train_physics_loss": 0.0
    },
    {
      "step": 29330,
      "train_flow_matching_loss": 0.1640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 29330,
      "train_flow_matching_loss": 0.10498046875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11177127808332443,
      "learning_rate": 5e-05,
      "loss": 0.1184,
      "step": 29340
    },
    {
      "step": 29340,
      "train_flow_matching_loss": 0.140625,
      "train_physics_loss": 0.0
    },
    {
      "step": 29340,
      "train_flow_matching_loss": 0.138671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 29340,
      "train_flow_matching_loss": 0.1083984375,
      "train_physics_loss": 0.0
    },
    {
      "step": 29340,
      "train_flow_matching_loss": 0.11767578125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.16046418249607086,
      "learning_rate": 5e-05,
      "loss": 0.1199,
      "step": 29350
    },
    {
      "step": 29350,
      "train_flow_matching_loss": 0.1396484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 29350,
      "train_flow_matching_loss": 0.220703125,
      "train_physics_loss": 0.0
    },
    {
      "step": 29350,
      "train_flow_matching_loss": 0.1865234375,
      "train_physics_loss": 0.0
    },
    {
      "step": 29350,
      "train_flow_matching_loss": 0.1357421875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.14747022092342377,
      "learning_rate": 5e-05,
      "loss": 0.1191,
      "step": 29360
    },
    {
      "step": 29360,
      "train_flow_matching_loss": 0.10498046875,
      "train_physics_loss": 0.0
    },
    {
      "step": 29360,
      "train_flow_matching_loss": 0.2158203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 29360,
      "train_flow_matching_loss": 0.169921875,
      "train_physics_loss": 0.0
    },
    {
      "step": 29360,
      "train_flow_matching_loss": 0.09326171875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.16259931027889252,
      "learning_rate": 5e-05,
      "loss": 0.1152,
      "step": 29370
    },
    {
      "step": 29370,
      "train_flow_matching_loss": 0.158203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 29370,
      "train_flow_matching_loss": 0.291015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 29370,
      "train_flow_matching_loss": 0.154296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 29370,
      "train_flow_matching_loss": 0.1005859375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.18756474554538727,
      "learning_rate": 5e-05,
      "loss": 0.1188,
      "step": 29380
    },
    {
      "step": 29380,
      "train_flow_matching_loss": 0.09716796875,
      "train_physics_loss": 0.0
    },
    {
      "step": 29380,
      "train_flow_matching_loss": 0.1728515625,
      "train_physics_loss": 0.0
    },
    {
      "step": 29380,
      "train_flow_matching_loss": 0.146484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 29380,
      "train_flow_matching_loss": 0.1484375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09886347502470016,
      "learning_rate": 5e-05,
      "loss": 0.1183,
      "step": 29390
    },
    {
      "step": 29390,
      "train_flow_matching_loss": 0.08349609375,
      "train_physics_loss": 0.0
    },
    {
      "step": 29390,
      "train_flow_matching_loss": 0.162109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 29390,
      "train_flow_matching_loss": 0.1494140625,
      "train_physics_loss": 0.0
    },
    {
      "step": 29390,
      "train_flow_matching_loss": 0.1240234375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11205873638391495,
      "learning_rate": 5e-05,
      "loss": 0.1244,
      "step": 29400
    },
    {
      "step": 29400,
      "train_flow_matching_loss": 0.09375,
      "train_physics_loss": 0.0
    },
    {
      "step": 29400,
      "train_flow_matching_loss": 0.1201171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 29400,
      "train_flow_matching_loss": 0.189453125,
      "train_physics_loss": 0.0
    },
    {
      "step": 29400,
      "train_flow_matching_loss": 0.12451171875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 1.6524797677993774,
      "learning_rate": 5e-05,
      "loss": 0.1158,
      "step": 29410
    },
    {
      "step": 29410,
      "train_flow_matching_loss": 0.220703125,
      "train_physics_loss": 0.0
    },
    {
      "step": 29410,
      "train_flow_matching_loss": 0.212890625,
      "train_physics_loss": 0.0
    },
    {
      "step": 29410,
      "train_flow_matching_loss": 0.08154296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 29410,
      "train_flow_matching_loss": 0.1201171875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11546137928962708,
      "learning_rate": 5e-05,
      "loss": 0.1154,
      "step": 29420
    },
    {
      "step": 29420,
      "train_flow_matching_loss": 0.1396484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 29420,
      "train_flow_matching_loss": 0.1669921875,
      "train_physics_loss": 0.0
    },
    {
      "step": 29420,
      "train_flow_matching_loss": 0.146484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 29420,
      "train_flow_matching_loss": 0.142578125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.0848233625292778,
      "learning_rate": 5e-05,
      "loss": 0.1184,
      "step": 29430
    },
    {
      "step": 29430,
      "train_flow_matching_loss": 0.1591796875,
      "train_physics_loss": 0.0
    },
    {
      "step": 29430,
      "train_flow_matching_loss": 0.1630859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 29430,
      "train_flow_matching_loss": 0.12060546875,
      "train_physics_loss": 0.0
    },
    {
      "step": 29430,
      "train_flow_matching_loss": 0.11328125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.07664811611175537,
      "learning_rate": 5e-05,
      "loss": 0.1152,
      "step": 29440
    },
    {
      "step": 29440,
      "train_flow_matching_loss": 0.09765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 29440,
      "train_flow_matching_loss": 0.177734375,
      "train_physics_loss": 0.0
    },
    {
      "step": 29440,
      "train_flow_matching_loss": 0.10986328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 29440,
      "train_flow_matching_loss": 0.158203125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08724138885736465,
      "learning_rate": 5e-05,
      "loss": 0.1171,
      "step": 29450
    },
    {
      "step": 29450,
      "train_flow_matching_loss": 0.0859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 29450,
      "train_flow_matching_loss": 0.205078125,
      "train_physics_loss": 0.0
    },
    {
      "step": 29450,
      "train_flow_matching_loss": 0.1484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 29450,
      "train_flow_matching_loss": 0.0771484375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.12524008750915527,
      "learning_rate": 5e-05,
      "loss": 0.1216,
      "step": 29460
    },
    {
      "step": 29460,
      "train_flow_matching_loss": 0.12451171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 29460,
      "train_flow_matching_loss": 0.154296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 29460,
      "train_flow_matching_loss": 0.2041015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 29460,
      "train_flow_matching_loss": 0.162109375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.07888191193342209,
      "learning_rate": 5e-05,
      "loss": 0.1134,
      "step": 29470
    },
    {
      "step": 29470,
      "train_flow_matching_loss": 0.201171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 29470,
      "train_flow_matching_loss": 0.2041015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 29470,
      "train_flow_matching_loss": 0.1630859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 29470,
      "train_flow_matching_loss": 0.1416015625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08505556732416153,
      "learning_rate": 5e-05,
      "loss": 0.1177,
      "step": 29480
    },
    {
      "step": 29480,
      "train_flow_matching_loss": 0.1376953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 29480,
      "train_flow_matching_loss": 0.1865234375,
      "train_physics_loss": 0.0
    },
    {
      "step": 29480,
      "train_flow_matching_loss": 0.173828125,
      "train_physics_loss": 0.0
    },
    {
      "step": 29480,
      "train_flow_matching_loss": 0.181640625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10843765735626221,
      "learning_rate": 5e-05,
      "loss": 0.1161,
      "step": 29490
    },
    {
      "step": 29490,
      "train_flow_matching_loss": 0.07080078125,
      "train_physics_loss": 0.0
    },
    {
      "step": 29490,
      "train_flow_matching_loss": 0.1552734375,
      "train_physics_loss": 0.0
    },
    {
      "step": 29490,
      "train_flow_matching_loss": 0.1396484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 29490,
      "train_flow_matching_loss": 0.1162109375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.12490518391132355,
      "learning_rate": 5e-05,
      "loss": 0.1127,
      "step": 29500
    },
    {
      "step": 29500,
      "train_flow_matching_loss": 0.11474609375,
      "train_physics_loss": 0.0
    },
    {
      "step": 29500,
      "train_flow_matching_loss": 0.1533203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 29500,
      "train_flow_matching_loss": 0.130859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 29500,
      "train_flow_matching_loss": 0.126953125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10641224682331085,
      "learning_rate": 5e-05,
      "loss": 0.1171,
      "step": 29510
    },
    {
      "step": 29510,
      "train_flow_matching_loss": 0.07275390625,
      "train_physics_loss": 0.0
    },
    {
      "step": 29510,
      "train_flow_matching_loss": 0.171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 29510,
      "train_flow_matching_loss": 0.1650390625,
      "train_physics_loss": 0.0
    },
    {
      "step": 29510,
      "train_flow_matching_loss": 0.11083984375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10698706656694412,
      "learning_rate": 5e-05,
      "loss": 0.1205,
      "step": 29520
    },
    {
      "step": 29520,
      "train_flow_matching_loss": 0.0869140625,
      "train_physics_loss": 0.0
    },
    {
      "step": 29520,
      "train_flow_matching_loss": 0.0830078125,
      "train_physics_loss": 0.0
    },
    {
      "step": 29520,
      "train_flow_matching_loss": 0.1474609375,
      "train_physics_loss": 0.0
    },
    {
      "step": 29520,
      "train_flow_matching_loss": 0.1025390625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11030145734548569,
      "learning_rate": 5e-05,
      "loss": 0.1174,
      "step": 29530
    },
    {
      "step": 29530,
      "train_flow_matching_loss": 0.07421875,
      "train_physics_loss": 0.0
    },
    {
      "step": 29530,
      "train_flow_matching_loss": 0.0810546875,
      "train_physics_loss": 0.0
    },
    {
      "step": 29530,
      "train_flow_matching_loss": 0.126953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 29530,
      "train_flow_matching_loss": 0.07861328125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10440317541360855,
      "learning_rate": 5e-05,
      "loss": 0.1145,
      "step": 29540
    },
    {
      "step": 29540,
      "train_flow_matching_loss": 0.07861328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 29540,
      "train_flow_matching_loss": 0.10009765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 29540,
      "train_flow_matching_loss": 0.10546875,
      "train_physics_loss": 0.0
    },
    {
      "step": 29540,
      "train_flow_matching_loss": 0.09912109375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08396976441144943,
      "learning_rate": 5e-05,
      "loss": 0.1125,
      "step": 29550
    },
    {
      "step": 29550,
      "train_flow_matching_loss": 0.10107421875,
      "train_physics_loss": 0.0
    },
    {
      "step": 29550,
      "train_flow_matching_loss": 0.11376953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 29550,
      "train_flow_matching_loss": 0.166015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 29550,
      "train_flow_matching_loss": 0.12255859375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10757853835821152,
      "learning_rate": 5e-05,
      "loss": 0.1193,
      "step": 29560
    },
    {
      "step": 29560,
      "train_flow_matching_loss": 0.0810546875,
      "train_physics_loss": 0.0
    },
    {
      "step": 29560,
      "train_flow_matching_loss": 0.10107421875,
      "train_physics_loss": 0.0
    },
    {
      "step": 29560,
      "train_flow_matching_loss": 0.1298828125,
      "train_physics_loss": 0.0
    },
    {
      "step": 29560,
      "train_flow_matching_loss": 0.08935546875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.0805845707654953,
      "learning_rate": 5e-05,
      "loss": 0.1108,
      "step": 29570
    },
    {
      "step": 29570,
      "train_flow_matching_loss": 0.1083984375,
      "train_physics_loss": 0.0
    },
    {
      "step": 29570,
      "train_flow_matching_loss": 0.08056640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 29570,
      "train_flow_matching_loss": 0.126953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 29570,
      "train_flow_matching_loss": 0.10400390625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.07646231353282928,
      "learning_rate": 5e-05,
      "loss": 0.1169,
      "step": 29580
    },
    {
      "step": 29580,
      "train_flow_matching_loss": 0.068359375,
      "train_physics_loss": 0.0
    },
    {
      "step": 29580,
      "train_flow_matching_loss": 0.0986328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 29580,
      "train_flow_matching_loss": 0.150390625,
      "train_physics_loss": 0.0
    },
    {
      "step": 29580,
      "train_flow_matching_loss": 0.07275390625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08775024116039276,
      "learning_rate": 5e-05,
      "loss": 0.1156,
      "step": 29590
    },
    {
      "step": 29590,
      "train_flow_matching_loss": 0.11181640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 29590,
      "train_flow_matching_loss": 0.09423828125,
      "train_physics_loss": 0.0
    },
    {
      "step": 29590,
      "train_flow_matching_loss": 0.1396484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 29590,
      "train_flow_matching_loss": 0.1357421875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.06992938369512558,
      "learning_rate": 5e-05,
      "loss": 0.1203,
      "step": 29600
    },
    {
      "step": 29600,
      "train_flow_matching_loss": 0.1396484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 29600,
      "train_flow_matching_loss": 0.146484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 29600,
      "train_flow_matching_loss": 0.12060546875,
      "train_physics_loss": 0.0
    },
    {
      "step": 29600,
      "train_flow_matching_loss": 0.1728515625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10157228261232376,
      "learning_rate": 5e-05,
      "loss": 0.1173,
      "step": 29610
    },
    {
      "step": 29610,
      "train_flow_matching_loss": 0.07666015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 29610,
      "train_flow_matching_loss": 0.0654296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 29610,
      "train_flow_matching_loss": 0.12890625,
      "train_physics_loss": 0.0
    },
    {
      "step": 29610,
      "train_flow_matching_loss": 0.1201171875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.10684161633253098,
      "learning_rate": 5e-05,
      "loss": 0.1149,
      "step": 29620
    },
    {
      "step": 29620,
      "train_flow_matching_loss": 0.052490234375,
      "train_physics_loss": 0.0
    },
    {
      "step": 29620,
      "train_flow_matching_loss": 0.09619140625,
      "train_physics_loss": 0.0
    },
    {
      "step": 29620,
      "train_flow_matching_loss": 0.1005859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 29620,
      "train_flow_matching_loss": 0.1103515625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08663863688707352,
      "learning_rate": 5e-05,
      "loss": 0.1112,
      "step": 29630
    },
    {
      "step": 29630,
      "train_flow_matching_loss": 0.11181640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 29630,
      "train_flow_matching_loss": 0.146484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 29630,
      "train_flow_matching_loss": 0.2060546875,
      "train_physics_loss": 0.0
    },
    {
      "step": 29630,
      "train_flow_matching_loss": 0.1591796875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11210799962282181,
      "learning_rate": 5e-05,
      "loss": 0.1127,
      "step": 29640
    },
    {
      "step": 29640,
      "train_flow_matching_loss": 0.1884765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 29640,
      "train_flow_matching_loss": 0.08203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 29640,
      "train_flow_matching_loss": 0.171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 29640,
      "train_flow_matching_loss": 0.11376953125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09960578382015228,
      "learning_rate": 5e-05,
      "loss": 0.1193,
      "step": 29650
    },
    {
      "step": 29650,
      "train_flow_matching_loss": 0.034912109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 29650,
      "train_flow_matching_loss": 0.134765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 29650,
      "train_flow_matching_loss": 0.142578125,
      "train_physics_loss": 0.0
    },
    {
      "step": 29650,
      "train_flow_matching_loss": 0.1181640625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09122616052627563,
      "learning_rate": 5e-05,
      "loss": 0.1187,
      "step": 29660
    },
    {
      "step": 29660,
      "train_flow_matching_loss": 0.2333984375,
      "train_physics_loss": 0.0
    },
    {
      "step": 29660,
      "train_flow_matching_loss": 0.10400390625,
      "train_physics_loss": 0.0
    },
    {
      "step": 29660,
      "train_flow_matching_loss": 0.09228515625,
      "train_physics_loss": 0.0
    },
    {
      "step": 29660,
      "train_flow_matching_loss": 0.12109375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09558714181184769,
      "learning_rate": 5e-05,
      "loss": 0.1145,
      "step": 29670
    },
    {
      "step": 29670,
      "train_flow_matching_loss": 0.130859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 29670,
      "train_flow_matching_loss": 0.11474609375,
      "train_physics_loss": 0.0
    },
    {
      "step": 29670,
      "train_flow_matching_loss": 0.1572265625,
      "train_physics_loss": 0.0
    },
    {
      "step": 29670,
      "train_flow_matching_loss": 0.09130859375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.12213035672903061,
      "learning_rate": 5e-05,
      "loss": 0.1194,
      "step": 29680
    },
    {
      "step": 29680,
      "train_flow_matching_loss": 0.10986328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 29680,
      "train_flow_matching_loss": 0.08203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 29680,
      "train_flow_matching_loss": 0.1708984375,
      "train_physics_loss": 0.0
    },
    {
      "step": 29680,
      "train_flow_matching_loss": 0.0869140625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.12093779444694519,
      "learning_rate": 5e-05,
      "loss": 0.1177,
      "step": 29690
    },
    {
      "step": 29690,
      "train_flow_matching_loss": 0.1337890625,
      "train_physics_loss": 0.0
    },
    {
      "step": 29690,
      "train_flow_matching_loss": 0.10986328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 29690,
      "train_flow_matching_loss": 0.1259765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 29690,
      "train_flow_matching_loss": 0.1171875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11179955303668976,
      "learning_rate": 5e-05,
      "loss": 0.1148,
      "step": 29700
    },
    {
      "step": 29700,
      "train_flow_matching_loss": 0.146484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 29700,
      "train_flow_matching_loss": 0.10400390625,
      "train_physics_loss": 0.0
    },
    {
      "step": 29700,
      "train_flow_matching_loss": 0.0830078125,
      "train_physics_loss": 0.0
    },
    {
      "step": 29700,
      "train_flow_matching_loss": 0.1015625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.0895310565829277,
      "learning_rate": 5e-05,
      "loss": 0.1199,
      "step": 29710
    },
    {
      "step": 29710,
      "train_flow_matching_loss": 0.2099609375,
      "train_physics_loss": 0.0
    },
    {
      "step": 29710,
      "train_flow_matching_loss": 0.0908203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 29710,
      "train_flow_matching_loss": 0.126953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 29710,
      "train_flow_matching_loss": 0.1630859375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.7663776278495789,
      "learning_rate": 5e-05,
      "loss": 0.1186,
      "step": 29720
    },
    {
      "step": 29720,
      "train_flow_matching_loss": 0.1533203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 29720,
      "train_flow_matching_loss": 0.09326171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 29720,
      "train_flow_matching_loss": 0.1279296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 29720,
      "train_flow_matching_loss": 0.11474609375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11475182324647903,
      "learning_rate": 5e-05,
      "loss": 0.1178,
      "step": 29730
    },
    {
      "step": 29730,
      "train_flow_matching_loss": 0.138671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 29730,
      "train_flow_matching_loss": 0.0986328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 29730,
      "train_flow_matching_loss": 0.1845703125,
      "train_physics_loss": 0.0
    },
    {
      "step": 29730,
      "train_flow_matching_loss": 0.09375,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.1155180111527443,
      "learning_rate": 5e-05,
      "loss": 0.1212,
      "step": 29740
    },
    {
      "step": 29740,
      "train_flow_matching_loss": 0.10986328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 29740,
      "train_flow_matching_loss": 0.03955078125,
      "train_physics_loss": 0.10786955803632736
    },
    {
      "step": 29740,
      "train_flow_matching_loss": 0.10205078125,
      "train_physics_loss": 0.0
    },
    {
      "step": 29740,
      "train_flow_matching_loss": 0.0947265625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.11996619403362274,
      "learning_rate": 5e-05,
      "loss": 0.1162,
      "step": 29750
    },
    {
      "step": 29750,
      "train_flow_matching_loss": 0.08837890625,
      "train_physics_loss": 0.0
    },
    {
      "step": 29750,
      "train_flow_matching_loss": 0.08642578125,
      "train_physics_loss": 0.10439769923686981
    },
    {
      "step": 29750,
      "train_flow_matching_loss": 0.1904296875,
      "train_physics_loss": 0.0
    },
    {
      "step": 29750,
      "train_flow_matching_loss": 0.1015625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.08002351969480515,
      "learning_rate": 5e-05,
      "loss": 0.1205,
      "step": 29760
    },
    {
      "step": 29760,
      "train_flow_matching_loss": 0.08837890625,
      "train_physics_loss": 0.0
    },
    {
      "step": 29760,
      "train_flow_matching_loss": 0.0322265625,
      "train_physics_loss": 0.09979456663131714
    },
    {
      "step": 29760,
      "train_flow_matching_loss": 0.1640625,
      "train_physics_loss": 0.0
    },
    {
      "step": 29760,
      "train_flow_matching_loss": 0.07763671875,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.07221616804599762,
      "learning_rate": 5e-05,
      "loss": 0.1157,
      "step": 29770
    },
    {
      "step": 29770,
      "train_flow_matching_loss": 0.11865234375,
      "train_physics_loss": 0.0
    },
    {
      "step": 29770,
      "train_flow_matching_loss": 0.052734375,
      "train_physics_loss": 0.10905647277832031
    },
    {
      "step": 29770,
      "train_flow_matching_loss": 0.1044921875,
      "train_physics_loss": 0.0
    },
    {
      "step": 29770,
      "train_flow_matching_loss": 0.1142578125,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.09812753647565842,
      "learning_rate": 5e-05,
      "loss": 0.1144,
      "step": 29780
    },
    {
      "step": 29780,
      "train_flow_matching_loss": 0.0849609375,
      "train_physics_loss": 0.0
    },
    {
      "step": 29780,
      "train_flow_matching_loss": 0.06591796875,
      "train_physics_loss": 0.10028912872076035
    },
    {
      "step": 29780,
      "train_flow_matching_loss": 0.1875,
      "train_physics_loss": 0.0
    },
    {
      "step": 29780,
      "train_flow_matching_loss": 0.12353515625,
      "train_physics_loss": 0.0
    },
    {
      "grad_norm": 0.07858746498823166,
      "learning_rate": 5e-05,
      "loss": 0.1225,
      "step": 29790
    },
    {
      "step": 29790,
      "train_flow_matching_loss": 0.140625,
      "train_physics_loss": 0.0
    },
    {
      "step": 29790,
      "train_flow_matching_loss": 0.049072265625,
      "train_physics_loss": 0.10078731179237366
    },
    {
      "step": 29790,
      "train_flow_matching_loss": 0.1455078125,
      "train_physics_loss": 0.0
    },
    {
      "step": 29790,
      "train_flow_matching_loss": 0.04296875,
      "train_physics_loss": 0.12932553887367249
    },
    {
      "grad_norm": 0.08086048811674118,
      "learning_rate": 5e-05,
      "loss": 0.1212,
      "step": 29800
    },
    {
      "step": 29800,
      "train_flow_matching_loss": 0.1123046875,
      "train_physics_loss": 0.0
    },
    {
      "step": 29800,
      "train_flow_matching_loss": 0.0751953125,
      "train_physics_loss": 0.08550303429365158
    },
    {
      "step": 29800,
      "train_flow_matching_loss": 0.11962890625,
      "train_physics_loss": 0.0
    },
    {
      "step": 29800,
      "train_flow_matching_loss": 0.0830078125,
      "train_physics_loss": 0.2872672975063324
    },
    {
      "grad_norm": 0.09026404470205307,
      "learning_rate": 5e-05,
      "loss": 0.1211,
      "step": 29810
    },
    {
      "step": 29810,
      "train_flow_matching_loss": 0.07861328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 29810,
      "train_flow_matching_loss": 0.08203125,
      "train_physics_loss": 0.30364784598350525
    },
    {
      "step": 29810,
      "train_flow_matching_loss": 0.234375,
      "train_physics_loss": 0.0
    },
    {
      "step": 29810,
      "train_flow_matching_loss": 0.06982421875,
      "train_physics_loss": 0.13097690045833588
    },
    {
      "grad_norm": 0.08469074219465256,
      "learning_rate": 5e-05,
      "loss": 0.122,
      "step": 29820
    },
    {
      "step": 29820,
      "train_flow_matching_loss": 0.0693359375,
      "train_physics_loss": 0.0
    },
    {
      "step": 29820,
      "train_flow_matching_loss": 0.0230712890625,
      "train_physics_loss": 0.08674494177103043
    },
    {
      "step": 29820,
      "train_flow_matching_loss": 0.1123046875,
      "train_physics_loss": 0.0
    },
    {
      "step": 29820,
      "train_flow_matching_loss": 0.05029296875,
      "train_physics_loss": 0.0825350359082222
    },
    {
      "grad_norm": 0.10610980540513992,
      "learning_rate": 5e-05,
      "loss": 0.1162,
      "step": 29830
    },
    {
      "step": 29830,
      "train_flow_matching_loss": 0.1484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 29830,
      "train_flow_matching_loss": 0.059326171875,
      "train_physics_loss": 0.10864072293043137
    },
    {
      "step": 29830,
      "train_flow_matching_loss": 0.0986328125,
      "train_physics_loss": 0.0
    },
    {
      "step": 29830,
      "train_flow_matching_loss": 0.043701171875,
      "train_physics_loss": 0.14092129468917847
    },
    {
      "grad_norm": 0.10198995471000671,
      "learning_rate": 5e-05,
      "loss": 0.121,
      "step": 29840
    },
    {
      "step": 29840,
      "train_flow_matching_loss": 0.146484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 29840,
      "train_flow_matching_loss": 0.05615234375,
      "train_physics_loss": 0.08511469513177872
    },
    {
      "step": 29840,
      "train_flow_matching_loss": 0.1533203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 29840,
      "train_flow_matching_loss": 0.055419921875,
      "train_physics_loss": 0.1252976357936859
    },
    {
      "grad_norm": 0.13017798960208893,
      "learning_rate": 5e-05,
      "loss": 0.1247,
      "step": 29850
    },
    {
      "step": 29850,
      "train_flow_matching_loss": 0.0732421875,
      "train_physics_loss": 0.0
    },
    {
      "step": 29850,
      "train_flow_matching_loss": 0.078125,
      "train_physics_loss": 0.09809834510087967
    },
    {
      "step": 29850,
      "train_flow_matching_loss": 0.1875,
      "train_physics_loss": 0.0
    },
    {
      "step": 29850,
      "train_flow_matching_loss": 0.09765625,
      "train_physics_loss": 0.18379971385002136
    },
    {
      "grad_norm": 0.0835212990641594,
      "learning_rate": 5e-05,
      "loss": 0.123,
      "step": 29860
    },
    {
      "step": 29860,
      "train_flow_matching_loss": 0.1044921875,
      "train_physics_loss": 0.0
    },
    {
      "step": 29860,
      "train_flow_matching_loss": 0.053466796875,
      "train_physics_loss": 0.21351690590381622
    },
    {
      "step": 29860,
      "train_flow_matching_loss": 0.166015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 29860,
      "train_flow_matching_loss": 0.08642578125,
      "train_physics_loss": 0.19432125985622406
    },
    {
      "grad_norm": 0.09164702147245407,
      "learning_rate": 5e-05,
      "loss": 0.1244,
      "step": 29870
    },
    {
      "step": 29870,
      "train_flow_matching_loss": 0.1103515625,
      "train_physics_loss": 0.0
    },
    {
      "step": 29870,
      "train_flow_matching_loss": 0.060791015625,
      "train_physics_loss": 0.08589979261159897
    },
    {
      "step": 29870,
      "train_flow_matching_loss": 0.1728515625,
      "train_physics_loss": 0.0
    },
    {
      "step": 29870,
      "train_flow_matching_loss": 0.08837890625,
      "train_physics_loss": 0.1710299253463745
    },
    {
      "grad_norm": 0.07800984382629395,
      "learning_rate": 5e-05,
      "loss": 0.1189,
      "step": 29880
    },
    {
      "step": 29880,
      "train_flow_matching_loss": 0.197265625,
      "train_physics_loss": 0.0
    },
    {
      "step": 29880,
      "train_flow_matching_loss": 0.028564453125,
      "train_physics_loss": 0.08989794552326202
    },
    {
      "step": 29880,
      "train_flow_matching_loss": 0.1376953125,
      "train_physics_loss": 0.0
    },
    {
      "step": 29880,
      "train_flow_matching_loss": 0.061279296875,
      "train_physics_loss": 0.2126852124929428
    },
    {
      "grad_norm": 0.07306242734193802,
      "learning_rate": 5e-05,
      "loss": 0.1186,
      "step": 29890
    },
    {
      "step": 29890,
      "train_flow_matching_loss": 0.1484375,
      "train_physics_loss": 0.0
    },
    {
      "step": 29890,
      "train_flow_matching_loss": 0.023193359375,
      "train_physics_loss": 0.084204763174057
    },
    {
      "step": 29890,
      "train_flow_matching_loss": 0.1337890625,
      "train_physics_loss": 0.0
    },
    {
      "step": 29890,
      "train_flow_matching_loss": 0.04541015625,
      "train_physics_loss": 0.14958684146404266
    },
    {
      "grad_norm": 0.08426348865032196,
      "learning_rate": 5e-05,
      "loss": 0.1188,
      "step": 29900
    },
    {
      "step": 29900,
      "train_flow_matching_loss": 0.1826171875,
      "train_physics_loss": 0.0
    },
    {
      "step": 29900,
      "train_flow_matching_loss": 0.0751953125,
      "train_physics_loss": 0.10321638733148575
    },
    {
      "step": 29900,
      "train_flow_matching_loss": 0.2158203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 29900,
      "train_flow_matching_loss": 0.061279296875,
      "train_physics_loss": 0.13334640860557556
    },
    {
      "grad_norm": 0.08961544185876846,
      "learning_rate": 5e-05,
      "loss": 0.1203,
      "step": 29910
    },
    {
      "step": 29910,
      "train_flow_matching_loss": 0.0849609375,
      "train_physics_loss": 0.0
    },
    {
      "step": 29910,
      "train_flow_matching_loss": 0.06298828125,
      "train_physics_loss": 0.0
    },
    {
      "step": 29910,
      "train_flow_matching_loss": 0.1298828125,
      "train_physics_loss": 0.0
    },
    {
      "step": 29910,
      "train_flow_matching_loss": 0.05810546875,
      "train_physics_loss": 0.1410350501537323
    },
    {
      "grad_norm": 0.12085505574941635,
      "learning_rate": 5e-05,
      "loss": 0.1158,
      "step": 29920
    },
    {
      "step": 29920,
      "train_flow_matching_loss": 0.1015625,
      "train_physics_loss": 0.0
    },
    {
      "step": 29920,
      "train_flow_matching_loss": 0.06396484375,
      "train_physics_loss": 0.12655223906040192
    },
    {
      "step": 29920,
      "train_flow_matching_loss": 0.1455078125,
      "train_physics_loss": 0.0
    },
    {
      "step": 29920,
      "train_flow_matching_loss": 0.07861328125,
      "train_physics_loss": 0.12163995206356049
    },
    {
      "grad_norm": 0.114077627658844,
      "learning_rate": 5e-05,
      "loss": 0.1179,
      "step": 29930
    },
    {
      "step": 29930,
      "train_flow_matching_loss": 0.0859375,
      "train_physics_loss": 0.0
    },
    {
      "step": 29930,
      "train_flow_matching_loss": 0.021240234375,
      "train_physics_loss": 0.10129138082265854
    },
    {
      "step": 29930,
      "train_flow_matching_loss": 0.1103515625,
      "train_physics_loss": 0.0
    },
    {
      "step": 29930,
      "train_flow_matching_loss": 0.06103515625,
      "train_physics_loss": 0.14073726534843445
    },
    {
      "grad_norm": 0.08127974718809128,
      "learning_rate": 5e-05,
      "loss": 0.1215,
      "step": 29940
    },
    {
      "step": 29940,
      "train_flow_matching_loss": 0.228515625,
      "train_physics_loss": 0.0
    },
    {
      "step": 29940,
      "train_flow_matching_loss": 0.07861328125,
      "train_physics_loss": 0.07329577207565308
    },
    {
      "step": 29940,
      "train_flow_matching_loss": 0.259765625,
      "train_physics_loss": 0.0
    },
    {
      "step": 29940,
      "train_flow_matching_loss": 0.03369140625,
      "train_physics_loss": 0.12724407017230988
    },
    {
      "grad_norm": 0.08807950466871262,
      "learning_rate": 5e-05,
      "loss": 0.1207,
      "step": 29950
    },
    {
      "step": 29950,
      "train_flow_matching_loss": 0.1650390625,
      "train_physics_loss": 0.0
    },
    {
      "step": 29950,
      "train_flow_matching_loss": 0.05712890625,
      "train_physics_loss": 0.11399371176958084
    },
    {
      "step": 29950,
      "train_flow_matching_loss": 0.138671875,
      "train_physics_loss": 0.0
    },
    {
      "step": 29950,
      "train_flow_matching_loss": 0.055419921875,
      "train_physics_loss": 0.3062489628791809
    },
    {
      "grad_norm": 0.09854834526777267,
      "learning_rate": 5e-05,
      "loss": 0.1226,
      "step": 29960
    },
    {
      "step": 29960,
      "train_flow_matching_loss": 0.1923828125,
      "train_physics_loss": 0.0
    },
    {
      "step": 29960,
      "train_flow_matching_loss": 0.032470703125,
      "train_physics_loss": 0.09915134310722351
    },
    {
      "step": 29960,
      "train_flow_matching_loss": 0.150390625,
      "train_physics_loss": 0.0
    },
    {
      "step": 29960,
      "train_flow_matching_loss": 0.07275390625,
      "train_physics_loss": 0.11774756759405136
    },
    {
      "grad_norm": 0.10756074637174606,
      "learning_rate": 5e-05,
      "loss": 0.1207,
      "step": 29970
    },
    {
      "step": 29970,
      "train_flow_matching_loss": 0.1650390625,
      "train_physics_loss": 0.0
    },
    {
      "step": 29970,
      "train_flow_matching_loss": 0.09912109375,
      "train_physics_loss": 0.11697299033403397
    },
    {
      "step": 29970,
      "train_flow_matching_loss": 0.1689453125,
      "train_physics_loss": 0.0
    },
    {
      "step": 29970,
      "train_flow_matching_loss": 0.03173828125,
      "train_physics_loss": 0.0955752432346344
    },
    {
      "grad_norm": 0.10593639314174652,
      "learning_rate": 5e-05,
      "loss": 0.121,
      "step": 29980
    },
    {
      "step": 29980,
      "train_flow_matching_loss": 0.1533203125,
      "train_physics_loss": 0.0
    },
    {
      "step": 29980,
      "train_flow_matching_loss": 0.0303955078125,
      "train_physics_loss": 0.09318944811820984
    },
    {
      "step": 29980,
      "train_flow_matching_loss": 0.189453125,
      "train_physics_loss": 0.0
    },
    {
      "step": 29980,
      "train_flow_matching_loss": 0.0732421875,
      "train_physics_loss": 0.14039652049541473
    },
    {
      "grad_norm": 0.06976176798343658,
      "learning_rate": 5e-05,
      "loss": 0.1197,
      "step": 29990
    },
    {
      "step": 29990,
      "train_flow_matching_loss": 0.09912109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 29990,
      "train_flow_matching_loss": 0.2197265625,
      "train_physics_loss": 0.0
    },
    {
      "step": 29990,
      "train_flow_matching_loss": 0.12109375,
      "train_physics_loss": 0.0
    },
    {
      "step": 29990,
      "train_flow_matching_loss": 0.045166015625,
      "train_physics_loss": 0.19628430902957916
    },
    {
      "grad_norm": 0.08687153458595276,
      "learning_rate": 5e-05,
      "loss": 0.124,
      "step": 30000
    }
  ],
  "logging_steps": 10,
  "max_steps": 30000,
  "num_input_tokens_seen": 0,
  "num_train_epochs": 9223372036854775807,
  "save_steps": 1000,
  "stateful_callbacks": {
    "TrainerControl": {
      "args": {
        "should_epoch_stop": false,
        "should_evaluate": false,
        "should_log": false,
        "should_save": true,
        "should_training_stop": true
      },
      "attributes": {}
    }
  },
  "total_flos": 0.0,
  "train_batch_size": 8,
  "trial_name": null,
  "trial_params": null
}