transformer
===========

.. automodule:: penzai.experimental.v2.models.transformer

   
   
   

   
   
   

   
   
   

   
   
   



.. rubric:: Submodules

.. autosummary::
   :toctree:
   :template: pzmodule.rst
   :recursive:

   penzai.experimental.v2.models.transformer.model_parts
   penzai.experimental.v2.models.transformer.sampling_mode
   penzai.experimental.v2.models.transformer.simple_decoding_loop
   penzai.experimental.v2.models.transformer.variants