jinja2
joblib
numpy
safetensors
tokenizers>=0.20
tqdm

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

[distill]
torch
transformers<5.4.0
scikit-learn
skeletoken<0.5.0,>=0.4.1

[integration]
mteb

[onnx]
onnx
torch

[quantization]
scikit-learn

[tests]
scikit-learn
skops

[train]
torch
