torch>=2.0.0
numpy>=1.24.0
scipy>=1.10.0
scikit-learn>=1.3.0
matplotlib>=3.7.0
pandas>=2.0.0

[all]
transformers>=4.30.0
datasets>=2.14.0
sentencepiece>=0.1.99
tiktoken>=0.5.0
pytest>=7.4.0
pytest-cov>=4.1.0
black>=23.0.0
ruff>=0.1.0
mypy>=1.5.0

[dev]
pytest>=7.4.0
pytest-cov>=4.1.0
black>=23.0.0
ruff>=0.1.0
mypy>=1.5.0

[nlp]
transformers>=4.30.0
datasets>=2.14.0
sentencepiece>=0.1.99
tiktoken>=0.5.0
