transformers>=4.28.0
optimum
huggingface_hub>=0.14.0
numpy<=1.21.6,>1.19
protobuf<4

[neuron]
wheel
torch-neuron==1.13.1.*
neuron-cc[tensorflow]
protobuf
torchvision

[neuronx]
wheel
neuronx-cc==2.*
torch-neuronx
torch==1.13.1.*
torchvision==0.14.*

[quality]
black
ruff
isort
hf_doc_builder

[tests]
pytest
psutil
parameterized
GitPython
sentencepiece
datasets
