MCPcopy Index your code
hub / github.com/NVIDIA-NeMo/Automodel

github.com/NVIDIA-NeMo/Automodel @v0.5.0

Chat with this repo
repository ↗ · DeepWiki ↗ · release v0.5.0 ↗ · + Follow
19,169 symbols 71,693 edges 1,085 files 8,882 documented · 46%
What it actually does AI analysis from the code graph — generated when you open this
loading…
README

🚀 NeMo AutoModel

codecov CICD NeMo Python 3.10+ Contributions Welcome GitHub Stars

📖 Documentation🔥 Ready-to-Use Recipes💡 ExamplesModel CoveragePerformance🤝 Contributing

📣 News and Discussions

  • [06/04/2026]Nemotron-3 Ultra We now support finetuning NVIDIA's Nemotron 3 Ultra 550B A55B. Check out our recipe and guide.
  • [06/03/2026]Gemma 4 12B We now support finetuning the dense google/gemma-4-12B model. Check out our recipe.
  • [05/27/2026]Step-3.7-Flash We added model coverage for Stepfun AI's 198B-A13B MoE vision-language model, targeting image/video agentic developer workflows with a 256k context language backbone and 1.8B ViT vision tower. See the model coverage page.
  • [05/19/2026]Ling 2.0 We now support finetuning the inclusionAI Ling 2.0 MoE family (inclusionAI/Ling-mini-2.0, inclusionAI/Ling-flash-2.0, and inclusionAI/Ling-1T), thanks to @Hayden727. Check out our recipes.
  • [05/17/2026]ERNIE 4.5 and MiMo-V2-Flash We now support finetuning baidu/ERNIE-4.5-0.3B-PT, baidu/ERNIE-4.5-21B-A3B-PT, and XiaomiMiMo/MiMo-V2-Flash. Check out our ERNIE dense recipe, ERNIE MoE recipe, and MiMo recipe.
  • [04/29/2026]Mistral Medium 3.5 We now support finetuning Mistral AI's 128B FP8-native VLM Mistral Medium 3.5. Check out our recipe and guide.
  • [04/28/2026]Nemotron-3-Nano-Omni We now support finetuning nvidia/Nemotron-3-Nano-Omni-30B-A3B-Reasoning-BF16, NVIDIA's 30B-A3B omnimodal MoE (text · image · audio) with NemotronH hybrid Mamba+Attention backbone. Check out our SFT recipe, LoRA recipe, and guide.
  • [04/28/2026]Hy3-preview We now support finetuning tencent/Hy3-preview, thanks to @Khazic. Check out our recipe.
  • [04/25/2026]DeepSeek V4 Flash We now support finetuning deepseek-ai/DeepSeek-V4-Flash, thanks to @Khazic. Check out our recipe and guide.
  • [04/22/2026]Qwen3.6-27B We now support finetuning Qwen/Qwen3.6-27B. Check out our recipe.
  • [04/20/2026]Qwen-Image We now support finetuning Qwen/Qwen-Image, thanks to @harshareddy832. Check out our recipe.
  • [04/16/2026]Qwen3.6 MoE We now support finetuning Qwen/Qwen3.6-35B-A3B. Check out our recipe.
  • [04/16/2026]LLaVA-OneVision-1.5 We now support finetuning lmms-lab/LLaVA-OneVision-1.5-4B-Instruct, thanks to @vgauraha62. Check out our recipe.
  • [04/12/2026]MiniMax-M2.7 We now support finetuning MiniMaxAI/MiniMax-M2.7. Check out our recipe.
  • [04/07/2026]GLM-5.1 We now support finetuning zai-org/GLM-5.1. GLM-5.1 is Zhipu AI's latest open-source MoE model featuring MLA + DeepSeek Sparse Attention. Check out our recipe and discussion.
  • [04/02/2026]Gemma 4 We support fine-tuning for Gemma4 (2B, 4B, 31B, 26BA4B)! Check out our recipes.
  • [03/30/2026]NeMo AutoModel ships with agent-friendly skills in skills/ to help you with common development tasks (e.g., running a recipe, model onboarding, development) across the repo. We welcome PRs that improve existing skills or add new ones.
  • [03/16/2026]Mistral Small 4 We support fine-tuning for Mistral4 119B! Check out our recipe.
  • [03/11/2026]Nemotron Super v3 We support fine-tuning for nvidia/NVIDIA-Nemotron-3-Super-120B-A12B-BF16. Check out our recipe.
  • [03/11/2026]GLM-5 We now support finetuning zai-org/GLM-5. Check out our recipe.
  • [03/02/2026]Qwen3.5 small models We support finetuning for Qwen3.5 small models 0.8B, 2B, 4B (recipe) and 9B (recipe)
  • [02/16/2026]Qwen3.5 MoE We support finetuning for Qwen/Qwen3.5-397B-A17B (recipe) and Qwen/Qwen3.5-35B-A3B (recipe)

Previous News

Extension points exported contracts — how you extend this code

BadgeLinksProps (Interface)
(no doc)
fern/components/BadgeLinks.tsx

Core symbols most depended-on inside this repo

info
called by 1008
nemo_automodel/components/distributed/pipelining/autopipeline.py
patch
called by 822
nemo_automodel/autonvtx/__init__.py
get
called by 617
nemo_automodel/recipes/_typed_config.py
get
called by 521
nemo_automodel/components/config/loader.py
keys
called by 385
nemo_automodel/_transformers/registry.py
get
called by 349
nemo_automodel/components/datasets/llm/megatron/indexed_dataset.py
size
called by 311
nemo_automodel/components/datasets/llm/megatron/indexed_dataset.py
exists
called by 229
nemo_automodel/components/checkpoint/_backports/filesystem.py

Shape

Method 10,492
Function 5,615
Class 2,870
Route 191
Interface 1

Languages

Python100%
C++1%
TypeScript1%

Modules by API surface

tests/unit_tests/datasets/vlm/test_collate_fns.py314 symbols
tests/unit_tests/recipes/test_finetune_vlm_helpers.py218 symbols
tests/unit_tests/distributed/test_parallelizer.py207 symbols
tests/unit_tests/moe/test_parallelizer.py200 symbols
tests/unit_tests/_transformers/test_auto_model.py157 symbols
tests/unit_tests/checkpoint/test_checkpointing.py154 symbols
tests/unit_tests/recipes/test_train_ft.py149 symbols
tests/unit_tests/models/kimi_k25_vl/test_kimi_k25_vl_model.py142 symbols
tests/unit_tests/moe/test_experts.py127 symbols
tests/unit_tests/distributed/pipelining/test_functional.py119 symbols
tests/unit_tests/distributed/pipelining/test_hf_utils.py116 symbols
tests/unit_tests/models/kimivl/test_kimivl.py114 symbols

For agents

$ claude mcp add Automodel \
  -- python -m otcore.mcp_server <graph>

⬇ download graph artifact