jinja2
joblib
numpy
safetensors
tokenizers>=0.20
tqdm
huggingface-hub>=1.0.0

[dev]
ipython
mypy
pre-commit
pytest
pytest-cov
ruff
setuptools

[distill]
torch
transformers<5.4.0
skeletoken<0.7.0,>=0.6.0

[integration]
mteb

[onnx]
onnx
torch
onnxruntime
onnxscript
skeletoken<0.7.0,>=0.6.0

[quantization]
scikit-learn

[tests]
scikit-learn
skops

[train]
torch
datasets>=4.0.0
