h5py>=3.8
numpy>=1.24
pandas>=2.0
scipy>=1.10
tabulate>=0.9
scikit-learn>=1.3
tqdm>=4.60
torch>=2.0
torchaudio>=2.0
torchvision>=0.15
transformers>=4.36
sentencepiece>=0.2
spacy>=2.3
PyYAML>=6.0
absl-py
kaggle
requests>=2.28
tables
fsspec[http]
dataclasses-json
jsonschema>=4.17
tensorboard
torchmetrics>=1.0
torchinfo
filelock
psutil
protobuf>=4.0
gpustat
rich~=12.4.4
packaging
retry
sacremoses
sentencepiece
bitsandbytes<0.41.0
xlwt
xlrd
openpyxl
pyarrow>=14.0
lxml
datasets

[benchmarking]
s3fs

[distributed]
dask[dataframe]
pyarrow>=14.0
ray[data,default,serve,tune]>=2.9
GPUtil
tblib
awscli

[explain]
captum

[extra]
modin[ray]
predibase>=2023.10.2

[full]
uvicorn
httpx
fastapi
python-multipart
matplotlib>=3.4
seaborn
hiplot
ptitprince
dask[dataframe]
pyarrow>=14.0
ray[data,default,serve,tune]>=2.9
GPUtil
tblib
awscli
ray[default,tune]>=2.9
hyperopt
future<1.0
sentence-transformers
faiss-cpu
accelerate
loralib
peft>=0.10.0
captum
s3fs

[hyperopt]
ray[default,tune]>=2.9
hyperopt
future<1.0

[llm]
sentence-transformers
faiss-cpu
accelerate
loralib
peft>=0.10.0

[serve]
uvicorn
httpx
fastapi
python-multipart

[test]
uvicorn
httpx
fastapi
python-multipart
matplotlib>=3.4
seaborn
hiplot
ptitprince
dask[dataframe]
pyarrow>=14.0
ray[data,default,serve,tune]>=2.9
GPUtil
tblib
awscli
ray[default,tune]>=2.9
hyperopt
future<1.0
sentence-transformers
faiss-cpu
accelerate
loralib
peft>=0.10.0
captum
s3fs
pytest
pytest-timeout
pytest-rerunfailures
tifffile
wget
aim
wandb
comet_ml
mlflow
hpbandster
ConfigSpace>=1.0
ax-platform
sqlalchemy
bayesian-optimization
flaml[blendsearch]
HEBO
nevergrad
optuna
scikit-optimize
zoopt
s3fs>=2022.8.2

[viz]
matplotlib>=3.4
seaborn
hiplot
ptitprince
