megatron.core.parallel_state.is_rank_in_embedding_group#

megatron.core.parallel_state.is_rank_in_embedding_group(ignore_virtual=False)#

Return true if current rank is in embedding group, False otherwise.