megatron.text\_generation.generation
====================================

.. rubric:: Description

.. automodule:: megatron.text_generation.generation

.. currentmodule:: megatron.text_generation.generation




.. rubric:: Functions

.. autosummary::
    :toctree: .
    
    beam_search_and_return_on_first_stage
    
    generate_tokens_probs_and_return_on_first_stage
    
    score_and_return_on_first_stage