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
DistributedType
DistributedType
class max.experimental.sharding.DistributedType(dtype, shape, mesh, placements)
Shared state and shard-shape logic for distributed type descriptors.
Not intended for direct use. See DistributedTensorType and
DistributedBufferType.
-
Parameters:
-
- dtype (DType)
- shape (Shape)
- mesh (DeviceMesh)
- placements (tuple[Placement, ...])
dtype
dtype: DType
local_types
abstract property local_types: list[T]
The per-device types in mesh order.
mesh
mesh: DeviceMesh
placements
rank
property rank: int
The rank (number of dimensions) of the global tensor.
shape
shape: Shape
Was this page helpful?
Thank you! We'll create more content like this.
Thank you for helping us improve!