Accelerate Large Model Training using DeepSpeed