step1_modeling/model.py
step2_process_group_manager/model.py

step1_modeling/utils.py
step2_process_group_manager/utils.py

step1_modeling/train.py
step2_process_group_manager/train.py