Container for the outcome of processing a single case.

Collate examples for supervised fine-tuning.

Collate instances for supervised dataset. Each instance is a tokenized dictionary with fields `input_ids`(List[int]), `labels`(List[int])

Stateful distributed sampler for multi-stage training.