transformers>=4.29
torch>=1.11
packaging
numpy
huggingface_hub>=0.8.0

[amd]
optimum-amd

[benchmark]
optuna
tqdm
scikit-learn
seqeval
torchvision
evaluate>=0.2.0

[dev]
pytest
accelerate
requests
parameterized
pytest-xdist
Pillow
sacremoses
torchvision
torchaudio
einops
timm
scikit-learn
sentencepiece
rjieba
hf_xet
black~=23.1
ruff==0.1.5

[doc-build]
accelerate

[furiosa]
optimum-furiosa

[graphcore]
optimum-graphcore

[habana]
optimum-habana>=1.17.0

[intel]
optimum-intel>=1.23.0

[ipex]
optimum-intel[ipex]>=1.23.0

[neural-compressor]
optimum-intel[neural-compressor]>=1.23.0

[nncf]
optimum-intel[nncf]>=1.23.0

[onnx]
optimum-onnx

[onnxruntime]
optimum-onnx[onnxruntime]

[onnxruntime-gpu]
optimum-onnx[onnxruntime-gpu]

[openvino]
optimum-intel[openvino]>=1.23.0

[quality]
black~=23.1
ruff==0.1.5

[quanto]
optimum-quanto>=0.2.4

[tests]
pytest
accelerate
requests
parameterized
pytest-xdist
Pillow
sacremoses
torchvision
torchaudio
einops
timm
scikit-learn
sentencepiece
rjieba
hf_xet
