setuptools==59.5.0
threadpoolctl>2.2.0
cython
numpy<1.24.0
pandas
scipy
pyarrow
fastparquet
matplotlib
seaborn>=0.11.0
ray<2.0.0
gensim
ray[tune]
tqdm
tensorboardX
hyperopt
scikit-learn>=1.1.0

[:platform_system == "Windows"]
redis

[dev]
mlflow
lightgbm
xgboost
pygam
flaml
bayesian_optimization==1.4.0
colorama==0.4.4
nevergrad
optuna
gensim
torch
pytorch_lightning
nni

[extended]
mlflow
lightgbm
xgboost
pygam
flaml
bayesian_optimization==1.4.0
colorama==0.4.4
nevergrad
optuna
rpy2

[legacy]
tqdm==4.62.3
mlflow==1.21.0
tensorboardX
hyperopt==0.2.5

[nn]
gensim
torch
pytorch_lightning
nni
rpy2

[normal]
