torch>=1.9.0
transformers>=4.20.0
sentence-transformers>=2.2.0
scikit-learn>=1.0.0
numpy>=1.21.0
pandas>=1.3.0
tqdm>=4.62.0

[dev]
pytest>=6.0
pytest-cov>=2.0
black>=21.0
flake8>=3.8
mypy>=0.800

[gpu]
torch>=1.9.0
torchvision>=0.10.0
