shap==0.40.0
scikit-learn==0.24.2
pandas
torch
tb-nightly
seaborn
future
h5py
wandb==0.12.7
google-cloud
google-cloud-storage
pyyaml
plotly~=5.1.0
pytz~=2021.3
setuptools~=59.0.1
numpy==1.20
requests
torchvision>=0.6.0
mpld3~=0.5
numba>=0.50
sphinx
sphinx-rtd-theme
sphinx-autodoc-typehints

[dev]
autopep8
flake8
