> For clean Markdown of any page, append .md to the page URL.
> For a complete documentation index, see https://docs.nvidia.com/nemo/automodel/v0.4/nemo-automodel/nemo_automodel/components/models/diffusion_gemma/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.diffusion_gemma

NeMo Automodel support for the `diffusion_gemma` block-diffusion model.

The config classes are the released `transformers` `DiffusionGemmaConfig` /
`DiffusionGemmaTextConfig` (transformers defines and `AutoConfig`-registers
`model_type="diffusion_gemma"` natively), re-exported here for convenience.
AM owns only the compute (model/layers/adapter).

## Submodules

* **[`nemo_automodel.components.models.diffusion_gemma.attention_mask`](/nemo-automodel/nemo_automodel/components/models/diffusion_gemma/attention_mask)**
* **[`nemo_automodel.components.models.diffusion_gemma.fsdp`](/nemo-automodel/nemo_automodel/components/models/diffusion_gemma/fsdp)**
* **[`nemo_automodel.components.models.diffusion_gemma.layers`](/nemo-automodel/nemo_automodel/components/models/diffusion_gemma/layers)**
* **[`nemo_automodel.components.models.diffusion_gemma.model`](/nemo-automodel/nemo_automodel/components/models/diffusion_gemma/model)**
* **[`nemo_automodel.components.models.diffusion_gemma.state_dict_adapter`](/nemo-automodel/nemo_automodel/components/models/diffusion_gemma/state_dict_adapter)**

## Docs

- [nemo_automodel.components.models.diffusion_gemma.attention_mask](https://docs.nvidia.com/nemo/automodel/v0.4/nemo-automodel/nemo_automodel/components/models/diffusion_gemma/attention_mask.md)
- [nemo_automodel.components.models.diffusion_gemma.fsdp](https://docs.nvidia.com/nemo/automodel/v0.4/nemo-automodel/nemo_automodel/components/models/diffusion_gemma/fsdp.md)
- [nemo_automodel.components.models.diffusion_gemma.layers](https://docs.nvidia.com/nemo/automodel/v0.4/nemo-automodel/nemo_automodel/components/models/diffusion_gemma/layers.md)
- [nemo_automodel.components.models.diffusion_gemma.model](https://docs.nvidia.com/nemo/automodel/v0.4/nemo-automodel/nemo_automodel/components/models/diffusion_gemma/model.md)
- [nemo_automodel.components.models.diffusion_gemma.state_dict_adapter](https://docs.nvidia.com/nemo/automodel/v0.4/nemo-automodel/nemo_automodel/components/models/diffusion_gemma/state_dict_adapter.md)