DeepSpeed
Move param_shapes to model files
#1732
Merged

Loading