torch
torchvision
transformers[torch]

[dev]
pytest
