> 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.components.models.laguna.state_dict_adapter

## Module Contents

### Classes

| Name                                                                                                           | Description                                                      |
| -------------------------------------------------------------------------------------------------------------- | ---------------------------------------------------------------- |
| [`LagunaStateDictAdapter`](#nemo_automodel-components-models-laguna-state_dict_adapter-LagunaStateDictAdapter) | Convert Laguna HF checkpoints to Automodel's grouped-MoE layout. |

### Functions

| Name                                                                                           | Description |
| ---------------------------------------------------------------------------------------------- | ----------- |
| [`_apply_renames`](#nemo_automodel-components-models-laguna-state_dict_adapter-_apply_renames) | -           |

### Data

[`_HF_TO_NATIVE_RENAMES`](#nemo_automodel-components-models-laguna-state_dict_adapter-_HF_TO_NATIVE_RENAMES)

[`_NATIVE_TO_HF_RENAMES`](#nemo_automodel-components-models-laguna-state_dict_adapter-_NATIVE_TO_HF_RENAMES)

[`__all__`](#nemo_automodel-components-models-laguna-state_dict_adapter-__all__)

### API

```python
class nemo_automodel.components.models.laguna.state_dict_adapter.LagunaStateDictAdapter(
    config: typing.Any,
    moe_config: nemo_automodel.components.moe.config.MoEConfig,
    backend: nemo_automodel.components.models.common.BackendConfig,
    dtype: torch.dtype = torch.bfloat16
)
```

**Bases:** [MoESplitExpertsStateDictMixin](/nemo-automodel/nemo_automodel/components/moe/state_dict_mixin#nemo_automodel-components-moe-state_dict_mixin-MoESplitExpertsStateDictMixin), [StateDictAdapter](/nemo-automodel/nemo_automodel/components/checkpoint/state_dict_adapter#nemo_automodel-components-checkpoint-state_dict_adapter-StateDictAdapter)

Convert Laguna HF checkpoints to Automodel's grouped-MoE layout.

```python
nemo_automodel.components.models.laguna.state_dict_adapter.LagunaStateDictAdapter.convert_single_tensor_to_hf(
    fqn: str,
    tensor: typing.Any,
    kwargs = {}
) -> list[tuple[str, typing.Any]]
```

```python
nemo_automodel.components.models.laguna.state_dict_adapter.LagunaStateDictAdapter.from_hf(
    hf_state_dict: dict[str, typing.Any],
    device_mesh: torch.distributed.device_mesh.DeviceMesh | None = None,
    kwargs = {}
) -> dict[str, typing.Any]
```

```python
nemo_automodel.components.models.laguna.state_dict_adapter.LagunaStateDictAdapter.to_hf(
    state_dict: dict[str, typing.Any],
    exclude_key_regex: str | None = None,
    kwargs = {}
) -> dict[str, typing.Any]
```

```python
nemo_automodel.components.models.laguna.state_dict_adapter._apply_renames(
    key: str,
    renames: tuple[tuple[re.Pattern[str], str], ...]
) -> str
```

```python
nemo_automodel.components.models.laguna.state_dict_adapter._HF_TO_NATIVE_RENAMES: tuple[tuple[Pattern[str], str], ...] = ((re.compile('\\.mlp\\.shared_expert\\.'), '.mlp.shared_experts.'), (re.compile(...
```

```python
nemo_automodel.components.models.laguna.state_dict_adapter._NATIVE_TO_HF_RENAMES: tuple[tuple[Pattern[str], str], ...] = ((re.compile('\\.mlp\\.shared_experts\\.'), '.mlp.shared_expert.'), (re.compile(...
```

```python
nemo_automodel.components.models.laguna.state_dict_adapter.__all__ = ['LagunaStateDictAdapter']
```