Megatron-Bridge

Hugging Face 模型一键转 Megatron 训练，训完再转回来

开源 free 模型训练

访问官网 ↗ GitHub ↗ 文档 ↗

GitHub 星标 ★ 951

维护状态活跃

是否开源是

定价模式 free

项目预览

项目数据

分类模型训练

开发团队NVIDIA-NeMo

所属国家

官网地址https://docs.nvidia.com/nemo/megatron-bridge/latest

GitHubhttps://github.com/NVIDIA-NeMo/Megatron-Bridge

定价模式free

价格说明开源免费，NVIDIA NeMo生态训练库

访问状态

是否开源是

开源协议Apache-2.0

主要语言Python

技术栈/模型

GitHub 星标★ 951

30天Star增速

HF 下载量

上线时间2025-05-21 00:00:00

代码更新—

仓库体积—

本站数据更新—

维护状态活跃

中文支持

访问方式

移动端支持

综合评分

收录时间2026-10-12

浏览次数0

项目介绍

Megatron-Bridge 是 NVIDIA 推出的训练库，用于训练基于 Megatron 的大规模语言模型，并支持与 Hugging Face 模型格式的双向转换。它解决的核心问题是：Megatron 训练框架性能强但生态封闭，模型权重难以直接用于 Hugging Face 生态；而 Hugging Face 模型又难以直接接入 Megatron 的高效并行训练。该库提供统一的桥接层，让用户可以把 Hugging Face 模型转换为 Megatron 格式进行大规模训练，训练完成后再转回 Hugging Face 格式用于推理、评测或发布。核心能力包括：双向权重转换、支持多种主流模型架构、与 Megatron-Core 并行策略集成、面向大规模分布式训练的配置管理。适合需要兼顾训练效率与生态兼容性的团队。