IMPORTANT: To view this page as Markdown, append `.md` to the URL (e.g. /get-started.md). For the complete documentation index, see llms.txt.
Skip to main content
For the complete documentation index, see llms.txt. Markdown versions of all pages are available by appending .md to any URL (e.g. /get-started.md).

Python data

Transition

Transition​

max.experimental.sharding.Transition

source

A placement change on one mesh axis, as (from, to) placement types.

(Partial, Replicated) is an allreduce, (Sharded, Replicated) an allgather, (Replicated, Sharded) a local slice, (Sharded, Sharded) a re-shard along another tensor axis, and (Partial, Sharded) a reduce-scatter. Moving an input from another mesh is not a transition.

alias of tuple[type[Placement], type[Placement]]