torch>=2.0
numpy>=1.24

[all]
matplotlib>=3.7
transformers>=4.30
accelerate>=1.1.0

[dev]
pytest>=7.0
matplotlib>=3.7

[hf]
transformers>=4.30
accelerate>=1.1.0

[viz]
matplotlib>=3.7
