numpy>=1.19.2
scikit-optimize==0.9.0
statsmodels==0.12.2
scipy==1.6.3

[dev]
black==20.8b1
flake8==3.9.0
pre-commit==2.11.1

[docs]
mkdocs==1.3.0
mike==1.1.2
mkdocs-material==8.2.8
markdown-include==0.6.0

[pydoop]
pydoop

[spark]
pyspark==2.4.3

[tf]
tensorflow==2.4.1

[torch]
torch==1.7.1

[zero]
deepspeed==0.3.13
fairscale==0.3.0
