deepfold.distributed.model_parallel.gather_from_model_parallel_region

deepfold.distributed.model_parallel.gather_from_model_parallel_region(tensor: Tensor, dim: int, bwd: str = 'split') → Tensor