DeepSpeed

d5ad09d0 - Add conditional on torch version for scaled_dot_product_attention