Optuna (wrapper)
flytekitplugins-optuna
This documentation provides a guide to a fully parallelized Flyte plugin for Optuna. This wrapper leverages Flyte's scalable and distributed workflow orchestration capabilities to parallelize Optuna's hyperparameter optimization across multiple trials efficiently.
pip install flytekitplugins-optunaQuick Start(example, may need adjustment)
pip install flytekitplugins-optuna
from flytekit import task, workflow
from flytekitplugins.optuna import Optimizer, optimize, suggest
@task
def my_task() -> None:
optimize(...)
@workflow
def my_workflow() -> None:
my_task()Available Imports (3)
Configuration type for Optuna (wrapper).
extends dataclass — configuration or data structure for plugin setup
from flytekitplugins.optuna import Optimizer
Task for Optuna (wrapper).
from flytekitplugins.optuna import optimize
Task for Optuna (wrapper).
from flytekitplugins.optuna import suggest
Dependencies
Related Plugins
Anthropic Claude
This plugin provides integration between Flyte and Anthropic's Claude API, enabling you to use Flyte tasks as tools for Claude agents.
Deck
This plugin provides additional renderers to improve task visibility within Flytekit.
Envd
envd is a command-line tool that helps you create the container-based development environment for AI/ML.
Flyte Interactive
FlyteInteractive plugin provides users' favorite interface to develop and debug a flyte task interactively. We support vscode, jupyter (WIP), and neovim (WIP).
