megatron.model.positional\_embeddings
=====================================

.. rubric:: Description

.. automodule:: megatron.model.positional_embeddings

.. currentmodule:: megatron.model.positional_embeddings




.. rubric:: Functions

.. autosummary::
    :toctree: .
    
    apply_rotary_emb
    
    precompute_freqs_cis
    
    reshape_for_broadcast