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.

scikit-optimize

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

autogluon-tabular

Fast and Accurate ML in 3 Lines of Code

flaml

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

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

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

optuna-dashboard

Real-time Web Dashboard for Optuna.

kubeflow

Universal Python SDK to run AI workloads on Kubernetes

autogluon-text

Fast and Accurate ML in 3 Lines of Code

autogluon-vision

Fast and Accurate ML in 3 Lines of Code

arm-mango

Parallel Hyperparameter Tuning in Python

polyaxon

AI Infra / AI Orchestration / AI Control Plane

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.

nni

An open source AutoML toolkit for automate machine learning lifecycle, including feature engineering, neural architecture search, model compression and hyper-parameter tuning.

smac

SMAC3: A Versatile Bayesian Optimization Package for Hyperparameter Optimization