megatron.checkpointing
======================

.. rubric:: Description

.. automodule:: megatron.checkpointing

.. currentmodule:: megatron.checkpointing




.. rubric:: Functions

.. autosummary::
    :toctree: .
    
    check_checkpoint_args
    
    ensure_directory_exists
    
    find_checkpoint_rank_0
    
    fix_query_key_value_ordering
    
    get_checkpoint_name
    
    get_checkpoint_names
    
    get_checkpoint_tracker_filename
    
    get_checkpoint_version
    
    get_rng_state
    
    load_args_from_checkpoint
    
    load_biencoder_checkpoint
    
    load_checkpoint
    
    read_metadata
    
    save_checkpoint
    
    set_checkpoint_version