DeepSpeed

d21534d7 - add a user-friendly assert to check the zero-linear input dimensions