DeepSpeed

717c3020 - Nested zero.Init() and dynamically defined model class (#2989)

Nested zero.Init() and dynamically defined model class (#2989) * support nesting zero.Init() and dynamically defined module * throw an error if model class defined in zero.Init is not wrapped * fix check on new classes that are not wrapped in zero.Init() * add tests of nesting zero.Init() and dynamically defined classes * fix tests for zero.Init * fix style --------- Co-authored-by: Logan Adams <114770087+loadams@users.noreply.github.com> Co-authored-by: Olatunji Ruwase <olruwase@microsoft.com>