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.3.3

[inference]
scikit-learn
skops

[onnx]
onnx
torch

[quantization]
scikit-learn

[train]
torch
lightning
scikit-learn
skops
