I don’t think that there are any official implementations available yet, but you can take a look at some of these repos for semantic segmentation implementations using pytorch:

https://github.com/bodokaiser/piwise 1.7k
https://github.com/ycszen/pytorch-ss 1.1k
https://github.com/mattmacy/vnet.pytorch 954