torch>=2.1.0
transformers>=4.40.0
numpy>=1.24.0
matplotlib>=3.7.0

[dev]
pytest>=7.4.0
