Hyperparameter Optimization Python Packages

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

ray

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

cmaes

Python library for CMA Evolution Strategy.

scikit-optimize

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

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-tabular

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!

optuna-dashboard

Real-time Web Dashboard for Optuna.

autogluon-timeseries

Fast and Accurate ML in 3 Lines of Code

autogluon

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

autogluon-multimodal

Fast and Accurate ML in 3 Lines of Code

google-vizier

Python-based research interface for blackbox and hyperparameter optimization, based on the internal Google Vizier Service.

hypertune

A library for performing hyperparameter optimization

polyaxon

AI Infra / AI Orchestration / AI Control Plane

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.

nni

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

arm-mango

Parallel Hyperparameter Tuning in Python

autogluon-text

Fast and Accurate ML in 3 Lines of Code