Shardy

Shardy is an MLIR-based tensor partitioning system for all dialects. Built from the collaboration of both the GSPMD and PartIR teams, it incorporates the best of both systems, and the shared experience of both teams and users.

Getting started

For model developers looking to use Shardy, take a look at Getting started with Shardy in JAX.

To start learning more about the components that make up Shardy, take a look at the Shardy Overview page.

Communication channels

Status

Shardy is a work in progress. The core dialect and C bindings are fully functional. The python bindings are under development.

Except as otherwise noted, the content of this page is licensed under the Creative Commons Attribution 4.0 License, and code samples are licensed under the Apache 2.0 License. For details, see the Google Developers Site Policies. Java is a registered trademark of Oracle and/or its affiliates.

Last updated 2024年12月03日 UTC.