numpy>=1.17
packaging>=20.0
psutil
pyyaml
torch>=1.4.0

[dev]
black~=22.0
isort>=5.5.4
flake8>=3.8.3
hf-doc-builder>=0.3.0
pytest
pytest-xdist
pytest-subtests
datasets<=2.2.2
evaluate
transformers
scipy
sklearn
parameterized
deepspeed

[docs]

[quality]
black~=22.0
isort>=5.5.4
flake8>=3.8.3
hf-doc-builder>=0.3.0

[sagemaker]
sagemaker

[test]
pytest
pytest-xdist
pytest-subtests
datasets<=2.2.2
evaluate
transformers
scipy
sklearn
parameterized
deepspeed

[test_trackers]
wandb
comet-ml
tensorboard
