gpt\_neox
=========

.. automodule:: penzai.models.transformer.variants.gpt_neox

   
   
   .. rubric:: Classes

   .. autosummary::
      :toctree: leaf
      :template: pzclass.rst
   
      GPTNeoXTransformerConfig
   
   

   
   
   .. rubric:: Functions

   .. autosummary::
      :toctree: leaf
      :template: pzbase.rst
   
      build_gpt_neox_attention
      build_gpt_neox_block
      build_gpt_neox_feedforward
      build_gpt_neox_transformer
      gpt_neox_from_huggingface_model