≡ AI工具

XTuner V1 – 上海AI Lab开源的新一代大模型训练引擎

分类:AI工具
XTuner V1 – 上海AI Lab开源的新一代大模型训练引擎

XTuner V1是什么

XTuner V1 是上海人工智能实验室开源的新一代大模型训练引擎,基于 PyTorch FSDP 开发,针对超大模型训练中的显存、通信和负载问题进行系统性优化,支持 1T 参数量级 MoE 模型训练,能在 200B 量级模型上实现训练吞吐超越传统 3D 并行方案。XTuner V1 与华为昇腾技术团队合作,进一步提升训练效率,实现更高的模型浮点运算利用率。XTuner V1 为学术界和工业界提供高性能、低门槛、易扩展的大模型训练方案。

XTuner V1