DeepSpeed

4fd79205 - Allow to compile collective for PT>2.3 (#6899)

Allow to compile collective for PT>2.3 (#6899) Allow to compile collective for PT>2.3 commit re-uploaded due to github CI issue originally uploaded by @nelyahu