DeepSpeed

981bc7d4 - Move workspace memory-allocation to PyTorch (#661)

Move workspace memory-allocation to PyTorch (#661) * move workspace memory-allocation to PyTorch * refine the code based on the comments * remove unnecessary options * remove bsz from set_seq_len function