Vision Transformers for Segmentation using PyTorch