step1_modeling/model.py
step8_pipeline_parallel_1f1b/model.py

step1_modeling/utils.py
step8_pipeline_parallel_1f1b/utils.py

step1_modeling/train.py
step8_pipeline_parallel_1f1b/train.py