megatron.model.utils
====================

.. rubric:: Description

.. automodule:: megatron.model.utils

.. currentmodule:: megatron.model.utils




.. rubric:: Functions

.. autosummary::
    :toctree: .
    
    attention_mask_func
    
    get_linear_layer
    
    init_method_normal
    
    scaled_init_method_normal