[DeepSpeed] simplify init
This PR simplifies `deepspeed.initialize` setup thanks to this PR https://github.com/microsoft/DeepSpeed/pull/825  

We already have the required version that includes that change in DeepSpeed in place.

@sgugger 