DeepSpeed

274c375c - Support Callable type for client optimizer and lr_scheduler (#1316)

Support Callable type for client optimizer and lr_scheduler (#1316) * Callable option for optimizer and scheduler * Add unit test * Formatting * Disable debug prints * Use base optimizer to construct lr scheduler * Formatting * Remove dead import