IMPORTANT: To view this page as Markdown, append `.md` to the URL (e.g. /get-started.md). For the complete documentation index, see llms.txt.
Skip to main content
For the complete documentation index, see llms.txt. Markdown versions of all pages are available by appending .md to any URL (e.g. /get-started.md).

Python module

max.pipelines.speculative

Speculative decoding pipelines and configuration for MAX.

Submodulesโ€‹

Configurationโ€‹

RejectionSamplingStrategyalias of Literal['greedy', 'residual', 'typical-acceptance', 'logit-comparison']
SpeculativeConfigConfigures speculative decoding for a pipeline.
SpeculativeMethodalias of Literal['eagle', 'mtp', 'dflash', 'dflash2']
VerifyWidthRangeOne inclusive decode-batch-size range and the drafts to verify in it.

Token mergingโ€‹

RaggedTokenMergerMerges prompt and draft token sequences into a single ragged batch.
ragged_token_mergerBuilds a graph that merges prompt and draft tokens into a single ragged sequence.