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

The AI Compute Platform for frontier teams. SkyPilot turns fragmented AI compute into one AI supercomputer, so frontier AI teams build custom intelligence faster.

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!

skypilot-nightly

The AI Compute Platform for frontier teams. SkyPilot turns fragmented AI compute into one AI supercomputer, so frontier AI teams build custom intelligence faster.

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

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

smac

SMAC3: A Versatile Bayesian Optimization Package for Hyperparameter Optimization

trainy-skypilot-nightly

The AI Compute Platform for frontier teams. SkyPilot turns fragmented AI compute into one AI supercomputer, so frontier AI teams build custom intelligence faster.

kubeflow-katib-api

Automated Machine Learning on Kubernetes

tune-sklearn

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

tune

An abstraction layer for parameter tuning

kubeflow-katib

Automated Machine Learning on Kubernetes

google-vizier-dev

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

lale

Library for Semi-Automated Data Science

evalml

EvalML is an AutoML library written in python.

pyruns

A lightweight Python CLI for batch experiments: auto-generate tasks from configs, run in parallel, and track results with real-time monitoring.

polystores

A library for performing hyperparameter optimization

rl-zoo3

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

gamma-facet

Human-explainable AI.

sambo

🎯 📈 Sequential and Model-Based Optimization in Python, feat. SCE-UA, SMBO & SHGO. SOTA perf; 0 deps

gradient-utils

Python utils and helpers library for Gradient