Skip to main content

Ctrl+K

NeMo-AutoModel

GitHub

NeMo-AutoModel

GitHub

Table of Contents

Get Started

About
Key Features
Installation
Configuration
🤗 HF Compatibility
Repo Structure
Release Notes

Announcements

Announcements

Performance

Performance Summary

Model Coverage

Overview
Release Log
Large Language Models
- Llama
- Gemma
- Qwen2
- Qwen2 MoE
- Qwen3
- Qwen3 MoE
- Qwen3-Next
- ERNIE 4.5
- DeepSeek
- DeepSeek-V3
- DeepSeek V4 Flash
- Mistral
- Mixtral
- Ministral3 / Devstral
- Phi
- Phi-3 / Phi-4
- Phi-3-Small
- Nemotron / Minitron
- Nemotron-H
- Nemotron-Flash
- Nemotron-Super (Llama-3.3-Nemotron-Super-49B)
- ChatGLM
- GLM-4
- GLM-4 MoE (GLM-4.5 / GLM-4.7)
- GLM-5 / GLM-5.1 (MoE + DSA)
- Granite
- Granite MoE
- Bamba
- OLMo
- OLMo2
- OLMoE
- GPT-OSS
- GPT-2
- GPT-J
- GPT-NeoX / Pythia
- StarCoder
- StarCoder2
- Aquila / Aquila2
- Baichuan / Baichuan2
- Command-R
- Falcon
- EXAONE
- InternLM
- Jais
- MiniMax-M2
- MiniCPM
- Moonlight
- Seed (ByteDance)
- Solar Pro
- Orion
- StableLM
- Step-3.5
- GritLM
- Hy3 (HunyuanLarge)
- Hy-MT2 (Hunyuan-MT2)
- MiMo-V2-Flash
- Ling 2.0
Vision Language Models
Omni
Diffusion
Embedding Models
- Llama (Bidirectional) for Embedding
- Ministral3 (Bidirectional) for Embedding
Reranking Models
- Llama (Bidirectional) for Reranking

Recipes & E2E Examples

Recipes and End-to-End Examples
SFT & PEFT
Function Calling
Knowledge Distillation
Large MoE Fine-Tuning
DeepSeek V4 Flash
Hy3-preview
Pretraining
NanoGPT Pretraining
Sequence Classification
Gemma 3 / 3n
Gemma 4
Qwen3.5-VL
Step-3.7-Flash
Nemotron-Omni
Mistral Medium 3.5 VL
Qwen3-Omni ASR
Diffusion Fine-Tuning
dLLM Fine-Tuning
QAT
Databricks
SGLang EAGLE Serving

Datasets

Overview
Text Dataset
Retrieval Dataset
ColumnMapped Dataset
ColumnMapped Iterable
Multi-Modal Dataset
Diffusion Dataset

Job Launchers

Overview
Local Workstation
SLURM Cluster
NeMo Run
SkyPilot
SkyPilot k8s

Development

Checkpointing
Gradient Checkpointing
Pipeline Parallelism
Mixed-Precision Training
FP8 Training
MLflow Logging
API Reference
- nemo_automodel
Breaking Changes

API Reference
nemo_automodel
nemo_automodel.components
nemo_automodel.components.models
nemo_automodel.components.models.baichuan

`nemo_automodel.components.models.baichuan`#

Submodules#

nemo_automodel.components.models.baichuan.model
nemo_automodel.components.models.baichuan.configuration

previous

nemo_automodel.components.models.hy_mt2.state_dict_adapter

next

nemo_automodel.components.models.baichuan.model

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 © 2026, NVIDIA Corporation.

Last updated on 2026-05-29.