Remove deprecated torch._six imports

[pruthvistony]

• It is modified version of PR#2863 https://github.com/microsoft/DeepSpeed/pull/2863
• Fixes https://ontrack-internal.amd.com/browse/SWDEV-396029 and few others.