jinja2
joblib
numpy
rich
safetensors
setuptools
tokenizers>=0.20
tqdm

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

[distill]
torch
transformers
scikit-learn
skeletoken>=0.3.2

[inference]
scikit-learn
skops

[onnx]
onnx
torch

[quantization]
scikit-learn

[tokenizer]
transformers

[train]
torch
lightning
scikit-learn
skops
