pytest
torchvision
transformers