veScale

veScale

基于PyTorch的大规模语言模型训练框架

veScale是一个基于PyTorch的大规模语言模型训练框架,专为简化LLM训练过程而设计。它支持零代码修改、单设备抽象和自动并行规划,实现了张量并行、序列并行和数据并行等多种策略。框架还提供自动检查点重分片和nD分布式时间线功能,大幅提升了训练效率。作为一个持续发展的项目,veScale计划在未来引入更多先进功能,为研究人员和开发者提供全面的LLM训练解决方案。

PyTorchLLM训练框架分布式训练并行计算模型执行Github开源项目
<div align="center"> <img src="https://yellow-cdn.veclightyear.com/ab5030c0/cfc4fe9e-0053-46a3-a56a-6614da17f803.png" width="150"/> </div>

PyTorch原生LLM训练框架

一个易用的工业级框架

  • 🔥 PyTorch原生:veScale基于PyTorch原生的数据结构、运算符和API,充分利用了PyTorch在机器学习领域的生态系统优势。

  • 🛡 零模型代码改动:veScale将分布式系统设计与模型架构解耦,用户几乎无需或完全无需修改模型代码。

  • 🚀 单设备抽象:veScale为用户提供单设备语义,自动在设备集群中分发和编排模型执行。

  • 🎯 自动并行规划:veScale通过多种策略(张量、序列、数据、ZeRO、流水线并行)的协同,在半自动或全自动模式下并行化模型执行[即将推出]。

  • 即时与编译模式:veScale不仅支持即时模式下的并行训练和推理自动化,还支持编译模式以实现极致性能[即将推出]。

  • 📀 自动检查点重分片:veScale自动管理分布式检查点,可在不同集群规模和不同并行策略间进行在线重分片。

最新动态

  • [2024-7-25] veScale的流水线并行功能开源,包括API、图解析器、阶段抽象、调度和执行运行时,以及nD分布式时间线

  • [2024-5-31] veScale的快速检查点系统开源,具备自动检查点重分片、缓存、负载均衡、快速复制、去重和异步IO功能。

  • [2024-5-21] veScale的示例(MixtralLLama2nanoGPT)开源,训练损失曲线达到逐位正确。

  • [2024-5-13] veScale在MLSys 2024以海报形式首次亮相。

  • [2024-4-16] 我们的内部LLM训练系统在NSDI 2024上发表。

即将推出

veScale仍处于早期阶段。我们正在重构内部LLM训练系统组件以满足开源标准。暂定时间表如下:

  • 高级nD并行API,极易使用

  • 高级用户计划API,便于定制nD并行训练

  • 端到端vescale/examples,支持5D并行训练(TP、SP、DP、ZeRO、PP)

目录(网页版

简介

快速开始

DTensor

并行

规划

检查点

我们正在招聘!

许可证

veScale项目采用Apache License v2.0许可证。

编辑推荐精选

博思AIPPT

博思AIPPT

AI一键生成PPT,就用博思AIPPT!

博思AIPPT,新一代的AI生成PPT平台,支持智能生成PPT、AI美化PPT、文本&链接生成PPT、导入Word/PDF/Markdown文档生成PPT等,内置海量精美PPT模板,涵盖商务、教育、科技等不同风格,同时针对每个页面提供多种版式,一键自适应切换,完美适配各种办公场景。

AI办公办公工具AI工具博思AIPPTAI生成PPT智能排版海量精品模板AI创作热门
潮际好麦

潮际好麦

AI赋能电商视觉革命,一站式智能商拍平台

潮际好麦深耕服装行业,是国内AI试衣效果最好的软件。使用先进AIGC能力为电商卖家批量提供优质的、低成本的商拍图。合作品牌有Shein、Lazada、安踏、百丽等65个国内外头部品牌,以及国内10万+淘宝、天猫、京东等主流平台的品牌商家,为卖家节省将近85%的出图成本,提升约3倍出图效率,让品牌能够快速上架。

iTerms

iTerms

企业专属的AI法律顾问

iTerms是法大大集团旗下法律子品牌,基于最先进的大语言模型(LLM)、专业的法律知识库和强大的智能体架构,帮助企业扫清合规障碍,筑牢风控防线,成为您企业专属的AI法律顾问。

SimilarWeb流量提升

SimilarWeb流量提升

稳定高效的流量提升解决方案,助力品牌曝光

稳定高效的流量提升解决方案,助力品牌曝光

Sora2视频免费生成

Sora2视频免费生成

最新版Sora2模型免费使用,一键生成无水印视频

最新版Sora2模型免费使用,一键生成无水印视频

Transly

Transly

实时语音翻译/同声传译工具

Transly是一个多场景的AI大语言模型驱动的同声传译、专业翻译助手,它拥有超精准的音频识别翻译能力,几乎零延迟的使用体验和支持多国语言可以让你带它走遍全球,无论你是留学生、商务人士、韩剧美剧爱好者,还是出国游玩、多国会议、跨国追星等等,都可以满足你所有需要同传的场景需求,线上线下通用,扫除语言障碍,让全世界的语言交流不再有国界。

讯飞绘文

讯飞绘文

选题、配图、成文,一站式创作,让内容运营更高效

讯飞绘文,一个AI集成平台,支持写作、选题、配图、排版和发布。高效生成适用于各类媒体的定制内容,加速品牌传播,提升内容营销效果。

热门AI辅助写作AI工具讯飞绘文内容运营AI创作个性化文章多平台分发AI助手
TRAE编程

TRAE编程

AI辅助编程,代码自动修复

Trae是一种自适应的集成开发环境(IDE),通过自动化和多元协作改变开发流程。利用Trae,团队能够更快速、精确地编写和部署代码,从而提高编程效率和项目交付速度。Trae具备上下文感知和代码自动完成功能,是提升开发效率的理想工具。

AI工具TraeAI IDE协作生产力转型热门
商汤小浣熊

商汤小浣熊

最强AI数据分析助手

小浣熊家族Raccoon,您的AI智能助手,致力于通过先进的人工智能技术,为用户提供高效、便捷的智能服务。无论是日常咨询还是专业问题解答,小浣熊都能以快速、准确的响应满足您的需求,让您的生活更加智能便捷。

imini AI

imini AI

像人一样思考的AI智能体

imini 是一款超级AI智能体,能根据人类指令,自主思考、自主完成、并且交付结果的AI智能体。

下拉加载更多