Automated cherry pick of #4281: WORLD_SIZE calculation for PyTorch Jobs

[JesseStutler]

Cherry pick of #4281 on release-1.10.

#4281: WORLD_SIZE calculation for PyTorch Jobs For details on the cherry pick process, see the cherry picks page.

NONE

[Monokaix]

/approve

[volcano-sh-bot]

[APPROVALNOTIFIER] This PR is APPROVED

This pull-request has been approved by: Monokaix

The full list of commands accepted by this bot can be found here.

The pull request process is described here

Needs approval from an approver in each of these files:

• ~~pkg/controllers/OWNERS~~ [Monokaix]

Approvers can indicate their approval by writing /approve in a comment Approvers can cancel approval by writing /approve cancel in a comment

[hwdef]

/lgtm