torchvision
infra
torchvision is the standard image toolbox for PyTorch, including ready-made transforms for altering images. We drew some of our test-time augmentations from it.
For engineers: PyTorch vision package; a source of the augmentation set we screened for TTA.