研究人员开发了 TrainBridge,一种在 Ascend 神经网络处理单元 (NPU) 上优化训练效率的新型智能体方法。该方法利用成熟的 GPU 编译栈中的优化洞察来指导 NPU 程序重写,旨在减少碎片化计算和执行开销。实验表明,TrainBridge 在包括 LLM 预训练和图像分类在内的各种工作负载上平均可实现 1.7 倍的加速,在 Modded-NanoGPT 速度运行中接近 2.0 倍。 AI
影响 这项研究可能导致在专用硬件上更高效的 AI 模型训练,从而降低成本并加速开发周期。
排序理由 该集群描述了一篇研究论文,其中详细介绍了一种在特定硬件上优化 AI 模型训练的新方法。[lever_c_demoted from research: ic=1 ai=1.0]
AI 生成摘要 · Google Gemini · 来自 1 个来源。 我们如何撰写摘要 →