timm
torch
torchvision

[test]
pytest
