step5_data_parallel_naive/tensor_parallel.py
step6_data_parallel_bucket/tensor_parallel.py

step5_data_parallel_naive/model.py
step6_data_parallel_bucket/model.py

step5_data_parallel_naive/dataloader.py
step6_data_parallel_bucket/dataloader.py

step5_data_parallel_naive/train.py
step6_data_parallel_bucket/train.py

step5_data_parallel_naive/process_group_manager.py
step6_data_parallel_bucket/process_group_manager.py

step5_data_parallel_naive/utils.py
step6_data_parallel_bucket/utils.py

step5_data_parallel_naive/data_parallel.py
step6_data_parallel_bucket/data_parallel.py