Skip to content

About

No description, website, or topics provided.

Resources

Stars

62 stars

Watchers

2 watching

Forks

Latest commit

 

History

1,799 Commits

Folders and files

NameName
Last commit message
Last commit date
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

HyperParallel Logo

WeChat 微信交流群 小红书 HyperParallel(小红书号:hyperparallel) Document 文档中心 Papers

简化超节点编程  ·  释放并行算力

快速开始  ·  文档中心  ·  API 参考  ·  技术报告


简介

HyperParallel 面向昇腾超节点提供分布式并行加速能力。针对资源池化、对等架构、分层网络拓扑和 FP8 低精度格式等特点,框架覆盖从集群分布式并行到芯片内多核并行,并支持 CPU DRAM 与 NPU HBM 池化管理、拓扑感知调度、通信路径规划和 FP8 混合精度训练。


核心设计

  • 模型与系统优化解耦:HyperParallel 将并行、重计算和 Offload 等系统优化从模型代码中分离,并通过声明式接口注入模型;并行范式覆盖集群 SPMD、集群 MPMD 和芯片内多核 MPMD,面向大语言模型和多模态模型训练。

  • 兼容 Hugging Face 生态:HyperParallel 支持直接读取 Hugging Face 模型配置和预训练权重,并复用 Transformers 中已有的模型实现;模型创建后,由 HyperParallel 以声明式方式将并行、重计算和算子替换等分布式逻辑注入模型。

  • 全流程确定性:HyperParallel 支持计算、通信、数据预处理和随机数等环节的确定性,并对已支持模型进行 bitwise 对齐验证。确定性模式可能带来性能开销,但有助于复现训练结果、发现 SDC 和定位问题。

  • 动静混合:静态图有利于通算并发、内存分析和执行序编排,但完整动转静对模型语法和编译能力要求较高。HyperParallel 在动态图基础上支持受约束的局部动转静。


📣 最新消息


使用入口

HyperParallel 模型训练

HyperParallel 模型 面向训练用户,通过 YAML 配置 Hugging Face 模型、数据、优化器、并行策略以及保存与恢复方式。

支持模型

HyperParallel Core

HyperParallel Core 面向模型作者和框架开发者,提供从分布式模型构建到训练执行与状态管理的底层能力。

核心能力

  • 模型切分与并行组合:以 DeviceMesh、DTensor 和 Shard 描述设备拓扑与 Tensor 布局,并组合 FSDP/HSDP、TP、CP、EP、PP 等并行策略。
  • MoE 与异构执行:覆盖专家切分、路由和 token dispatch,以及流水线调度、集群 MPMD 和芯片内多核 MPMD。
  • 内存与性能优化:提供 Activation Checkpoint、Activation Swap、通信与计算重叠、通信融合和自动并行能力。
  • 训练与状态管理:提供优化器、分布式检查点、异步保存、离线格式转换和确定性训练能力。
相关文档

未来计划

安装

完整安装方式、源码构建参数和环境要求见 安装指南。


🚀 快速开始

使用 HyperParallel 启动模型训练

运行仓库提供的单机 8 卡训练示例 hyper_parallel/models/qwen3_moe/recipes/train.yaml:

torchrun --standalone --nproc_per_node=8 \
  scripts/train_lm.py hyper_parallel/models/qwen3_moe/recipes/train.yaml

附加命令行参数可按 YAML 字段路径覆盖配置值,无需修改配置文件:

torchrun --standalone --nproc_per_node=8 \
  scripts/train_lm.py hyper_parallel/models/qwen3_moe/recipes/train.yaml \
  --model.pretrained_model_name_or_path=/path/to/model \
  --training.train_iters=10

字段定义与解析规则见 训练配置与组件。

使用 HyperParallel Core API

以下示例在数据并行 DeviceMesh 上应用 fully_shard,实现模型参数分片:

from hyper_parallel import fully_shard, init_device_mesh

dp_mesh = init_device_mesh("npu", (dp_size,), mesh_dim_names=("dp",))
model = fully_shard(model, mesh=dp_mesh)

TP、CP、EP、PP 等组合方式见 特性使用指南,公开接口见 API 参考。


📖 文档


🤝 加入我们

欢迎提交 Issue 和 Pull Request。开发环境、测试规范和提交流程见 社区贡献指南。

Parallel Training System SIG

如果对 HyperParallel 的技术方向感兴趣,欢迎加入 Parallel Training System SIG。

扫描下方二维码加入 Parallel Training System SIG 微信交流群,与社区开发者交流使用经验、技术方案和项目进展。

Parallel Training System SIG 微信交流群二维码

HyperParallel 技术博客

扫描下方二维码关注 HyperParallel 小红书账号(小红书号:hyperparallel),获取技术博客与实践分享。

HyperParallel 小红书技术博客二维码


📄 技术报告

  • HyperParallel-FSDP: Topology-Aware Fully Sharded Training with Layout-Driven Muon on Ascend SuperPods [paper]
  • HyperParallel-Mpipe: A Composable Algebra System for Optimizing MLLM Training over Supernode Clusters [paper]
  • HyperParallel-MoE: Multi-Core Interleaved Scheduling for Fast MoE Training on Ascend NPUs [paper]
  • HyperParallel: A Supernode-Affinity AI Framework [paper]
  • HyperOffload: Graph-Driven Hierarchical Memory Management for Large Language Models on SuperNode Architectures [paper]

⚖️ 许可证

Apache 2.0许可证

About

No description, website, or topics provided.

Resources

Stars

62 stars

Watchers

2 watching

Forks

Releases

Packages

Contributors

Languages