DeepSpeed
Fixes code and tests skipping/asserting incorrectly on torch 2+.
#3136
Merged
Merged