DeepSpeed

e8136931 - [minor] improve code quality and readablilty (#5011)

[minor] improve code quality and readablilty (#5011) as title --------- Signed-off-by: byhsu <byhsu@linkedin.com> Co-authored-by: byhsu <byhsu@linkedin.com>