DeepSpeed

8aeab68d - add bwd support for list/dict types returned in fwd