torch>=1.10.0
numpy>=1.20.0

[benchmarks]
torchvision>=0.11.0
matplotlib>=3.5.0
transformers>=4.20.0
datasets>=2.0.0

[dev]
pytest>=7.0.0
pytest-cov>=3.0.0
black>=22.0.0
flake8>=4.0.0
mypy>=0.950
