For the complete documentation index, see llms.txt. Markdown versions of all pages are available by appending .md to any URL (e.g. /max/get-started.md).
Python class
Action
Action
class max.experimental.sharding.Action(inputs, outputs)
Bases: object
A rule’s picked decision for one op call.
inputs has one entry per op argument in positional order:
DeviceMapping at tensor positions; bare scalar (treated as
uniform across ranks) or PerShard at non-tensor positions.
outputs has one DeviceMapping per op output. The
dispatcher inserts transfer_to collectives to match
inputs before per-shard dispatch; outputs is the
post-op mapping each result wears.
-
Parameters:
-
- inputs (tuple[Any, ...])
- outputs (tuple[DeviceMapping, ...])
inputs
outputs
outputs: tuple[DeviceMapping, ...]
Was this page helpful?
Thank you! We'll create more content like this.
Thank you for helping us improve!