ray/python/requirements_tune.txt

33 lines
543 B
Text

ax-platform
bayesian-optimization
ConfigSpace==0.4.10
dragonfly-opt
gluoncv
gym[atari]
h5py
hpbandster
hyperopt==0.1.2
jupyter
keras
kubernetes
lightgbm
matplotlib
mlflow
mxnet
nevergrad
optuna
pytest-remotedata>=0.3.1
pytorch-lightning
scikit-optimize
sigopt
smart_open
tensorflow-probability
timm
torch>=1.5.0
torchvision>=0.6.0
# transformers
git+git://github.com/huggingface/transformers.git@bdcc4b78a27775d1ec8f3fd297cb679c257289db#transformers
git+git://github.com/ray-project/tune-sklearn@master#tune-sklearn
wandb
xgboost
zoopt>=0.4.0