astra-sim

单机仿真万卡训练，提前看清网络瓶颈

开源 free 基础设施

访问官网 ↗ GitHub ↗ 文档 ↗

GitHub 星标 ★ 702

维护状态较活跃

是否开源是

定价模式 free

项目预览

项目数据

分类基础设施

开发团队astra-sim

所属国家

官网地址https://astra-sim.github.io

GitHubhttps://github.com/astra-sim/astra-sim

定价模式free

价格说明开源免费，需自行部署

访问状态

是否开源是

开源协议MIT

主要语言C++

技术栈/模型

GitHub 星标★ 702

30天Star增速

HF 下载量

上线时间2020-07-10 00:00:00

代码更新—

仓库体积—

本站数据更新—

维护状态较活跃

中文支持

访问方式

移动端支持

综合评分

收录时间2026-10-12

浏览次数0

项目介绍

ASTRA-sim 是一个面向大规模模型训练的分布式系统仿真器，由佐治亚理工等团队开发。它解决的是：在真实集群上验证千卡/万卡训练的网络拓扑、并行策略和调度算法成本极高，而纯解析模型又难以刻画层次化网络与解耦架构。ASTRA-sim 通过将计算、内存、通信行为抽象为工作负载图，结合可插拔的网络后端（如 ns-3、Garnet）和集合通信库模型，在单机上模拟 DNN 训练在层次化网络与解耦系统上的端到端性能。核心能力包括：支持数据/流水/张量/专家并行组合，建模层次化互联（节点内 NVLink、节点间 InfiniBand/RoCE），支持解耦式（disaggregated）内存与计算池，输出迭代时间、通信占比等指标，帮助研究者在设计阶段快速比较拓扑与算法。