So one should explore solutions such as DeepSpeed, to train/fine-tune the model.