DeepSpeed

f3ffc7ab - Apply memory bloat fix

Apply memory bloat fix Backport memory bloat fix from https://github.com/microsoft/DeepSpeed/commit/d24d3de901fa8b110923f1ee9c2ccf0efd56d31f