path	# lines of code
extract_metrics.py	157
picotron/pipeline_parallel/pp_communications.py	43
picotron/pipeline_parallel/pipeline_parallel.py	114
picotron/checkpoint.py	212
picotron/context_parallel/cp_communications.py	45
picotron/context_parallel/context_parallel.py	143
picotron/tensor_parallel/tensor_parallel.py	186
picotron/tensor_parallel/tp_communications.py	73
picotron/utils.py	78
picotron/process_group_manager.py	47
picotron/data_parallel/data_parallel.py	76
picotron/data_parallel/bucket.py	83
picotron/model.py	231
picotron/data.py	115
train.py	215
create_config.py	114
setup.py	10
submit_slurm_jobs.py	157