megatron.core.parallel_state.get_pipeline_model_parallel_rank#

megatron.core.parallel_state.get_pipeline_model_parallel_rank()#

Return my rank for the pipeline model parallel group.