torch
torchvision
pytest
timm
transformers
