NumPy>=1.17.0rc1

[autogen]
openai==0.27.8
diskcache
termcolor

[automl]
lightgbm>=2.3.1
xgboost>=0.90
scipy>=1.4.1
pandas>=1.1.4
scikit-learn>=0.24

[autozero]
scikit-learn
pandas
packaging

[azureml]
azureml-mlflow

[benchmark]
catboost>=0.26
psutil==5.8.0
xgboost==1.3.3
pandas==1.1.4

[blendsearch]
optuna==2.8.0
packaging

[catboost]
catboost>=0.26

[forecast]
holidays<0.14
prophet>=1.0.1
statsmodels>=0.12.2
hcrystalball==0.1.10
pytorch-forecasting>=0.9.0
pytorch-lightning==1.9.0
tensorboardX==2.6

[hf]
transformers[torch]==4.26
datasets
nltk
rouge_score
seqeval

[mathchat]
openai==0.27.8
diskcache
termcolor
sympy
pydantic==1.10.9
wolframalpha

[nlp]
transformers[torch]==4.26
datasets
nltk
rouge_score
seqeval

[nni]
nni

[notebook]
jupyter

[openai]
openai==0.27.8
diskcache

[ray]
ray[tune]~=1.13

[retrievechat]
openai==0.27.8
diskcache
termcolor
chromadb
tiktoken
sentence_transformers

[spark]
pyspark>=3.2.0
joblibspark>=0.5.0
joblib<1.3.0

[synapse]
joblibspark>=0.5.0
optuna==2.8.0
pyspark>=3.2.0
joblib<1.3.0

[test]
lightgbm>=2.3.1
xgboost>=0.90
scipy>=1.4.1
pandas>=1.1.4
scikit-learn>=0.24
thop
pytest>=6.1.1
coverage>=5.3
pre-commit
torch
torchvision
catboost<1.2,>=0.26
rgf-python
optuna==2.8.0
openml
statsmodels>=0.12.2
psutil==5.8.0
dataclasses
transformers[torch]==4.26
datasets
nltk
rouge_score
hcrystalball==0.1.10
seqeval
pytorch-forecasting<=0.10.1,>=0.9.0
mlflow
pyspark>=3.2.0
joblibspark>=0.5.0
nbconvert
nbformat
ipykernel
pytorch-lightning<1.9.1
tensorboardX==2.6
requests<2.29.0
packaging
pydantic==1.10.9
sympy
wolframalpha
joblib<1.3.0

[ts_forecast]
holidays<0.14
prophet>=1.0.1
statsmodels>=0.12.2
hcrystalball==0.1.10

[vw]
vowpalwabbit<9.0.0,>=8.10.0
scikit-learn
