{
    "lm loss": [
        11.50431,
        11.56956,
        11.55847,
        11.4692,
        11.44926,
        11.27195,
        11.28963,
        11.22943,
        11.21624,
        11.05591,
        10.90876,
        10.91855,
        10.85786,
        10.8296,
        10.95429
    ],
    "grad norm": [
        31.663,
        29.068,
        28.144,
        27.955,
        27.733,
        30.824,
        29.434,
        32.166,
        28.058,
        26.7,
        29.918,
        29.413,
        29.132,
        28.203,
        25.576
    ],
    "memo info": [
        {
            "rank": 0,
            "allocated memory": 10474.21875,
            "max allocated memory": 20688.57177734375
        }
    ]
}