coloredlogs
sympy
transformers[sentencepiece]>=4.26.0
torch>=1.9
torchvision
packaging
numpy
huggingface_hub>=0.8.0
datasets

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

[dev]
pytest
requests
parameterized
pytest-xdist
Pillow
sacremoses
torchvision
diffusers>=0.17.0
torchaudio
black~=23.1
ruff>=0.0.241

[exporters]
onnx<1.14.0
onnxruntime
timm

[exporters-gpu]
onnx<1.14.0
onnxruntime-gpu
timm

[exporters-tf]
tensorflow<2.11,>=2.4
tf2onnx
onnx
onnxruntime
timm
h5py
numpy<1.24.0

[graphcore]
optimum-graphcore

[habana]
transformers<4.29.0
optimum-habana

[intel]
optimum-intel

[neural-compressor]
optimum-intel[neural-compressor]

[neuron]
optimum-neuron[neuron]

[neuronx]
optimum-neuron[neuronx]

[nncf]
optimum-intel[nncf]

[onnxruntime]
onnx<1.14.0
onnxruntime>=1.9.0
datasets>=1.2.1
evaluate
protobuf>=3.20.1

[onnxruntime-gpu]
onnx<1.14.0
onnxruntime-gpu>=1.9.0
datasets>=1.2.1
evaluate
protobuf>=3.20.1

[openvino]
optimum-intel[openvino]

[quality]
black~=23.1
ruff>=0.0.241

[tests]
pytest
requests
parameterized
pytest-xdist
Pillow
sacremoses
torchvision
diffusers>=0.17.0
torchaudio
