step7_pipeline_parallel_afab/tensor_parallel.py
step8_pipeline_parallel_1f1b/tensor_parallel.py

step7_pipeline_parallel_afab/data_parallel.py
step8_pipeline_parallel_1f1b/data_parallel.py

step7_pipeline_parallel_afab/model.py
step8_pipeline_parallel_1f1b/model.py

step7_pipeline_parallel_afab/train.py
step8_pipeline_parallel_1f1b/train.py

step7_pipeline_parallel_afab/dataloader.py
step8_pipeline_parallel_1f1b/dataloader.py

step7_pipeline_parallel_afab/pipeline_parallel.py
step8_pipeline_parallel_1f1b/pipeline_parallel.py

step7_pipeline_parallel_afab/process_group_manager.py
step8_pipeline_parallel_1f1b/process_group_manager.py

step7_pipeline_parallel_afab/utils.py
step8_pipeline_parallel_1f1b/utils.py