DeepSpeed

a64b0abb - fixed a typo; this was fixed before but seems like it has been lost in the refactor (#364)