DeepSpeed

aea67957 - Revert "Update container to newer version rather than updating specific packages"