DeepSpeed

d2b1b790 - remove deepcopy to avoid incompatible torch.jit.script modules.