jax.experimental.pallas.tpu.ChipVersion
=======================================

.. currentmodule:: jax.experimental.pallas.tpu

.. autoclass:: ChipVersion

   
   .. automethod:: __init__

   
   

   
   
   .. rubric:: Attributes

   .. autosummary::
   
      ~ChipVersion.num_physical_tensor_cores_per_chip
      ~ChipVersion.supports_megacore
      ~ChipVersion.is_lite
      ~ChipVersion.TPU_V2
      ~ChipVersion.TPU_V3
      ~ChipVersion.TPU_V4I
      ~ChipVersion.TPU_V4
      ~ChipVersion.TPU_V5E
      ~ChipVersion.TPU_V5P
      ~ChipVersion.TPU_V6E
      ~ChipVersion.TPU_7
      ~ChipVersion.TPU_7X
      ~ChipVersion.TPU_8I