nemo_automodel.components.speculative.dspark

DSpark speculative-decoding draft model and training objective.

A semi-autoregressive parallel drafter: a parallel backbone produces every position of a block in one pass, a lightweight serial Markov head injects intra-block token dependency, and a confidence head predicts per-position acceptance probability for scheduled verification.

Submodules

• nemo_automodel.components.speculative.dspark._sampling
• nemo_automodel.components.speculative.dspark.common
• nemo_automodel.components.speculative.dspark.config
• nemo_automodel.components.speculative.dspark.core
• nemo_automodel.components.speculative.dspark.draft_gemma4
• nemo_automodel.components.speculative.dspark.draft_qwen3
• nemo_automodel.components.speculative.dspark.loss
• nemo_automodel.components.speculative.dspark.markov_head
• nemo_automodel.components.speculative.dspark.registry
• nemo_automodel.components.speculative.dspark.target

Package Contents

Data

__all__

API

nemo_automodel.components.speculative.dspark.__all__ = ['DSparkForwardOutput', 'Qwen3DSparkModel', 'build_draft_config', 'compute_dspar...