GitHub · 灵魂体仓库·· 20 天前AI 评分48
AMD 开源 hip-kernel-llm-lab:面向 HIP kernel 的 LLM 训练框架
AMD-AGI/hip_kernel_llm_lab
来源与原文
GitHub · 灵魂体仓库
发布时间:(北京时间)
阅读原始来源 · github.com ↗完整链接与内容说明
https://github.com/AMD-AGI/hip_kernel_llm_lab
本站目前仅有 AI 导读与摘要,未展示原文正文;技术细节、代码和完整结论请查看原始来源。
AI 导读
AMD 在 GitHub 开源 hip-kernel-llm-lab,一个涵盖持续预训练(CPT)、监督微调(SFT)和强化学习(RL)三个阶段的 LLM 训练框架,用于训练模型生成、理解和优化 HIP kernel。仓库还提供 hip-kernel-generator 数据生成工具,支持 PyTorch 到 HIP kernel 的转换与优化,训练数据托管在 Hugging Face 的 amd/hip_llm_dataset。CPT 和 SFT 基于 LLaMA-Factory,RL 部分提供基于 veRL 的 HIP 训练栈及 32-MI325X 的多轮 Triton RL 配置。
来源:GitHub · 灵魂体仓库 · github.com