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.

26.2M 11K 872

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).

1.8M 10K 1K

scikit-optimize

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

795K 24 6

wandb-osh

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

646K 91 10

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).

407K 10K 1K

arm-mango

Parallel Hyperparameter Tuning in Python

43K 417 48

smac

SMAC3: A Versatile Bayesian Optimization Package for Hyperparameter Optimization

36K 1K 242

hypertune

A library for performing hyperparameter optimization

33K 63 9

nni

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

31K 14K 2K

google-vizier

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

21K 2K 110

tune-sklearn

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

18K 468 49

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).

17K 10K 1K