> 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.

# API Reference

> Auto-generated Python API reference for the nemo_automodel library.

Auto-generated reference documentation for the `nemo_automodel` Python package.

This reference is built from docstrings in the [source code](https://github.com/NVIDIA-NeMo/Automodel/tree/main/nemo_automodel). Browse the sidebar to explore modules, classes, and functions.

## Key Modules

| Module                                   | Description                                                                                                                                      |
| ---------------------------------------- | ------------------------------------------------------------------------------------------------------------------------------------------------ |
| `nemo_automodel._transformers`           | HuggingFace bridge — `NeMoAutoModelForCausalLM`, `NeMoAutoModelForImageTextToText`, model registry, capability flags, device-mesh infrastructure |
| `nemo_automodel._diffusers`              | Diffusion pipeline wrapper (`NeMoAutoDiffusionPipeline`)                                                                                         |
| `nemo_automodel.components.models`       | 27+ model families (LLM, VLM, MoE, retrieval) and their state-dict adapters                                                                      |
| `nemo_automodel.components.datasets`     | LLM, VLM, and diffusion data pipelines                                                                                                           |
| `nemo_automodel.components.distributed`  | FSDP2, HSDP, DDP utilities and device-mesh helpers                                                                                               |
| `nemo_automodel.components.checkpoint`   | Async DCP and SafeTensors checkpointing                                                                                                          |
| `nemo_automodel.components.quantization` | FP8, QAT, calibration                                                                                                                            |
| `nemo_automodel.components._peft`        | LoRA, QLoRA adapters                                                                                                                             |
| `nemo_automodel.components.launcher`     | Slurm and SkyPilot job submission                                                                                                                |
| `nemo_automodel.cli`                     | `automodel <command> <domain> -c <config.yaml>` entry point                                                                                      |