DeepSpeed

8950731e - guard for when cuda is not available & test with cpu-only environment