step3_dataloader/model.py
step5_data_parallel_naive/model.py

step3_dataloader/dataloader.py
step5_data_parallel_naive/dataloader.py

step3_dataloader/train.py
step5_data_parallel_naive/train.py

step3_dataloader/process_group_manager.py
step5_data_parallel_naive/process_group_manager.py

step3_dataloader/utils.py
step5_data_parallel_naive/utils.py