DeepSpeed

a298a43a - [zero-3] Handle forward parameter return correctly in nested cases (#2642)

[zero-3] Handle forward parameter return correctly in nested cases (#2642) Co-authored-by: Stas Bekman <stas00@users.noreply.github.com> Co-authored-by: Olatunji Ruwase <olruwase@microsoft.com> Co-authored-by: Jeff Rasley <jerasley@microsoft.com>