pytest
torchvision
transformers
