transformers>=4.30.0
datasets>=2.0.0
torch>=1.10.0
accelerate>=0.20.0
scikit-learn>=1.0.0
numpy>=1.20.0
pandas>=1.0.0
matplotlib>=3.0.0
seaborn>=0.11.0
nltk>=3.6.0
optuna>=3.0.0
psutil>=5.0.0
pynvml>=11.0.0
scipy>=1.5.0
tqdm>=4.0.0
tensorboard>=2.5.0

[api-serving]
fastapi
uvicorn[standard]

[automl]
xgboost

[dev]
pytest>=7.0
flake8
black

[optimization]
onnx>=1.10.0
onnxruntime>=1.8.0
optimum>=1.0.0

[peft]
peft>=0.5.0

[xai]
lime
shap
