Cython>=0.25
h5py!=3.0.0,>=2.6
numpy>=1.15
pandas!=1.1.5,>=1.0
scipy>=0.18
tabulate>=0.7
scikit-learn
tqdm
torch>=1.13.0
torchaudio
torchtext
torchvision
pydantic<2.0
transformers
tifffile
imagecodecs
tokenizers>=0.13.3
spacy>=2.3
PyYAML!=5.4.*,<6.0.1,>=3.12
absl-py
kaggle
requests
fsspec[http]<=2023.10.0
dataclasses-json
jsonschema<4.7,>=4.5.0
marshmallow
marshmallow-jsonschema
marshmallow-dataclass==8.5.4
tensorboard
nltk
torchmetrics>=0.11.0
torchinfo
filelock
psutil==5.9.4
protobuf==3.20.3
py-cpuinfo==9.0.0
gpustat
rich~=12.4.4
packaging
retry
sacremoses
sentencepiece
getdaft<0.2
bitsandbytes<0.41.0
xlwt
xlrd>=2.0.1
xlsxwriter>=1.4.3
openpyxl>=3.0.7
pyxlsb>=1.0.8
pyarrow
lxml
html5lib
datasets

[benchmarking]
s3fs

[distributed]
dask[dataframe]<2023.4.0
pyarrow
ray[data,default,serve,tune]<2.4,>=2.2.0
tensorboardX<2.3
GPUtil
tblib
awscli
deepspeed!=0.11.0
getdaft[ray]

[explain]
captum

[extra]
horovod[pytorch]!=0.26.0,>=0.24.0
modin[ray]
predibase>=2023.10.2

[full]
uvicorn
httpx
fastapi
python-multipart
seaborn<0.12,>=0.7
hiplot
ptitprince
dask[dataframe]<2023.4.0
pyarrow
ray[data,default,serve,tune]<2.4,>=2.2.0
tensorboardX<2.3
GPUtil
tblib
awscli
deepspeed!=0.11.0
getdaft[ray]
ray[default,tune]>=2.0.0
hyperopt
hummingbird-ml>=0.4.8
lightgbm
lightgbm-ray
sentence-transformers
faiss-cpu
accelerate
loralib
peft
captum
s3fs

[full:platform_system != "Windows" and python_version < "3.9"]
neuropod==0.3.0rc6

[full:python_version <= "3.6"]
matplotlib<3.4,>=3.0

[full:python_version > "3.6"]
matplotlib>=3.4

[hyperopt]
ray[default,tune]>=2.0.0
hyperopt

[llm]
sentence-transformers
faiss-cpu
accelerate
loralib
peft

[serve]
uvicorn
httpx
fastapi
python-multipart

[serve:platform_system != "Windows" and python_version < "3.9"]
neuropod==0.3.0rc6

[test]
uvicorn
httpx
fastapi
python-multipart
seaborn<0.12,>=0.7
hiplot
ptitprince
dask[dataframe]<2023.4.0
pyarrow
ray[data,default,serve,tune]<2.4,>=2.2.0
tensorboardX<2.3
GPUtil
tblib
awscli
deepspeed!=0.11.0
getdaft[ray]
ray[default,tune]>=2.0.0
hyperopt
hummingbird-ml>=0.4.8
lightgbm
lightgbm-ray
sentence-transformers
faiss-cpu
accelerate
loralib
peft
captum
s3fs
pytest
pytest-timeout
tifffile
wget
six>=1.13.0
aim
wandb<0.12.11
comet_ml
mlflow
hpbandster
ConfigSpace
ax-platform
sqlalchemy<2
bayesian-optimization
flaml[blendsearch]
HEBO
nevergrad
optuna
scikit-optimize
zoopt
s3fs>=2022.8.2

[test:platform_system != "Windows" and python_version < "3.9"]
neuropod==0.3.0rc6

[test:python_version <= "3.6"]
matplotlib<3.4,>=3.0

[test:python_version > "3.6"]
matplotlib>=3.4

[tree]
hummingbird-ml>=0.4.8
lightgbm
lightgbm-ray

[viz]
seaborn<0.12,>=0.7
hiplot
ptitprince

[viz:python_version <= "3.6"]
matplotlib<3.4,>=3.0

[viz:python_version > "3.6"]
matplotlib>=3.4
