megatron.checkpointing.fix\_query\_key\_value\_ordering
=======================================================

.. currentmodule:: megatron.checkpointing

.. autofunction:: fix_query_key_value_ordering