DeepSpeed

203f5b71 - Update container to newer version rather than updating specific packages