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.

24.7M 11K 864

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

848K 24 6

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

482K 10K 1K

wandb-osh

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

217K 90 10

arm-mango

Parallel Hyperparameter Tuning in Python

42K 417 47

hypertune

A library for performing hyperparameter optimization

38K 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.

34K 14K 2K

smac

SMAC3: A Versatile Bayesian Optimization Package for Hyperparameter Optimization

31K 1K 241

google-vizier

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

25K 2K 110

tune-sklearn

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

18K 469 49

layer

Metadata store for Production ML

17K 88 6

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