shap==0.41.0
scikit-learn>=1.0.1
pandas
torch
tb-nightly
seaborn
future
h5py
wandb==0.15.0
google-cloud
google-cloud-storage
plotly~=5.14.1
pytz>=2022.1
setuptools~=67.7.0
numpy>=1.21
requests
torchvision>=0.6.0
mpld3>=0.5
numba>=0.50
sphinx
sphinx-rtd-theme
sphinx-autodoc-typehints
sphinx
einops