jax.experimental.pallas.tpu.ChipVersion#

class jax.experimental.pallas.tpu.ChipVersion(value, names=<not given>, *values, module=None, qualname=None, type=None, start=1, boundary=None)[source]#
__init__(*args, **kwds)#

Attributes

TPU_V2

TPU_V3

TPU_V4I

TPU_V4

TPU_V5E

TPU_V5P

TPU_V6E

TPU_7X