Pallas TPU
==========
TPU specific documentation.

.. toctree::
   :caption: Guides
   :maxdepth: 2

   details
   pipelining
   matmul
   sparse
   distributed