maxtext.kernels.attention.splash\_attention\_kernel module
==========================================================

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