DeepSpeed

8c477624 - Add check to exempt 'torch.cuda' from pre-commit formatting checks - this is needed to determine which torch accelerator, cpu or cuda we are using