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"
    ]




    "[train.py]" -- "[picotron/utils.py]" [label=" 4 ", penwidth="4", color="#00688b93"];
    "[train.py]" -- "[create_config.py]" [label=" 4 ", penwidth="4", color="#00688b93"];
    "[create_config.py]" -- "[picotron/utils.py]" [label=" 3 ", penwidth="3", color="#00688b82"];
    "[train.py]" -- "[picotron/data_parallel/data_parallel.py]" [label=" 3 ", penwidth="3", color="#00688b82"];
    "[train.py]" -- "[picotron/model.py]" [label=" 3 ", penwidth="3", color="#00688b82"];
    "[picotron/data_parallel/data_parallel.py]" -- "[picotron/model.py]" [label=" 3 ", penwidth="3", color="#00688b82"];
    "[train.py]" -- "[picotron/data_parallel/bucket.py]" [label=" 3 ", penwidth="3", color="#00688b82"];
    "[picotron/data_parallel/data_parallel.py]" -- "[picotron/data_parallel/bucket.py]" [label=" 3 ", penwidth="3", color="#00688b82"];
    "[picotron/data_parallel/bucket.py]" -- "[picotron/model.py]" [label=" 3 ", penwidth="3", color="#00688b82"];
    "[picotron/model.py]" -- "[picotron/checkpoint.py]" [label=" 2 ", penwidth="2", color="#00688b70"];
    "[picotron/data_parallel/data_parallel.py]" -- "[picotron/pipeline_parallel/pipeline_parallel.py]" [label=" 1 ", penwidth="1", color="#00688b5E"];
    "[picotron/data_parallel/data_parallel.py]" -- "[picotron/utils.py]" [label=" 1 ", penwidth="1", color="#00688b5E"];
    "[picotron/data_parallel/data_parallel.py]" -- "[create_config.py]" [label=" 1 ", penwidth="1", color="#00688b5E"];
    "[picotron/model.py]" -- "[picotron/utils.py]" [label=" 1 ", penwidth="1", color="#00688b5E"];
    "[picotron/model.py]" -- "[picotron/pipeline_parallel/pipeline_parallel.py]" [label=" 1 ", penwidth="1", color="#00688b5E"];
    "[picotron/utils.py]" -- "[picotron/pipeline_parallel/pipeline_parallel.py]" [label=" 1 ", penwidth="1", color="#00688b5E"];
    "[create_config.py]" -- "[picotron/model.py]" [label=" 1 ", penwidth="1", color="#00688b5E"];
    "[picotron/data_parallel/bucket.py]" -- "[picotron/pipeline_parallel/pipeline_parallel.py]" [label=" 1 ", penwidth="1", color="#00688b5E"];
    "[train.py]" -- "[picotron/pipeline_parallel/pipeline_parallel.py]" [label=" 1 ", penwidth="1", color="#00688b5E"];
    "[picotron/data_parallel/bucket.py]" -- "[create_config.py]" [label=" 1 ", penwidth="1", color="#00688b5E"];
    "[create_config.py]" -- "[picotron/pipeline_parallel/pipeline_parallel.py]" [label=" 1 ", penwidth="1", color="#00688b5E"];
    "[picotron/data_parallel/bucket.py]" -- "[picotron/utils.py]" [label=" 1 ", penwidth="1", color="#00688b5E"];

}