maxtext.kernels.attention.jax\_flash\_attention module
======================================================

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