Python class

Placement

Placement

class max.experimental.sharding.Placement

source

Bases: ABC

Abstract base for all placement types.

Every placement describes what a single mesh axis does to a tensor:

• Replicated: full copy on every device.
• Sharded: split along a tensor dimension.
• Partial: partial result needing reduction.

Custom subclasses (for example, StridedShard or InterleavedShard) can extend the vocabulary for non-standard partitioning patterns.

Was this page helpful?

Thank you! We'll create more content like this.

Thank you for helping us improve!

😔 What went wrong?

Some code doesn’t work

It includes inaccurate information

It's missing information I need

It was difficult to understand

Other

Submit