apricot-select>=0.6.0
numba>=0.43.0
scipy>=1.5.0
scikit-learn==1.1.2
torch==1.13.1
torchvision
tqdm>=4.24.0
sphinxcontrib-napoleon
sphinxcontrib-bibtex
sphinx-rtd-theme
matplotlib
numpy==1.23.4
torchvision
pillow>=8.4.0
pandas>=1.1.0
torchtext
scikit-image>=0.17.0
pyyaml~=5.4.1
dotmap~=1.3.24
# setuptools~=58.0.4
ray[tune]
ray[default]
submodlib
datasets
transformers
sentence-transformers
setuptools>=65.5.1
h5py
wandb