torch>=1.8.1
packaging

[all]
accelerate==0.5.1
dataclasses
pytest==6.2.5
pytest-flake8==1.0.7
pytest-cov==3.0.0
pytest-mypy==0.8.1
pytest-xdist==2.4.0
autopep8
pydocstyle
pre-commit>=2.0
yapf
isort
numpy
scipy

[dev]
pytest==6.2.5
pytest-flake8==1.0.7
pytest-cov==3.0.0
pytest-mypy==0.8.1
pytest-xdist==2.4.0
autopep8
pydocstyle
pre-commit>=2.0
yapf
isort
numpy
scipy

[x]
accelerate==0.5.1
dataclasses
