attention
=========

.. automodule:: penzai.experimental.v2.nn.attention

   
   
   .. rubric:: Classes

   .. autosummary::
      :toctree: leaf
      :template: pzclass.rst
   
      ApplyCausalAttentionMask
      ApplyCausalSlidingWindowAttentionMask
      ApplyExplicitAttentionMask
      Attention
      KVCachingAttention