Skip to main content

Module causal

Module causal 

Source
Available on crate feature std only.
Expand description

Explicit-label causal-LM examples and packed/padded device collation. Causal data collation independent of architecture and tokenizer selection.

Structs§

CausalBatcher
Explicit supervision/layout policy shared by every call of a data batcher.
CausalDataError
Invalid actual input/label geometry or explicit batch configuration.
CausalDataErrorItem
CausalExample
One already-encoded document with caller-declared supervision.
PackedCausalBatch
Flat actual documents, ready for an explicit packed-attention model.
PaddedCausalBatch
Rectangular actual documents with explicitly masked batch-padding cells.

Enums§

CausalBatch
A device batch carrying its actual geometry, not an inferred model family.
CausalBatchLayout
Device batch layout explicitly selected by the caller’s model contract.
CausalPadding
Direction of rectangular batch padding, not a truncation policy.
CausalTargetAlignment
Alignment used by the downstream causal loss.