跳到正文
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