Hyperparameter Optimization Python Packages

ray

Ray is an AI compute engine. Ray consists of a core distributed runtime and a set of AI Libraries for accelerating ML workloads.

wandb

The AI developer platform. Use Weights & Biases to train and fine-tune models, and manage models from experimentation to production.

optuna

A hyperparameter optimization framework

cmaes

Python library for CMA Evolution Strategy.

autogluon-tabular

Fast and Accurate ML in 3 Lines of Code

autogluon-core

Fast and Accurate ML in 3 Lines of Code

autogluon-features

Fast and Accurate ML in 3 Lines of Code

autogluon-common

Fast and Accurate ML in 3 Lines of Code

flaml

A fast library for AutoML and tuning. Join our Discord: https://discord.gg/Cppx2vSPVP.

scikit-optimize

Sequential model-based optimization with a `scipy.optimize` interface

wandb-osh

A convenient way to trigger synchronizations to wandb / Weights & Biases if your compute nodes don't have internet!

autogluon-timeseries

Fast and Accurate ML in 3 Lines of Code

autogluon

Fast and Accurate ML in 3 Lines of Code

autogluon-multimodal

Fast and Accurate ML in 3 Lines of Code

keras-tuner

A Hyperparameter Tuning Library for Keras

kubeflow

Universal Python SDK to run AI workloads on Kubernetes

optuna-dashboard

Real-time Web Dashboard for Optuna.

autogluon-vision

Fast and Accurate ML in 3 Lines of Code

autogluon-text

Fast and Accurate ML in 3 Lines of Code

arm-mango

Parallel Hyperparameter Tuning in Python

ant-ray-cpp-nightly

Ray is an AI compute engine. Ray consists of a core distributed runtime and a set of AI Libraries for accelerating ML workloads.

ray-cpp

Ray is an AI compute engine. Ray consists of a core distributed runtime and a set of AI Libraries for accelerating ML workloads.

smac

SMAC3: A Versatile Bayesian Optimization Package for Hyperparameter Optimization

polyaxon

MLOps Tools For Managing & Orchestrating The Machine Learning LifeCycle