-record(checkpoint, {
    input :: viva_tensor@tensor:tensor(),
    forward_fn_id :: integer(),
    memory_saved_gb :: float()
}).