build\_gpt\_neox\_attention
===========================

.. currentmodule:: penzai.models.transformer.variants.gpt_neox

.. autofunction:: build_gpt_neox_attention