transformers
accelerate
datasets
hydra-core
omegaconf
psutil
typing-extensions
flatten_dict
colorlog
pandas
rich
nvidia-ml-py

[auto-gptq]
optimum
auto-gptq

[autoawq]
autoawq

[bitsandbytes]
bitsandbytes

[codecarbon]
codecarbon

[deepspeed]
deepspeed

[diffusers]
diffusers

[flash-attn]
flash-attn

[ipex]
optimum[ipex]>=1.18.0

[llama-cpp]
llama-cpp-python

[llm-swarm]
llm-swarm

[onnxruntime]
optimum[onnxruntime]>=1.18.0

[onnxruntime-gpu]
optimum[onnxruntime-gpu]>=1.18.0

[openvino]
optimum[nncf,openvino]>=1.18.0

[peft]
peft

[py-txi]
py-txi

[quality]
ruff

[sentence-transformers]
sentence-transformers

[tensorrt-llm]
optimum[nvidia]>=1.18.0

[testing]
pytest
hydra-joblib-launcher

[timm]
timm

[torch-ort]
torch-ort
onnxruntime-training
optimum>=1.18.0

[torchao]
torchao

[vllm]
vllm
