昇思 MindSpore是华为于 2019 年 8 月推出的全场景 AI 框架,2020 年 3 月正式开源,旨在通过技术创新和生态共建推动 AI 产业发展。
-
动静统一编程
- 支持动态图(调试友好)与静态图(高效执行)统一编码,通过一行代码切换模式(
set_context(mode=GRAPH_MODE/PYNATIVE_MODE)
),兼顾开发效率与性能。
- 基于 Python 原生语法构建神经网络,支持控制流、Lambda 表达式等灵活表达。
-
全场景部署能力
- 端 – 边 – 云协同:统一架构支持手机、边缘设备、服务器及云端部署,适配昇腾、GPU、CPU 等硬件,提供轻量化推理引擎 MindSpore Lite。
- 隐私保护:支持端侧训练和联邦学习,减少数据传输与隐私风险。
-
分布式训练原生优化
- 自动并行策略:通过
set_auto_parallel_context
自动选择并行模式,降低分布式训练门槛。
- 多维度并行技术:支持流水线并行、优化器并行、重计算等,提升大模型训练效率。
-
AI 与科学计算融合
- 支持函数式微分编程,满足科学计算需求,如气象预报、蛋白质结构预测等,发布 AI4S 套件(MindSpore SciAI)和地球科学套件(MindSpore Earth)。
- 四层架构设计:
- 模型层:预置大模型套件、领域套件(如推荐、生成式 AI)及科学计算库。
- 表达层(MindExpression):提供 Python/C++ 统一接口,支持动静态图混合开发。
- 编译优化层(MindCompiler):通过中间表达 MindIR 实现硬件无关优化(代数化简、图算融合)和硬件相关优化(算子生成)。
- 运行时(MindRT):对接底层硬件,支持联邦学习等协同任务。
- v2.3(2024 年 3 月):
- 优化大模型训练效率,支持快速故障恢复(编译快照、确定性 CKPT)。
- 部署性能提升:推理吞吐量翻倍,千亿模型压缩至十分之一。
- 推出 MindSpore One 生成式套件,简化大模型开发流程。
- v2.2.0:
- 引入 Lazy Inline 模式,降低大模型编译耗时。
- 强化科学计算能力,发布 AI4S 高频模型套件。
- v1.8.0:
- 开源大模型训练套件 MindSpore Transformers,支持推荐网络(如 Wide&Deep)。
- 统一图像处理接口,优化 API 易用性。
- 大模型开发:支持 50 + 主流大模型(如 Baichuan2-13B),提供全流程工具链。
- AI + 科学计算:气象预报、药物研发、材料设计等,缩短研发周期。
- 企业智能化:金融风控、智慧城市、工业设计(如气动设计周期从月级缩短至分钟级)。
- 边缘与端侧:手机、耳机等设备的轻量化推理与个性化模型微调。
- 市场份额:2024 年中国 AI 框架新增市场份额 30.26%,位列第一(IDC 数据)。
- 社区规模:全球下载量超 1100 万次,覆盖 130 + 国家,开发者超 3.7 万,学术论文发表量全球第二(Papers With Code)。
- 行业合作:服务 5500 + 企业,包括鹏城实验室、星舟志屹等,推动能源、交通、医疗等领域创新。
昇思 MindSpore 通过动静统一编程、全场景部署和分布式训练优化,成为大模型时代的核心框架之一。未来将持续强化 AI 与科学计算融合、多模态模型支持及硬件兼容性,推动国产 AI 生态的全球竞争力。