A Probabilistic U-Net for segmentation of ambiguous images implemented in PyTorch - View it on GitHub