AMD-LLM

在 AMD GPU 上快速跑通 1.35 亿参数语言模型训练

开源 🏷️完全免费 🔧 仅开源项目,无在线服务,需自行部署 开源模型 🍴 复刻 13 👀 关注 4
GitHub ↗
GitHub 星标 ★ 195
维护状态 低维护
是否开源 开源
定价模式 🏷️完全免费 开源项目,训练代码和资源,可自行部署使用,完全免费。

项目预览

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

项目数据

分类开源模型
开发团队AMD-AGI
定价模式🏷️完全免费
价格说明开源项目,训练代码和资源,可自行部署使用,完全免费。
访问状态🔧 仅开源项目,无在线服务,需自行部署
是否开源是
开源协议NOASSERTION
主要语言Python
技术栈/模型amd-agi,large-language-models,rocm
GitHub 星标★ 195
上线时间2024-05-30 00:00:00
代码更新2026-06-23
仓库体积77 KB
本站数据更新2026-11-08
维护状态低维护
收录时间2026-08-09
浏览次数4

项目介绍

AMD-LLM 是 AMD 推出的面向其 GPU 的 135M 参数语言模型训练代码与资源库。该项目旨在解决 AMD GPU 上训练大语言模型时软件栈不成熟、生态碎片化的问题,提供了一套开箱即用的训练流程。核心能力包括:针对 AMD GPU(如 MI300 系列)优化的训练脚本、数据预处理管线、模型配置与检查点管理,以及基于 ROCm 的底层适配。项目采用 PyTorch 框架,代码结构清晰,便于研究者复现和扩展。通过该仓库,开发者可以快速在 AMD 硬件上启动小规模语言模型的预训练或微调,验证算法或进行教学实验。项目目前处于早期阶段,模型规模较小(135M),但为 AMD 平台上的 LLM 训练提供了重要参考实现。

Star 增长趋势

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

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

核心亮点

  • 专为 AMD GPU(ROCm)优化,填补了 AMD 平台 LLM 训练教程空白
  • 代码简洁,依赖少,基于 PyTorch 即可上手,适合学习研究
  • 提供了完整的数据处理到训练检查点的全流程示例

不足之处

  • 模型规模仅 135M,实用性有限,主要面向验证和教学
  • 文档/社区待观察

适用场景

  • 在 AMD 显卡上尝试训练小型语言模型的科研人员
  • 高校或企业进行 LLM 训练流程教学或内部培训
  • 验证 ROCm 软件栈在生成式 AI 任务中的表现

替代项目

NanoGPT、LitGPT、TinyLlama

上一篇:residual-context-diffusion

下一篇:LLM-ESR

同类项目推荐

lora-speedrun 开源

固定任务硬件,比谁把 LoRA 微调跑得最快

Speedrunning LoRA fine-tuning: frozen task, frozen hardware, public wall-clock leade···

★ 151 2026-09-29
RAG-Retrieval 开源

统一微调 RAG 检索模型,让召回更精准

Unify Efficient Fine-tuning of RAG Retrieval, including Embedding, ColBERT, ReRanke···

★ 1137 2026-08-09
SimPO 开源

去掉参考模型,用更简单的公式让模型更懂人类偏好

[NeurIPS 2024] SimPO: Simple Preference Optimization with a Reference-Free Reward

★ 964 2026-08-09
RAGEN 开源

多轮智能体强化学习训练与推理退化诊断

Agent RL framework for LLM agents: multi-turn reinforcement learning with StarPO and···

★ 2840 2026-11-03
FedML 开源

一键跨云调度,轻松搞定分布式与联邦学习

FEDML - The unified and scalable ML library for large-scale distributed training, mo···

★ 4071 2026-08-09
grafting 开源

像嫁接植物一样组合扩散模型架构,快速找到最优设计。

[NeurIPS 2025 Oral] Official Code for Exploring Diffusion Transformer Designs via Gr···

★ 77 2026-08-09
FeynRL 开源

换目标、换采样,后训练实验一套框架跑通

Post-training framework for large models, from new objectives to new rollout systems···

★ 228 2026-10-31
Matlab-GAN 开源

用 MATLAB 跑通多种 GAN,从入门到 CycleGAN 一步到位

MATLAB implementations of Generative Adversarial Networks -- from GAN to Pixel2Pixel···

★ 209 2026-08-09
sd-leap-booster 开源

让 Stable Diffusion 微调更快,从局部最优起步,省时省力。

Fast finetuning using a booster model that puts the initial state to a local minimum

★ 112 2026-08-09