Shardy

Shardy 是一种基于 MLIR 的张量分区系统,适用于所有方言。该工具由 GSPMDPartIR 团队联合打造,融合了这两个系统的优势,以及这两个团队和用户的共同经验。

使用入门

如果模型开发者想要使用 Shardy,请参阅 JAX 中的 Shardy 使用入门

如需开始详细了解构成 Shardy 的组件,请参阅 Shardy 概览页面。

通信通道

状态

Shardy 仍在开发中。核心方言和 C 绑定完全正常运行。Python 绑定正在开发中。