Transformer wrappers
====================

.. currentmodule:: attnax

.. autosummary::

   TransformerEncoder
   TransformerDecoder

Transformer encoder
-------------------

.. autoclass:: TransformerEncoder

Transformer decoder
-------------------

.. autoclass:: TransformerDecoder