Polyaxon v3 is coming →

Find better model configurations

Define a search space, run parallel trials on your Kubernetes clusters, and compare parameters, metrics, and artifacts before selecting a candidate.

The managed optimization engine is available with Polyaxon Cloud and self-hosted Enterprise. Plan and queue limits apply to trial concurrency.

From search space to reviewed candidate

  1. Define the search

    Choose inputs, an algorithm, and execution limits.

  2. Run trials

    Reuse a training component with different parameter values.

  3. Compare candidates

    Inspect metrics and outputs, then validate the selected run.

Each trial keeps its own configuration and results. Your training code defines the model and logs the measurements used for comparison.

See it in code

Define the space. Submit the trials.

Add a matrix to a training operation. Save a tab as tuning.yaml, then submit it with polyaxon run -f tuning.yaml.

Follow the tuning guide
version: 1.1
kind: operation
hubRef: acme/train-model:v1
matrix:
  kind: grid
  concurrency: 2
  params:
    learning_rate:
      kind: choice
      value: [0.0001, 0.001, 0.01]
    dropout:
      kind: choice
      value: [0.1, 0.2, 0.3, 0.4]

Replace acme/train-model:v1 with your registered component and its inputs. The controls example stops the group when a trial's logged val_loss meets the threshold; choose a target for your workload.

Compare more than a final score

Keep each trial's inputs, metrics, and artifacts together. Move from a table of runs to metric histories and visual outputs to understand what changed and choose a candidate to validate.

Explore run comparison

Compare metric histories to inspect convergence, training and validation behavior, and the steps behind a final score.

Polyaxon comparison dashboard overlaying learning rate, training and validation loss, accuracy, and overfitting gap for three training-simulator runs

These documentation screenshots use a training simulator to demonstrate the comparison views. Follow the tracking quick start. Open an image to inspect the details.

Keep trials within shared compute limits

Carry the selected trial into the next workflow