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

.. automodule:: parts.transformer
    :members:
    :undoc-members:
    :show-inheritance:

attention\_layer
-------------------------------------------------------

.. automodule:: parts.transformer.attention_layer
    :members:
    :undoc-members:
    :show-inheritance:

beam\_search
---------------------------------------------------

.. automodule:: parts.transformer.beam_search
    :members:
    :undoc-members:
    :show-inheritance:

common
---------------------------------------------

.. automodule:: parts.transformer.common
    :members:
    :undoc-members:
    :show-inheritance:

embedding\_layer
-------------------------------------------------------

.. automodule:: parts.transformer.embedding_layer
    :members:
    :undoc-members:
    :show-inheritance:

ffn\_layer
-------------------------------------------------

.. automodule:: parts.transformer.ffn_layer
    :members:
    :undoc-members:
    :show-inheritance:

utils
--------------------------------------------

.. automodule:: parts.transformer.utils
    :members:
    :undoc-members:
    :show-inheritance: