megatron.checkpointing.get\_rng\_state ====================================== .. currentmodule:: megatron.checkpointing .. autofunction:: get_rng_state