DeepSpeed

a52cbf80 - [zero-3] add bwd support for list/dict types returned in fwd (#1857)