transformers==4.18.0
datasets==2.1.0
tokenizers==0.12.1
scipy==1.5.4
pyyaml==5.4.1
protobuf==3.19.4
wandb==0.12.8
pytest==6.2.5
pytest-pythonpath==0.7.4
tfrecord==1.14.1
filelock==3.11.0
mpi4py==3.1.3
horovod[pytorch]==0.24.0
examples-utils[benchmark] @ git+https://github.com/graphcore/examples-utils@v3.4
cmake==3.22.4
grpcio==1.54.2
tritonclient[grpc]==2.16.0
torch==1.10.0+cpu; python_version=='3.6'
torch==2.0.1+cpu; python_version>'3.6'
numpy==1.23.5; python_version > '3.7'
numpy==1.19.5; python_version <= '3.7'