一个名为KOSMOS的新库已被开发出来,以提高在GPU上训练大型语言模型的效率。KOSMOS通过使用自调度持久性内核来优化Transformer层内的通信和计算阶段,这些内核会适应数据就绪情况。在AMD Instinct MI350X和MI355X GPU上进行评估时,与现有方法相比,KOSMOS在Megatron-LM、DeepSeek-V3和Qwen3-235B等模型上实现了1.10倍至1.83倍的显著加速。 AI
影响 该库可以显著缩短LLM的训练时间并降低成本,从而使大型模型开发更加易于获得。
排序理由 该集群描述了一个新库及其在arXiv论文中提出的性能评估。[lever_c_demoted from research: ic=1 ai=1.0]
- AMD Instinct MI350X
- AMD Instinct MI355X
- arXiv
- DeepSeek-V3
- Hugging Face
- KOSMOS
- Megatron-LM
- Qwen3 235B
AI 生成摘要 · Google Gemini · 来自 1 个来源。 我们如何撰写摘要 →