google/vizier

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

What it solves

OSS Vizier は、ブラックボックス最適化の信頼性と柔軟性を兼ね備えたシステムで、特に大規模ハイパーパラメータチューニング向けに設計されています。ユーザーは目的関数の内部構造を知らなくても、最適なパラメータを見つけることができます。

How it works

分散型クライアント‑サーバーシステムとして動作します。ユーザーは検索空間（float、integer、離散、カテゴリカルパラメータを含む）と最大化または最小化すべき指標を定義します。システムは評価すべきパラメータを提案し、ユーザー側のコードが評価して結果をサービスに返すことで、最適化ループが構築されます。

Who it’s for

  • ML Engineers: 機械学習モデルのハイパーパラメータをチューニングするため。
  • AI Researchers: Developer および Benchmarking API を用いて新しい最適化アルゴリズムを実装・ベンチマークするため。
  • Developers: 大規模な進化的実験やプログラム探索のために分散バックエンドが必要な方。

Highlights

  • Distributed Architecture: マルチクライアントに対応したクライアント‑サーバーモデル。
  • Comprehensive API: ユーザー、アルゴリズム実装開発者、アルゴリズムベンチマーク研究者向けに分かれた API を提供。
  • Flexible Search Spaces: カテゴリカルや離散値を含む多様なパラメータタイプに対応。
  • Integration: TensorFlow Probability、Flax、PyGlove への高度な統合サポート。
  • JAX-based Optimizer: JAX をベースにした最先端のベイズ最適化器を搭載。