step2_process_group_manager/model.py (1 line):
	- line 97: #TODO: dont do single line operations as it is harder to debug


step7_pipeline_parallel_afab/model.py (1 line):
	- line 101: #TODO: dont do single line operations as it is harder to debug


step4_tensor_parallel/model.py (1 line):
	- line 101: #TODO: dont do single line operations as it is harder to debug


step8_pipeline_parallel_1f1b/model.py (1 line):
	- line 101: #TODO: dont do single line operations as it is harder to debug


step3_dataloader/model.py (1 line):
	- line 97: #TODO: dont do single line operations as it is harder to debug


step5_data_parallel_naive/model.py (1 line):
	- line 101: #TODO: dont do single line operations as it is harder to debug


step1_modeling/model.py (1 line):
	- line 97: #TODO: dont do single line operations as it is harder to debug


step6_data_parallel_bucket/model.py (1 line):
	- line 101: #TODO: dont do single line operations as it is harder to debug