DeepSpeed

03629e3f - Fix adam leak (#918)

Fix adam leak (#918) Co-authored-by: Stas Bekman <stas@stason.org> Co-authored-by: Olatunji Ruwase <olruwase@microsoft.com> Co-authored-by: Reza Yazdani <reyazda@microsoft.com>