Primus

在 AMD GPU 上高效训练大模型,开箱即用

开源 🏷️完全免费 🔧 仅开源项目,无在线服务,需自行部署 模型训练 🍴 复刻 63 🐛 待处理 64 👀 关注 3
访问官网 ↗ GitHub ↗
GitHub 星标 ★ 139
维护状态 活跃
是否开源 开源
定价模式 🏷️完全免费 开源免费,基于ROCm生态

项目预览

Primus 项目界面预览图
图片来源:GitHub 官方仓库名片

项目数据

分类模型训练
开发团队AMD-AGI
定价模式🏷️完全免费
价格说明开源免费,基于ROCm生态
访问状态🔧 仅开源项目,无在线服务,需自行部署
是否开源是
开源协议NOASSERTION
主要语言Python
技术栈/模型amd-gpu,distributed-training,training,training-at-scale
GitHub 星标★ 139
上线时间2025-02-24 00:00:00
代码更新2026-11-03
仓库体积119.5 MB
本站数据更新2026-11-03
维护状态活跃
收录时间2026-09-19
浏览次数1

项目介绍

Primus 是面向 AMD GPU 的大规模基础模型训练框架,目标是在 AMD 硬件上提供灵活且高性能的分布式训练能力。它解决的核心问题是:主流训练框架(如 Megatron、DeepSpeed)对 AMD ROCm 生态支持有限,用户难以直接在 AMD 集群上高效训练大模型。Primus 通过整合分布式并行策略(数据并行、张量并行、流水线并行等)、混合精度训练、通信优化和显存管理,让开发者能在 AMD GPU 集群上完成从中小规模到超大规模模型的预训练与微调。项目以 Python 实现,强调可扩展性与性能,适合需要构建自主可控训练栈的团队。当前项目处于早期阶段,星标数较少,但方向明确,填补了 AMD 生态在大模型训练框架上的空白。

Star 增长趋势

本站每日记录该项目 GitHub 星标变化,观测越久曲线越完整

数据来源:GitHub API 逐日快照 · 收录于 2026-09-19 · 当前星标 ★ 139

核心亮点

  • 专为 AMD ROCm 生态设计,填补主流框架对 AMD 支持不足的空白
  • 集成多种分布式并行策略,支持大规模集群训练
  • Python 实现,便于二次开发和自定义训练流程

不足之处

  • 项目处于早期,文档和示例可能不完善
  • 社区规模小,问题反馈和生态集成有待观察

适用场景

  • 在 AMD GPU 集群上预训练百亿参数级大语言模型
  • 为 AMD 平台定制分布式训练流程的科研团队
  • 需要自主可控训练框架的企业进行模型微调

替代项目

Megatron-LM、DeepSpeed

上一篇:long-context-attention

下一篇:modalities

同类项目推荐

ml-engineering 开源

大模型训练推理踩坑指南,照着调优省半年时间

Machine Learning Engineering Open Book

★ 19657 2026-08-09
Dreambooth 开源

用几张照片微调扩散模型,让AI学会画你的专属对象

Fine-tuning of diffusion models

★ 97 2026-08-09
Fast-LLM 开源

让 LLM 训练跑满 GPU,少等快出结果

Accelerating your LLM training to full speed! Made with ❤️ by ServiceNow Research

★ 339 2026-09-26
RPG 开源

给大模型 RL 训练加上安全护栏,稳定提升性能

[ICLR 2026] RPG: KL-Regularized Policy Gradient (https://arxiv.org/abs/2505.17508)

★ 78 2026-08-09
OmniQuant 开源

把大模型压到4比特,速度更快内存更省,精度还不掉链子。

[ICLR2024 spotlight] OmniQuant is a simple and powerful quantization technique for L···

★ 968 2026-08-09
Medusa 开源

给大模型加几个解码头,生成速度翻倍,不用改架构

Medusa: Simple Framework for Accelerating LLM Generation with Multiple Decoding Head···

★ 2773 2026-08-09
LlamaFactory 开源

上百种模型随意挑,教出你专属的 AI

Unified Efficient Fine-Tuning of 100+ LLMs & VLMs (ACL 2024)

★ 75868 2026-08-09
VADER 开源

用奖励梯度微调视频模型,让生成结果更合你心意

Video Diffusion Alignment via Reward Gradients. We improve a variety of video diffus···

★ 318 2026-08-09