Module partition 

Backend-neutral rank-local architecture ownership. Backend-neutral ownership of one rank-local architecture partition.

Structs

ArchitectureBoundaryValue

One architecture-tagged auxiliary boundary value.

ArchitectureParameterDescription

Complete, validated parameter-ownership declaration for an architecture.

ArchitecturePartition

Complete backend-neutral realization of one rank’s architecture ownership.

BoundaryTensorSpec

Semantic role, symbolic shape, and logical dtype of one boundary tensor.

BoundaryWireSchema

Complete primary and ordered auxiliary wire schema for one architecture boundary.

LayeredPartitionDriver

Validated execution metadata for one rank-local layered partition.

NoAuxiliaryBoundary

Explicit declaration that an architecture partition carries no auxiliary tensors across its boundary.

NoAuxiliaryBoundarySchema

Schema for an evolving decoder activation with no auxiliary tensors.

OwnedParameterGroupSpec

One neutral parameter group tagged with its architecture-owned location.

PartitionGroup

One validated architecture group and its group-local global unit range.

PartitionOwnership

Input, output, and pinned static-module ownership for one partition.

PartitionState

Rank-local mutable-state geometry and its architecture-global layer range.

PipelineWireContract

Backend-neutral wire contract shared by every stage of one pipeline.

ResolvedBoundaryTensorSpec

One boundary tensor after invocation-dependent dimensions are resolved.

ResolvedBoundaryWireSchema

One architecture boundary after invocation-dependent dimensions are resolved.

Enums

ArchitectureBoundaryError

Invalid architecture-owned partition boundary declaration or payload.

ArchitectureParameterError

Invalid architecture-owned parameter ownership declaration.

ArchitecturePartitionError

Invalid backend-neutral architecture partition declaration.

BoundaryTensorDimension

One symbolic dimension in an architecture-owned boundary tensor.

BoundaryTensorDtype

Logical scalar kind carried by one architecture-owned boundary tensor.

LayeredPartitionBeginError

Failure to enter one concrete rank-local partition group.

LayeredPartitionError

Invalid concrete realization or boundary use of a layered partition.

ParameterGroupOwner

Architecture-owned location of one neutral parameter group.

PipelineActivationDtype

Portable floating-point dtype carried between pipeline stages.

Traits

ArchitectureBoundary

Typed architecture-owned tensor state and wire geometry carried across one partition boundary.

Functions

validate_boundary_tensor_count

Validates the number of tensors before a family boundary decodes any positional value.