torch>=2.0
transformers>=4.48
numpy>=1.24
pandas>=2.0
matplotlib>=3.7
scikit-learn>=1.3
optuna>=3.0
mlflow>=2.9
datasets>=2.14
tensorboard>=2.15
accelerate>=0.26
sentencepiece>=0.1.99

[dev]
pytest>=7.0
build
twine

[train]
