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

Submodules
Package Contents
Data
API

Development API Reference Full Library Reference Nemo Automodel Nemo Automodel Components

nemo_automodel.components.loss

||View as Markdown|

Submodules

nemo_automodel.components.loss.chunked_ce
nemo_automodel.components.loss.dllm_loss
nemo_automodel.components.loss.kd_loss
nemo_automodel.components.loss.linear_ce
nemo_automodel.components.loss.loss
nemo_automodel.components.loss.masked_ce
nemo_automodel.components.loss.mtp
nemo_automodel.components.loss.soft_ce
nemo_automodel.components.loss.te_parallel_ce
nemo_automodel.components.loss.utils

Package Contents

Data

API

nemo_automodel.components.loss.__all__ = ['LOSS_CONFIG_REGISTRY', 'FusedLinearCEConfig', 'KDLossConfig', 'LossConfig', 'L...

nemo_automodel.components.loggers.wandb_utils

nemo_automodel.components.loss.chunked_ce

Developer-friendly docs for your API

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

Copyright © 2026, NVIDIA Corporation.