DeepSpeed

Refactor the injection to accept policy when using kernels

#4267

Open