{
    "lm loss": [
        12.13663,
        12.12253,
        12.11792,
        12.06363,
        12.03781,
        11.8663,
        11.81957,
        11.82628,
        11.75255,
        11.42776,
        11.31091,
        11.2725,
        11.28029,
        11.26136,
        11.31776
    ],
    "time info": [
        5766.0,
        387.2,
        367.0,
        368.9,
        365.6,
        365.1,
        369.6,
        372.5,
        368.2,
        364.4,
        368.6,
        367.0,
        365.9,
        365.1,
        364.1
    ],
    "memo info": [
        {
            "rank": 0,
            "allocated memory": 4733.1123046875,
            "max allocated memory": 5291.009765625
        },
        {
            "rank": 1,
            "allocated memory": 4733.1123046875,
            "max allocated memory": 5291.009765625
        },
        {
            "rank": 4,
            "allocated memory": 4740.1181640625,
            "max allocated memory": 5526.6494140625
        },
        {
            "rank": 5,
            "allocated memory": 4740.1181640625,
            "max allocated memory": 5526.6494140625
        }
    ]
}