megatron.utils.print_rank_0

megatron.utils.print_rank_0(message)

If distributed is initialized, print only on rank 0.