Torch does not seem to support torch._six anymore and it has been removed.
Refer - pytorch/pytorch#94709

DeepSpeed still has dependency on it. Example in runtime/utils.py

from torch._six import inf

Torch version to be precise - 2.0.0a0+git5a54537