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

[dev]
autopep8
flake8
