nemo_automodel.components.speculative.eagle

EAGLE-3 training components.

Only the EAGLE-3-native model and trainer classes live here. Generic helpers were moved to the project’s canonical locations:

• masked_soft_cross_entropy -> nemo_automodel.components.loss.soft_ce
• build_eagle3_dataloader / build_eagle3_token_mapping -> nemo_automodel.components.datasets.llm.eagle3

Import them directly from those modules.

Subpackages

• nemo_automodel.components.speculative.eagle.remote

Submodules

• nemo_automodel.components.speculative.eagle.backend
• nemo_automodel.components.speculative.eagle.core
• nemo_automodel.components.speculative.eagle.core_v12
• nemo_automodel.components.speculative.eagle.draft_gpt_oss
• nemo_automodel.components.speculative.eagle.draft_llama
• nemo_automodel.components.speculative.eagle.draft_llama_v12
• nemo_automodel.components.speculative.eagle.peagle_attention
• nemo_automodel.components.speculative.eagle.peagle_data
• nemo_automodel.components.speculative.eagle.peagle_draft
• nemo_automodel.components.speculative.eagle.peagle_trainer
• nemo_automodel.components.speculative.eagle.registry
• nemo_automodel.components.speculative.eagle.target
• nemo_automodel.components.speculative.eagle.target_v12

Package Contents

Data

__all__

API

nemo_automodel.components.speculative.eagle.__all__ = ['EagleTrainerModule', 'Eagle3TrainerModule', 'PEagleTrainerModule', 'Eagle3Targ...