Hyperparameter Search Python Packages

wandb

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

ray

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

scikit-optimize

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

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.

smac

SMAC3: A Versatile Bayesian Optimization Package for Hyperparameter Optimization

ant-ray-cpp-nightly

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

lale

Library for Semi-Automated Data Science

rl-zoo3

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

ant-ray-nightly

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

sambo

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

ant-ray

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

octis

OCTIS: Comparing Topic Models is Simple! A python package to optimize and evaluate topic models (accepted at EACL2021 demo track)

pyrameter

Structure, sample, and savor hyperparameter searches.

neuraxle

The world's cleanest AutoML library ✨ - Do hyperparameter tuning with the right pipeline abstractions to write clean deep learning production pipelines. Let your pipeline steps have hyperparameter spaces. Design steps in your pipeline like components. Compatible with Scikit-Learn, TensorFlow, and most other libraries, frameworks and MLOps environments.

wandb-ng

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

maggy

Distribution transparent Machine Learning experiments on Apache Spark

syftr

syftr is an agent optimizer that helps you find the best agentic workflows for your budget.

phasesweep

YAML-driven phase-chained hyperparameter sweeps with Optuna: each phase's winner is set for all following sweeps [of other params]

ve-ray

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

parameter-sherpa

Hyperparameter optimization that enables researchers to experiment, visualize, and scale quickly.

secretflow-ray

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

pygenetics

Genetic algorithm framework for tuning arbitrary functions

malet

Hyperparameter search and metric visualization tool for personal research.

auto-surprise

An AutoRecSys library for Surprise. Automate algorithm selection and hyperparameter tuning :rocket: