For AI agents: a documentation index is available at the root level at /llms.txt. Append /llms.txt to any URL for a page-level index, or .md for the markdown version of any page.

- Home
Get Started
Announcements
- Announcements
NeMo AutoModel Performance
- Performance Summary
Model Coverage
Recipes & E2E Examples
Datasets
- Dataset Overview: LLM, VLM, and Retrieval Datasets in NeMo AutoModel
- Integrate Your Own Text Dataset
- Retrieval Dataset (Embedding Fine-tuning)
- Use the ColumnMappedTextInstructionDataset
- Use the ColumnMappedTextInstructionIterableDataset (Streaming)
- Integrate Your Own Multi-Modal Dataset
- Diffusion Dataset Preparation
Job Launchers
Development
- Checkpointing in NeMo Automodel
- Gradient (Activation) Checkpointing in NeMo AutoModel
- Pipeline Parallelism with AutoPipeline
- FP8 Training in NeMo AutoModel
- MLflow Logging in NeMo AutoModel
- Breaking Changes
- - Overview
  - Nemo Automodel
    Autonvtx
    Cli
    Components
    Attention
    Checkpoint
    Config
    Datasets
    Distributed
    Eval
    Flow Matching
    Launcher
    Loggers
    Loss
    Models
    Moe
    Optim
    Quantization
    Speculative
    Training
    Utils
    Package Info
    Shared

On this page

Subpackages
Submodules

Development API Reference Full Library Reference Nemo Automodel Nemo Automodel Components

nemo_automodel.components.moe

||View as Markdown|

Subpackages

nemo_automodel.components.moe.megatron
nemo_automodel.components.moe.uccl_ep

Submodules

nemo_automodel.components.moe.config
nemo_automodel.components.moe.experts
nemo_automodel.components.moe.fsdp_mixin
nemo_automodel.components.moe.layers
nemo_automodel.components.moe.load_balance_metrics
nemo_automodel.components.moe.mxfp8
nemo_automodel.components.moe.parallelizer
nemo_automodel.components.moe.state_dict_mixin
nemo_automodel.components.moe.state_dict_utils

nemo_automodel.components.models.step3p7.vision_encoder

nemo_automodel.components.moe.config

Developer-friendly docs for your API

Privacy Policy | Your Privacy Choices | Terms of Service | Accessibility | Corporate Policies | Product Security | Contact

Copyright © 2026, NVIDIA Corporation.