adding-model-support

Official

Extend Megatron-Bridge to support new models.

AuthorNVIDIA
Version1.0.0
Installs0

System Documentation

What problem does it solve?

This guide helps engineers extend Megatron-Bridge to support new HF model families by providing a repeatable blueprint for bridges, providers, mappings, tests, and docs.

Core Features & Use Cases

  • Structured extension workflow: Phase-based guidance for discovery, bridge/provider/mappings, recipe scaffolding, tests, and documentation.
  • Model-agnostic patterns: Applicable to both LLMs and VLMs with recommended strategies for private HF configs, top-level vs sub-config mappings, and optional VLM providers.
  • Improved reliability: Ensures consistent mappings, provider hooks, and test coverage to minimize integration risk.

Quick Start

Follow Phase 1–5 to add support for a new HF model family by implementing the bridge, mapping, provider (if needed), recipes, tests, and docs.

Dependency Matrix

Required Modules

None required

Components

Standard package

💻 Claude Code Installation

Recommended: Let Claude install automatically. Simply copy and paste the text below to Claude Code.

Please help me install this Skill:
Name: adding-model-support
Download link: https://github.com/NVIDIA/skills/archive/main.zip#adding-model-support

Please download this .zip file, extract it, and install it in the .claude/skills/ directory.
View Source Repository

Agent Skills Search Helper

Install a tiny helper to your Agent, search and equip skill from 471,000+ vetted skills library on demand.