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/conf/config_schemas.py]" -- "[src/hyperpod_nemo_adapter/collections/model/sagemaker_base_model.py]" [label=" 7 ", penwidth="7", color="#00688bC9"];
    "[src/hyperpod_nemo_adapter/utils/callbacks/peft_ckpt_io.py]" -- "[src/hyperpod_nemo_adapter/collections/parts/fsdp_strategy.py]" [label=" 7 ", penwidth="7", color="#00688bC9"];
    "[src/hyperpod_nemo_adapter/conf/config_schemas.py]" -- "[src/hyperpod_nemo_adapter/collections/parts/fsdp_strategy.py]" [label=" 7 ", penwidth="7", color="#00688bC9"];
    "[src/hyperpod_nemo_adapter/version/__version__.py]" -- "[src/hyperpod_nemo_adapter/collections/parts/fsdp_strategy.py]" [label=" 7 ", penwidth="7", color="#00688bC9"];
    "[src/hyperpod_nemo_adapter/collections/parts/fsdp_strategy.py]" -- "[src/hyperpod_nemo_adapter/collections/model/sagemaker_base_model.py]" [label=" 7 ", penwidth="7", color="#00688bC9"];
    "[src/hyperpod_nemo_adapter/utils/fsdp_utils.py]" -- "[src/hyperpod_nemo_adapter/collections/parts/fsdp_strategy.py]" [label=" 7 ", penwidth="7", color="#00688bC9"];
    "[src/hyperpod_nemo_adapter/version/__version__.py]" -- "[src/hyperpod_nemo_adapter/conf/config_schemas.py]" [label=" 7 ", penwidth="7", color="#00688bC9"];
    "[src/hyperpod_nemo_adapter/version/__version__.py]" -- "[src/hyperpod_nemo_adapter/collections/model/sagemaker_base_model.py]" [label=" 7 ", penwidth="7", color="#00688bC9"];
    "[src/hyperpod_nemo_adapter/utils/fsdp_utils.py]" -- "[src/hyperpod_nemo_adapter/constants.py]" [label=" 5 ", penwidth="5", color="#00688bA5"];
    "[src/hyperpod_nemo_adapter/collections/model/nlp/__init__.py]" -- "[src/hyperpod_nemo_adapter/collections/parts/fsdp_strategy.py]" [label=" 5 ", penwidth="5", color="#00688bA5"];
    "[src/hyperpod_nemo_adapter/conf/config_schemas.py]" -- "[src/hyperpod_nemo_adapter/collections/parts/sagemaker_trainer_builder.py]" [label=" 5 ", penwidth="5", color="#00688bA5"];
    "[src/hyperpod_nemo_adapter/collections/parts/sagemaker_trainer_builder.py]" -- "[src/hyperpod_nemo_adapter/collections/model/sagemaker_base_model.py]" [label=" 5 ", penwidth="5", color="#00688bA5"];
    "[src/hyperpod_nemo_adapter/utils/callbacks/peft_ckpt_io.py]" -- "[src/hyperpod_nemo_adapter/collections/model/nlp/sagemaker_llama_model.py]" [label=" 5 ", penwidth="5", color="#00688bA5"];
    "[src/hyperpod_nemo_adapter/conf/config_schemas.py]" -- "[src/hyperpod_nemo_adapter/collections/data/__init__.py]" [label=" 5 ", penwidth="5", color="#00688bA5"];
    "[src/hyperpod_nemo_adapter/collections/model/nlp/__init__.py]" -- "[src/hyperpod_nemo_adapter/utils/callbacks/peft_ckpt_io.py]" [label=" 5 ", penwidth="5", color="#00688bA5"];
    "[src/hyperpod_nemo_adapter/collections/model/sagemaker_base_model.py]" -- "[src/hyperpod_nemo_adapter/collections/data/__init__.py]" [label=" 5 ", penwidth="5", color="#00688bA5"];
    "[src/hyperpod_nemo_adapter/utils/fsdp_utils.py]" -- "[src/hyperpod_nemo_adapter/utils/callbacks/peft_ckpt_io.py]" [label=" 5 ", penwidth="5", color="#00688bA5"];
    "[src/hyperpod_nemo_adapter/utils/callbacks/peft_ckpt_io.py]" -- "[src/hyperpod_nemo_adapter/collections/model/sagemaker_base_model.py]" [label=" 5 ", penwidth="5", color="#00688bA5"];
    "[src/hyperpod_nemo_adapter/utils/fsdp_utils.py]" -- "[src/hyperpod_nemo_adapter/collections/model/nlp/__init__.py]" [label=" 5 ", penwidth="5", color="#00688bA5"];
    "[src/hyperpod_nemo_adapter/constants.py]" -- "[src/hyperpod_nemo_adapter/collections/model/nlp/__init__.py]" [label=" 5 ", penwidth="5", color="#00688bA5"];
    "[src/hyperpod_nemo_adapter/collections/parts/sagemaker_trainer_builder.py]" -- "[src/hyperpod_nemo_adapter/collections/parts/fsdp_strategy.py]" [label=" 5 ", penwidth="5", color="#00688bA5"];
    "[src/hyperpod_nemo_adapter/version/__version__.py]" -- "[src/hyperpod_nemo_adapter/utils/callbacks/peft_ckpt_io.py]" [label=" 5 ", penwidth="5", color="#00688bA5"];
    "[src/hyperpod_nemo_adapter/utils/fsdp_utils.py]" -- "[src/hyperpod_nemo_adapter/collections/model/sagemaker_base_model.py]" [label=" 5 ", penwidth="5", color="#00688bA5"];
    "[src/hyperpod_nemo_adapter/collections/parts/fsdp_strategy.py]" -- "[src/hyperpod_nemo_adapter/collections/model/nlp/sagemaker_llama_model.py]" [label=" 5 ", penwidth="5", color="#00688bA5"];
    "[src/hyperpod_nemo_adapter/collections/parts/sagemaker_trainer_builder.py]" -- "[src/hyperpod_nemo_adapter/collections/data/__init__.py]" [label=" 5 ", penwidth="5", color="#00688bA5"];
    "[src/hyperpod_nemo_adapter/collections/parts/fsdp_strategy.py]" -- "[src/hyperpod_nemo_adapter/collections/data/__init__.py]" [label=" 5 ", penwidth="5", color="#00688bA5"];
    "[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/collections/model/nlp/sagemaker_deepseek_model.py]" -- "[src/hyperpod_nemo_adapter/constants.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/collections/model/nlp/sagemaker_deepseek_model.py]" -- "[src/hyperpod_nemo_adapter/utils/fsdp_utils.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/collections/model/nlp/sagemaker_qwen_model.py]" -- "[src/hyperpod_nemo_adapter/collections/model/nlp/sagemaker_deepseek_model.py]" [label=" 4 ", penwidth="4", color="#00688b93"];
    "[src/hyperpod_nemo_adapter/collections/model/nlp/sagemaker_deepseek_model.py]" -- "[src/hyperpod_nemo_adapter/collections/model/nlp/__init__.py]" [label=" 4 ", penwidth="4", color="#00688b93"];
    "[src/hyperpod_nemo_adapter/collections/model/nlp/sagemaker_qwen_model.py]" -- "[src/hyperpod_nemo_adapter/constants.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/utils/callbacks/sharded_ckpt_io.py]" -- "[src/hyperpod_nemo_adapter/utils/callbacks/full_ckpt_io.py]" [label=" 3 ", penwidth="3", color="#00688b82"];
    "[src/hyperpod_nemo_adapter/collections/model/nlp/sagemaker_llama_model.py]" -- "[src/hyperpod_nemo_adapter/collections/data/__init__.py]" [label=" 3 ", penwidth="3", color="#00688b82"];
    "[src/hyperpod_nemo_adapter/collections/parts/fsdp_strategy.py]" -- "[src/hyperpod_nemo_adapter/collections/data/hf_data_module.py]" [label=" 3 ", penwidth="3", color="#00688b82"];
    "[src/hyperpod_nemo_adapter/utils/callbacks/sharded_ckpt_io.py]" -- "[src/hyperpod_nemo_adapter/patches/patch_llama_flash_attn_cp.py]" [label=" 3 ", penwidth="3", color="#00688b82"];
    "[src/hyperpod_nemo_adapter/utils/callbacks/ckpt_io.py]" -- "[src/hyperpod_nemo_adapter/collections/model/sagemaker_base_model.py]" [label=" 3 ", penwidth="3", color="#00688b82"];
    "[src/hyperpod_nemo_adapter/collections/model/sagemaker_base_model.py]" -- "[src/hyperpod_nemo_adapter/collections/model/nlp/sagemaker_llama_model.py]" [label=" 3 ", penwidth="3", color="#00688b82"];
    "[src/hyperpod_nemo_adapter/collections/model/nlp/__init__.py]" -- "[src/hyperpod_nemo_adapter/conf/config_schemas.py]" [label=" 3 ", penwidth="3", color="#00688b82"];
    "[src/hyperpod_nemo_adapter/scripts/merge_peft_checkpoint.py]" -- "[src/hyperpod_nemo_adapter/conf/config_schemas.py]" [label=" 3 ", penwidth="3", color="#00688b82"];
    "[src/hyperpod_nemo_adapter/utils/callbacks/full_ckpt_io.py]" -- "[src/hyperpod_nemo_adapter/collections/model/sagemaker_base_model.py]" [label=" 3 ", penwidth="3", color="#00688b82"];
    "[src/hyperpod_nemo_adapter/collections/data/datasets/__init__.py]" -- "[src/hyperpod_nemo_adapter/collections/parts/sagemaker_trainer_builder.py]" [label=" 3 ", penwidth="3", color="#00688b82"];
    "[src/hyperpod_nemo_adapter/patches/patch_llama_flash_attn_cp.py]" -- "[src/hyperpod_nemo_adapter/version/__version__.py]" [label=" 3 ", penwidth="3", color="#00688b82"];
    "[src/hyperpod_nemo_adapter/utils/config_utils.py]" -- "[src/hyperpod_nemo_adapter/constants.py]" [label=" 3 ", penwidth="3", color="#00688b82"];
    "[src/hyperpod_nemo_adapter/patches/patch_llama_flash_attn_cp.py]" -- "[src/hyperpod_nemo_adapter/collections/model/sagemaker_base_model.py]" [label=" 3 ", penwidth="3", color="#00688b82"];
    "[src/hyperpod_nemo_adapter/utils/callbacks/ckpt_io.py]" -- "[src/hyperpod_nemo_adapter/collections/parts/fsdp_strategy.py]" [label=" 3 ", penwidth="3", color="#00688b82"];
    "[src/hyperpod_nemo_adapter/utils/callbacks/sharded_ckpt_io.py]" -- "[src/hyperpod_nemo_adapter/collections/model/sagemaker_base_model.py]" [label=" 3 ", penwidth="3", color="#00688b82"];

}