> This page is for version Nightly (default).
> For other versions, use one of these documentation indexes:
> - Nightly (default): https://docs.nvidia.com/nemo/automodel/nightly/llms.txt
> - Latest: https://docs.nvidia.com/nemo/automodel/latest/llms.txt
> - 0.5.0 · 26.06: https://docs.nvidia.com/nemo/automodel/v0.5/llms.txt
> - 0.4.0 · 26.04: https://docs.nvidia.com/nemo/automodel/v0.4/llms.txt

> For clean Markdown of any page, append .md to the page URL.
> For a complete documentation index, see https://docs.nvidia.com/nemo/automodel/llms.txt.
> For AI client integration (Claude Code, Cursor, etc.), connect to the MCP server at https://docs.nvidia.com/nemo/automodel/_mcp/server.

# nemo_automodel.recipes.llm.train_eagle2

Minimal Llama-only EAGLE-2 training recipe.

EAGLE-2 reuses the same draft-model training objective as EAGLE-1. The
difference is in the speculative decoding tree policy at inference time.
This recipe exists as an explicit entrypoint so users can train a draft
checkpoint intended for an EAGLE-2 deployment workflow without having to
remember that the underlying optimization is identical to EAGLE-1.

## Module Contents

### Classes

| Name                                                                              | Description                              |
| --------------------------------------------------------------------------------- | ---------------------------------------- |
| [`TrainEagle2Recipe`](#nemo_automodel-recipes-llm-train_eagle2-TrainEagle2Recipe) | Recipe alias for EAGLE-2 draft training. |

### Functions

| Name                                                    | Description                         |
| ------------------------------------------------------- | ----------------------------------- |
| [`main`](#nemo_automodel-recipes-llm-train_eagle2-main) | Entrypoint for `TrainEagle2Recipe`. |

### API

```python
class nemo_automodel.recipes.llm.train_eagle2.TrainEagle2Recipe()
```

**Bases:** [TrainEagle1Recipe](/nemo-automodel/nemo_automodel/recipes/llm/train_eagle1#nemo_automodel-recipes-llm-train_eagle1-TrainEagle1Recipe)

Recipe alias for EAGLE-2 draft training.

```python
nemo_automodel.recipes.llm.train_eagle2.main(
    config_path: str | None = None
)
```

Entrypoint for `TrainEagle2Recipe`.