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.

Submodules

• nemo_automodel.components.speculative.eagle.draft_llama
• nemo_automodel.components.speculative.eagle.draft_llama_v12
• nemo_automodel.components.speculative.eagle.target_v12
• nemo_automodel.components.speculative.eagle.core
• nemo_automodel.components.speculative.eagle.target
• nemo_automodel.components.speculative.eagle.core_v12

Package Contents

Data

__all__

API

nemo_automodel.components.speculative.eagle.__all__

[‘EagleTrainerModule’, ‘Eagle3TrainerModule’, ‘HFEagleTargetModel’, ‘HFEagle3TargetModel’, ‘LlamaEag…