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 class
Partial
Partial
class max.experimental.sharding.Partial
Bases: Placement
Every device holds a part of the value, and the value is their sum.
transition_to()
transition_to(other)
Returns the Collective that resolves the partial sum.
Replicated takes the allreduce, and Sharded takes
the reduce-scatter.
-
Parameters:
-
other (Placement) – The placement this one moves to.
-
Returns:
-
The
Collectivethat performs the move. -
Return type:
-
Collective