coloredlogs
sympy
transformers[sentencepiece]<4.43.0,>=4.26.0
torch>=1.11
packaging
numpy<2.0
huggingface_hub>=0.8.0
datasets

[amd]
optimum-amd

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

[dev]
accelerate
pytest<=8.0.0
requests
parameterized
pytest-xdist
Pillow
sacremoses
torchvision
diffusers>=0.17.0
torchaudio
einops
invisible-watermark
timm
scikit-learn
rjieba
black~=23.1
ruff==0.1.5

[diffusers]
diffusers

[doc-build]
accelerate

[exporters]
onnx
onnxruntime
timm

[exporters-gpu]
onnx
onnxruntime-gpu
timm

[exporters-tf]
tensorflow<=2.12.1,>=2.4
tf2onnx
onnx
onnxruntime
timm
h5py
numpy<1.24.0
transformers[sentencepiece]<4.38.0,>=4.26.0

[furiosa]
optimum-furiosa

[graphcore]
optimum-graphcore

[habana]
optimum-habana
transformers<4.39.0,>=4.38.0

[intel]
optimum-intel>=1.16.0

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

[neuron]
optimum-neuron[neuron]>=0.0.20
transformers<4.42.0,>=4.36.2

[neuronx]
optimum-neuron[neuronx]>=0.0.20
transformers<4.42.0,>=4.36.2

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

[onnxruntime]
onnx
onnxruntime>=1.11.0
datasets>=1.2.1
evaluate
protobuf>=3.20.1

[onnxruntime-gpu]
onnx
onnxruntime-gpu>=1.11.0
datasets>=1.2.1
evaluate
protobuf>=3.20.1
accelerate

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

[quality]
black~=23.1
ruff==0.1.5

[tests]
accelerate
pytest<=8.0.0
requests
parameterized
pytest-xdist
Pillow
sacremoses
torchvision
diffusers>=0.17.0
torchaudio
einops
invisible-watermark
timm
scikit-learn
rjieba
