

TITLE
Use DS callable API to allow hf_scheduler + ds_optimizer

COMMENTS
1

REACTIONS
+1: 0
-1: 0
laugh: 0
hooray: 0
heart: 1
rocket: 0
eyes: 0

BODY
This PR:

- Used the (new) Callable api of deepspeed.initialize() to enable combining hf schedulers with deepspeed optimizers.  
- `create_scheduler` now has an optional `optimizer` arg
- Updates relevant unit test.

Blocking events: All unblocked now.

- [x] depends on deepspeed PR [1316](https://github.com/microsoft/DeepSpeed/pull/1316).
- [x] needs new deepspeed version after PR is merged and need to update the dependencies when that happens.

deepspeed: @stas00. 
