Skip to main content
Ctrl+K
NeMo-AutoModel - Home NeMo-AutoModel - Home

NeMo-AutoModel

NeMo-AutoModel - Home NeMo-AutoModel - Home

NeMo-AutoModel

Table of Contents

🖥️ Environment Start

  • Run on Your Local Workstation

🚀 E2E Examples

  • Full Model Fine-Tuning on HellaSwag

📚 Guides

  • Bringing Your Own Dataset

🛠️ Development

  • Documentation Development
  • API Reference
    • nemo_automodel
      • nemo_automodel._transformers
        • nemo_automodel._transformers.auto_model
      • nemo_automodel.shared
        • nemo_automodel.shared.import_utils
      • nemo_automodel.training
        • nemo_automodel.training.utils
        • nemo_automodel.training.rng
        • nemo_automodel.training.base_recipe
        • nemo_automodel.training.step_scheduler
        • nemo_automodel.training.timers
      • nemo_automodel.utils
        • nemo_automodel.utils.sig_utils
        • nemo_automodel.utils.model_utils
        • nemo_automodel.utils.dist_utils
        • nemo_automodel.utils.yaml_utils
      • nemo_automodel.datasets
        • nemo_automodel.datasets.llm
        • nemo_automodel.datasets.vlm
        • nemo_automodel.datasets.utils
      • nemo_automodel.optim
        • nemo_automodel.optim.scheduler
      • nemo_automodel.distributed
        • nemo_automodel.distributed.fsdp2
        • nemo_automodel.distributed.parallelizer
        • nemo_automodel.distributed.ddp
        • nemo_automodel.distributed.cp_utils
        • nemo_automodel.distributed.nvfsdp
        • nemo_automodel.distributed.init_utils
      • nemo_automodel.config
        • nemo_automodel.config.loader
        • nemo_automodel.config.cli
      • nemo_automodel.loss
        • nemo_automodel.loss.linear_ce
        • nemo_automodel.loss.chunked_ce
        • nemo_automodel.loss.masked_ce
      • nemo_automodel.loggers
        • nemo_automodel.loggers.wandb_utils
        • nemo_automodel.loggers.log_utils
      • nemo_automodel.checkpoint
        • nemo_automodel.checkpoint._backports
        • nemo_automodel.checkpoint.stateful_wrappers
        • nemo_automodel.checkpoint.checkpointing
        • nemo_automodel.checkpoint._torch_backports
  • API Reference
  • nemo_automodel
  • nemo_automodel._transformers

nemo_automodel._transformers#

Submodules#

  • nemo_automodel._transformers.auto_model

Package Contents#

Data#

__all__

API#

nemo_automodel._transformers.__all__#

[‘NeMoAutoModelForCausalLM’, ‘NeMoAutoModelForImageTextToText’]

previous

nemo_automodel

next

nemo_automodel._transformers.auto_model
On this page
  • Submodules
  • Package Contents
    • Data
    • API
      • __all__
NVIDIA NVIDIA
Privacy Policy | Manage My Privacy | Do Not Sell or Share My Data | Terms of Service | Accessibility | Corporate Policies | Product Security | Contact

Copyright © 2025, NVIDIA Corporation.