graph G {
    compound="true"
    rankdir="TB"
    bgcolor="white"
    fontname="Tahoma"

    node [
        fixedsize="false"
        fontname="Tahoma"
        color="white"
        fillcolor="deepskyblue2"
        fontcolor="black"
        shape="box"
        style="filled"
        penwidth="1.0"
    ]
    edge [
        fontname="Arial"
        color="#00688b"
        fontcolor="black"
        fontsize="12"
        arrowsize="0.5"
        penwidth="1.0"
    ]




    "[src/hyperpod_nemo_adapter/utils/fsdp_utils.py]" -- "[src/hyperpod_nemo_adapter/collections/parts/fsdp_strategy.py]" [label=" 6 ", penwidth="6", color="#00688bB7"];
    "[src/hyperpod_nemo_adapter/version/__version__.py]" -- "[src/hyperpod_nemo_adapter/conf/config_schemas.py]" [label=" 6 ", penwidth="6", color="#00688bB7"];
    "[src/hyperpod_nemo_adapter/conf/config_schemas.py]" -- "[src/hyperpod_nemo_adapter/collections/parts/fsdp_strategy.py]" [label=" 6 ", penwidth="6", color="#00688bB7"];
    "[src/hyperpod_nemo_adapter/conf/config_schemas.py]" -- "[src/hyperpod_nemo_adapter/collections/model/sagemaker_base_model.py]" [label=" 6 ", penwidth="6", color="#00688bB7"];
    "[src/hyperpod_nemo_adapter/version/__version__.py]" -- "[src/hyperpod_nemo_adapter/collections/parts/fsdp_strategy.py]" [label=" 6 ", penwidth="6", color="#00688bB7"];
    "[src/hyperpod_nemo_adapter/version/__version__.py]" -- "[src/hyperpod_nemo_adapter/collections/model/sagemaker_base_model.py]" [label=" 6 ", penwidth="6", color="#00688bB7"];
    "[src/hyperpod_nemo_adapter/collections/parts/fsdp_strategy.py]" -- "[src/hyperpod_nemo_adapter/collections/model/sagemaker_base_model.py]" [label=" 6 ", penwidth="6", color="#00688bB7"];
    "[src/hyperpod_nemo_adapter/utils/callbacks/peft_ckpt_io.py]" -- "[src/hyperpod_nemo_adapter/collections/parts/fsdp_strategy.py]" [label=" 6 ", penwidth="6", color="#00688bB7"];
    "[src/hyperpod_nemo_adapter/conf/config_schemas.py]" -- "[src/hyperpod_nemo_adapter/collections/parts/sagemaker_trainer_builder.py]" [label=" 4 ", penwidth="4", color="#00688b93"];
    "[src/hyperpod_nemo_adapter/utils/callbacks/peft_ckpt_io.py]" -- "[src/hyperpod_nemo_adapter/collections/model/nlp/sagemaker_llama_model.py]" [label=" 4 ", penwidth="4", color="#00688b93"];
    "[src/hyperpod_nemo_adapter/collections/parts/fsdp_strategy.py]" -- "[src/hyperpod_nemo_adapter/collections/model/nlp/sagemaker_llama_model.py]" [label=" 4 ", penwidth="4", color="#00688b93"];
    "[src/hyperpod_nemo_adapter/collections/model/nlp/sagemaker_deepseek_model.py]" -- "[src/hyperpod_nemo_adapter/version/__version__.py]" [label=" 4 ", penwidth="4", color="#00688b93"];
    "[src/hyperpod_nemo_adapter/scripts/fp8_cast_bf16.py]" -- "[src/hyperpod_nemo_adapter/collections/parts/fsdp_strategy.py]" [label=" 4 ", penwidth="4", color="#00688b93"];
    "[src/hyperpod_nemo_adapter/scripts/fp8_cast_bf16.py]" -- "[src/hyperpod_nemo_adapter/collections/model/sagemaker_base_model.py]" [label=" 4 ", penwidth="4", color="#00688b93"];
    "[src/hyperpod_nemo_adapter/utils/fsdp_utils.py]" -- "[src/hyperpod_nemo_adapter/collections/model/nlp/__init__.py]" [label=" 4 ", penwidth="4", color="#00688b93"];
    "[src/hyperpod_nemo_adapter/collections/parts/fsdp_strategy.py]" -- "[src/hyperpod_nemo_adapter/collections/data/__init__.py]" [label=" 4 ", penwidth="4", color="#00688b93"];
    "[src/hyperpod_nemo_adapter/collections/parts/sagemaker_trainer_builder.py]" -- "[src/hyperpod_nemo_adapter/collections/model/sagemaker_base_model.py]" [label=" 4 ", penwidth="4", color="#00688b93"];
    "[src/hyperpod_nemo_adapter/version/__version__.py]" -- "[src/hyperpod_nemo_adapter/utils/callbacks/peft_ckpt_io.py]" [label=" 4 ", penwidth="4", color="#00688b93"];
    "[src/hyperpod_nemo_adapter/conf/config_schemas.py]" -- "[src/hyperpod_nemo_adapter/collections/data/__init__.py]" [label=" 4 ", penwidth="4", color="#00688b93"];
    "[src/hyperpod_nemo_adapter/collections/parts/sagemaker_trainer_builder.py]" -- "[src/hyperpod_nemo_adapter/collections/parts/fsdp_strategy.py]" [label=" 4 ", penwidth="4", color="#00688b93"];
    "[src/hyperpod_nemo_adapter/utils/callbacks/peft_ckpt_io.py]" -- "[src/hyperpod_nemo_adapter/collections/model/sagemaker_base_model.py]" [label=" 4 ", penwidth="4", color="#00688b93"];
    "[src/hyperpod_nemo_adapter/collections/parts/sagemaker_trainer_builder.py]" -- "[src/hyperpod_nemo_adapter/collections/data/__init__.py]" [label=" 4 ", penwidth="4", color="#00688b93"];
    "[src/hyperpod_nemo_adapter/collections/model/nlp/__init__.py]" -- "[src/hyperpod_nemo_adapter/utils/callbacks/peft_ckpt_io.py]" [label=" 4 ", penwidth="4", color="#00688b93"];
    "[src/hyperpod_nemo_adapter/collections/model/sagemaker_base_model.py]" -- "[src/hyperpod_nemo_adapter/collections/data/__init__.py]" [label=" 4 ", penwidth="4", color="#00688b93"];
    "[src/hyperpod_nemo_adapter/utils/fsdp_utils.py]" -- "[src/hyperpod_nemo_adapter/collections/model/sagemaker_base_model.py]" [label=" 4 ", penwidth="4", color="#00688b93"];
    "[src/hyperpod_nemo_adapter/collections/model/nlp/__init__.py]" -- "[src/hyperpod_nemo_adapter/collections/parts/fsdp_strategy.py]" [label=" 4 ", penwidth="4", color="#00688b93"];
    "[src/hyperpod_nemo_adapter/scripts/fp8_cast_bf16.py]" -- "[src/hyperpod_nemo_adapter/utils/fsdp_utils.py]" [label=" 4 ", penwidth="4", color="#00688b93"];
    "[src/hyperpod_nemo_adapter/utils/fsdp_utils.py]" -- "[src/hyperpod_nemo_adapter/utils/callbacks/peft_ckpt_io.py]" [label=" 4 ", penwidth="4", color="#00688b93"];
    "[src/hyperpod_nemo_adapter/scripts/casting_utils/kernel.py]" -- "[src/hyperpod_nemo_adapter/collections/data/__init__.py]" [label=" 2 ", penwidth="2", color="#00688b70"];
    "[src/hyperpod_nemo_adapter/collections/model/nlp/custom_models/configuration_deepseek.py]" -- "[src/hyperpod_nemo_adapter/collections/model/sagemaker_base_model.py]" [label=" 2 ", penwidth="2", color="#00688b70"];
    "[src/hyperpod_nemo_adapter/collections/data/datasets/__init__.py]" -- "[src/hyperpod_nemo_adapter/collections/data/__init__.py]" [label=" 2 ", penwidth="2", color="#00688b70"];
    "[src/hyperpod_nemo_adapter/constants.py]" -- "[src/hyperpod_nemo_adapter/collections/model/nlp/sagemaker_llama_model.py]" [label=" 2 ", penwidth="2", color="#00688b70"];
    "[src/hyperpod_nemo_adapter/collections/data/datasets/hf_dpo_dataset.py]" -- "[src/hyperpod_nemo_adapter/collections/parts/sagemaker_trainer_builder.py]" [label=" 2 ", penwidth="2", color="#00688b70"];
    "[src/hyperpod_nemo_adapter/collections/data/hf_dpo_data_module.py]" -- "[src/hyperpod_nemo_adapter/conf/config_schemas.py]" [label=" 2 ", penwidth="2", color="#00688b70"];
    "[src/hyperpod_nemo_adapter/scripts/casting_utils/kernel.py]" -- "[src/hyperpod_nemo_adapter/collections/parts/fsdp_strategy.py]" [label=" 2 ", penwidth="2", color="#00688b70"];
    "[src/hyperpod_nemo_adapter/constants.py]" -- "[src/hyperpod_nemo_adapter/utils/callbacks/peft_ckpt_io.py]" [label=" 2 ", penwidth="2", color="#00688b70"];
    "[src/hyperpod_nemo_adapter/collections/model/sagemaker_base_model.py]" -- "[src/hyperpod_nemo_adapter/collections/model/nlp/sagemaker_llama_model.py]" [label=" 2 ", penwidth="2", color="#00688b70"];
    "[src/hyperpod_nemo_adapter/collections/data/dummy_dpo_data_module.py]" -- "[src/hyperpod_nemo_adapter/collections/parts/sagemaker_trainer_builder.py]" [label=" 2 ", penwidth="2", color="#00688b70"];
    "[src/hyperpod_nemo_adapter/collections/parts/sagemaker_trainer_builder.py]" -- "[src/hyperpod_nemo_adapter/collections/data/hf_data_module.py]" [label=" 2 ", penwidth="2", color="#00688b70"];
    "[src/hyperpod_nemo_adapter/collections/model/nlp/custom_models/modeling_deepseek.py]" -- "[src/hyperpod_nemo_adapter/utils/fsdp_utils.py]" [label=" 2 ", penwidth="2", color="#00688b70"];
    "[src/hyperpod_nemo_adapter/patches/patch_llama_flash_attn_cp.py]" -- "[src/hyperpod_nemo_adapter/collections/model/sagemaker_base_model.py]" [label=" 2 ", penwidth="2", color="#00688b70"];
    "[src/hyperpod_nemo_adapter/utils/fsdp_utils.py]" -- "[src/hyperpod_nemo_adapter/version/__version__.py]" [label=" 2 ", penwidth="2", color="#00688b70"];
    "[src/hyperpod_nemo_adapter/utils/callbacks/sharded_ckpt_io.py]" -- "[src/hyperpod_nemo_adapter/utils/callbacks/full_ckpt_io.py]" [label=" 2 ", penwidth="2", color="#00688b70"];
    "[src/hyperpod_nemo_adapter/collections/model/nlp/custom_models/modeling_deepseek.py]" -- "[src/hyperpod_nemo_adapter/collections/model/nlp/sagemaker_deepseek_model.py]" [label=" 2 ", penwidth="2", color="#00688b70"];
    "[src/hyperpod_nemo_adapter/collections/data/hf_data_module.py]" -- "[src/hyperpod_nemo_adapter/collections/data/__init__.py]" [label=" 2 ", penwidth="2", color="#00688b70"];
    "[src/hyperpod_nemo_adapter/scripts/fp8_cast_bf16.py]" -- "[src/hyperpod_nemo_adapter/scripts/casting_utils/kernel.py]" [label=" 2 ", penwidth="2", color="#00688b70"];
    "[src/hyperpod_nemo_adapter/collections/model/nlp/sagemaker_llama_model.py]" -- "[src/hyperpod_nemo_adapter/collections/data/__init__.py]" [label=" 2 ", penwidth="2", color="#00688b70"];
    "[src/hyperpod_nemo_adapter/collections/data/datasets/dummy_dpo_dataset.py]" -- "[src/hyperpod_nemo_adapter/collections/parts/fsdp_strategy.py]" [label=" 2 ", penwidth="2", color="#00688b70"];
    "[src/hyperpod_nemo_adapter/utils/dpo_utils.py]" -- "[src/hyperpod_nemo_adapter/collections/data/datasets/hf_dpo_dataset.py]" [label=" 2 ", penwidth="2", color="#00688b70"];
    "[src/hyperpod_nemo_adapter/scripts/fp8_cast_bf16.py]" -- "[src/hyperpod_nemo_adapter/collections/data/datasets/dummy_dpo_dataset.py]" [label=" 2 ", penwidth="2", color="#00688b70"];

}