torch>=2.0.0
numpy>=1.21.0
scikit-learn>=1.0.0
torchvision>=0.15.0
datasets>=2.0.0
transformers>=4.0.0
matplotlib>=3.5.0
tqdm>=4.60.0

[dev]
pytest>=7.0
twine>=4.0.2
build>=0.10.0
nbformat>=5.0.0
