TPU

Create a TPU device instance.

def TPU(
    device: TPUType,
    partition: V5PParts | V6EParts | None = None
) - > Device

Create a TPU device instance.

Parameters

Name	Type	Description
device	TPUType	The type of the TPU device to create, such as "V5P" or "V6E".
partition	`V5PParts	V6EParts

Returns

Type	Description
Device	Device instance.