numpy>=1.21
pyyaml>=6.0

[all]
torch>=2.0
transformers>=4.30
scikit-learn>=1.0
tqdm
sentencepiece

[dev]
pytest>=7.0
pytest-cov
jupyter

[input-gate]
torch>=2.0
transformers>=4.30
scikit-learn>=1.0
tqdm
sentencepiece

[output-gate]
transformers>=4.30
torch>=2.0
