'''Copyright The Microsoft DeepSpeed Team'''