-record(run_batch, {
    batch_id :: integer(),
    batch :: viva_tensor@data@dataloader:batch(),
    params :: list(viva_tensor@nn@optim:param())
}).