Skip to main content

Ctrl+K

NeMo-AutoModel

GitHub

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)

Blogs

NeMo Framework Now Supports Google Gemma 3n: Efficient Multimodal Fine-Tuning Made Simple

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.shared
    
    nemo_automodel.shared.utils
    
    nemo_automodel.shared.import_utils
  - nemo_automodel.components
    
    nemo_automodel.components._transformers
    
    nemo_automodel.components.training
    
    nemo_automodel.components.launcher
    
    nemo_automodel.components.quantization
    
    nemo_automodel.components.utils
    
    nemo_automodel.components._peft
    
    nemo_automodel.components.datasets
    
    nemo_automodel.components.optim
    
    nemo_automodel.components.distributed
    
    nemo_automodel.components.config
    
    nemo_automodel.components.loss
    
    nemo_automodel.components.loggers
    
    nemo_automodel.components.checkpoint

API Reference
nemo_automodel
nemo_automodel.components
nemo_automodel.components.loggers

`nemo_automodel.components.loggers`#

Submodules#

nemo_automodel.components.loggers.wandb_utils
nemo_automodel.components.loggers.log_utils

previous

nemo_automodel.components.loss.masked_ce

next

nemo_automodel.components.loggers.wandb_utils

On this page

Submodules

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.