megatron.optimizer.clip_grads.count_zeros_fp32# megatron.optimizer.clip_grads.count_zeros_fp32(parameters, model_parallel_group)#