ModuleNotFoundError: No module named 'megatron.core.dist_checkpointing'
I found the issue. Instead of installing from `main` branch, it wokred when I reinstall from `r2.*`
**ModuleNotFoundError: No module named 'megatron.core.dist_checkpointing'**

I am getting `ModuleNotFoundError: No module named 'megatron.core.dist_checkpointing'`


**Steps/Code to reproduce bug**
```bash
python /home/ubuntu/NeMo/scripts/checkpoint_converters/convert_llama_hf_to_nemo.py     --input_name_or_path meta-llama/Llama-3.2-1B-Instruct     --output_path Llama-3.2-1B-Instruct.nemo     --precision=bf16

Traceback (most recent call last):
  File "/home/ubuntu/NeMo/scripts/checkpoint_converters/convert_llama_hf_to_nemo.py", line 34, in <module>
    from nemo.collections.nlp.models.language_modeling.megatron_gpt_model import MegatronGPTModel
  File "/home/ubuntu/NeMo/nemo/collections/nlp/__init__.py", line 15, in <module>
    from nemo.collections.nlp import data, losses, models, modules
  File "/home/ubuntu/NeMo/nemo/collections/nlp/models/__init__.py", line 16, in <module>
    from nemo.collections.nlp.models.duplex_text_normalization import (
  File "/home/ubuntu/NeMo/nemo/collections/nlp/models/duplex_text_normalization/__init__.py", line 15, in <module>
    from nemo.collections.nlp.models.duplex_text_normalization.duplex_decoder import DuplexDecoderModel
  File "/home/ubuntu/NeMo/nemo/collections/nlp/models/duplex_text_normalization/duplex_decoder.py", line 34, in <module>
    from nemo.collections.nlp.models.nlp_model import NLPModel
  File "/home/ubuntu/NeMo/nemo/collections/nlp/models/nlp_model.py", line 40, in <module>
    from nemo.collections.nlp.parts.nlp_overrides import NLPSaveRestoreConnector
  File "/home/ubuntu/NeMo/nemo/collections/nlp/parts/nlp_overrides.py", line 68, in <module>
    from nemo.collections.nlp.modules.common.megatron.transformer import AutocastTransformerLayer, ParallelTransformerLayer
  File "/home/ubuntu/NeMo/nemo/collections/nlp/modules/common/megatron/transformer.py", line 28, in <module>
    from nemo.collections.nlp.modules.common.megatron.adapters.parallel_adapters import (
  File "/home/ubuntu/NeMo/nemo/collections/nlp/modules/common/megatron/adapters/parallel_adapters.py", line 27, in <module>
    from megatron.core.dist_checkpointing.mapping import ShardedStateDict
ModuleNotFoundError: No module named 'megatron.core.dist_checkpointing'
```

**Expected behavior**

A clear and concise description of what you expected to happen.

**Environment overview (please complete the following information)**

 - Environment location: Cloud(LambdaLabs)
 - Method of NeMo install: from source

**Environment details**

If NVIDIA docker image is used you don't need to specify these.
Otherwise, please provide:
- OS version: debian
- PyTorch version: 2.4.1+cu124
- Python version: Python 3.10.16

**Additional context**
GPU model: A100 SXM4
