API Reference | NVIDIA NeMo AutoModel

Auto-generated reference documentation for the nemo_automodel Python package.

This reference is built from docstrings in the source code. 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