DeepSpeed

79643351 - Add generic load checkpoint to replace_layer, still need to add to containers