torchmaxflow (pypi): torchmaxflow: Max-flow/Min-cut in PyTorch for 2D images and 3D volumes
Keywords: python|pytorch|segmentation