maxtext.kernels package
=======================

.. automodule:: maxtext.kernels
   :members:
   :show-inheritance:
   :undoc-members:

Subpackages
-----------

.. toctree::
   :maxdepth: 4

   maxtext.kernels.attention
   maxtext.kernels.megablox

Submodules
----------

.. toctree::
   :maxdepth: 4

   maxtext.kernels.gather_reduce_pallas
   maxtext.kernels.gather_reduce_sc
   maxtext.kernels.sort_activations