src/transformers/sagemaker/training_args_sm.py/SageMakerTrainingArguments/world_size
class SageMakerTrainingArguments:    def world_size(self):
        if is_sagemaker_model_parallel_available():
            return smp.dp_size()

        return super().world_size