DeepSpeed

Improve leaf module interface (enable via config, relax matching criteria, add document, etc.)

#7604

Merged