hf-xet
numpy
onnx
onnx_ir>=0.1.2
onnxscript>=0.5.3
opentelemetry-sdk>=1.39.1
optuna
pandas
pydantic>=2.0
pyyaml
questionary
torch
torchmetrics>=1.0.0
transformers

[aimet-onnx]
aimet-onnx>=2.15.0

[auto-opt]
optimum

[azureml]
azure-ai-ml>=1.11.1
azure-identity

[bnb]
bitsandbytes
triton

[capture-onnx-graph]
optimum

[cpu]
onnxruntime

[diffusers]
accelerate>=0.30.0
peft
diffusers

[directml]
onnxruntime-directml

[docker]
docker

[finetune]
optimum
accelerate>=0.30.0
peft
scipy
bitsandbytes
triton

[flash-attn]
flash_attn

[gpu]
onnxruntime-gpu

[inc]
neural-compressor

[lora]
accelerate>=0.30.0
peft
scipy

[mobius-ai]
mobius-ai

[nvmo]
nvidia-modelopt[onnx]

[openvino]
openvino>=2025.4.1
nncf>=2.19.0
numpy<2.0
optimum[openvino]>=2.1.0
onnxruntime-openvino

[optimum]
optimum

[qairt]
qairt-dev[onnx]

[qnn]
onnxruntime-qnn

[shared-cache]
azure-identity
azure-storage-blob

[speech]
jiwer
librosa
soundfile

[tf]
tensorflow==1.15.0

[torch-tensorrt]
torch-tensorrt

[tune-session-params]
psutil

[vision]
pillow
