torch
transformers
numpy
tqdm
matplotlib

[dev]
pip-tools
pytest
