Hyperparameter Tuning Python Packages

wandb

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

skypilot

Run, manage, and scale AI workloads on any AI infrastructure. Use one system to access & manage all AI compute (Kubernetes, Slurm, 20+ clouds, on-prem).

scikit-optimize

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

skypilot-nightly

Run, manage, and scale AI workloads on any AI infrastructure. Use one system to access & manage all AI compute (Kubernetes, Slurm, 20+ clouds, on-prem).

wandb-osh

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

arm-mango

Parallel Hyperparameter Tuning in Python

hypertune

A library for performing hyperparameter optimization

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

google-vizier

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

tune-sklearn

A drop-in replacement for Scikit-Learn’s GridSearchCV / RandomizedSearchCV -- but with cutting edge hyperparameter tuning techniques.

trainy-skypilot-nightly

Run, manage, and scale AI workloads on any AI infrastructure. Use one system to access & manage all AI compute (Kubernetes, Slurm, 20+ clouds, on-prem).

layer

Metadata store for Production ML

kubeflow-katib-api

Automated Machine Learning on Kubernetes

evalml

EvalML is an AutoML library written in python.

lale

Library for Semi-Automated Data Science

kubeflow-katib

Automated Machine Learning on Kubernetes

mlprimitives

Primitives for machine learning and data science.

gamma-facet

Human-explainable AI.

google-vizier-dev

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

syne-tune

Large scale and asynchronous Hyperparameter and Architecture Optimization at your fingertips.

rl-zoo3

A training framework for Stable Baselines3 reinforcement learning agents, with hyperparameter optimization and pre-trained agents included.

polystores

A library for performing hyperparameter optimization

tune

An abstraction layer for hyper parameter tuning