``jax.experimental.pallas.tpu`` module
======================================

.. automodule:: jax.experimental.pallas.tpu

Classes
-------

.. autosummary::
   :toctree: _autosummary

Functions
---------

.. autosummary::
   :toctree: _autosummary