jax.experimental.pallas.tpu.TpuInfo
===================================

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

.. autoclass:: TpuInfo

   
   .. automethod:: __init__

   
   .. rubric:: Methods

   .. autosummary::
   
      ~TpuInfo.__init__
      ~TpuInfo.get_sublane_tiling
      ~TpuInfo.is_matmul_supported
   
   

   
   
   .. rubric:: Attributes

   .. autosummary::
   
      ~TpuInfo.is_lite
      ~TpuInfo.is_megacore
      ~TpuInfo.is_split_chip
      ~TpuInfo.sparse_core
      ~TpuInfo.chip_version
      ~TpuInfo.generation
      ~TpuInfo.num_cores
      ~TpuInfo.num_lanes
      ~TpuInfo.num_sublanes
      ~TpuInfo.mxu_column_size
      ~TpuInfo.num_mxus
      ~TpuInfo.num_accumulators
      ~TpuInfo.vmem_capacity_bytes
      ~TpuInfo.cmem_capacity_bytes
      ~TpuInfo.smem_capacity_bytes
      ~TpuInfo.hbm_capacity_bytes
      ~TpuInfo.mem_bw_bytes_per_second
      ~TpuInfo.bf16_ops_per_second
      ~TpuInfo.int8_ops_per_second
      ~TpuInfo.fp8_ops_per_second
      ~TpuInfo.int4_ops_per_second