megatron.model.enums
====================

.. rubric:: Description

.. automodule:: megatron.model.enums

.. currentmodule:: megatron.model.enums


.. rubric:: Classes

.. autosummary::
    :toctree: .
    
    AttnMaskType
    
    AttnType
    
    LayerType
    
    ModelType
    
    PositionEmbeddingType