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

NeMo-AutoModel

  • GitHub
NeMo-AutoModel - Home NeMo-AutoModel - Home

NeMo-AutoModel
  • GitHub

Table of Contents

Get Started

  • Introduction to the NeMo Automodel Repository
  • Install NeMo Automodel
  • Run on Your Local Workstation

Guides

  • Fine-Tune Gemma 3 and Gemma 3n

E2E Examples

  • Supervised Fine-Tuning (SFT) with NeMo Automodel
  • Parameter-Efficient Fine-Tuning (PEFT) with NeMo Automodel

Model Coverage

  • Large Language Models (LLMs)
  • Vision Language Models (VLMs)

Development

  • Checkpointing in NeMo AutoModel
  • Integrate Your Own Text Dataset
  • Integrate Your Own Multi-Modal Dataset
  • Use the ColumnMappedTextInstructionDataset
  • API Reference
    • nemo_automodel
      • nemo_automodel.components
        • nemo_automodel.components.distributed
        • nemo_automodel.components.checkpoint
        • nemo_automodel.components.config
        • nemo_automodel.components.datasets
        • nemo_automodel.components.quantization
        • nemo_automodel.components.launcher
        • nemo_automodel.components.optim
        • nemo_automodel.components.loss
        • nemo_automodel.components.loggers
        • nemo_automodel.components._transformers
        • nemo_automodel.components.utils
        • nemo_automodel.components._peft
        • nemo_automodel.components.training
      • nemo_automodel.shared
        • nemo_automodel.shared.import_utils
        • nemo_automodel.shared.utils
  • API Reference
  • nemo_automodel
  • nemo_automodel.components
  • nemo_automodel.components._transformers

nemo_automodel.components._transformers#

Submodules#

  • nemo_automodel.components._transformers.auto_model
  • nemo_automodel.components._transformers.utils

Package Contents#

Data#

__all__

API#

nemo_automodel.components._transformers.__all__#

[‘NeMoAutoModelForCausalLM’, ‘NeMoAutoModelForImageTextToText’, ‘NeMoAutoModelForSequenceClassificat…

previous

nemo_automodel.components.loggers.wandb_utils

next

nemo_automodel.components._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.