transformers
torch

[dev]
pytest
flake8
mypy
black
isort
