sampling\_mode
==============

.. automodule:: penzai.example_models.gemma.sampling_mode

   
   
   .. rubric:: Classes

   .. autosummary::
      :toctree: leaf
      :template: pzclass.rst
   
      GemmaKVCachingAttention
      GemmaKVCachingInputs
      GemmaKVCachingState
      GemmaKVCachingTransformer