chore: release v0.9.0

This reverts commit ed80c59795.

# Conflicts:
#	src/lib.rs
#	src/storage/sqlite.rs

.github/workflows/ci.yml

@@ -8,6 +8,7 @@ on:
 
 permissions:
   contents: read
+  pull-requests: write
 
 env:
   CARGO_TERM_COLOR: always
@@ -41,11 +42,6 @@ jobs:
   test:
     name: Test
     runs-on: ubuntu-latest
-    strategy:
-      matrix:
-        features:
-          - ""
-          - "async"
     steps:
       - uses: actions/checkout@v6
       - name: Install Rust
@@ -54,12 +50,7 @@ jobs:
           rustup update stable
       - uses: Swatinem/rust-cache@v2
       - name: Run tests
-        run: |
-          if [ -z "${{ matrix.features }}" ]; then
-            cargo test --verbose
-          else
-            cargo test --verbose --features "${{ matrix.features }}"
-          fi
+        run: cargo test --verbose --all-features --all-targets
 
   examples:
     name: Examples
@@ -73,8 +64,14 @@ jobs:
       - uses: Swatinem/rust-cache@v2
       - name: Run sync example (ml_hyperparameter_tuning)
         run: cargo run --example ml_hyperparameter_tuning
+      - name: Run sync example (benchmark_convergence)
+        run: cargo run --example benchmark_convergence
+      - name: Run derive example (parameter_api)
+        run: cargo run --example parameter_api --features derive
       - name: Run async example (async_api_optimization)
         run: cargo run --example async_api_optimization --features async
+      - name: Run visualization example (visualization_demo)
+        run: cargo run --example visualization_demo
 
   docs:
     name: Docs
@@ -91,8 +88,12 @@ jobs:
           RUSTDOCFLAGS: -D warnings
 
   feature-check:
-    name: Feature Combinations
+    name: Feature Combinations (${{ matrix.partition }}/4)
     runs-on: ubuntu-latest
+    strategy:
+      fail-fast: false
+      matrix:
+        partition: [1, 2, 3, 4]
     steps:
       - uses: actions/checkout@v6
       - name: Install Rust
@@ -103,7 +104,7 @@ jobs:
         uses: taiki-e/install-action@cargo-hack
       - uses: Swatinem/rust-cache@v2
       - name: Check feature powerset
-        run: cargo hack check --feature-powerset --no-dev-deps
+        run: cargo hack check --feature-powerset --no-dev-deps --partition ${{ matrix.partition }}/4
 
   coverage:
     name: Coverage
@@ -128,14 +129,14 @@ jobs:
           fail_ci_if_error: true
 
   msrv:
-    name: MSRV (1.88)
+    name: MSRV (1.89)
     runs-on: ubuntu-latest
     steps:
       - uses: actions/checkout@v6
-      - name: Install Rust 1.88
+      - name: Install Rust 1.89
         run: |
-          rustup override set 1.88
-          rustup update 1.88
+          rustup override set 1.89
+          rustup update 1.89
       - uses: Swatinem/rust-cache@v2
       - name: Check compilation
         run: cargo check --all-features
@@ -250,6 +251,99 @@ jobs:
       - name: Typos Check
         run: typos src/
 
+  bench-check:
+    name: Benchmark Smoke Test
+    if: github.event_name == 'push'
+    runs-on: ubuntu-latest
+    steps:
+      - uses: actions/checkout@v6
+      - name: Install Rust
+        run: |
+          rustup override set stable
+          rustup update stable
+      - uses: Swatinem/rust-cache@v2
+      - name: Run benchmarks (smoke test)
+        run: cargo bench --all-features --bench samplers --bench optimization
+
+  bench-compare:
+    name: Benchmark Comparison
+    if: github.event_name == 'pull_request'
+    runs-on: ubuntu-latest
+    steps:
+      - name: Install Rust
+        run: |
+          rustup override set stable
+          rustup update stable
+
+      - name: Install critcmp
+        run: cargo install critcmp
+
+      # --- Run benchmarks on the BASE (target) branch ---
+      - uses: actions/checkout@v6
+        with:
+          ref: ${{ github.event.pull_request.base.sha }}
+          clean: false
+
+      - uses: Swatinem/rust-cache@v2
+        with:
+          key: bench-base
+
+      - name: Bench baseline
+        run: cargo bench --all-features --bench samplers --bench optimization -- --save-baseline base
+
+      # --- Run benchmarks on the HEAD (PR) branch ---
+      - uses: actions/checkout@v6
+        with:
+          ref: ${{ github.event.pull_request.head.sha }}
+          clean: false
+
+      - uses: Swatinem/rust-cache@v2
+        with:
+          key: bench-head
+
+      - name: Bench PR head
+        run: cargo bench --all-features --bench samplers --bench optimization -- --save-baseline head
+
+      # --- Compare and post comment ---
+      - name: Compare benchmarks
+        id: compare
+        run: |
+          EOF=$(dd if=/dev/urandom bs=15 count=1 status=none | base64)
+          echo "result<<$EOF" >> "$GITHUB_OUTPUT"
+          critcmp base head --color never >> "$GITHUB_OUTPUT"
+          echo "$EOF" >> "$GITHUB_OUTPUT"
+
+      - name: Find existing comment
+        uses: peter-evans/find-comment@v4
+        id: find-comment
+        with:
+          issue-number: ${{ github.event.pull_request.number }}
+          comment-author: github-actions[bot]
+          body-includes: "## Benchmark Comparison"
+
+      - name: Post or update PR comment
+        uses: peter-evans/create-or-update-comment@v5
+        with:
+          comment-id: ${{ steps.find-comment.outputs.comment-id }}
+          issue-number: ${{ github.event.pull_request.number }}
+          edit-mode: replace
+          body: |
+            ## Benchmark Comparison
+
+            **Base:** `${{ github.event.pull_request.base.sha }}` | **Head:** `${{ github.event.pull_request.head.sha }}`
+
+            <details>
+            <summary>Click to expand full results</summary>
+
+            ```
+            ${{ steps.compare.outputs.result }}
+            ```
+
+            </details>
+
+            > Benchmarks run on `ubuntu-latest` via [Criterion](https://github.com/bheisler/criterion.rs) + [critcmp](https://github.com/BurntSushi/critcmp).
+            > Results may vary due to shared CI runners. Look for consistent >5% changes.
+
   publish:
     name: Publish to crates.io
     runs-on: ubuntu-latest
@@ -270,6 +364,7 @@ jobs:
       - cross-platform
       - cross-targets
       - typos
+      - bench-check
     steps:
       - uses: actions/checkout@v6
       - name: Install Rust

Cargo.toml

@@ -3,9 +3,9 @@ members = ["optimizer-derive"]
 
 [package]
 name = "optimizer"
-version = "0.5.1"
+version = "0.9.0"
 edition = "2024"
-rust-version = "1.88"
+rust-version = "1.89"
 license = "MIT"
 authors = ["Manuel Raimann <raimannma@outlook.de"]
 description = "A Rust library for optimization algorithms."
@@ -16,20 +16,42 @@ categories = ["algorithm", "science", "data-structures"]
 readme = "README.md"
 
 [dependencies]
-rand = "0.10"
+fastrand = "2.3"
 thiserror = "2"
 parking_lot = "0.12"
 tokio = { version = "1", features = ["sync", "rt-multi-thread"], optional = true }
 optimizer-derive = { version = "0.1.0", path = "optimizer-derive", optional = true }
+serde = { version = "1", features = ["derive"], optional = true }
+serde_json = { version = "1", optional = true }
+tracing = { version = "0.1.29", optional = true }
+sobol_burley = { version = "0.5", optional = true }
+nalgebra = { version = "0.34", optional = true }
+fs2 = { version = "0.4", optional = true }
 
 [features]
 default = []
 async = ["dep:tokio"]
 derive = ["dep:optimizer-derive"]
+serde = ["dep:serde", "dep:serde_json"]
+journal = ["dep:fs2", "serde"]
+tracing = ["dep:tracing"]
+sobol = ["dep:sobol_burley"]
+cma-es = ["dep:nalgebra"]
+gp = ["dep:nalgebra"]
 
 [dev-dependencies]
 tokio = { version = "1", features = ["rt-multi-thread", "macros", "time"] }
 optimizer-derive = { version = "0.1.0", path = "optimizer-derive" }
+serde_json = "1"
+criterion = { version = "0.8", features = ["html_reports"] }
+
+[[bench]]
+name = "samplers"
+harness = false
+
+[[bench]]
+name = "optimization"
+harness = false
 
 [[example]]
 name = "async_api_optimization"
@@ -44,3 +66,11 @@ path = "examples/ml_hyperparameter_tuning.rs"
 name = "parameter_api"
 path = "examples/parameter_api.rs"
 required-features = ["derive"]
+
+[[example]]
+name = "visualization_demo"
+path = "examples/visualization_demo.rs"
+
+[[test]]
+name = "journal_tests"
+required-features = ["journal"]

benches/optimization.rs

@@ -0,0 +1,112 @@
+#[allow(dead_code)]
+mod test_functions;
+
+use criterion::{BenchmarkId, Criterion, criterion_group, criterion_main};
+use optimizer::parameter::Parameter;
+use optimizer::sampler::random::RandomSampler;
+use optimizer::sampler::tpe::TpeSampler;
+use optimizer::{FloatParam, Study};
+
+fn make_params(dims: usize) -> Vec<FloatParam> {
+    (0..dims)
+        .map(|i| FloatParam::new(-5.0, 5.0).name(format!("x{i}")))
+        .collect()
+}
+
+fn bench_tpe_sphere(c: &mut Criterion) {
+    let mut group = c.benchmark_group("tpe_sphere");
+    group.sample_size(10);
+
+    for dims in [2, 10, 50] {
+        let params = make_params(dims);
+        group.bench_with_input(BenchmarkId::new("dims", dims), &params, |b, params| {
+            b.iter(|| {
+                let study = Study::minimize(TpeSampler::builder().seed(42).build().unwrap());
+                study
+                    .optimize(100, |trial| {
+                        let x: Vec<f64> = params
+                            .iter()
+                            .map(|p| p.suggest(trial))
+                            .collect::<Result<_, _>>()
+                            .unwrap();
+                        Ok::<_, optimizer::Error>(test_functions::sphere(&x))
+                    })
+                    .unwrap();
+            });
+        });
+    }
+    group.finish();
+}
+
+fn bench_tpe_rosenbrock(c: &mut Criterion) {
+    let mut group = c.benchmark_group("tpe_rosenbrock");
+    group.sample_size(10);
+
+    for dims in [2, 10] {
+        let params = make_params(dims);
+        group.bench_with_input(BenchmarkId::new("dims", dims), &params, |b, params| {
+            b.iter(|| {
+                let study = Study::minimize(TpeSampler::builder().seed(42).build().unwrap());
+                study
+                    .optimize(100, |trial| {
+                        let x: Vec<f64> = params
+                            .iter()
+                            .map(|p| p.suggest(trial))
+                            .collect::<Result<_, _>>()
+                            .unwrap();
+                        Ok::<_, optimizer::Error>(test_functions::rosenbrock(&x))
+                    })
+                    .unwrap();
+            });
+        });
+    }
+    group.finish();
+}
+
+fn bench_random_vs_tpe(c: &mut Criterion) {
+    let mut group = c.benchmark_group("random_vs_tpe");
+    group.sample_size(10);
+    let params = make_params(5);
+
+    group.bench_function("random_5d", |b| {
+        b.iter(|| {
+            let study = Study::minimize(RandomSampler::with_seed(42));
+            study
+                .optimize(100, |trial| {
+                    let x: Vec<f64> = params
+                        .iter()
+                        .map(|p| p.suggest(trial))
+                        .collect::<Result<_, _>>()
+                        .unwrap();
+                    Ok::<_, optimizer::Error>(test_functions::sphere(&x))
+                })
+                .unwrap();
+        });
+    });
+
+    group.bench_function("tpe_5d", |b| {
+        b.iter(|| {
+            let study = Study::minimize(TpeSampler::builder().seed(42).build().unwrap());
+            study
+                .optimize(100, |trial| {
+                    let x: Vec<f64> = params
+                        .iter()
+                        .map(|p| p.suggest(trial))
+                        .collect::<Result<_, _>>()
+                        .unwrap();
+                    Ok::<_, optimizer::Error>(test_functions::sphere(&x))
+                })
+                .unwrap();
+        });
+    });
+
+    group.finish();
+}
+
+criterion_group!(
+    benches,
+    bench_tpe_sphere,
+    bench_tpe_rosenbrock,
+    bench_random_vs_tpe
+);
+criterion_main!(benches);

benches/samplers.rs

@@ -0,0 +1,118 @@
+use std::collections::HashMap;
+
+use criterion::{BenchmarkId, Criterion, criterion_group, criterion_main};
+use optimizer::parameter::Parameter;
+use optimizer::sampler::Sampler;
+use optimizer::sampler::grid::GridSearchSampler;
+use optimizer::sampler::random::RandomSampler;
+use optimizer::sampler::tpe::TpeSampler;
+use optimizer::{CompletedTrial, FloatParam};
+
+/// Build a synthetic history of `n` completed trials over `dims` float parameters.
+fn build_history(n: usize, dims: usize) -> Vec<CompletedTrial<f64>> {
+    let params: Vec<FloatParam> = (0..dims)
+        .map(|i| FloatParam::new(-5.0, 5.0).name(format!("x{i}")))
+        .collect();
+
+    let mut history = Vec::with_capacity(n);
+    let sampler = RandomSampler::with_seed(42);
+
+    for trial_id in 0..n {
+        let id = trial_id as u64;
+        let mut param_values = HashMap::new();
+        let mut distributions = HashMap::new();
+        let mut param_labels = HashMap::new();
+        for p in &params {
+            let dist = p.distribution();
+            let val = sampler.sample(&dist, id, &history);
+            param_values.insert(p.id(), val);
+            distributions.insert(p.id(), dist);
+            param_labels.insert(p.id(), p.label());
+        }
+        // Use sphere function value as objective
+        let value: f64 = param_values
+            .values()
+            .map(|v| {
+                let optimizer::ParamValue::Float(f) = v else {
+                    unreachable!()
+                };
+                f * f
+            })
+            .sum();
+        history.push(CompletedTrial::new(
+            id,
+            param_values,
+            distributions,
+            param_labels,
+            value,
+        ));
+    }
+    history
+}
+
+fn bench_tpe_sample(c: &mut Criterion) {
+    let mut group = c.benchmark_group("tpe_sample");
+    let dist = FloatParam::new(-5.0, 5.0).distribution();
+    let tpe = TpeSampler::builder().seed(42).build().unwrap();
+
+    for history_size in [10, 100, 1000] {
+        let history = build_history(history_size, 2);
+        group.bench_with_input(
+            BenchmarkId::new("history", history_size),
+            &history,
+            |b, history| {
+                b.iter(|| tpe.sample(&dist, history.len() as u64, history));
+            },
+        );
+    }
+    group.finish();
+}
+
+fn bench_random_sample(c: &mut Criterion) {
+    let mut group = c.benchmark_group("random_sample");
+    let dist = FloatParam::new(-5.0, 5.0).distribution();
+    let sampler = RandomSampler::with_seed(42);
+
+    for history_size in [10, 100, 1000] {
+        let history = build_history(history_size, 2);
+        group.bench_with_input(
+            BenchmarkId::new("history", history_size),
+            &history,
+            |b, history| {
+                b.iter(|| sampler.sample(&dist, history.len() as u64, history));
+            },
+        );
+    }
+    group.finish();
+}
+
+fn bench_grid_sample(c: &mut Criterion) {
+    let mut group = c.benchmark_group("grid_sample");
+    let dist = FloatParam::new(-5.0, 5.0).distribution();
+    let history: Vec<CompletedTrial<f64>> = Vec::new();
+
+    for grid_points in [5, 10, 50] {
+        group.bench_with_input(
+            BenchmarkId::new("points", grid_points),
+            &grid_points,
+            |b, _| {
+                b.iter(|| {
+                    // Fresh sampler each iteration since grid tracks used points
+                    let sampler = GridSearchSampler::builder()
+                        .n_points_per_param(grid_points)
+                        .build();
+                    sampler.sample(&dist, 0, &history)
+                });
+            },
+        );
+    }
+    group.finish();
+}
+
+criterion_group!(
+    benches,
+    bench_tpe_sample,
+    bench_random_sample,
+    bench_grid_sample
+);
+criterion_main!(benches);

benches/test_functions.rs

@@ -0,0 +1,84 @@
+//! Standard optimization test functions for benchmarking.
+
+/// Sphere function: unimodal, convex. Global minimum f(0,...,0) = 0.
+pub fn sphere(x: &[f64]) -> f64 {
+    x.iter().map(|xi| xi * xi).sum()
+}
+
+/// Rosenbrock function: narrow valley. Global minimum f(1,...,1) = 0.
+pub fn rosenbrock(x: &[f64]) -> f64 {
+    x.windows(2)
+        .map(|w| 100.0 * (w[1] - w[0] * w[0]).powi(2) + (1.0 - w[0]).powi(2))
+        .sum()
+}
+
+/// Rastrigin function: highly multimodal. Global minimum f(0,...,0) = 0.
+pub fn rastrigin(x: &[f64]) -> f64 {
+    let n = x.len() as f64;
+    10.0 * n
+        + x.iter()
+            .map(|xi| xi * xi - 10.0 * (2.0 * std::f64::consts::PI * xi).cos())
+            .sum::<f64>()
+}
+
+/// Ackley function: nearly flat with a deep well. Global minimum f(0,...,0) = 0.
+pub fn ackley(x: &[f64]) -> f64 {
+    let n = x.len() as f64;
+    let sum_sq: f64 = x.iter().map(|xi| xi * xi).sum();
+    let sum_cos: f64 = x
+        .iter()
+        .map(|xi| (2.0 * std::f64::consts::PI * xi).cos())
+        .sum();
+    -20.0 * (-0.2 * (sum_sq / n).sqrt()).exp() - (sum_cos / n).exp() + 20.0 + std::f64::consts::E
+}
+
+/// Branin function (2D only). Three global minima with f* ≈ 0.397887.
+///
+/// # Panics
+///
+/// Panics if `x` does not have exactly 2 elements.
+pub fn branin(x: &[f64]) -> f64 {
+    assert!(x.len() == 2, "Branin requires exactly 2 dimensions");
+    let (x1, x2) = (x[0], x[1]);
+    let pi = std::f64::consts::PI;
+    let a = 1.0;
+    let b = 5.1 / (4.0 * pi * pi);
+    let c = 5.0 / pi;
+    let r = 6.0;
+    let s = 10.0;
+    let t = 1.0 / (8.0 * pi);
+    a * (x2 - b * x1 * x1 + c * x1 - r).powi(2) + s * (1.0 - t) * x1.cos() + s
+}
+
+/// Hartmann 6D function. Global minimum f* ≈ -3.3224.
+///
+/// # Panics
+///
+/// Panics if `x` does not have exactly 6 elements.
+pub fn hartmann6(x: &[f64]) -> f64 {
+    assert!(x.len() == 6, "Hartmann6 requires exactly 6 dimensions");
+
+    let alpha = [1.0, 1.2, 3.0, 3.2];
+    let a_matrix = [
+        [10.0, 3.0, 17.0, 3.5, 1.7, 8.0],
+        [0.05, 10.0, 17.0, 0.1, 8.0, 14.0],
+        [3.0, 3.5, 1.7, 10.0, 17.0, 8.0],
+        [17.0, 8.0, 0.05, 10.0, 0.1, 14.0],
+    ];
+    let p_matrix = [
+        [0.1312, 0.1696, 0.5569, 0.0124, 0.8283, 0.5886],
+        [0.2329, 0.4135, 0.8307, 0.3736, 0.1004, 0.9991],
+        [0.2348, 0.1451, 0.3522, 0.2883, 0.3047, 0.6650],
+        [0.4047, 0.8828, 0.8732, 0.5743, 0.1091, 0.0381],
+    ];
+
+    let mut result = 0.0;
+    for i in 0..4 {
+        let mut inner = 0.0;
+        for (j, xj) in x.iter().enumerate() {
+            inner += a_matrix[i][j] * (xj - p_matrix[i][j]).powi(2);
+        }
+        result -= alpha[i] * (-inner).exp();
+    }
+    result
+}

deny.toml

@@ -1,7 +1,11 @@
 [advisories]
 version = 2
 db-path = "~/.cargo/advisory-db"
-ignore = []
+ignore = [
+    # paste is unmaintained but it's a transitive dep via simba -> nalgebra
+    # with no fix available upstream yet
+    "RUSTSEC-2024-0436",
+]
 
 [licenses]
 version = 2
@@ -9,8 +13,8 @@ allow = [
     "MIT",
     "Apache-2.0",
     "Apache-2.0 WITH LLVM-exception",
-    "BSD-2-Clause",
     "Unicode-3.0",
+    "Zlib",
 ]
 confidence-threshold = 0.8
 

examples/benchmark_convergence.rs

@@ -0,0 +1,124 @@
+use std::ops::ControlFlow;
+use std::time::Instant;
+
+use optimizer::parameter::Parameter;
+use optimizer::sampler::random::RandomSampler;
+use optimizer::sampler::tpe::TpeSampler;
+use optimizer::{FloatParam, Study};
+
+/// Standard optimization test functions.
+mod functions {
+    pub fn sphere(x: &[f64]) -> f64 {
+        x.iter().map(|xi| xi * xi).sum()
+    }
+
+    pub fn rosenbrock(x: &[f64]) -> f64 {
+        x.windows(2)
+            .map(|w| 100.0 * (w[1] - w[0] * w[0]).powi(2) + (1.0 - w[0]).powi(2))
+            .sum()
+    }
+
+    pub fn rastrigin(x: &[f64]) -> f64 {
+        let n = x.len() as f64;
+        10.0 * n
+            + x.iter()
+                .map(|xi| xi * xi - 10.0 * (2.0 * std::f64::consts::PI * xi).cos())
+                .sum::<f64>()
+    }
+}
+
+fn run_convergence(
+    name: &str,
+    sampler_name: &str,
+    study: Study<f64>,
+    params: &[FloatParam],
+    objective: fn(&[f64]) -> f64,
+    n_trials: usize,
+) {
+    let start = Instant::now();
+
+    study
+        .optimize_with_callback(
+            n_trials,
+            |trial| {
+                let x: Vec<f64> = params
+                    .iter()
+                    .map(|p| p.suggest(trial))
+                    .collect::<Result<_, _>>()
+                    .unwrap();
+                Ok::<_, optimizer::Error>(objective(&x))
+            },
+            |study, _trial| {
+                let elapsed = start.elapsed().as_millis();
+                let best = study.best_value().unwrap();
+                let n = study.n_trials();
+                println!("{n},{best},{elapsed},{sampler_name},{name}");
+                ControlFlow::Continue(())
+            },
+        )
+        .unwrap();
+}
+
+fn main() {
+    println!("trial,best_value,elapsed_ms,sampler,function");
+
+    let dims = 5;
+    let params: Vec<FloatParam> = (0..dims)
+        .map(|i| FloatParam::new(-5.0, 5.0).name(format!("x{i}")))
+        .collect();
+    let n_trials = 200;
+
+    // Sphere: Random vs TPE
+    run_convergence(
+        "sphere_5d",
+        "random",
+        Study::minimize(RandomSampler::with_seed(1)),
+        &params,
+        functions::sphere,
+        n_trials,
+    );
+    run_convergence(
+        "sphere_5d",
+        "tpe",
+        Study::minimize(TpeSampler::builder().seed(1).build().unwrap()),
+        &params,
+        functions::sphere,
+        n_trials,
+    );
+
+    // Rosenbrock: Random vs TPE
+    run_convergence(
+        "rosenbrock_5d",
+        "random",
+        Study::minimize(RandomSampler::with_seed(2)),
+        &params,
+        functions::rosenbrock,
+        n_trials,
+    );
+    run_convergence(
+        "rosenbrock_5d",
+        "tpe",
+        Study::minimize(TpeSampler::builder().seed(2).build().unwrap()),
+        &params,
+        functions::rosenbrock,
+        n_trials,
+    );
+
+    // Rastrigin: Random vs TPE
+    run_convergence(
+        "rastrigin_5d",
+        "random",
+        Study::minimize(RandomSampler::with_seed(3)),
+        &params,
+        functions::rastrigin,
+        n_trials,
+    );
+    run_convergence(
+        "rastrigin_5d",
+        "tpe",
+        Study::minimize(TpeSampler::builder().seed(3).build().unwrap()),
+        &params,
+        functions::rastrigin,
+        n_trials,
+    );
+}

examples/visualization_demo.rs

@@ -0,0 +1,45 @@
+use optimizer::prelude::*;
+
+fn main() {
+    // Multi-parameter optimization with TPE sampler.
+    let sampler = TpeSampler::builder().seed(42).build().unwrap();
+    let mut study: Study<f64> = Study::with_sampler(Direction::Minimize, sampler);
+    study.set_pruner(MedianPruner::new(Direction::Minimize));
+
+    let lr = FloatParam::new(1e-5, 1e-1)
+        .log_scale()
+        .name("learning_rate");
+    let n_layers = IntParam::new(1, 5).name("n_layers");
+    let dropout = FloatParam::new(0.0, 0.5).step(0.05).name("dropout");
+    let batch_size = CategoricalParam::new(vec![16, 32, 64, 128]).name("batch_size");
+
+    study
+        .optimize(80, |trial| {
+            let lr_val = lr.suggest(trial)?;
+            let layers = n_layers.suggest(trial)?;
+            let drop = dropout.suggest(trial)?;
+            let bs = batch_size.suggest(trial)?;
+
+            // Simulate training with intermediate reporting.
+            let mut loss = 1.0;
+            for epoch in 0..10 {
+                loss *= 0.7 + 0.3 * lr_val.ln().abs() / 12.0;
+                loss += drop * 0.05;
+                loss += (1.0 / bs as f64) * 0.1;
+                loss -= layers as f64 * 0.02;
+                trial.report(epoch, loss);
+                if trial.should_prune() {
+                    return Err(TrialPruned.into());
+                }
+            }
+
+            Ok::<_, Error>(loss)
+        })
+        .unwrap();
+
+    println!("{}", study.summary());
+
+    let path = "optimization_report.html";
+    generate_html_report(&study, path).unwrap();
+    println!("\nReport saved to {path}");
+}

optimizer-derive/Cargo.toml

@@ -2,7 +2,7 @@
 name = "optimizer-derive"
 version = "0.1.0"
 edition = "2024"
-rust-version = "1.88"
+rust-version = "1.89"
 license = "MIT"
 description = "Derive macros for the optimizer crate"
 repository = "https://github.com/raimannma/rust-optimizer"
@@ -13,4 +13,3 @@ proc-macro = true
 [dependencies]
 syn = { version = "2", features = ["full"] }
 quote = "1"
-proc-macro2 = "1"

src/distribution.rs

@@ -2,6 +2,7 @@
 
 /// Distribution for floating-point parameters.
 #[derive(Clone, Debug, PartialEq)]
+#[cfg_attr(feature = "serde", derive(serde::Serialize, serde::Deserialize))]
 pub struct FloatDistribution {
     /// Lower bound (inclusive).
     pub low: f64,
@@ -15,6 +16,7 @@ pub struct FloatDistribution {
 
 /// Distribution for integer parameters.
 #[derive(Clone, Debug, PartialEq)]
+#[cfg_attr(feature = "serde", derive(serde::Serialize, serde::Deserialize))]
 pub struct IntDistribution {
     /// Lower bound (inclusive).
     pub low: i64,
@@ -28,6 +30,7 @@ pub struct IntDistribution {
 
 /// Distribution for categorical parameters.
 #[derive(Clone, Debug, PartialEq)]
+#[cfg_attr(feature = "serde", derive(serde::Serialize, serde::Deserialize))]
 pub struct CategoricalDistribution {
     /// Number of choices available.
     pub n_choices: usize,
@@ -35,6 +38,7 @@ pub struct CategoricalDistribution {
 
 /// Enum wrapping all parameter distribution types.
 #[derive(Clone, Debug, PartialEq)]
+#[cfg_attr(feature = "serde", derive(serde::Serialize, serde::Deserialize))]
 pub enum Distribution {
     /// A floating-point distribution.
     Float(FloatDistribution),

src/error.rs

@@ -72,6 +72,19 @@ pub enum Error {
         got: usize,
     },
 
+    /// Returned when a trial is pruned (stopped early by the objective function).
+    #[error("trial was pruned")]
+    TrialPruned,
+
+    /// Returned when the objective returns the wrong number of values.
+    #[error("objective dimension mismatch: expected {expected} values, got {got}")]
+    ObjectiveDimensionMismatch {
+        /// The expected number of objective values.
+        expected: usize,
+        /// The actual number of objective values returned.
+        got: usize,
+    },
+
     /// Returned when an internal invariant is violated.
     #[error("internal error: {0}")]
     Internal(&'static str),
@@ -80,6 +93,41 @@ pub enum Error {
     #[cfg(feature = "async")]
     #[error("async task error: {0}")]
     TaskError(String),
+
+    /// Returned when a storage operation fails.
+    #[cfg(feature = "journal")]
+    #[error("storage error: {0}")]
+    Storage(String),
 }
 
 pub type Result<T> = core::result::Result<T, Error>;
+
+/// Convenience type for signalling a pruned trial from an objective function.
+///
+/// Implements `Into<Error>` so it can be used with `?` in objectives that
+/// return `Result<V, Error>`.
+///
+/// # Examples
+///
+/// ```
+/// use optimizer::{Error, TrialPruned};
+///
+/// fn objective_that_prunes() -> Result<f64, Error> {
+///     // ... some computation ...
+///     Err(TrialPruned)?
+/// }
+/// ```
+#[derive(Debug)]
+pub struct TrialPruned;
+
+impl core::fmt::Display for TrialPruned {
+    fn fmt(&self, f: &mut core::fmt::Formatter<'_>) -> core::fmt::Result {
+        write!(f, "trial was pruned")
+    }
+}
+
+impl From<TrialPruned> for Error {
+    fn from(_: TrialPruned) -> Self {
+        Error::TrialPruned
+    }
+}

src/fanova.rs

@@ -0,0 +1,547 @@
+//! fANOVA (functional ANOVA) parameter importance via random forest.
+//!
+//! Decomposes the variance of the objective function into contributions
+//! from individual parameters (main effects) and parameter interactions.
+//!
+//! The algorithm:
+//! 1. Fits a random forest to `(parameters) -> objective_value`
+//! 2. Applies functional ANOVA decomposition to the forest
+//! 3. Computes main effects (single-parameter importance)
+//! 4. Computes interaction effects (pairwise parameter importance)
+
+/// Result of fANOVA analysis.
+#[derive(Debug, Clone)]
+pub struct FanovaResult {
+    /// Per-parameter importance (fraction of total variance explained).
+    /// Sorted by descending importance.
+    pub main_effects: Vec<(String, f64)>,
+    /// Pairwise interaction importance (fraction of total variance explained).
+    /// Sorted by descending importance.
+    pub interactions: Vec<((String, String), f64)>,
+}
+
+/// Configuration for fANOVA analysis.
+#[derive(Debug, Clone)]
+pub struct FanovaConfig {
+    /// Number of trees in the random forest (default: 64).
+    pub n_trees: usize,
+    /// Maximum depth of each tree. `None` for unlimited (default: `None`).
+    pub max_depth: Option<usize>,
+    /// Minimum samples required to split a node (default: 2).
+    pub min_samples_split: usize,
+    /// Minimum samples required in a leaf node (default: 1).
+    pub min_samples_leaf: usize,
+    /// Random seed for reproducibility (default: `Some(42)`).
+    pub seed: Option<u64>,
+}
+
+impl Default for FanovaConfig {
+    fn default() -> Self {
+        Self {
+            n_trees: 64,
+            max_depth: None,
+            min_samples_split: 2,
+            min_samples_leaf: 1,
+            seed: Some(42),
+        }
+    }
+}
+
+// --- Decision Tree ---
+
+/// A node in the regression tree (arena-allocated).
+#[derive(Debug, Clone)]
+enum TreeNode {
+    Leaf {
+        value: f64,
+        n_samples: usize,
+    },
+    Split {
+        feature: usize,
+        threshold: f64,
+        left: usize,
+        right: usize,
+        n_samples: usize,
+    },
+}
+
+/// A regression decision tree for fANOVA.
+#[derive(Debug, Clone)]
+struct DecisionTree {
+    nodes: Vec<TreeNode>,
+}
+
+impl DecisionTree {
+    /// Build a tree from the given data using the specified bootstrap indices.
+    fn build(
+        data: &[Vec<f64>],
+        targets: &[f64],
+        indices: &[usize],
+        config: &FanovaConfig,
+        rng: &mut fastrand::Rng,
+    ) -> Self {
+        let mut tree = Self { nodes: Vec::new() };
+        tree.build_node(data, targets, indices, 0, config, rng);
+        tree
+    }
+
+    #[allow(clippy::cast_precision_loss)]
+    fn build_node(
+        &mut self,
+        data: &[Vec<f64>],
+        targets: &[f64],
+        indices: &[usize],
+        depth: usize,
+        config: &FanovaConfig,
+        rng: &mut fastrand::Rng,
+    ) -> usize {
+        let n = indices.len();
+        let mean = indices.iter().map(|&i| targets[i]).sum::<f64>() / n as f64;
+
+        // Stopping conditions
+        if n < config.min_samples_split || config.max_depth.is_some_and(|d| depth >= d) {
+            let idx = self.nodes.len();
+            self.nodes.push(TreeNode::Leaf {
+                value: mean,
+                n_samples: n,
+            });
+            return idx;
+        }
+
+        // Pure node check (all targets identical)
+        #[allow(clippy::float_cmp)]
+        if indices.iter().all(|&i| targets[i] == targets[indices[0]]) {
+            let idx = self.nodes.len();
+            self.nodes.push(TreeNode::Leaf {
+                value: mean,
+                n_samples: n,
+            });
+            return idx;
+        }
+
+        let n_features = data[0].len();
+        #[allow(clippy::cast_possible_truncation, clippy::cast_sign_loss)]
+        let max_features = ((n_features as f64).sqrt().ceil() as usize)
+            .max(1)
+            .min(n_features);
+        let candidates = partial_shuffle(n_features, max_features, rng);
+
+        // Total variance at this node
+        let total_var: f64 = indices.iter().map(|&i| (targets[i] - mean).powi(2)).sum();
+        if total_var == 0.0 {
+            let idx = self.nodes.len();
+            self.nodes.push(TreeNode::Leaf {
+                value: mean,
+                n_samples: n,
+            });
+            return idx;
+        }
+
+        let mut best_score = f64::NEG_INFINITY;
+        let mut best_feature = 0;
+        let mut best_threshold = 0.0;
+
+        for &feat in &candidates {
+            let mut values: Vec<f64> = indices.iter().map(|&i| data[i][feat]).collect();
+            values.sort_by(|a, b| a.partial_cmp(b).unwrap_or(core::cmp::Ordering::Equal));
+            values.dedup();
+
+            if values.len() < 2 {
+                continue;
+            }
+
+            for w in values.windows(2) {
+                let threshold = f64::midpoint(w[0], w[1]);
+                let (l_sum, l_sq, l_n, r_sum, r_sq, r_n) =
+                    split_stats(data, targets, indices, feat, threshold);
+
+                if l_n < config.min_samples_leaf || r_n < config.min_samples_leaf {
+                    continue;
+                }
+
+                let l_var = l_sq - l_sum * l_sum / l_n as f64;
+                let r_var = r_sq - r_sum * r_sum / r_n as f64;
+                let score = total_var - l_var - r_var;
+
+                if score > best_score {
+                    best_score = score;
+                    best_feature = feat;
+                    best_threshold = threshold;
+                }
+            }
+        }
+
+        if best_score <= 0.0 {
+            let idx = self.nodes.len();
+            self.nodes.push(TreeNode::Leaf {
+                value: mean,
+                n_samples: n,
+            });
+            return idx;
+        }
+
+        let (left_indices, right_indices): (Vec<usize>, Vec<usize>) = indices
+            .iter()
+            .partition(|&&i| data[i][best_feature] <= best_threshold);
+
+        if left_indices.is_empty() || right_indices.is_empty() {
+            let idx = self.nodes.len();
+            self.nodes.push(TreeNode::Leaf {
+                value: mean,
+                n_samples: n,
+            });
+            return idx;
+        }
+
+        // Reserve slot for this split node (placeholder replaced below)
+        let node_idx = self.nodes.len();
+        self.nodes.push(TreeNode::Leaf {
+            value: 0.0,
+            n_samples: 0,
+        });
+
+        let left = self.build_node(data, targets, &left_indices, depth + 1, config, rng);
+        let right = self.build_node(data, targets, &right_indices, depth + 1, config, rng);
+
+        self.nodes[node_idx] = TreeNode::Split {
+            feature: best_feature,
+            threshold: best_threshold,
+            left,
+            right,
+            n_samples: n,
+        };
+
+        node_idx
+    }
+
+    /// Compute marginal prediction for a given feature subset.
+    ///
+    /// Features in `subset` use values from `feature_values`.
+    /// Features not in `subset` are marginalized by weighting branches
+    /// proportionally to their training-data fractions.
+    fn marginal_predict(&self, subset: &[usize], feature_values: &[f64]) -> f64 {
+        self.marginal_predict_at(0, subset, feature_values)
+    }
+
+    #[allow(clippy::cast_precision_loss)]
+    fn marginal_predict_at(&self, idx: usize, subset: &[usize], vals: &[f64]) -> f64 {
+        match self.nodes[idx] {
+            TreeNode::Leaf { value, .. } => value,
+            TreeNode::Split {
+                feature,
+                threshold,
+                left,
+                right,
+                n_samples,
+            } => {
+                if subset.contains(&feature) {
+                    if vals[feature] <= threshold {
+                        self.marginal_predict_at(left, subset, vals)
+                    } else {
+                        self.marginal_predict_at(right, subset, vals)
+                    }
+                } else {
+                    let l_n = self.n_samples(left) as f64;
+                    let r_n = self.n_samples(right) as f64;
+                    let total = n_samples as f64;
+                    (l_n / total) * self.marginal_predict_at(left, subset, vals)
+                        + (r_n / total) * self.marginal_predict_at(right, subset, vals)
+                }
+            }
+        }
+    }
+
+    fn n_samples(&self, idx: usize) -> usize {
+        match self.nodes[idx] {
+            TreeNode::Leaf { n_samples, .. } | TreeNode::Split { n_samples, .. } => n_samples,
+        }
+    }
+}
+
+// --- Helper Functions ---
+
+/// Select `k` random indices from `0..n` using partial Fisher-Yates shuffle.
+fn partial_shuffle(n: usize, k: usize, rng: &mut fastrand::Rng) -> Vec<usize> {
+    let mut indices: Vec<usize> = (0..n).collect();
+    let k = k.min(n);
+    for i in 0..k {
+        let j = rng.usize(i..n);
+        indices.swap(i, j);
+    }
+    indices.truncate(k);
+    indices
+}
+
+/// Compute left/right split statistics for variance reduction.
+#[allow(clippy::cast_precision_loss)]
+fn split_stats(
+    data: &[Vec<f64>],
+    targets: &[f64],
+    indices: &[usize],
+    feature: usize,
+    threshold: f64,
+) -> (f64, f64, usize, f64, f64, usize) {
+    let (mut l_sum, mut l_sq, mut l_n) = (0.0, 0.0, 0usize);
+    let (mut r_sum, mut r_sq, mut r_n) = (0.0, 0.0, 0usize);
+
+    for &i in indices {
+        let y = targets[i];
+        if data[i][feature] <= threshold {
+            l_sum += y;
+            l_sq += y * y;
+            l_n += 1;
+        } else {
+            r_sum += y;
+            r_sq += y * y;
+            r_n += 1;
+        }
+    }
+
+    (l_sum, l_sq, l_n, r_sum, r_sq, r_n)
+}
+
+/// Population variance of a slice.
+#[allow(clippy::cast_precision_loss)]
+fn variance(values: &[f64]) -> f64 {
+    if values.is_empty() {
+        return 0.0;
+    }
+    let n = values.len() as f64;
+    let mean = values.iter().sum::<f64>() / n;
+    values.iter().map(|v| (v - mean).powi(2)).sum::<f64>() / n
+}
+
+// --- Public API ---
+
+/// Run fANOVA analysis on pre-processed numerical data.
+///
+/// `data` is `n_samples` rows, each with `n_features` columns.
+/// `targets` has one entry per sample.
+/// `feature_names` maps feature index to human-readable name.
+#[allow(clippy::cast_precision_loss)]
+pub(crate) fn compute_fanova(
+    data: &[Vec<f64>],
+    targets: &[f64],
+    feature_names: &[String],
+    config: &FanovaConfig,
+) -> FanovaResult {
+    let n_samples = data.len();
+    let n_features = data[0].len();
+
+    let mut rng: fastrand::Rng = config
+        .seed
+        .map_or_else(fastrand::Rng::new, fastrand::Rng::with_seed);
+
+    // Build random forest with bootstrap sampling
+    let trees: Vec<DecisionTree> = (0..config.n_trees)
+        .map(|_| {
+            let bootstrap: Vec<usize> = (0..n_samples).map(|_| rng.usize(0..n_samples)).collect();
+            DecisionTree::build(data, targets, &bootstrap, config, &mut rng)
+        })
+        .collect();
+
+    // Compute main effects: V_j = Var[E[f | x_j]]
+    let main_var: Vec<f64> = (0..n_features)
+        .map(|j| {
+            let subset = [j];
+            let preds: Vec<f64> = (0..n_samples)
+                .map(|i| {
+                    trees
+                        .iter()
+                        .map(|t| t.marginal_predict(&subset, &data[i]))
+                        .sum::<f64>()
+                        / trees.len() as f64
+                })
+                .collect();
+            variance(&preds)
+        })
+        .collect();
+
+    // Compute pairwise interaction effects: V_{j,k} - V_j - V_k
+    let mut interactions: Vec<((String, String), f64)> = Vec::new();
+    for j in 0..n_features {
+        for k in (j + 1)..n_features {
+            let subset = [j, k];
+            let preds: Vec<f64> = (0..n_samples)
+                .map(|i| {
+                    trees
+                        .iter()
+                        .map(|t| t.marginal_predict(&subset, &data[i]))
+                        .sum::<f64>()
+                        / trees.len() as f64
+                })
+                .collect();
+            let joint = variance(&preds);
+            let interaction = (joint - main_var[j] - main_var[k]).max(0.0);
+            if interaction > 1e-10 {
+                interactions.push((
+                    (feature_names[j].clone(), feature_names[k].clone()),
+                    interaction,
+                ));
+            }
+        }
+    }
+
+    // Normalize so all importances sum to 1.0
+    let total: f64 =
+        main_var.iter().sum::<f64>() + interactions.iter().map(|(_, v)| *v).sum::<f64>();
+
+    let mut main_effects: Vec<(String, f64)> = feature_names
+        .iter()
+        .zip(&main_var)
+        .map(|(name, &v)| (name.clone(), if total > 0.0 { v / total } else { 0.0 }))
+        .collect();
+    main_effects.sort_by(|a, b| b.1.partial_cmp(&a.1).unwrap_or(core::cmp::Ordering::Equal));
+
+    if total > 0.0 {
+        for entry in &mut interactions {
+            entry.1 /= total;
+        }
+    }
+    interactions.sort_by(|a, b| b.1.partial_cmp(&a.1).unwrap_or(core::cmp::Ordering::Equal));
+
+    FanovaResult {
+        main_effects,
+        interactions,
+    }
+}
+
+#[cfg(test)]
+mod tests {
+    use super::*;
+    use crate::rng_util;
+
+    #[test]
+    fn single_dominant_parameter() {
+        // f(x, y) = x — only x matters
+        let mut rng = fastrand::Rng::with_seed(0);
+        let n = 100;
+        let data: Vec<Vec<f64>> = (0..n)
+            .map(|_| {
+                vec![
+                    rng_util::f64_range(&mut rng, 0.0, 10.0),
+                    rng_util::f64_range(&mut rng, 0.0, 10.0),
+                ]
+            })
+            .collect();
+        let targets: Vec<f64> = data.iter().map(|row| row[0]).collect();
+
+        let result = compute_fanova(
+            &data,
+            &targets,
+            &["x".into(), "y".into()],
+            &FanovaConfig::default(),
+        );
+
+        assert_eq!(result.main_effects[0].0, "x");
+        assert!(
+            result.main_effects[0].1 > 0.8,
+            "x importance = {}",
+            result.main_effects[0].1
+        );
+    }
+
+    #[test]
+    fn interaction_detection() {
+        // f(x, y) = x * y — both matter and interact
+        let mut rng = fastrand::Rng::with_seed(42);
+        let n = 200;
+        let data: Vec<Vec<f64>> = (0..n)
+            .map(|_| {
+                vec![
+                    rng_util::f64_range(&mut rng, 0.0, 10.0),
+                    rng_util::f64_range(&mut rng, 0.0, 10.0),
+                ]
+            })
+            .collect();
+        let targets: Vec<f64> = data.iter().map(|row| row[0] * row[1]).collect();
+
+        let config = FanovaConfig {
+            n_trees: 128,
+            ..FanovaConfig::default()
+        };
+        let result = compute_fanova(&data, &targets, &["x".into(), "y".into()], &config);
+
+        assert!(
+            !result.interactions.is_empty(),
+            "should detect x*y interaction"
+        );
+        assert!(
+            result.interactions[0].1 > 0.05,
+            "interaction importance = {}",
+            result.interactions[0].1
+        );
+    }
+
+    #[test]
+    fn variance_computation() {
+        assert!((variance(&[1.0, 2.0, 3.0, 4.0, 5.0]) - 2.0).abs() < 1e-10);
+        assert!(variance(&[5.0, 5.0, 5.0]).abs() < 1e-10);
+        assert!(variance(&[]).abs() < 1e-10);
+    }
+
+    #[test]
+    fn three_params_one_dominant() {
+        // f(x, y, z) = 3*x + 0.1*y + 0*z
+        let mut rng = fastrand::Rng::with_seed(7);
+        let n = 150;
+        let data: Vec<Vec<f64>> = (0..n)
+            .map(|_| {
+                vec![
+                    rng_util::f64_range(&mut rng, 0.0, 10.0),
+                    rng_util::f64_range(&mut rng, 0.0, 10.0),
+                    rng_util::f64_range(&mut rng, 0.0, 10.0),
+                ]
+            })
+            .collect();
+        let targets: Vec<f64> = data.iter().map(|r| 3.0 * r[0] + 0.1 * r[1]).collect();
+
+        let result = compute_fanova(
+            &data,
+            &targets,
+            &["x".into(), "y".into(), "z".into()],
+            &FanovaConfig::default(),
+        );
+
+        // x should be the most important
+        assert_eq!(result.main_effects[0].0, "x");
+        assert!(result.main_effects[0].1 > 0.5);
+
+        // z should have near-zero importance
+        let z_imp = result
+            .main_effects
+            .iter()
+            .find(|(name, _)| name == "z")
+            .map_or(0.0, |(_, v)| *v);
+        assert!(z_imp < 0.1, "z importance = {z_imp}");
+    }
+
+    #[test]
+    fn importances_sum_to_one() {
+        let mut rng = fastrand::Rng::with_seed(3);
+        let n = 100;
+        let data: Vec<Vec<f64>> = (0..n)
+            .map(|_| {
+                vec![
+                    rng_util::f64_range(&mut rng, 0.0, 10.0),
+                    rng_util::f64_range(&mut rng, 0.0, 10.0),
+                ]
+            })
+            .collect();
+        let targets: Vec<f64> = data.iter().map(|r| r[0] + r[1]).collect();
+
+        let result = compute_fanova(
+            &data,
+            &targets,
+            &["x".into(), "y".into()],
+            &FanovaConfig::default(),
+        );
+
+        let total: f64 = result.main_effects.iter().map(|(_, v)| *v).sum::<f64>()
+            + result.interactions.iter().map(|(_, v)| *v).sum::<f64>();
+        assert!(
+            (total - 1.0).abs() < 1e-10,
+            "importances should sum to 1.0, got {total}"
+        );
+    }
+}

src/importance.rs

@@ -0,0 +1,93 @@
+//! Parameter importance via Spearman rank correlation.
+
+/// Assign average ranks to a slice of `f64` values (handles ties).
+#[allow(clippy::cast_precision_loss, clippy::float_cmp)]
+pub(crate) fn rank(values: &[f64]) -> Vec<f64> {
+    let n = values.len();
+    let mut indexed: Vec<(usize, f64)> = values.iter().copied().enumerate().collect();
+    indexed.sort_by(|a, b| a.1.partial_cmp(&b.1).unwrap_or(core::cmp::Ordering::Equal));
+
+    let mut ranks = vec![0.0; n];
+    let mut i = 0;
+    while i < n {
+        // Find the run of tied values.
+        let mut j = i + 1;
+        while j < n && indexed[j].1 == indexed[i].1 {
+            j += 1;
+        }
+        // Average rank for the tie group (1-based ranks).
+        let avg = (i + 1..=j).sum::<usize>() as f64 / (j - i) as f64;
+        for item in &indexed[i..j] {
+            ranks[item.0] = avg;
+        }
+        i = j;
+    }
+    ranks
+}
+
+/// Pearson correlation coefficient on two equal-length slices.
+#[allow(clippy::cast_precision_loss)]
+fn pearson(x: &[f64], y: &[f64]) -> f64 {
+    let n = x.len() as f64;
+    let mean_x = x.iter().sum::<f64>() / n;
+    let mean_y = y.iter().sum::<f64>() / n;
+
+    let mut cov = 0.0;
+    let mut var_x = 0.0;
+    let mut var_y = 0.0;
+    for (xi, yi) in x.iter().zip(y.iter()) {
+        let dx = xi - mean_x;
+        let dy = yi - mean_y;
+        cov += dx * dy;
+        var_x += dx * dx;
+        var_y += dy * dy;
+    }
+
+    let denom = (var_x * var_y).sqrt();
+    if denom == 0.0 { 0.0 } else { cov / denom }
+}
+
+/// Spearman rank correlation (Pearson on ranks).
+pub(crate) fn spearman(x: &[f64], y: &[f64]) -> f64 {
+    pearson(&rank(x), &rank(y))
+}
+
+#[cfg(test)]
+mod tests {
+    use super::*;
+
+    #[test]
+    fn rank_no_ties() {
+        let ranks = rank(&[30.0, 10.0, 20.0]);
+        assert_eq!(ranks, vec![3.0, 1.0, 2.0]);
+    }
+
+    #[test]
+    fn rank_with_ties() {
+        let ranks = rank(&[10.0, 20.0, 20.0, 30.0]);
+        assert_eq!(ranks, vec![1.0, 2.5, 2.5, 4.0]);
+    }
+
+    #[test]
+    fn perfect_positive_correlation() {
+        let x = vec![1.0, 2.0, 3.0, 4.0, 5.0];
+        let y = vec![2.0, 4.0, 6.0, 8.0, 10.0];
+        let r = spearman(&x, &y);
+        assert!((r - 1.0).abs() < 1e-10);
+    }
+
+    #[test]
+    fn perfect_negative_correlation() {
+        let x = vec![1.0, 2.0, 3.0, 4.0, 5.0];
+        let y = vec![10.0, 8.0, 6.0, 4.0, 2.0];
+        let r = spearman(&x, &y);
+        assert!((r + 1.0).abs() < 1e-10);
+    }
+
+    #[test]
+    fn zero_variance_returns_zero() {
+        let x = vec![5.0, 5.0, 5.0];
+        let y = vec![1.0, 2.0, 3.0];
+        assert!(spearman(&x, &y).abs() < f64::EPSILON);
+    }
+}

src/kde/multivariate.rs

@@ -5,8 +5,6 @@
 //! parameter independently, the multivariate KDE models the joint distribution
 //! to better capture correlations between parameters.
 
-use rand::{Rng, RngExt};
-
 use crate::error::{Error, Result};
 
 /// A multivariate Gaussian kernel density estimator for joint distributions.
@@ -308,9 +306,9 @@ impl MultivariateKDE {
     /// # Returns
     ///
     /// A `Vec<f64>` of length `n_dims` representing a sample from the KDE.
-    pub(crate) fn sample<R: Rng>(&self, rng: &mut R) -> Vec<f64> {
+    pub(crate) fn sample(&self, rng: &mut fastrand::Rng) -> Vec<f64> {
         // Select a random sample to center the kernel on
-        let idx = rng.random_range(0..self.samples.len());
+        let idx = rng.usize(0..self.samples.len());
         let center = &self.samples[idx];
 
         // Add independent Gaussian noise to each dimension
@@ -319,8 +317,8 @@ impl MultivariateKDE {
             .iter()
             .zip(self.bandwidths.iter())
             .map(|(&center_j, &bandwidth_j)| {
-                let u1: f64 = rng.random();
-                let u2: f64 = rng.random();
+                let u1: f64 = rng.f64();
+                let u2: f64 = rng.f64();
 
                 // Box-Muller transform: generates standard normal variate
                 let z = (-2.0 * u1.ln()).sqrt() * (2.0 * core::f64::consts::PI * u2).cos();
@@ -724,7 +722,7 @@ mod tests {
     fn test_multivariate_kde_sample_basic() {
         let samples = vec![vec![0.0, 0.0], vec![1.0, 1.0], vec![2.0, 2.0]];
         let kde = MultivariateKDE::new(samples).unwrap();
-        let mut rng = rand::rng();
+        let mut rng = fastrand::Rng::new();
 
         // Sample should have correct dimensionality
         let sample = kde.sample(&mut rng);
@@ -741,7 +739,7 @@ mod tests {
             vec![4.0, 4.0],
         ];
         let kde = MultivariateKDE::new(samples).unwrap();
-        let mut rng = rand::rng();
+        let mut rng = fastrand::Rng::new();
 
         // Samples should generally be in a reasonable range around the data
         for _ in 0..100 {
@@ -766,7 +764,7 @@ mod tests {
         // When KDE has only one sample, all samples should be centered around it
         let samples = vec![vec![5.0, 10.0]];
         let kde = MultivariateKDE::new(samples).unwrap();
-        let mut rng = rand::rng();
+        let mut rng = fastrand::Rng::new();
 
         // Generate many samples and check they cluster around (5.0, 10.0)
         let n_samples = 100;
@@ -803,7 +801,7 @@ mod tests {
             })
             .collect();
         let kde = MultivariateKDE::new(samples).unwrap();
-        let mut rng = rand::rng();
+        let mut rng = fastrand::Rng::new();
 
         // Sample should have correct dimensionality
         for _ in 0..50 {
@@ -823,7 +821,7 @@ mod tests {
         let data = vec![vec![0.0, 0.0], vec![0.0, 0.0], vec![0.0, 0.0]];
         let bandwidths = vec![0.1, 10.0]; // Small bandwidth in x, large in y
         let kde = MultivariateKDE::with_bandwidths(data, bandwidths).unwrap();
-        let mut rng = rand::rng();
+        let mut rng = fastrand::Rng::new();
 
         // Generate samples and check variance in each dimension
         let n_samples = 1000;
@@ -868,7 +866,7 @@ mod tests {
             vec![4.0, 4.0],
         ];
         let kde = MultivariateKDE::new(data).unwrap();
-        let mut rng = rand::rng();
+        let mut rng = fastrand::Rng::new();
 
         // Sample many points and verify the mean is near the center
         let n_samples = 500;
@@ -896,14 +894,12 @@ mod tests {
 
     #[test]
     fn test_multivariate_kde_sample_deterministic_with_seeded_rng() {
-        use rand::SeedableRng;
-
         let data = vec![vec![1.0, 2.0], vec![3.0, 4.0]];
         let kde = MultivariateKDE::new(data).unwrap();
 
         // Use a seeded RNG for reproducibility
-        let mut rng1 = rand::rngs::StdRng::seed_from_u64(42);
-        let mut rng2 = rand::rngs::StdRng::seed_from_u64(42);
+        let mut rng1 = fastrand::Rng::with_seed(42);
+        let mut rng2 = fastrand::Rng::with_seed(42);
 
         // Same seed should produce same samples
         let result1 = kde.sample(&mut rng1);

src/kde/univariate.rs

@@ -3,8 +3,6 @@
 //! This module provides a Gaussian kernel density estimator used by the TPE
 //! sampler to model probability distributions over good and bad trial regions.
 
-use rand::{Rng, RngExt};
-
 use crate::error::{Error, Result};
 
 /// A Gaussian kernel density estimator for continuous distributions.
@@ -26,7 +24,7 @@ use crate::error::{Error, Result};
 /// assert!(density > 0.0);
 ///
 /// // Sample from the estimated distribution
-/// let mut rng = rand::rng();
+/// let mut rng = fastrand::Rng::new();
 /// let sample = kde.sample(&mut rng);
 /// ```
 #[derive(Clone, Debug)]
@@ -132,15 +130,15 @@ impl KernelDensityEstimator {
     /// Sampling works by:
     /// 1. Uniformly selecting one of the kernel centers (samples)
     /// 2. Adding Gaussian noise with the bandwidth as standard deviation
-    pub(crate) fn sample<R: Rng>(&self, rng: &mut R) -> f64 {
+    pub(crate) fn sample(&self, rng: &mut fastrand::Rng) -> f64 {
         // Select a random sample to center the kernel on
-        let idx = rng.random_range(0..self.samples.len());
+        let idx = rng.usize(0..self.samples.len());
         let center = self.samples[idx];
 
         // Add Gaussian noise with bandwidth as standard deviation
         // Using Box-Muller transform for Gaussian sampling
-        let u1: f64 = rng.random();
-        let u2: f64 = rng.random();
+        let u1: f64 = rng.f64();
+        let u2: f64 = rng.f64();
 
         let z = (-2.0 * u1.ln()).sqrt() * (2.0 * core::f64::consts::PI * u2).cos();
         center + z * self.bandwidth
@@ -211,7 +209,7 @@ mod tests {
     fn test_kde_sample_in_reasonable_range() {
         let samples = vec![0.0, 1.0, 2.0, 3.0, 4.0];
         let kde = KernelDensityEstimator::new(samples).unwrap();
-        let mut rng = rand::rng();
+        let mut rng = fastrand::Rng::new();
 
         // Samples should generally be in a reasonable range around the data
         for _ in 0..100 {

src/lib.rs

@@ -17,6 +17,15 @@
 //! - **Random Search** - Simple random sampling for baseline comparisons
 //! - **TPE (Tree-Parzen Estimator)** - Bayesian optimization for efficient search
 //! - **Grid Search** - Exhaustive search over a specified parameter grid
+//! - **Sobol (QMC)** - Quasi-random sampling for better space coverage (requires `sobol` feature)
+//! - **CMA-ES** - Covariance Matrix Adaptation Evolution Strategy for continuous optimization (requires `cma-es` feature)
+//! - **DE** - Differential Evolution for population-based global optimization
+//! - **GP** - Gaussian Process Bayesian optimization with Expected Improvement (requires `gp` feature)
+//! - **BOHB** - Bayesian Optimization + `HyperBand` for budget-aware TPE sampling
+//! - **NSGA-II** - Non-dominated Sorting Genetic Algorithm II for multi-objective optimization
+//! - **NSGA-III** - Reference-point-based NSGA for many-objective (3+) optimization
+//! - **MOEA/D** - Decomposition-based multi-objective with Tchebycheff, Weighted Sum, or PBI
+//! - **MOTPE** - Multi-Objective Tree-Parzen Estimator for Bayesian multi-objective optimization
 //!
 //! Additional features include:
 //!
@@ -182,31 +191,95 @@
 //!
 //! - `async`: Enable async optimization methods (requires tokio)
 //! - `derive`: Enable `#[derive(Categorical)]` for enum parameters
+//! - `serde`: Enable `Serialize`/`Deserialize` on public types and `Study::save()`/`Study::load()`
+//! - `sobol`: Enable the Sobol quasi-random sampler for better space coverage
+//! - `cma-es`: Enable the CMA-ES sampler for continuous optimization
+//! - `gp`: Enable the Gaussian Process sampler for Bayesian optimization
+//! - `visualization`: Generate self-contained HTML reports with interactive Plotly.js charts
+//! - `tracing`: Emit structured log events via the [`tracing`](https://docs.rs/tracing) crate at key optimization points
+
+/// Emit a `tracing::info!` event when the `tracing` feature is enabled.
+/// No-op otherwise.
+#[cfg(feature = "tracing")]
+macro_rules! trace_info {
+    ($($arg:tt)*) => { tracing::info!($($arg)*) };
+}
+
+#[cfg(not(feature = "tracing"))]
+macro_rules! trace_info {
+    ($($arg:tt)*) => {};
+}
+
+/// Emit a `tracing::debug!` event when the `tracing` feature is enabled.
+/// No-op otherwise.
+#[cfg(feature = "tracing")]
+macro_rules! trace_debug {
+    ($($arg:tt)*) => { tracing::debug!($($arg)*) };
+}
+
+#[cfg(not(feature = "tracing"))]
+macro_rules! trace_debug {
+    ($($arg:tt)*) => {};
+}
 
 mod distribution;
 mod error;
+mod fanova;
+mod importance;
 mod kde;
+pub mod multi_objective;
 mod param;
 pub mod parameter;
+pub mod pareto;
+pub mod pruner;
+mod rng_util;
 pub mod sampler;
+pub mod storage;
 mod study;
 mod trial;
 mod types;
+mod visualization;
 
-pub use error::{Error, Result};
+pub use error::{Error, Result, TrialPruned};
+pub use fanova::{FanovaConfig, FanovaResult};
+pub use multi_objective::{MultiObjectiveSampler, MultiObjectiveStudy, MultiObjectiveTrial};
 #[cfg(feature = "derive")]
 pub use optimizer_derive::Categorical;
 pub use param::ParamValue;
 pub use parameter::{
     BoolParam, Categorical, CategoricalParam, EnumParam, FloatParam, IntParam, ParamId, Parameter,
 };
+pub use pruner::{
+    HyperbandPruner, MedianPruner, NopPruner, PatientPruner, PercentilePruner, Pruner,
+    SuccessiveHalvingPruner, ThresholdPruner, WilcoxonPruner,
+};
 pub use sampler::CompletedTrial;
+pub use sampler::bohb::BohbSampler;
+#[cfg(feature = "cma-es")]
+pub use sampler::cma_es::CmaEsSampler;
+pub use sampler::differential_evolution::{
+    DifferentialEvolutionSampler, DifferentialEvolutionStrategy,
+};
+#[cfg(feature = "gp")]
+pub use sampler::gp::GpSampler;
 pub use sampler::grid::GridSearchSampler;
+pub use sampler::moead::{Decomposition, MoeadSampler};
+pub use sampler::motpe::MotpeSampler;
+pub use sampler::nsga2::Nsga2Sampler;
+pub use sampler::nsga3::Nsga3Sampler;
 pub use sampler::random::RandomSampler;
+#[cfg(feature = "sobol")]
+pub use sampler::sobol::SobolSampler;
 pub use sampler::tpe::TpeSampler;
-pub use study::Study;
-pub use trial::Trial;
+#[cfg(feature = "journal")]
+pub use storage::JournalStorage;
+pub use storage::{MemoryStorage, Storage};
+#[cfg(feature = "serde")]
+pub use study::StudySnapshot;
+pub use study::{Study, StudyBuilder};
+pub use trial::{AttrValue, Trial};
 pub use types::{Direction, TrialState};
+pub use visualization::generate_html_report;
 
 /// Convenient wildcard import for the most common types.
 ///
@@ -217,16 +290,42 @@ pub mod prelude {
     #[cfg(feature = "derive")]
     pub use optimizer_derive::Categorical as DeriveCategory;
 
-    pub use crate::error::{Error, Result};
+    pub use crate::error::{Error, Result, TrialPruned};
+    pub use crate::fanova::{FanovaConfig, FanovaResult};
+    pub use crate::multi_objective::{MultiObjectiveStudy, MultiObjectiveTrial};
     pub use crate::param::ParamValue;
     pub use crate::parameter::{
         BoolParam, Categorical, CategoricalParam, EnumParam, FloatParam, IntParam, Parameter,
     };
+    pub use crate::pruner::{
+        HyperbandPruner, MedianPruner, NopPruner, PatientPruner, PercentilePruner, Pruner,
+        SuccessiveHalvingPruner, ThresholdPruner,
+    };
     pub use crate::sampler::CompletedTrial;
+    pub use crate::sampler::bohb::BohbSampler;
+    #[cfg(feature = "cma-es")]
+    pub use crate::sampler::cma_es::CmaEsSampler;
+    pub use crate::sampler::differential_evolution::{
+        DifferentialEvolutionSampler, DifferentialEvolutionStrategy,
+    };
+    #[cfg(feature = "gp")]
+    pub use crate::sampler::gp::GpSampler;
     pub use crate::sampler::grid::GridSearchSampler;
+    pub use crate::sampler::moead::{Decomposition, MoeadSampler};
+    pub use crate::sampler::motpe::MotpeSampler;
+    pub use crate::sampler::nsga2::Nsga2Sampler;
+    pub use crate::sampler::nsga3::Nsga3Sampler;
     pub use crate::sampler::random::RandomSampler;
+    #[cfg(feature = "sobol")]
+    pub use crate::sampler::sobol::SobolSampler;
     pub use crate::sampler::tpe::TpeSampler;
-    pub use crate::study::Study;
-    pub use crate::trial::Trial;
+    #[cfg(feature = "journal")]
+    pub use crate::storage::JournalStorage;
+    pub use crate::storage::{MemoryStorage, Storage};
+    #[cfg(feature = "serde")]
+    pub use crate::study::StudySnapshot;
+    pub use crate::study::{Study, StudyBuilder};
+    pub use crate::trial::{AttrValue, Trial};
     pub use crate::types::Direction;
+    pub use crate::visualization::generate_html_report;
 }

src/multi_objective.rs

@@ -0,0 +1,412 @@
+//! Multi-objective optimization via a dedicated study type.
+//!
+//! [`MultiObjectiveStudy`] manages trials that return multiple objective
+//! values. It supports arbitrary numbers of objectives with per-objective
+//! directions (minimize or maximize). Use [`pareto_front()`](MultiObjectiveStudy::pareto_front)
+//! to retrieve the Pareto-optimal solutions.
+//!
+//! # Examples
+//!
+//! ```
+//! use optimizer::Direction;
+//! use optimizer::multi_objective::MultiObjectiveStudy;
+//! use optimizer::parameter::{FloatParam, Parameter};
+//!
+//! let study = MultiObjectiveStudy::new(vec![Direction::Minimize, Direction::Minimize]);
+//! let x = FloatParam::new(0.0, 1.0);
+//!
+//! study
+//!     .optimize(20, |trial| {
+//!         let xv = x.suggest(trial)?;
+//!         Ok::<_, optimizer::Error>(vec![xv, 1.0 - xv])
+//!     })
+//!     .unwrap();
+//!
+//! let front = study.pareto_front();
+//! assert!(!front.is_empty());
+//! ```
+
+use core::sync::atomic::{AtomicU64, Ordering};
+use std::collections::HashMap;
+use std::sync::Arc;
+
+use parking_lot::RwLock;
+
+use crate::distribution::Distribution;
+use crate::param::ParamValue;
+use crate::parameter::{ParamId, Parameter};
+use crate::pruner::NopPruner;
+use crate::sampler::random::RandomSampler;
+use crate::sampler::{CompletedTrial, Sampler};
+use crate::trial::{AttrValue, Trial};
+use crate::types::{Direction, TrialState};
+
+// ---------------------------------------------------------------------------
+// MultiObjectiveTrial
+// ---------------------------------------------------------------------------
+
+/// A completed trial with multiple objective values.
+#[derive(Clone, Debug)]
+#[cfg_attr(feature = "serde", derive(serde::Serialize, serde::Deserialize))]
+pub struct MultiObjectiveTrial {
+    /// The unique identifier for this trial.
+    pub id: u64,
+    /// The sampled parameter values, keyed by parameter id.
+    pub params: HashMap<ParamId, ParamValue>,
+    /// The parameter distributions used, keyed by parameter id.
+    pub distributions: HashMap<ParamId, Distribution>,
+    /// Human-readable labels for parameters, keyed by parameter id.
+    pub param_labels: HashMap<ParamId, String>,
+    /// The objective values (one per objective).
+    pub values: Vec<f64>,
+    /// The state of the trial.
+    pub state: TrialState,
+    /// User-defined attributes stored during the trial.
+    pub user_attrs: HashMap<String, AttrValue>,
+    /// Constraint values for this trial (<=0.0 means feasible).
+    #[cfg_attr(feature = "serde", serde(default))]
+    pub constraints: Vec<f64>,
+}
+
+impl MultiObjectiveTrial {
+    /// Returns the typed value for the given parameter.
+    ///
+    /// Returns `None` if the parameter was not used in this trial.
+    ///
+    /// # Panics
+    ///
+    /// Panics if the stored value is incompatible with the parameter type.
+    pub fn get<P: Parameter>(&self, param: &P) -> Option<P::Value> {
+        self.params.get(&param.id()).map(|v| {
+            param
+                .cast_param_value(v)
+                .expect("parameter type mismatch: stored value incompatible with parameter")
+        })
+    }
+
+    /// Returns `true` if all constraints are satisfied (values <= 0.0).
+    ///
+    /// A trial with no constraints is considered feasible.
+    #[must_use]
+    pub fn is_feasible(&self) -> bool {
+        self.constraints.iter().all(|&c| c <= 0.0)
+    }
+
+    /// Gets a user attribute by key.
+    #[must_use]
+    pub fn user_attr(&self, key: &str) -> Option<&AttrValue> {
+        self.user_attrs.get(key)
+    }
+
+    /// Returns all user attributes.
+    #[must_use]
+    pub fn user_attrs(&self) -> &HashMap<String, AttrValue> {
+        &self.user_attrs
+    }
+}
+
+// ---------------------------------------------------------------------------
+// MultiObjectiveSampler trait
+// ---------------------------------------------------------------------------
+
+/// Trait for samplers aware of multi-objective history.
+///
+/// Separate from [`Sampler`] because NSGA-II needs access to
+/// `&[MultiObjectiveTrial]` (with vector-valued objectives) and
+/// `&[Direction]` (one direction per objective).
+pub trait MultiObjectiveSampler: Send + Sync {
+    /// Samples a parameter value from the given distribution.
+    fn sample(
+        &self,
+        distribution: &Distribution,
+        trial_id: u64,
+        history: &[MultiObjectiveTrial],
+        directions: &[Direction],
+    ) -> ParamValue;
+}
+
+// ---------------------------------------------------------------------------
+// RandomMultiObjectiveSampler
+// ---------------------------------------------------------------------------
+
+/// Default MO sampler that delegates to [`RandomSampler`].
+pub(crate) struct RandomMultiObjectiveSampler(RandomSampler);
+
+impl RandomMultiObjectiveSampler {
+    pub(crate) fn new() -> Self {
+        Self(RandomSampler::new())
+    }
+}
+
+impl MultiObjectiveSampler for RandomMultiObjectiveSampler {
+    fn sample(
+        &self,
+        distribution: &Distribution,
+        trial_id: u64,
+        _history: &[MultiObjectiveTrial],
+        _directions: &[Direction],
+    ) -> ParamValue {
+        self.0.sample(distribution, trial_id, &[])
+    }
+}
+
+// ---------------------------------------------------------------------------
+// MoSamplerBridge — bridges MultiObjectiveSampler to Sampler trait
+// ---------------------------------------------------------------------------
+
+/// Bridges a [`MultiObjectiveSampler`] to the [`Sampler`] trait so that
+/// `Trial::with_sampler()` can use it.
+struct MoSamplerBridge {
+    inner: Arc<dyn MultiObjectiveSampler>,
+    history: Arc<RwLock<Vec<MultiObjectiveTrial>>>,
+    directions: Vec<Direction>,
+}
+
+impl Sampler for MoSamplerBridge {
+    fn sample(
+        &self,
+        distribution: &Distribution,
+        trial_id: u64,
+        _history: &[CompletedTrial],
+    ) -> ParamValue {
+        let mo_history = self.history.read();
+        self.inner
+            .sample(distribution, trial_id, &mo_history, &self.directions)
+    }
+}
+
+// ---------------------------------------------------------------------------
+// MultiObjectiveStudy
+// ---------------------------------------------------------------------------
+
+/// A study for multi-objective optimization.
+///
+/// Manages trials that return multiple objective values. Supports
+/// arbitrary numbers of objectives with independent minimize/maximize
+/// directions.
+///
+/// # Examples
+///
+/// ```
+/// use optimizer::Direction;
+/// use optimizer::multi_objective::MultiObjectiveStudy;
+/// use optimizer::parameter::{FloatParam, Parameter};
+///
+/// // Bi-objective: minimize both
+/// let study = MultiObjectiveStudy::new(vec![Direction::Minimize, Direction::Minimize]);
+/// let x = FloatParam::new(0.0, 1.0);
+///
+/// study
+///     .optimize(30, |trial| {
+///         let xv = x.suggest(trial)?;
+///         Ok::<_, optimizer::Error>(vec![xv, 1.0 - xv])
+///     })
+///     .unwrap();
+///
+/// let front = study.pareto_front();
+/// assert!(!front.is_empty());
+/// ```
+pub struct MultiObjectiveStudy {
+    directions: Vec<Direction>,
+    sampler: Arc<dyn MultiObjectiveSampler>,
+    completed_trials: Arc<RwLock<Vec<MultiObjectiveTrial>>>,
+    next_trial_id: AtomicU64,
+}
+
+impl MultiObjectiveStudy {
+    /// Creates a new multi-objective study with the given directions.
+    ///
+    /// Uses a random sampler by default.
+    ///
+    /// # Arguments
+    ///
+    /// * `directions` - One direction per objective (minimize or maximize).
+    #[must_use]
+    pub fn new(directions: Vec<Direction>) -> Self {
+        Self {
+            directions,
+            sampler: Arc::new(RandomMultiObjectiveSampler::new()),
+            completed_trials: Arc::new(RwLock::new(Vec::new())),
+            next_trial_id: AtomicU64::new(0),
+        }
+    }
+
+    /// Creates a new study with a custom multi-objective sampler.
+    #[must_use]
+    pub fn with_sampler(
+        directions: Vec<Direction>,
+        sampler: impl MultiObjectiveSampler + 'static,
+    ) -> Self {
+        Self {
+            directions,
+            sampler: Arc::new(sampler),
+            completed_trials: Arc::new(RwLock::new(Vec::new())),
+            next_trial_id: AtomicU64::new(0),
+        }
+    }
+
+    /// Returns the optimization directions.
+    #[must_use]
+    pub fn directions(&self) -> &[Direction] {
+        &self.directions
+    }
+
+    /// Returns the number of objectives.
+    #[must_use]
+    pub fn n_objectives(&self) -> usize {
+        self.directions.len()
+    }
+
+    /// Returns the number of completed trials.
+    #[must_use]
+    pub fn n_trials(&self) -> usize {
+        self.completed_trials.read().len()
+    }
+
+    /// Returns all completed trials.
+    #[must_use]
+    pub fn trials(&self) -> Vec<MultiObjectiveTrial> {
+        self.completed_trials.read().clone()
+    }
+
+    /// Returns the Pareto-optimal trials (front 0).
+    #[must_use]
+    pub fn pareto_front(&self) -> Vec<MultiObjectiveTrial> {
+        let trials = self.completed_trials.read();
+        let complete: Vec<_> = trials
+            .iter()
+            .filter(|t| t.state == TrialState::Complete)
+            .collect();
+
+        if complete.is_empty() {
+            return Vec::new();
+        }
+
+        let values: Vec<Vec<f64>> = complete.iter().map(|t| t.values.clone()).collect();
+        let fronts = crate::pareto::fast_non_dominated_sort(&values, &self.directions);
+
+        if fronts.is_empty() {
+            return Vec::new();
+        }
+
+        fronts[0].iter().map(|&i| complete[i].clone()).collect()
+    }
+
+    /// Creates a new trial wired to the study's MO sampler.
+    fn create_trial(&self) -> Trial {
+        let id = self.next_trial_id.fetch_add(1, Ordering::SeqCst);
+
+        let bridge: Arc<dyn Sampler> = Arc::new(MoSamplerBridge {
+            inner: Arc::clone(&self.sampler),
+            history: Arc::clone(&self.completed_trials),
+            directions: self.directions.clone(),
+        });
+
+        // Dummy f64 history — the bridge ignores it.
+        let dummy_history: Arc<RwLock<Vec<CompletedTrial<f64>>>> =
+            Arc::new(RwLock::new(Vec::new()));
+
+        Trial::with_sampler(id, bridge, dummy_history, Arc::new(NopPruner))
+    }
+
+    /// Records a completed trial.
+    fn complete_trial(&self, mut trial: Trial, values: Vec<f64>) {
+        trial.set_complete();
+        let mo_trial = MultiObjectiveTrial {
+            id: trial.id(),
+            params: trial.params().clone(),
+            distributions: trial.distributions().clone(),
+            param_labels: trial.param_labels().clone(),
+            values,
+            state: TrialState::Complete,
+            user_attrs: trial.user_attrs().clone(),
+            constraints: trial.constraint_values().to_vec(),
+        };
+        self.completed_trials.write().push(mo_trial);
+    }
+
+    /// Records a failed trial (not stored in history).
+    fn fail_trial(trial: &mut Trial) {
+        trial.set_failed();
+    }
+
+    /// Request a new trial for the ask/tell interface.
+    ///
+    /// After creating the trial, suggest parameters on it, evaluate your
+    /// objective externally, then pass the trial back to [`tell()`](Self::tell).
+    pub fn ask(&self) -> Trial {
+        self.create_trial()
+    }
+
+    /// Report the result of a trial obtained from [`ask()`](Self::ask).
+    ///
+    /// Pass `Ok(values)` for a successful evaluation or `Err(reason)` for a failure.
+    ///
+    /// # Errors
+    ///
+    /// Returns `ObjectiveDimensionMismatch` if the number of values doesn't
+    /// match the number of directions.
+    pub fn tell(
+        &self,
+        mut trial: Trial,
+        result: core::result::Result<Vec<f64>, impl ToString>,
+    ) -> crate::Result<()> {
+        if let Ok(values) = result {
+            if values.len() != self.directions.len() {
+                return Err(crate::Error::ObjectiveDimensionMismatch {
+                    expected: self.directions.len(),
+                    got: values.len(),
+                });
+            }
+            self.complete_trial(trial, values);
+        } else {
+            Self::fail_trial(&mut trial);
+        }
+        Ok(())
+    }
+
+    /// Runs multi-objective optimization for `n_trials` trials.
+    ///
+    /// The objective function must return a `Vec<f64>` with one value per
+    /// objective.
+    ///
+    /// # Errors
+    ///
+    /// Returns `ObjectiveDimensionMismatch` if the objective returns the wrong
+    /// number of values. Returns `NoCompletedTrials` if all trials fail.
+    pub fn optimize<F, E>(&self, n_trials: usize, mut objective: F) -> crate::Result<()>
+    where
+        F: FnMut(&mut Trial) -> core::result::Result<Vec<f64>, E>,
+        E: ToString,
+    {
+        for _ in 0..n_trials {
+            let mut trial = self.create_trial();
+
+            match objective(&mut trial) {
+                Ok(values) => {
+                    if values.len() != self.directions.len() {
+                        return Err(crate::Error::ObjectiveDimensionMismatch {
+                            expected: self.directions.len(),
+                            got: values.len(),
+                        });
+                    }
+                    self.complete_trial(trial, values);
+                }
+                Err(_) => {
+                    Self::fail_trial(&mut trial);
+                }
+            }
+        }
+
+        let has_complete = self
+            .completed_trials
+            .read()
+            .iter()
+            .any(|t| t.state == TrialState::Complete);
+        if !has_complete {
+            return Err(crate::Error::NoCompletedTrials);
+        }
+
+        Ok(())
+    }
+}

src/param.rs

@@ -6,6 +6,7 @@
 /// For categorical parameters, the `Categorical` variant stores
 /// the index into the choices array.
 #[derive(Clone, Debug, PartialEq)]
+#[cfg_attr(feature = "serde", derive(serde::Serialize, serde::Deserialize))]
 pub enum ParamValue {
     /// A floating-point parameter value.
     Float(f64),

src/parameter.rs

@@ -21,6 +21,7 @@
 //! ```
 
 use core::fmt::Debug;
+use core::ops::RangeInclusive;
 use core::sync::atomic::{AtomicU64, Ordering};
 
 use crate::distribution::{
@@ -36,7 +37,8 @@ static NEXT_PARAM_ID: AtomicU64 = AtomicU64::new(0);
 ///
 /// Each parameter is assigned a unique `ParamId` at creation time. Cloning a parameter
 /// copies its `ParamId`, so clones refer to the same logical parameter.
-#[derive(Clone, Copy, Debug, PartialEq, Eq, Hash)]
+#[derive(Clone, Copy, Debug, PartialEq, Eq, PartialOrd, Ord, Hash)]
+#[cfg_attr(feature = "serde", derive(serde::Serialize, serde::Deserialize))]
 pub struct ParamId(u64);
 
 impl ParamId {
@@ -187,6 +189,12 @@ impl FloatParam {
     }
 }
 
+impl From<RangeInclusive<f64>> for FloatParam {
+    fn from(range: RangeInclusive<f64>) -> Self {
+        FloatParam::new(*range.start(), *range.end())
+    }
+}
+
 impl Parameter for FloatParam {
     type Value = f64;
 
@@ -306,6 +314,12 @@ impl IntParam {
     }
 }
 
+impl From<RangeInclusive<i64>> for IntParam {
+    fn from(range: RangeInclusive<i64>) -> Self {
+        IntParam::new(*range.start(), *range.end())
+    }
+}
+
 impl Parameter for IntParam {
     type Value = i64;
 
@@ -992,4 +1006,34 @@ mod tests {
         let cloned = param.clone();
         assert_eq!(param.id(), cloned.id());
     }
+
+    #[test]
+    fn float_param_from_range() {
+        let param = FloatParam::from(0.0..=1.0);
+        assert_eq!(
+            param.distribution(),
+            Distribution::Float(FloatDistribution {
+                low: 0.0,
+                high: 1.0,
+                log_scale: false,
+                step: None,
+            })
+        );
+        assert_eq!(param.label(), format!("{param:?}"));
+    }
+
+    #[test]
+    fn int_param_from_range() {
+        let param = IntParam::from(1..=10);
+        assert_eq!(
+            param.distribution(),
+            Distribution::Int(IntDistribution {
+                low: 1,
+                high: 10,
+                log_scale: false,
+                step: None,
+            })
+        );
+        assert_eq!(param.label(), format!("{param:?}"));
+    }
 }

src/pareto.rs

@@ -0,0 +1,593 @@
+//! Pareto front analysis utilities for multi-objective optimization.
+//!
+//! Provides functions for analyzing and working with Pareto fronts:
+//!
+//! - [`hypervolume`] — measure the quality of a Pareto front
+//! - [`non_dominated_sort`] — rank solutions into successive fronts
+//! - [`pareto_front_indices`] — filter to non-dominated solutions only
+//! - [`crowding_distance`] — measure diversity within a front
+//!
+//! Internally also provides fast non-dominated sorting (Deb et al., 2002)
+//! used by [`MultiObjectiveStudy::pareto_front()`](crate::MultiObjectiveStudy::pareto_front)
+//! and [`Nsga2Sampler`](crate::Nsga2Sampler).
+
+use crate::types::Direction;
+
+/// Returns `true` if solution `a` Pareto-dominates solution `b`.
+///
+/// A solution dominates another if it is at least as good in all objectives
+/// and strictly better in at least one, respecting the given directions.
+#[allow(clippy::module_name_repetitions)]
+pub(crate) fn dominates(a: &[f64], b: &[f64], directions: &[Direction]) -> bool {
+    debug_assert_eq!(a.len(), b.len());
+    debug_assert_eq!(a.len(), directions.len());
+
+    let mut strictly_better = false;
+    for ((&av, &bv), dir) in a.iter().zip(b.iter()).zip(directions.iter()) {
+        let better = match dir {
+            Direction::Minimize => av < bv,
+            Direction::Maximize => av > bv,
+        };
+        let worse = match dir {
+            Direction::Minimize => av > bv,
+            Direction::Maximize => av < bv,
+        };
+        if worse {
+            return false;
+        }
+        if better {
+            strictly_better = true;
+        }
+    }
+    strictly_better
+}
+
+/// Constrained dominance: feasible beats infeasible, among infeasible
+/// prefer lower total constraint violation, among feasible use Pareto dominance.
+pub(crate) fn constrained_dominates(
+    a_values: &[f64],
+    b_values: &[f64],
+    a_constraints: &[f64],
+    b_constraints: &[f64],
+    directions: &[Direction],
+) -> bool {
+    let a_feasible = a_constraints.iter().all(|&c| c <= 0.0);
+    let b_feasible = b_constraints.iter().all(|&c| c <= 0.0);
+
+    match (a_feasible, b_feasible) {
+        (true, false) => true,
+        (false, true) => false,
+        (false, false) => {
+            let a_violation: f64 = a_constraints.iter().map(|c| c.max(0.0)).sum();
+            let b_violation: f64 = b_constraints.iter().map(|c| c.max(0.0)).sum();
+            a_violation < b_violation
+        }
+        (true, true) => dominates(a_values, b_values, directions),
+    }
+}
+
+/// Fast non-dominated sorting (Deb et al., 2002).
+///
+/// Returns `Vec<Vec<usize>>` where `fronts[0]` is the Pareto front,
+/// each inner vec contains indices into `values`.
+///
+/// Complexity: O(M * N^2) where M = objectives, N = solutions.
+#[allow(clippy::cast_possible_truncation)]
+pub(crate) fn fast_non_dominated_sort(
+    values: &[Vec<f64>],
+    directions: &[Direction],
+) -> Vec<Vec<usize>> {
+    fast_non_dominated_sort_constrained(values, directions, &[])
+}
+
+/// Fast non-dominated sorting with constraint support.
+///
+/// `constraints` is either empty (no constraints) or has the same length
+/// as `values`, where each entry is the constraint vector for that solution.
+#[allow(clippy::cast_possible_truncation)]
+pub(crate) fn fast_non_dominated_sort_constrained(
+    values: &[Vec<f64>],
+    directions: &[Direction],
+    constraints: &[Vec<f64>],
+) -> Vec<Vec<usize>> {
+    let n = values.len();
+    if n == 0 {
+        return Vec::new();
+    }
+
+    let has_constraints = !constraints.is_empty();
+    let empty_constraints: Vec<f64> = Vec::new();
+
+    // S_p: set of solutions dominated by p
+    let mut dominated_by: Vec<Vec<usize>> = vec![Vec::new(); n];
+    // n_p: domination count for p
+    let mut domination_count: Vec<usize> = vec![0; n];
+
+    for i in 0..n {
+        for j in (i + 1)..n {
+            let (a_c, b_c) = if has_constraints {
+                (&constraints[i], &constraints[j])
+            } else {
+                (&empty_constraints, &empty_constraints)
+            };
+
+            let i_dom_j = if has_constraints {
+                constrained_dominates(&values[i], &values[j], a_c, b_c, directions)
+            } else {
+                dominates(&values[i], &values[j], directions)
+            };
+            let j_dom_i = if has_constraints {
+                constrained_dominates(&values[j], &values[i], b_c, a_c, directions)
+            } else {
+                dominates(&values[j], &values[i], directions)
+            };
+
+            if i_dom_j {
+                dominated_by[i].push(j);
+                domination_count[j] += 1;
+            } else if j_dom_i {
+                dominated_by[j].push(i);
+                domination_count[i] += 1;
+            }
+        }
+    }
+
+    let mut fronts: Vec<Vec<usize>> = Vec::new();
+    let mut current_front: Vec<usize> = (0..n).filter(|&i| domination_count[i] == 0).collect();
+
+    while !current_front.is_empty() {
+        let mut next_front: Vec<usize> = Vec::new();
+        for &p in &current_front {
+            for &q in &dominated_by[p] {
+                domination_count[q] -= 1;
+                if domination_count[q] == 0 {
+                    next_front.push(q);
+                }
+            }
+        }
+        fronts.push(current_front);
+        current_front = next_front;
+    }
+
+    fronts
+}
+
+/// Crowding distance for one front (index-based, internal API).
+///
+/// Boundary solutions get `f64::INFINITY`. Returns one distance value per
+/// solution in the front, in the same order as `front_indices`.
+#[allow(clippy::cast_precision_loss)]
+pub(crate) fn crowding_distance_indexed(front_indices: &[usize], values: &[Vec<f64>]) -> Vec<f64> {
+    let n = front_indices.len();
+    if n <= 2 {
+        return vec![f64::INFINITY; n];
+    }
+
+    let m = values[front_indices[0]].len(); // number of objectives
+    let mut distances = vec![0.0_f64; n];
+
+    // Helper to look up objective value for a front member.
+    let val = |front_pos: usize, obj: usize| -> f64 { values[front_indices[front_pos]][obj] };
+
+    for obj in 0..m {
+        // Sort front positions by this objective
+        let mut sorted: Vec<usize> = (0..n).collect();
+        sorted.sort_by(|&a, &b| {
+            val(a, obj)
+                .partial_cmp(&val(b, obj))
+                .unwrap_or(core::cmp::Ordering::Equal)
+        });
+
+        // Boundary solutions get infinity
+        distances[sorted[0]] = f64::INFINITY;
+        distances[sorted[n - 1]] = f64::INFINITY;
+
+        let range = val(sorted[n - 1], obj) - val(sorted[0], obj);
+        if range > 0.0 {
+            for i in 1..(n - 1) {
+                distances[sorted[i]] += (val(sorted[i + 1], obj) - val(sorted[i - 1], obj)) / range;
+            }
+        }
+    }
+
+    distances
+}
+
+// ---------------------------------------------------------------------------
+// Public API
+// ---------------------------------------------------------------------------
+
+/// Compute the hypervolume indicator of a Pareto front.
+///
+/// The hypervolume is the volume of the objective space dominated by
+/// the Pareto front and bounded by a reference point. Higher values
+/// indicate a better front.
+///
+/// Each entry in `front` is one solution's objective values.
+/// `reference_point` should be worse than all front members in every
+/// objective (e.g., the worst acceptable values).
+///
+/// # Panics
+///
+/// Panics (in debug) if dimensions of `front`, `reference_point`, and
+/// `directions` are inconsistent.
+#[must_use]
+#[allow(clippy::cast_precision_loss)]
+pub fn hypervolume(front: &[Vec<f64>], reference_point: &[f64], directions: &[Direction]) -> f64 {
+    if front.is_empty() {
+        return 0.0;
+    }
+    let d = reference_point.len();
+    debug_assert!(front.iter().all(|p| p.len() == d));
+    debug_assert_eq!(d, directions.len());
+
+    // Normalize to minimize-space (negate maximized objectives).
+    let normalized: Vec<Vec<f64>> = front
+        .iter()
+        .map(|p| {
+            p.iter()
+                .zip(directions)
+                .map(|(&v, dir)| match dir {
+                    Direction::Minimize => v,
+                    Direction::Maximize => -v,
+                })
+                .collect()
+        })
+        .collect();
+
+    let ref_norm: Vec<f64> = reference_point
+        .iter()
+        .zip(directions)
+        .map(|(&v, dir)| match dir {
+            Direction::Minimize => v,
+            Direction::Maximize => -v,
+        })
+        .collect();
+
+    // Keep only points strictly dominated by the reference point.
+    let filtered: Vec<Vec<f64>> = normalized
+        .into_iter()
+        .filter(|p| p.iter().zip(&ref_norm).all(|(&pv, &rv)| pv < rv))
+        .collect();
+
+    if filtered.is_empty() {
+        return 0.0;
+    }
+
+    hv_recursive(&filtered, &ref_norm)
+}
+
+/// Recursive hypervolume via slicing on the last objective.
+///
+/// All points are in minimize-space and dominated by `reference`.
+#[allow(clippy::cast_precision_loss)]
+fn hv_recursive(points: &[Vec<f64>], reference: &[f64]) -> f64 {
+    let d = reference.len();
+
+    // Base case: 1-D hypervolume is just the gap from the best point to ref.
+    if d == 1 {
+        let min_val = points.iter().map(|p| p[0]).fold(f64::INFINITY, f64::min);
+        return (reference[0] - min_val).max(0.0);
+    }
+
+    // Single point: hypervolume is the product of gaps.
+    if points.len() == 1 {
+        return points[0]
+            .iter()
+            .zip(reference)
+            .map(|(&p, &r)| (r - p).max(0.0))
+            .product();
+    }
+
+    // Sort by last objective ascending.
+    let mut sorted: Vec<&Vec<f64>> = points.iter().collect();
+    sorted.sort_by(|a, b| {
+        a[d - 1]
+            .partial_cmp(&b[d - 1])
+            .unwrap_or(core::cmp::Ordering::Equal)
+    });
+
+    let sub_ref: Vec<f64> = reference[..d - 1].to_vec();
+    let mut result = 0.0;
+
+    for i in 0..sorted.len() {
+        let height = if i + 1 < sorted.len() {
+            sorted[i + 1][d - 1] - sorted[i][d - 1]
+        } else {
+            reference[d - 1] - sorted[i][d - 1]
+        };
+
+        if height <= 0.0 {
+            continue;
+        }
+
+        // Project points[0..=i] onto the first d-1 dimensions and
+        // keep only the non-dominated subset.
+        let projected: Vec<Vec<f64>> = sorted[..=i].iter().map(|p| p[..d - 1].to_vec()).collect();
+        let non_dom = non_dominated_minimize(&projected);
+
+        if !non_dom.is_empty() {
+            result += height * hv_recursive(&non_dom, &sub_ref);
+        }
+    }
+
+    result
+}
+
+/// Return the non-dominated subset of `points` in minimize-space.
+fn non_dominated_minimize(points: &[Vec<f64>]) -> Vec<Vec<f64>> {
+    let mut result = Vec::new();
+    'outer: for (i, p) in points.iter().enumerate() {
+        for (j, q) in points.iter().enumerate() {
+            if i == j {
+                continue;
+            }
+            // Check if q dominates p (all <=, at least one <).
+            let mut all_leq = true;
+            let mut any_lt = false;
+            for (&qv, &pv) in q.iter().zip(p.iter()) {
+                if qv > pv {
+                    all_leq = false;
+                    break;
+                }
+                if qv < pv {
+                    any_lt = true;
+                }
+            }
+            if all_leq && any_lt {
+                continue 'outer;
+            }
+        }
+        result.push(p.clone());
+    }
+    result
+}
+
+/// Compute non-dominated sorting of a set of solutions.
+///
+/// Returns a vec of fronts, where `fronts[0]` is the Pareto front,
+/// `fronts[1]` is the next best, etc. Each inner vec contains indices
+/// into the original `solutions` slice.
+///
+/// Uses the fast non-dominated sorting algorithm from
+/// Deb et al. (2002) with O(M N²) complexity.
+#[must_use]
+pub fn non_dominated_sort(solutions: &[Vec<f64>], directions: &[Direction]) -> Vec<Vec<usize>> {
+    fast_non_dominated_sort(solutions, directions)
+}
+
+/// Filter solutions to return only non-dominated (Pareto-optimal) indices.
+///
+/// Equivalent to `non_dominated_sort(solutions, directions)[0]` but
+/// communicates the intent more clearly.
+#[must_use]
+pub fn pareto_front_indices(solutions: &[Vec<f64>], directions: &[Direction]) -> Vec<usize> {
+    let fronts = fast_non_dominated_sort(solutions, directions);
+    fronts.into_iter().next().unwrap_or_default()
+}
+
+/// Compute crowding distance for diversity measurement.
+///
+/// Returns one distance value per solution in `front` (same order).
+/// Boundary solutions (best/worst in any objective) receive
+/// [`f64::INFINITY`]. Interior solutions get a finite positive value
+/// proportional to the gap between their neighbors.
+///
+/// `directions` is accepted for API consistency but does not affect
+/// the result, since crowding distance measures spacing regardless of
+/// optimization direction.
+#[must_use]
+#[allow(clippy::cast_precision_loss, clippy::needless_range_loop)]
+pub fn crowding_distance(front: &[Vec<f64>], _directions: &[Direction]) -> Vec<f64> {
+    let n = front.len();
+    if n <= 2 {
+        return vec![f64::INFINITY; n];
+    }
+
+    let m = front[0].len();
+    let mut distances = vec![0.0_f64; n];
+
+    for obj in 0..m {
+        let mut sorted: Vec<usize> = (0..n).collect();
+        sorted.sort_by(|&a, &b| {
+            front[a][obj]
+                .partial_cmp(&front[b][obj])
+                .unwrap_or(core::cmp::Ordering::Equal)
+        });
+
+        distances[sorted[0]] = f64::INFINITY;
+        distances[sorted[n - 1]] = f64::INFINITY;
+
+        let range = front[sorted[n - 1]][obj] - front[sorted[0]][obj];
+        if range > 0.0 {
+            for i in 1..(n - 1) {
+                distances[sorted[i]] +=
+                    (front[sorted[i + 1]][obj] - front[sorted[i - 1]][obj]) / range;
+            }
+        }
+    }
+
+    distances
+}
+
+#[cfg(test)]
+mod tests {
+    use super::*;
+
+    #[test]
+    fn test_dominates_basic() {
+        let dirs = [Direction::Minimize, Direction::Minimize];
+        assert!(dominates(&[1.0, 1.0], &[2.0, 2.0], &dirs));
+        assert!(!dominates(&[2.0, 2.0], &[1.0, 1.0], &dirs));
+        // Equal does not dominate
+        assert!(!dominates(&[1.0, 1.0], &[1.0, 1.0], &dirs));
+    }
+
+    #[test]
+    fn test_dominates_incomparable() {
+        let dirs = [Direction::Minimize, Direction::Minimize];
+        assert!(!dominates(&[1.0, 3.0], &[3.0, 1.0], &dirs));
+        assert!(!dominates(&[3.0, 1.0], &[1.0, 3.0], &dirs));
+    }
+
+    #[test]
+    fn test_dominates_maximize() {
+        let dirs = [Direction::Maximize, Direction::Minimize];
+        // a = (5, 1) vs b = (3, 2): a is better in both
+        assert!(dominates(&[5.0, 1.0], &[3.0, 2.0], &dirs));
+        assert!(!dominates(&[3.0, 2.0], &[5.0, 1.0], &dirs));
+    }
+
+    #[test]
+    fn test_nds_known() {
+        let values = vec![
+            vec![1.0, 5.0], // front 0
+            vec![5.0, 1.0], // front 0
+            vec![3.0, 3.0], // front 0 (non-dominated)
+            vec![4.0, 4.0], // front 1 (dominated by #2)
+            vec![6.0, 6.0], // front 2
+        ];
+        let dirs = [Direction::Minimize, Direction::Minimize];
+        let fronts = fast_non_dominated_sort(&values, &dirs);
+
+        assert_eq!(fronts.len(), 3);
+        let mut f0 = fronts[0].clone();
+        f0.sort_unstable();
+        assert_eq!(f0, vec![0, 1, 2]);
+        assert_eq!(fronts[1], vec![3]);
+        assert_eq!(fronts[2], vec![4]);
+    }
+
+    #[test]
+    fn test_crowding_indexed_boundaries() {
+        let values = vec![vec![1.0, 5.0], vec![3.0, 3.0], vec![5.0, 1.0]];
+        let front = vec![0, 1, 2];
+        let cd = crowding_distance_indexed(&front, &values);
+        assert!(cd[0].is_infinite());
+        assert!(cd[2].is_infinite());
+        assert!(cd[1].is_finite());
+        assert!(cd[1] > 0.0);
+    }
+
+    // ---- Public API tests ----
+
+    #[test]
+    fn test_hypervolume_2d_minimize() {
+        // Front: (1,3), (2,2), (3,1) with ref (4,4) — all minimize
+        let front = vec![vec![1.0, 3.0], vec![2.0, 2.0], vec![3.0, 1.0]];
+        let dirs = [Direction::Minimize, Direction::Minimize];
+        let hv = hypervolume(&front, &[4.0, 4.0], &dirs);
+        // Strip 1: x=[1,2), h=4-3=1 → area=1
+        // Strip 2: x=[2,3), h=4-2=2 → area=2
+        // Strip 3: x=[3,4], h=4-1=3 → area=3
+        // Total = 6
+        assert!((hv - 6.0).abs() < 1e-10);
+    }
+
+    #[test]
+    fn test_hypervolume_2d_maximize() {
+        // Front: (3,1), (2,2), (1,3) with ref (0,0) — all maximize
+        let front = vec![vec![3.0, 1.0], vec![2.0, 2.0], vec![1.0, 3.0]];
+        let dirs = [Direction::Maximize, Direction::Maximize];
+        let hv = hypervolume(&front, &[0.0, 0.0], &dirs);
+        // In negate-space: points become (-3,-1),(-2,-2),(-1,-3), ref=(0,0)
+        // Same geometry as minimize test above → area = 6
+        assert!((hv - 6.0).abs() < 1e-10);
+    }
+
+    #[test]
+    fn test_hypervolume_single_point() {
+        let front = vec![vec![1.0, 1.0]];
+        let dirs = [Direction::Minimize, Direction::Minimize];
+        let hv = hypervolume(&front, &[3.0, 3.0], &dirs);
+        // Rectangle: (3-1) * (3-1) = 4
+        assert!((hv - 4.0).abs() < 1e-10);
+    }
+
+    #[test]
+    fn test_hypervolume_empty_front() {
+        let front: Vec<Vec<f64>> = vec![];
+        let dirs = [Direction::Minimize];
+        assert!(hypervolume(&front, &[1.0], &dirs).abs() < f64::EPSILON);
+    }
+
+    #[test]
+    fn test_hypervolume_point_at_ref() {
+        // Point not strictly better than ref → contributes nothing
+        let front = vec![vec![5.0, 5.0]];
+        let dirs = [Direction::Minimize, Direction::Minimize];
+        let hv = hypervolume(&front, &[5.0, 5.0], &dirs);
+        assert!(hv.abs() < f64::EPSILON);
+    }
+
+    #[test]
+    fn test_hypervolume_3d() {
+        // Single point in 3D: (1,1,1) with ref (2,2,2)
+        let front = vec![vec![1.0, 1.0, 1.0]];
+        let dirs = [
+            Direction::Minimize,
+            Direction::Minimize,
+            Direction::Minimize,
+        ];
+        let hv = hypervolume(&front, &[2.0, 2.0, 2.0], &dirs);
+        assert!((hv - 1.0).abs() < 1e-10);
+    }
+
+    #[test]
+    fn test_non_dominated_sort_public() {
+        let values = vec![
+            vec![1.0, 5.0],
+            vec![5.0, 1.0],
+            vec![3.0, 3.0],
+            vec![4.0, 4.0],
+        ];
+        let dirs = [Direction::Minimize, Direction::Minimize];
+        let fronts = non_dominated_sort(&values, &dirs);
+        assert_eq!(fronts.len(), 2);
+        let mut f0 = fronts[0].clone();
+        f0.sort_unstable();
+        assert_eq!(f0, vec![0, 1, 2]);
+        assert_eq!(fronts[1], vec![3]);
+    }
+
+    #[test]
+    fn test_pareto_front_indices_basic() {
+        let values = vec![
+            vec![1.0, 5.0],
+            vec![5.0, 1.0],
+            vec![3.0, 3.0],
+            vec![4.0, 4.0],
+        ];
+        let dirs = [Direction::Minimize, Direction::Minimize];
+        let mut idx = pareto_front_indices(&values, &dirs);
+        idx.sort_unstable();
+        assert_eq!(idx, vec![0, 1, 2]);
+    }
+
+    #[test]
+    fn test_pareto_front_indices_empty() {
+        let values: Vec<Vec<f64>> = vec![];
+        let dirs = [Direction::Minimize];
+        assert!(pareto_front_indices(&values, &dirs).is_empty());
+    }
+
+    #[test]
+    fn test_crowding_distance_public() {
+        let front = vec![vec![1.0, 5.0], vec![3.0, 3.0], vec![5.0, 1.0]];
+        let dirs = [Direction::Minimize, Direction::Minimize];
+        let cd = crowding_distance(&front, &dirs);
+        assert!(cd[0].is_infinite());
+        assert!(cd[2].is_infinite());
+        assert!(cd[1].is_finite());
+        assert!(cd[1] > 0.0);
+    }
+
+    #[test]
+    fn test_crowding_distance_single_point() {
+        let front = vec![vec![2.0, 3.0]];
+        let dirs = [Direction::Minimize, Direction::Minimize];
+        let cd = crowding_distance(&front, &dirs);
+        assert_eq!(cd.len(), 1);
+        assert!(cd[0].is_infinite());
+    }
+}

src/pruner/hyperband.rs

@@ -0,0 +1,559 @@
+use core::sync::atomic::{AtomicU64, Ordering};
+use std::collections::HashMap;
+use std::sync::Mutex;
+
+use super::Pruner;
+use crate::sampler::CompletedTrial;
+use crate::types::{Direction, TrialState};
+
+/// Hyperband pruner that manages multiple Successive Halving brackets.
+///
+/// Hyperband addresses SHA's sensitivity to the `min_resource` choice by
+/// running multiple brackets, each with a different tradeoff between the
+/// number of configurations and the starting budget:
+///
+/// - Bracket 0: many trials, very small starting budget (aggressive pruning)
+/// - Bracket 1: fewer trials, larger starting budget (moderate pruning)
+/// - ...
+/// - Bracket `s_max`: few trials, full budget (no pruning)
+///
+/// Trials are assigned to brackets in round-robin fashion. Each bracket
+/// runs SHA with its own `min_resource` and rung schedule.
+///
+/// # Examples
+///
+/// ```
+/// use optimizer::Direction;
+/// use optimizer::pruner::HyperbandPruner;
+///
+/// let pruner = HyperbandPruner::new()
+///     .min_resource(1)
+///     .max_resource(81)
+///     .reduction_factor(3)
+///     .direction(Direction::Minimize);
+/// ```
+pub struct HyperbandPruner {
+    min_resource: u64,
+    max_resource: u64,
+    reduction_factor: u64,
+    direction: Direction,
+    /// Tracks which bracket each trial belongs to.
+    trial_brackets: Mutex<HashMap<u64, usize>>,
+    /// Counter for round-robin bracket assignment.
+    next_bracket: AtomicU64,
+}
+
+impl HyperbandPruner {
+    /// Create a new `HyperbandPruner` with default parameters.
+    ///
+    /// Defaults: `min_resource=1`, `max_resource=81`, `reduction_factor=3`,
+    /// `direction=Minimize`.
+    #[must_use]
+    pub fn new() -> Self {
+        Self {
+            min_resource: 1,
+            max_resource: 81,
+            reduction_factor: 3,
+            direction: Direction::Minimize,
+            trial_brackets: Mutex::new(HashMap::new()),
+            next_bracket: AtomicU64::new(0),
+        }
+    }
+
+    /// Set the minimum resource (budget) per trial.
+    ///
+    /// # Panics
+    ///
+    /// Panics if `r` is 0.
+    #[must_use]
+    pub fn min_resource(mut self, r: u64) -> Self {
+        assert!(r > 0, "min_resource must be > 0, got {r}");
+        self.min_resource = r;
+        self
+    }
+
+    /// Set the maximum resource (budget) per trial.
+    ///
+    /// # Panics
+    ///
+    /// Panics if `r` is 0.
+    #[must_use]
+    pub fn max_resource(mut self, r: u64) -> Self {
+        assert!(r > 0, "max_resource must be > 0, got {r}");
+        self.max_resource = r;
+        self
+    }
+
+    /// Set the reduction factor (eta). At each rung, the top 1/eta trials survive.
+    ///
+    /// # Panics
+    ///
+    /// Panics if `eta` is less than 2.
+    #[must_use]
+    pub fn reduction_factor(mut self, eta: u64) -> Self {
+        assert!(eta >= 2, "reduction_factor must be >= 2, got {eta}");
+        self.reduction_factor = eta;
+        self
+    }
+
+    /// Set the optimization direction.
+    #[must_use]
+    pub fn direction(mut self, d: Direction) -> Self {
+        self.direction = d;
+        self
+    }
+
+    /// Compute `s_max = floor(log(max_resource / min_resource) / log(eta))`.
+    #[allow(
+        clippy::cast_precision_loss,
+        clippy::cast_possible_truncation,
+        clippy::cast_sign_loss
+    )]
+    fn s_max(&self) -> u64 {
+        let eta = self.reduction_factor as f64;
+        let ratio = self.max_resource as f64 / self.min_resource as f64;
+        (ratio.ln() / eta.ln()).floor() as u64
+    }
+
+    /// Compute the rung steps for a given bracket `s`.
+    ///
+    /// For bracket `s`, the starting resource is `max_resource / eta^(s_max - s)`,
+    /// and rungs are spaced at powers of eta from there up to `max_resource`.
+    #[allow(
+        clippy::cast_precision_loss,
+        clippy::cast_possible_truncation,
+        clippy::cast_sign_loss
+    )]
+    fn rung_steps_for_bracket(&self, bracket: usize) -> Vec<u64> {
+        let s_max = self.s_max();
+        let eta = self.reduction_factor as f64;
+
+        // Starting resource for this bracket
+        let exponent = s_max.saturating_sub(bracket as u64);
+        let min_resource_bracket =
+            (self.max_resource as f64 / eta.powi(exponent as i32)).ceil() as u64;
+
+        let mut steps = Vec::new();
+        let mut rung: u32 = 0;
+        while let Some(power) = self.reduction_factor.checked_pow(rung) {
+            let step = min_resource_bracket.saturating_mul(power);
+            if step > self.max_resource {
+                break;
+            }
+            steps.push(step);
+            rung += 1;
+        }
+        steps
+    }
+
+    /// Assign a trial to a bracket (round-robin) and return the bracket index.
+    #[allow(clippy::cast_possible_truncation)]
+    fn assign_bracket(&self, trial_id: u64) -> usize {
+        let n_brackets = (self.s_max() + 1) as usize;
+        let mut map = self.trial_brackets.lock().expect("lock poisoned");
+        *map.entry(trial_id).or_insert_with(|| {
+            let idx = self.next_bracket.fetch_add(1, Ordering::Relaxed);
+            (idx as usize) % n_brackets
+        })
+    }
+}
+
+impl Default for HyperbandPruner {
+    fn default() -> Self {
+        Self::new()
+    }
+}
+
+#[allow(clippy::cast_precision_loss)]
+impl Pruner for HyperbandPruner {
+    fn should_prune(
+        &self,
+        trial_id: u64,
+        step: u64,
+        intermediate_values: &[(u64, f64)],
+        completed_trials: &[CompletedTrial],
+    ) -> bool {
+        let bracket = self.assign_bracket(trial_id);
+        let rungs = self.rung_steps_for_bracket(bracket);
+
+        // Find the highest rung step <= current step
+        let Some(&rung_step) = rungs.iter().rev().find(|&&r| r <= step) else {
+            return false;
+        };
+
+        // Never prune at the last rung (full budget)
+        if rung_step >= self.max_resource {
+            return false;
+        }
+
+        // Get the current trial's value at this rung step
+        let current_value =
+            if let Some(&(_, v)) = intermediate_values.iter().find(|(s, _)| *s == rung_step) {
+                v
+            } else if let Some(&(_, v)) = intermediate_values
+                .iter()
+                .rev()
+                .find(|(s, _)| *s <= rung_step)
+            {
+                v
+            } else {
+                return false;
+            };
+
+        self.is_pruned_at_rung(current_value, rung_step, bracket, completed_trials)
+    }
+}
+
+impl HyperbandPruner {
+    /// Determine whether a trial should be pruned at the given rung within its bracket.
+    ///
+    /// Only compares against other trials in the same bracket.
+    #[allow(
+        clippy::cast_precision_loss,
+        clippy::cast_possible_truncation,
+        clippy::cast_sign_loss
+    )]
+    fn is_pruned_at_rung(
+        &self,
+        current_value: f64,
+        rung_step: u64,
+        bracket: usize,
+        completed_trials: &[CompletedTrial],
+    ) -> bool {
+        let eta = self.reduction_factor as usize;
+
+        // Collect values at this rung step from trials in the same bracket
+        let map = self.trial_brackets.lock().expect("lock poisoned");
+        let mut values_at_rung: Vec<f64> = completed_trials
+            .iter()
+            .filter(|t| t.state == TrialState::Complete || t.state == TrialState::Pruned)
+            .filter(|t| map.get(&t.id).copied() == Some(bracket))
+            .filter_map(|t| {
+                t.intermediate_values
+                    .iter()
+                    .find(|(s, _)| *s == rung_step)
+                    .map(|(_, v)| *v)
+            })
+            .collect();
+        drop(map);
+
+        // Need at least eta trials to make a meaningful comparison
+        if values_at_rung.len() < eta {
+            return false;
+        }
+
+        values_at_rung.push(current_value);
+
+        values_at_rung
+            .sort_unstable_by(|a, b| a.partial_cmp(b).unwrap_or(core::cmp::Ordering::Equal));
+        if self.direction == Direction::Maximize {
+            values_at_rung.reverse();
+        }
+
+        let n_keep = (values_at_rung.len() as f64 / eta as f64).ceil() as usize;
+        let threshold_idx = n_keep.max(1) - 1;
+        let threshold = values_at_rung[threshold_idx];
+
+        match self.direction {
+            Direction::Minimize => current_value > threshold,
+            Direction::Maximize => current_value < threshold,
+        }
+    }
+}
+
+#[cfg(test)]
+#[allow(clippy::cast_precision_loss)]
+mod tests {
+    use super::*;
+
+    fn make_trial(id: u64, values: &[(u64, f64)]) -> CompletedTrial {
+        use std::collections::HashMap;
+
+        use crate::parameter::ParamId;
+
+        CompletedTrial::with_intermediate_values(
+            id,
+            HashMap::<ParamId, crate::ParamValue>::new(),
+            HashMap::new(),
+            HashMap::new(),
+            0.0,
+            values.to_vec(),
+            HashMap::new(),
+        )
+    }
+
+    fn make_pruned_trial(id: u64, values: &[(u64, f64)]) -> CompletedTrial {
+        let mut t = make_trial(id, values);
+        t.state = TrialState::Pruned;
+        t
+    }
+
+    #[test]
+    fn s_max_default() {
+        let pruner = HyperbandPruner::new();
+        // s_max = floor(ln(81/1) / ln(3)) = floor(4.0) = 4
+        assert_eq!(pruner.s_max(), 4);
+    }
+
+    #[test]
+    fn s_max_custom() {
+        let pruner = HyperbandPruner::new()
+            .min_resource(1)
+            .max_resource(16)
+            .reduction_factor(2);
+        // s_max = floor(ln(16) / ln(2)) = floor(4.0) = 4
+        assert_eq!(pruner.s_max(), 4);
+    }
+
+    #[test]
+    fn bracket_count() {
+        let pruner = HyperbandPruner::new();
+        // s_max=4, so brackets 0..=4 → 5 brackets
+        assert_eq!(pruner.s_max() + 1, 5);
+    }
+
+    #[test]
+    fn rung_steps_bracket_0_default() {
+        let pruner = HyperbandPruner::new();
+        // Bracket 0: min_resource_bracket = ceil(81 / 3^4) = ceil(81/81) = 1
+        // Rungs: 1, 3, 9, 27, 81
+        assert_eq!(pruner.rung_steps_for_bracket(0), vec![1, 3, 9, 27, 81]);
+    }
+
+    #[test]
+    fn rung_steps_bracket_2_default() {
+        let pruner = HyperbandPruner::new();
+        // Bracket 2: min_resource_bracket = ceil(81 / 3^(4-2)) = ceil(81/9) = 9
+        // Rungs: 9, 27, 81
+        assert_eq!(pruner.rung_steps_for_bracket(2), vec![9, 27, 81]);
+    }
+
+    #[test]
+    fn rung_steps_bracket_4_default() {
+        let pruner = HyperbandPruner::new();
+        // Bracket 4 (s_max): min_resource_bracket = ceil(81 / 3^0) = 81
+        // Rungs: 81 only (no pruning, full budget)
+        assert_eq!(pruner.rung_steps_for_bracket(4), vec![81]);
+    }
+
+    #[test]
+    fn rung_steps_eta2() {
+        let pruner = HyperbandPruner::new()
+            .min_resource(1)
+            .max_resource(16)
+            .reduction_factor(2);
+        // s_max = 4
+        // Bracket 0: min=ceil(16/2^4)=1, rungs: 1,2,4,8,16
+        assert_eq!(pruner.rung_steps_for_bracket(0), vec![1, 2, 4, 8, 16]);
+        // Bracket 2: min=ceil(16/2^2)=4, rungs: 4,8,16
+        assert_eq!(pruner.rung_steps_for_bracket(2), vec![4, 8, 16]);
+        // Bracket 4: min=16, rungs: 16
+        assert_eq!(pruner.rung_steps_for_bracket(4), vec![16]);
+    }
+
+    #[test]
+    fn round_robin_bracket_assignment() {
+        let pruner = HyperbandPruner::new(); // 5 brackets (0..=4)
+        // Trials get assigned in round-robin: 0→0, 1→1, 2→2, 3→3, 4→4, 5→0, ...
+        assert_eq!(pruner.assign_bracket(100), 0);
+        assert_eq!(pruner.assign_bracket(101), 1);
+        assert_eq!(pruner.assign_bracket(102), 2);
+        assert_eq!(pruner.assign_bracket(103), 3);
+        assert_eq!(pruner.assign_bracket(104), 4);
+        assert_eq!(pruner.assign_bracket(105), 0); // wraps around
+
+        // Repeated calls for same trial return same bracket
+        assert_eq!(pruner.assign_bracket(100), 0);
+        assert_eq!(pruner.assign_bracket(103), 3);
+    }
+
+    #[test]
+    fn no_prune_before_first_rung() {
+        let pruner = HyperbandPruner::new().direction(Direction::Minimize);
+        // Assign trial 0 to bracket 0 (rungs: 1, 3, 9, 27, 81)
+        pruner.assign_bracket(0);
+
+        // Register completed trials in bracket 0
+        let mut completed = Vec::new();
+        for i in 1..=9 {
+            pruner.assign_bracket(i);
+            completed.push(make_trial(i, &[(1, i as f64)]));
+        }
+
+        // Trial at step 0 (before rung 1) → don't prune
+        assert!(!pruner.should_prune(0, 0, &[(0, 100.0)], &completed));
+    }
+
+    #[test]
+    fn no_prune_at_max_resource() {
+        let pruner = HyperbandPruner::new().direction(Direction::Minimize);
+
+        // Put all trials in bracket 0
+        let mut completed = Vec::new();
+        for i in 0..9 {
+            pruner.assign_bracket(i);
+            completed.push(make_trial(i, &[(81, (i + 1) as f64)]));
+        }
+
+        let trial_id = 9;
+        pruner.assign_bracket(trial_id);
+        // At max_resource (81), never prune
+        assert!(!pruner.should_prune(trial_id, 81, &[(81, 100.0)], &completed));
+    }
+
+    #[test]
+    fn prune_worst_in_bracket_minimize() {
+        let pruner = HyperbandPruner::new().direction(Direction::Minimize);
+
+        // Force all trials into bracket 0 by assigning sequentially
+        // With 5 brackets, trials 0,5,10,... go to bracket 0
+        let bracket_0_ids: Vec<u64> = (0..5).map(|i| i * 5).collect();
+        // Assign all 25 trial IDs to fill brackets
+        for i in 0..25 {
+            pruner.assign_bracket(i);
+        }
+
+        // Create 9 completed trials in bracket 0 at rung step=1
+        let completed: Vec<_> = bracket_0_ids
+            .iter()
+            .take(3)
+            .enumerate()
+            .map(|(idx, &id)| make_trial(id, &[(1, (idx + 1) as f64)]))
+            .collect();
+
+        // Trial 25 → bracket 0 (25 % 5 == 0)
+        let test_id = 25;
+        pruner.assign_bracket(test_id);
+        assert_eq!(pruner.assign_bracket(test_id), 0);
+
+        // 3 completed + 1 current = 4. eta=3. ceil(4/3)=2. Threshold = 2.0
+        // Value 2.0 → keep
+        assert!(!pruner.should_prune(test_id, 1, &[(1, 2.0)], &completed));
+        // Value 3.0 → prune
+        assert!(pruner.should_prune(test_id, 1, &[(1, 3.0)], &completed));
+    }
+
+    #[test]
+    fn prune_worst_in_bracket_maximize() {
+        let pruner = HyperbandPruner::new().direction(Direction::Maximize);
+
+        // Assign trials so they end up in bracket 0
+        for i in 0..25 {
+            pruner.assign_bracket(i);
+        }
+
+        let completed: Vec<_> = [0u64, 5, 10]
+            .iter()
+            .enumerate()
+            .map(|(idx, &id)| make_trial(id, &[(1, (idx + 1) as f64)]))
+            .collect();
+
+        let test_id = 25;
+        pruner.assign_bracket(test_id);
+
+        // For maximize, best = highest. Values: 1,2,3 + current
+        // Value 2.0 → keep (threshold = 2.0 when sorted desc: 3,2,current,1)
+        assert!(!pruner.should_prune(test_id, 1, &[(1, 2.0)], &completed));
+        // Value 1.0 → prune
+        assert!(pruner.should_prune(test_id, 1, &[(1, 0.5)], &completed));
+    }
+
+    #[test]
+    fn different_brackets_have_different_aggressiveness() {
+        let pruner = HyperbandPruner::new()
+            .min_resource(1)
+            .max_resource(81)
+            .reduction_factor(3)
+            .direction(Direction::Minimize);
+
+        let rungs_0 = pruner.rung_steps_for_bracket(0);
+        let rungs_2 = pruner.rung_steps_for_bracket(2);
+        let rungs_4 = pruner.rung_steps_for_bracket(4);
+
+        // Bracket 0 has the most rungs (most aggressive)
+        assert!(rungs_0.len() > rungs_2.len());
+        // Bracket 4 has just 1 rung (no pruning)
+        assert_eq!(rungs_4.len(), 1);
+        // Bracket 0 starts earliest
+        assert!(rungs_0[0] < rungs_2[0]);
+    }
+
+    #[test]
+    fn trials_in_different_brackets_independent() {
+        let pruner = HyperbandPruner::new().direction(Direction::Minimize);
+
+        // Assign trials: bracket 0 gets IDs 0,5,10,15,20
+        for i in 0..25 {
+            pruner.assign_bracket(i);
+        }
+
+        // Bracket 0 trials: bad values at rung step=1
+        let bracket_0_trials: Vec<_> = [0u64, 5, 10]
+            .iter()
+            .map(|&id| make_trial(id, &[(1, 100.0)]))
+            .collect();
+
+        // Bracket 1 trials: good values at rung step=1
+        let bracket_1_trials: Vec<_> = [1u64, 6, 11]
+            .iter()
+            .map(|&id| make_trial(id, &[(1, 1.0)]))
+            .collect();
+
+        let mut all_trials = bracket_0_trials;
+        all_trials.extend(bracket_1_trials);
+
+        // A new bracket-0 trial with value 50 should be compared against
+        // bracket-0 peers (100,100,100), not bracket-1 peers (1,1,1)
+        let test_id = 25; // bracket 0
+        pruner.assign_bracket(test_id);
+        // 3 peers at 100.0 + current at 50.0. ceil(4/3)=2. Sorted: 50,100,100,100. Threshold=100.0
+        // Value 50.0 < 100.0 → keep
+        assert!(!pruner.should_prune(test_id, 1, &[(1, 50.0)], &all_trials));
+    }
+
+    #[test]
+    fn includes_pruned_trials() {
+        let pruner = HyperbandPruner::new().direction(Direction::Minimize);
+
+        for i in 0..25 {
+            pruner.assign_bracket(i);
+        }
+
+        let completed = vec![
+            make_trial(0, &[(1, 1.0)]),
+            make_pruned_trial(5, &[(1, 8.0)]),
+            make_pruned_trial(10, &[(1, 9.0)]),
+        ];
+
+        let test_id = 25;
+        pruner.assign_bracket(test_id);
+
+        // Values: 1.0, 8.0, 9.0 + current. eta=3.
+        // Value 1.0 → keep
+        assert!(!pruner.should_prune(test_id, 1, &[(1, 1.0)], &completed));
+        // Value 5.0 → prune (sorted: 1,5,8,9 → keep ceil(4/3)=2 → threshold=5.0, 5.0 not > 5.0 → keep)
+        assert!(!pruner.should_prune(test_id, 1, &[(1, 5.0)], &completed));
+        // Value 6.0 → prune (sorted: 1,6,8,9 → threshold=6.0, 6.0 not > 6.0 → keep)
+        assert!(!pruner.should_prune(test_id, 1, &[(1, 6.0)], &completed));
+        // Value 9.5 → prune
+        assert!(pruner.should_prune(test_id, 1, &[(1, 9.5)], &completed));
+    }
+
+    #[test]
+    #[should_panic(expected = "min_resource must be > 0")]
+    fn rejects_zero_min_resource() {
+        let _ = HyperbandPruner::new().min_resource(0);
+    }
+
+    #[test]
+    #[should_panic(expected = "max_resource must be > 0")]
+    fn rejects_zero_max_resource() {
+        let _ = HyperbandPruner::new().max_resource(0);
+    }
+
+    #[test]
+    #[should_panic(expected = "reduction_factor must be >= 2")]
+    fn rejects_reduction_factor_one() {
+        let _ = HyperbandPruner::new().reduction_factor(1);
+    }
+}

src/pruner/median.rs

@@ -0,0 +1,127 @@
+use super::Pruner;
+use super::percentile::compute_percentile;
+use crate::sampler::CompletedTrial;
+use crate::types::{Direction, TrialState};
+
+/// Prune trials that are performing worse than the median of completed trials
+/// at the same step.
+///
+/// This is the most commonly used pruner. It compares the current trial's
+/// intermediate value at each step with the median of all completed trials'
+/// values at that same step.
+///
+/// Equivalent to `PercentilePruner::new(50.0, direction)`.
+///
+/// # Examples
+///
+/// ```
+/// use optimizer::Direction;
+/// use optimizer::pruner::MedianPruner;
+///
+/// // Prune trials worse than median when minimizing, after 5 warmup steps
+/// let pruner = MedianPruner::new(Direction::Minimize)
+///     .n_warmup_steps(5)
+///     .n_min_trials(3);
+/// ```
+pub struct MedianPruner {
+    /// The optimization direction.
+    direction: Direction,
+    /// Don't prune in the first N steps (let the trial warm up).
+    n_warmup_steps: u64,
+    /// Require at least N completed trials before pruning.
+    n_min_trials: usize,
+}
+
+impl MedianPruner {
+    /// Create a new `MedianPruner` for the given optimization direction.
+    ///
+    /// By default, `n_warmup_steps` is 0 and `n_min_trials` is 1.
+    #[must_use]
+    pub fn new(direction: Direction) -> Self {
+        Self {
+            direction,
+            n_warmup_steps: 0,
+            n_min_trials: 1,
+        }
+    }
+
+    /// Set the number of warmup steps. No pruning occurs before this step.
+    #[must_use]
+    pub fn n_warmup_steps(mut self, n: u64) -> Self {
+        self.n_warmup_steps = n;
+        self
+    }
+
+    /// Set the minimum number of completed trials required before pruning.
+    #[must_use]
+    pub fn n_min_trials(mut self, n: usize) -> Self {
+        self.n_min_trials = n;
+        self
+    }
+}
+
+impl Pruner for MedianPruner {
+    fn should_prune(
+        &self,
+        _trial_id: u64,
+        step: u64,
+        intermediate_values: &[(u64, f64)],
+        completed_trials: &[CompletedTrial],
+    ) -> bool {
+        // 1. Don't prune during warmup
+        if step < self.n_warmup_steps {
+            return false;
+        }
+
+        // Get the current trial's latest value
+        let Some(&(_, current_value)) = intermediate_values.last() else {
+            return false;
+        };
+
+        // 2. Collect values at this step from completed (non-pruned) trials
+        let mut values_at_step: Vec<f64> = completed_trials
+            .iter()
+            .filter(|t| t.state == TrialState::Complete)
+            .filter_map(|t| {
+                t.intermediate_values
+                    .iter()
+                    .find(|(s, _)| *s == step)
+                    .map(|(_, v)| *v)
+            })
+            .collect();
+
+        // 3. Not enough trials
+        if values_at_step.len() < self.n_min_trials {
+            return false;
+        }
+
+        // 4. Compute median (50th percentile)
+        let median = compute_percentile(&mut values_at_step, 50.0);
+
+        // 5. Compare against median based on direction
+        match self.direction {
+            Direction::Minimize => current_value > median,
+            Direction::Maximize => current_value < median,
+        }
+    }
+}
+
+#[cfg(test)]
+mod tests {
+    use super::*;
+
+    #[test]
+    fn compute_median_odd() {
+        assert!((compute_percentile(&mut [3.0, 1.0, 2.0], 50.0) - 2.0).abs() < f64::EPSILON);
+    }
+
+    #[test]
+    fn compute_median_even() {
+        assert!((compute_percentile(&mut [4.0, 1.0, 3.0, 2.0], 50.0) - 2.5).abs() < f64::EPSILON);
+    }
+
+    #[test]
+    fn compute_median_single() {
+        assert!((compute_percentile(&mut [5.0], 50.0) - 5.0).abs() < f64::EPSILON);
+    }
+}

src/pruner/mod.rs

@@ -0,0 +1,74 @@
+//! Pruner trait and implementations for trial pruning.
+//!
+//! Pruners decide whether to stop (prune) a trial early based on its
+//! intermediate values compared to other trials. This is useful for
+//! discarding unpromising trials before they complete, saving compute.
+
+mod hyperband;
+mod median;
+mod nop;
+mod patient;
+pub(crate) mod percentile;
+mod successive_halving;
+mod threshold;
+mod wilcoxon;
+
+pub use hyperband::HyperbandPruner;
+pub use median::MedianPruner;
+pub use nop::NopPruner;
+pub use patient::PatientPruner;
+pub use percentile::PercentilePruner;
+pub use successive_halving::SuccessiveHalvingPruner;
+pub use threshold::ThresholdPruner;
+pub use wilcoxon::WilcoxonPruner;
+
+use crate::sampler::CompletedTrial;
+
+/// Trait for pluggable trial pruning strategies.
+///
+/// Pruners are consulted after each intermediate value is reported to
+/// decide whether the trial should be stopped early. The trait requires
+/// `Send + Sync` to support concurrent and async optimization.
+///
+/// # Implementing a custom pruner
+///
+/// ```
+/// use optimizer::pruner::Pruner;
+/// use optimizer::sampler::CompletedTrial;
+///
+/// struct MyPruner {
+///     threshold: f64,
+/// }
+///
+/// impl Pruner for MyPruner {
+///     fn should_prune(
+///         &self,
+///         _trial_id: u64,
+///         _step: u64,
+///         intermediate_values: &[(u64, f64)],
+///         _completed_trials: &[CompletedTrial],
+///     ) -> bool {
+///         // Prune if the latest value exceeds the threshold
+///         intermediate_values
+///             .last()
+///             .is_some_and(|&(_, v)| v > self.threshold)
+///     }
+/// }
+/// ```
+pub trait Pruner: Send + Sync {
+    /// Decide whether to prune a trial at the given step.
+    ///
+    /// # Arguments
+    ///
+    /// * `trial_id` - The current trial's ID.
+    /// * `step` - The step at which the intermediate value was reported.
+    /// * `intermediate_values` - All `(step, value)` pairs reported so far for this trial.
+    /// * `completed_trials` - History of all completed trials (for comparison).
+    fn should_prune(
+        &self,
+        trial_id: u64,
+        step: u64,
+        intermediate_values: &[(u64, f64)],
+        completed_trials: &[CompletedTrial],
+    ) -> bool;
+}

src/pruner/nop.rs

@@ -0,0 +1,17 @@
+use super::Pruner;
+use crate::sampler::CompletedTrial;
+
+/// A pruner that never prunes. This is the default when no pruner is configured.
+pub struct NopPruner;
+
+impl Pruner for NopPruner {
+    fn should_prune(
+        &self,
+        _trial_id: u64,
+        _step: u64,
+        _intermediate_values: &[(u64, f64)],
+        _completed_trials: &[CompletedTrial],
+    ) -> bool {
+        false
+    }
+}

src/pruner/patient.rs

@@ -0,0 +1,160 @@
+use std::collections::HashMap;
+use std::sync::Mutex;
+
+use super::Pruner;
+use crate::sampler::CompletedTrial;
+
+/// Wraps another pruner and adds a patience window.
+///
+/// The inner pruner must recommend pruning for `patience` consecutive
+/// steps before this pruner actually prunes the trial. This is useful
+/// to prevent premature pruning when intermediate values are noisy.
+///
+/// # Examples
+///
+/// ```
+/// use optimizer::pruner::{PatientPruner, ThresholdPruner};
+///
+/// // Only prune after the threshold pruner recommends pruning 3 times in a row
+/// let inner = ThresholdPruner::new().upper(100.0);
+/// let pruner = PatientPruner::new(inner, 3);
+/// ```
+pub struct PatientPruner {
+    inner: Box<dyn Pruner>,
+    patience: u64,
+    /// Track consecutive prune recommendations per trial.
+    consecutive_counts: Mutex<HashMap<u64, u64>>,
+}
+
+impl PatientPruner {
+    /// Create a new `PatientPruner` wrapping the given inner pruner.
+    ///
+    /// The inner pruner must recommend pruning for `patience` consecutive
+    /// calls before this pruner returns `true`.
+    pub fn new(inner: impl Pruner + 'static, patience: u64) -> Self {
+        Self {
+            inner: Box::new(inner),
+            patience,
+            consecutive_counts: Mutex::new(HashMap::new()),
+        }
+    }
+}
+
+impl Pruner for PatientPruner {
+    fn should_prune(
+        &self,
+        trial_id: u64,
+        step: u64,
+        intermediate_values: &[(u64, f64)],
+        completed_trials: &[CompletedTrial],
+    ) -> bool {
+        let inner_says_prune =
+            self.inner
+                .should_prune(trial_id, step, intermediate_values, completed_trials);
+        let mut counts = self.consecutive_counts.lock().expect("lock poisoned");
+        let count = counts.entry(trial_id).or_insert(0);
+        if inner_says_prune {
+            *count += 1;
+            *count >= self.patience
+        } else {
+            *count = 0;
+            false
+        }
+    }
+}
+
+#[cfg(test)]
+mod tests {
+    use super::*;
+    use crate::pruner::ThresholdPruner;
+
+    /// A test pruner that always returns the given value.
+    struct ConstPruner(bool);
+
+    impl Pruner for ConstPruner {
+        fn should_prune(
+            &self,
+            _trial_id: u64,
+            _step: u64,
+            _intermediate_values: &[(u64, f64)],
+            _completed_trials: &[CompletedTrial],
+        ) -> bool {
+            self.0
+        }
+    }
+
+    /// A pruner that returns values from a sequence.
+    struct SequencePruner(Mutex<Vec<bool>>);
+
+    impl Pruner for SequencePruner {
+        fn should_prune(
+            &self,
+            _trial_id: u64,
+            _step: u64,
+            _intermediate_values: &[(u64, f64)],
+            _completed_trials: &[CompletedTrial],
+        ) -> bool {
+            self.0.lock().expect("lock poisoned").remove(0)
+        }
+    }
+
+    fn call(pruner: &PatientPruner, trial_id: u64, step: u64) -> bool {
+        pruner.should_prune(trial_id, step, &[(step, 0.0)], &[])
+    }
+
+    #[test]
+    fn patience_1_behaves_like_inner() {
+        let pruner = PatientPruner::new(ConstPruner(true), 1);
+        assert!(call(&pruner, 0, 0));
+        assert!(call(&pruner, 0, 1));
+
+        let pruner = PatientPruner::new(ConstPruner(false), 1);
+        assert!(!call(&pruner, 0, 0));
+        assert!(!call(&pruner, 0, 1));
+    }
+
+    #[test]
+    fn patience_3_requires_consecutive_recommendations() {
+        let pruner = PatientPruner::new(ConstPruner(true), 3);
+        assert!(!call(&pruner, 0, 0)); // count=1
+        assert!(!call(&pruner, 0, 1)); // count=2
+        assert!(call(&pruner, 0, 2)); // count=3 → prune
+    }
+
+    #[test]
+    fn counter_resets_on_no_prune() {
+        // Sequence: prune, prune, no-prune, prune, prune, prune
+        let seq = vec![true, true, false, true, true, true];
+        let pruner = PatientPruner::new(SequencePruner(Mutex::new(seq)), 3);
+
+        assert!(!call(&pruner, 0, 0)); // count=1
+        assert!(!call(&pruner, 0, 1)); // count=2
+        assert!(!call(&pruner, 0, 2)); // reset → count=0
+        assert!(!call(&pruner, 0, 3)); // count=1
+        assert!(!call(&pruner, 0, 4)); // count=2
+        assert!(call(&pruner, 0, 5)); // count=3 → prune
+    }
+
+    #[test]
+    fn independent_per_trial() {
+        let pruner = PatientPruner::new(ConstPruner(true), 2);
+        assert!(!call(&pruner, 0, 0)); // trial 0: count=1
+        assert!(!call(&pruner, 1, 0)); // trial 1: count=1
+        assert!(call(&pruner, 0, 1)); // trial 0: count=2 → prune
+        assert!(!call(&pruner, 2, 0)); // trial 2: count=1
+        assert!(call(&pruner, 1, 1)); // trial 1: count=2 → prune
+    }
+
+    #[test]
+    fn works_with_threshold_pruner() {
+        let inner = ThresholdPruner::new().upper(10.0);
+        let pruner = PatientPruner::new(inner, 2);
+
+        // Value below threshold → inner says no
+        assert!(!pruner.should_prune(0, 0, &[(0, 5.0)], &[]));
+        // Value above threshold → inner says yes, count=1
+        assert!(!pruner.should_prune(0, 1, &[(0, 5.0), (1, 15.0)], &[]));
+        // Value above threshold again → count=2 → prune
+        assert!(pruner.should_prune(0, 2, &[(0, 5.0), (1, 15.0), (2, 20.0)], &[]));
+    }
+}

src/pruner/percentile.rs

@@ -0,0 +1,294 @@
+use super::Pruner;
+use crate::sampler::CompletedTrial;
+use crate::types::{Direction, TrialState};
+
+/// Prune trials that are not in the top `percentile`% of completed trials
+/// at the same training step.
+///
+/// `PercentilePruner::new(50.0, direction)` is equivalent to `MedianPruner`.
+/// `PercentilePruner::new(25.0, direction)` keeps only the top 25% of trials.
+///
+/// # Examples
+///
+/// ```
+/// use optimizer::Direction;
+/// use optimizer::pruner::PercentilePruner;
+///
+/// // Keep only the top 25% of trials (aggressive pruning)
+/// let pruner = PercentilePruner::new(25.0, Direction::Minimize)
+///     .n_warmup_steps(5)
+///     .n_min_trials(3);
+/// ```
+pub struct PercentilePruner {
+    /// Keep trials in the top `percentile`%. Range: (0.0, 100.0).
+    percentile: f64,
+    /// Don't prune in the first N steps (let the trial warm up).
+    n_warmup_steps: u64,
+    /// Require at least N completed trials before pruning.
+    n_min_trials: usize,
+    /// The optimization direction.
+    direction: Direction,
+}
+
+impl PercentilePruner {
+    /// Create a new `PercentilePruner` for the given percentile and direction.
+    ///
+    /// The `percentile` value must be in `(0.0, 100.0)`.
+    /// A percentile of 50.0 is equivalent to median pruning.
+    ///
+    /// # Panics
+    ///
+    /// Panics if `percentile` is not in `(0.0, 100.0)`.
+    #[must_use]
+    pub fn new(percentile: f64, direction: Direction) -> Self {
+        assert!(
+            percentile > 0.0 && percentile < 100.0,
+            "percentile must be in (0.0, 100.0), got {percentile}"
+        );
+        Self {
+            percentile,
+            n_warmup_steps: 0,
+            n_min_trials: 1,
+            direction,
+        }
+    }
+
+    /// Set the number of warmup steps. No pruning occurs before this step.
+    #[must_use]
+    pub fn n_warmup_steps(mut self, n: u64) -> Self {
+        self.n_warmup_steps = n;
+        self
+    }
+
+    /// Set the minimum number of completed trials required before pruning.
+    #[must_use]
+    pub fn n_min_trials(mut self, n: usize) -> Self {
+        self.n_min_trials = n;
+        self
+    }
+}
+
+impl Pruner for PercentilePruner {
+    fn should_prune(
+        &self,
+        _trial_id: u64,
+        step: u64,
+        intermediate_values: &[(u64, f64)],
+        completed_trials: &[CompletedTrial],
+    ) -> bool {
+        // 1. Don't prune during warmup
+        if step < self.n_warmup_steps {
+            return false;
+        }
+
+        // Get the current trial's latest value
+        let Some(&(_, current_value)) = intermediate_values.last() else {
+            return false;
+        };
+
+        // 2. Collect values at this step from completed (non-pruned) trials
+        let mut values_at_step: Vec<f64> = completed_trials
+            .iter()
+            .filter(|t| t.state == TrialState::Complete)
+            .filter_map(|t| {
+                t.intermediate_values
+                    .iter()
+                    .find(|(s, _)| *s == step)
+                    .map(|(_, v)| *v)
+            })
+            .collect();
+
+        // 3. Not enough trials
+        if values_at_step.len() < self.n_min_trials {
+            return false;
+        }
+
+        // 4. Compute percentile threshold
+        let threshold = compute_percentile(&mut values_at_step, self.percentile);
+
+        // 5. Compare against threshold based on direction
+        match self.direction {
+            Direction::Minimize => current_value > threshold,
+            Direction::Maximize => current_value < threshold,
+        }
+    }
+}
+
+/// Compute the given percentile of a non-empty slice. Sorts the slice in place.
+///
+/// Uses linear interpolation between the two nearest ranks.
+#[allow(
+    clippy::cast_precision_loss,
+    clippy::cast_possible_truncation,
+    clippy::cast_sign_loss
+)]
+pub(crate) fn compute_percentile(values: &mut [f64], percentile: f64) -> f64 {
+    values.sort_unstable_by(|a, b| a.partial_cmp(b).unwrap_or(core::cmp::Ordering::Equal));
+    let len = values.len();
+    if len == 1 {
+        return values[0];
+    }
+    // Rank in [0, len-1] range
+    let rank = percentile / 100.0 * (len - 1) as f64;
+    let lower = rank.floor() as usize;
+    let upper = rank.ceil() as usize;
+    if lower == upper {
+        values[lower]
+    } else {
+        let frac = rank - lower as f64;
+        values[lower] * (1.0 - frac) + values[upper] * frac
+    }
+}
+
+#[cfg(test)]
+mod tests {
+    use super::*;
+
+    #[test]
+    fn compute_percentile_median_odd() {
+        // Percentile 50 on odd-length slice = median
+        let val = compute_percentile(&mut [3.0, 1.0, 2.0], 50.0);
+        assert!((val - 2.0).abs() < f64::EPSILON);
+    }
+
+    #[test]
+    fn compute_percentile_median_even() {
+        // Percentile 50 on even-length slice = median (interpolated)
+        let val = compute_percentile(&mut [4.0, 1.0, 3.0, 2.0], 50.0);
+        assert!((val - 2.5).abs() < f64::EPSILON);
+    }
+
+    #[test]
+    fn compute_percentile_25() {
+        // [1.0, 2.0, 3.0, 4.0], rank = 0.25 * 3 = 0.75
+        // interpolate: 1.0 * 0.25 + 2.0 * 0.75 = 1.75
+        let val = compute_percentile(&mut [4.0, 1.0, 3.0, 2.0], 25.0);
+        assert!((val - 1.75).abs() < f64::EPSILON);
+    }
+
+    #[test]
+    fn compute_percentile_75() {
+        // [1.0, 2.0, 3.0, 4.0], rank = 0.75 * 3 = 2.25
+        // interpolate: 3.0 * 0.75 + 4.0 * 0.25 = 3.25
+        let val = compute_percentile(&mut [4.0, 1.0, 3.0, 2.0], 75.0);
+        assert!((val - 3.25).abs() < f64::EPSILON);
+    }
+
+    #[test]
+    fn compute_percentile_single() {
+        let val = compute_percentile(&mut [5.0], 50.0);
+        assert!((val - 5.0).abs() < f64::EPSILON);
+    }
+
+    #[test]
+    #[should_panic(expected = "percentile must be in (0.0, 100.0)")]
+    fn new_rejects_zero() {
+        let _ = PercentilePruner::new(0.0, Direction::Minimize);
+    }
+
+    #[test]
+    #[should_panic(expected = "percentile must be in (0.0, 100.0)")]
+    fn new_rejects_hundred() {
+        let _ = PercentilePruner::new(100.0, Direction::Minimize);
+    }
+
+    fn make_completed_trial(id: u64, values: &[(u64, f64)]) -> CompletedTrial {
+        use std::collections::HashMap;
+
+        use crate::parameter::ParamId;
+
+        CompletedTrial::with_intermediate_values(
+            id,
+            HashMap::<ParamId, crate::ParamValue>::new(),
+            HashMap::new(),
+            HashMap::new(),
+            0.0,
+            values.to_vec(),
+            HashMap::new(),
+        )
+    }
+
+    #[test]
+    fn percentile_50_matches_median_behavior() {
+        let pruner = PercentilePruner::new(50.0, Direction::Minimize);
+        let completed = vec![
+            make_completed_trial(0, &[(0, 1.0), (1, 2.0)]),
+            make_completed_trial(1, &[(0, 3.0), (1, 4.0)]),
+            make_completed_trial(2, &[(0, 5.0), (1, 6.0)]),
+        ];
+        // Median at step 1 is 4.0
+        // Value 5.0 > 4.0 → prune
+        assert!(pruner.should_prune(3, 1, &[(0, 3.0), (1, 5.0)], &completed));
+        // Value 3.0 < 4.0 → keep
+        assert!(!pruner.should_prune(3, 1, &[(0, 3.0), (1, 3.0)], &completed));
+    }
+
+    #[test]
+    fn percentile_25_is_more_aggressive() {
+        let pruner_25 = PercentilePruner::new(25.0, Direction::Minimize);
+        let pruner_75 = PercentilePruner::new(75.0, Direction::Minimize);
+        let completed = vec![
+            make_completed_trial(0, &[(0, 1.0)]),
+            make_completed_trial(1, &[(0, 2.0)]),
+            make_completed_trial(2, &[(0, 3.0)]),
+            make_completed_trial(3, &[(0, 4.0)]),
+        ];
+        // 25th percentile at step 0: 1.75
+        // 75th percentile at step 0: 3.25
+        // Value 2.5: above 25th (prune), below 75th (keep)
+        assert!(pruner_25.should_prune(4, 0, &[(0, 2.5)], &completed));
+        assert!(!pruner_75.should_prune(4, 0, &[(0, 2.5)], &completed));
+    }
+
+    #[test]
+    fn warmup_prevents_pruning() {
+        let pruner = PercentilePruner::new(50.0, Direction::Minimize).n_warmup_steps(5);
+        let completed = vec![make_completed_trial(0, &[(0, 1.0)])];
+        // Step 3 < warmup 5 → no prune even with bad value
+        assert!(!pruner.should_prune(1, 3, &[(3, 100.0)], &completed));
+    }
+
+    #[test]
+    fn n_min_trials_prevents_pruning() {
+        let pruner = PercentilePruner::new(50.0, Direction::Minimize).n_min_trials(5);
+        let completed = vec![
+            make_completed_trial(0, &[(0, 1.0)]),
+            make_completed_trial(1, &[(0, 2.0)]),
+        ];
+        // Only 2 trials, need 5 → no prune
+        assert!(!pruner.should_prune(2, 0, &[(0, 100.0)], &completed));
+    }
+
+    #[test]
+    fn maximize_direction() {
+        let pruner = PercentilePruner::new(50.0, Direction::Maximize);
+        let completed = vec![
+            make_completed_trial(0, &[(0, 1.0)]),
+            make_completed_trial(1, &[(0, 3.0)]),
+            make_completed_trial(2, &[(0, 5.0)]),
+        ];
+        // Median at step 0 is 3.0
+        // Value 2.0 < 3.0 → prune (maximize wants higher)
+        assert!(pruner.should_prune(3, 0, &[(0, 2.0)], &completed));
+        // Value 4.0 > 3.0 → keep
+        assert!(!pruner.should_prune(3, 0, &[(0, 4.0)], &completed));
+    }
+
+    #[test]
+    fn near_boundary_percentiles() {
+        let pruner_low = PercentilePruner::new(1.0, Direction::Minimize);
+        let pruner_high = PercentilePruner::new(99.0, Direction::Minimize);
+        let completed = vec![
+            make_completed_trial(0, &[(0, 1.0)]),
+            make_completed_trial(1, &[(0, 2.0)]),
+            make_completed_trial(2, &[(0, 3.0)]),
+            make_completed_trial(3, &[(0, 100.0)]),
+        ];
+        // Percentile 1 is very aggressive (threshold near 1.0)
+        // Value 1.5 should be pruned
+        assert!(pruner_low.should_prune(4, 0, &[(0, 1.5)], &completed));
+        // Percentile 99 is very lenient (threshold near 100.0)
+        // Value 50.0 should not be pruned
+        assert!(!pruner_high.should_prune(4, 0, &[(0, 50.0)], &completed));
+    }
+}

src/pruner/successive_halving.rs

@@ -0,0 +1,466 @@
+use super::Pruner;
+use crate::sampler::CompletedTrial;
+use crate::types::{Direction, TrialState};
+
+/// Successive Halving pruner based on the SHA algorithm.
+///
+/// Trials are evaluated at exponentially-spaced "rungs". At each rung,
+/// only the top 1/eta fraction of trials survive to the next rung.
+///
+/// For example, with `min_resource=1`, `max_resource=81`, `reduction_factor=3`:
+/// - Rung 0: evaluate at step 1, keep top 1/3
+/// - Rung 1: evaluate at step 3, keep top 1/3
+/// - Rung 2: evaluate at step 9, keep top 1/3
+/// - Rung 3: evaluate at step 27, keep top 1/3
+/// - Rung 4: evaluate at step 81 (full budget)
+///
+/// # Examples
+///
+/// ```
+/// use optimizer::Direction;
+/// use optimizer::pruner::SuccessiveHalvingPruner;
+///
+/// let pruner = SuccessiveHalvingPruner::new()
+///     .min_resource(1)
+///     .max_resource(81)
+///     .reduction_factor(3)
+///     .direction(Direction::Minimize);
+/// ```
+pub struct SuccessiveHalvingPruner {
+    min_resource: u64,
+    max_resource: u64,
+    reduction_factor: u64,
+    min_early_stopping_rate: u64,
+    direction: Direction,
+}
+
+impl SuccessiveHalvingPruner {
+    /// Create a new `SuccessiveHalvingPruner` with default parameters.
+    ///
+    /// Defaults: `min_resource=1`, `max_resource=81`, `reduction_factor=3`,
+    /// `min_early_stopping_rate=0`, `direction=Minimize`.
+    #[must_use]
+    pub fn new() -> Self {
+        Self {
+            min_resource: 1,
+            max_resource: 81,
+            reduction_factor: 3,
+            min_early_stopping_rate: 0,
+            direction: Direction::Minimize,
+        }
+    }
+
+    /// Set the minimum resource (budget) per trial.
+    ///
+    /// # Panics
+    ///
+    /// Panics if `r` is 0.
+    #[must_use]
+    pub fn min_resource(mut self, r: u64) -> Self {
+        assert!(r > 0, "min_resource must be > 0, got {r}");
+        self.min_resource = r;
+        self
+    }
+
+    /// Set the maximum resource (budget) per trial.
+    ///
+    /// # Panics
+    ///
+    /// Panics if `r` is 0.
+    #[must_use]
+    pub fn max_resource(mut self, r: u64) -> Self {
+        assert!(r > 0, "max_resource must be > 0, got {r}");
+        self.max_resource = r;
+        self
+    }
+
+    /// Set the reduction factor (eta). At each rung, the top 1/eta trials survive.
+    ///
+    /// # Panics
+    ///
+    /// Panics if `eta` is less than 2.
+    #[must_use]
+    pub fn reduction_factor(mut self, eta: u64) -> Self {
+        assert!(eta >= 2, "reduction_factor must be >= 2, got {eta}");
+        self.reduction_factor = eta;
+        self
+    }
+
+    /// Set the minimum early stopping rate. Skips the first N rungs.
+    #[must_use]
+    pub fn min_early_stopping_rate(mut self, n: u64) -> Self {
+        self.min_early_stopping_rate = n;
+        self
+    }
+
+    /// Set the optimization direction.
+    #[must_use]
+    pub fn direction(mut self, d: Direction) -> Self {
+        self.direction = d;
+        self
+    }
+
+    /// Compute the rung steps: `[min_resource * eta^(s), ...]` up to `max_resource`,
+    /// skipping the first `min_early_stopping_rate` rungs.
+    fn rung_steps(&self) -> Vec<u64> {
+        let eta = self.reduction_factor;
+        let mut steps = Vec::new();
+        let mut rung: u32 = 0;
+        while let Some(power) = eta.checked_pow(rung) {
+            let step = self.min_resource.saturating_mul(power);
+            if step > self.max_resource {
+                break;
+            }
+            if u64::from(rung) >= self.min_early_stopping_rate {
+                steps.push(step);
+            }
+            rung += 1;
+        }
+        steps
+    }
+}
+
+impl Default for SuccessiveHalvingPruner {
+    fn default() -> Self {
+        Self::new()
+    }
+}
+
+#[allow(clippy::cast_precision_loss)]
+impl Pruner for SuccessiveHalvingPruner {
+    fn should_prune(
+        &self,
+        _trial_id: u64,
+        step: u64,
+        intermediate_values: &[(u64, f64)],
+        completed_trials: &[CompletedTrial],
+    ) -> bool {
+        let rungs = self.rung_steps();
+
+        // Find the highest rung step <= current step
+        let Some(&rung_step) = rungs.iter().rev().find(|&&r| r <= step) else {
+            // No rung matches (before the first rung) → don't prune
+            return false;
+        };
+
+        // If this is the last rung (full budget), don't prune
+        if rung_step >= self.max_resource {
+            return false;
+        }
+
+        // Get the current trial's value at this rung step
+        let Some(&(_, current_value)) = intermediate_values.iter().find(|(s, _)| *s == rung_step)
+        else {
+            // Trial hasn't reported a value at this exact rung step.
+            // Use the latest intermediate value at or before the rung step instead.
+            let Some(&(_, current_value)) = intermediate_values
+                .iter()
+                .rev()
+                .find(|(s, _)| *s <= rung_step)
+            else {
+                return false;
+            };
+            return self.is_pruned_at_rung(current_value, rung_step, completed_trials);
+        };
+
+        self.is_pruned_at_rung(current_value, rung_step, completed_trials)
+    }
+}
+
+impl SuccessiveHalvingPruner {
+    /// Determine whether a trial with `current_value` should be pruned at the given rung.
+    #[allow(
+        clippy::cast_precision_loss,
+        clippy::cast_possible_truncation,
+        clippy::cast_sign_loss
+    )]
+    fn is_pruned_at_rung(
+        &self,
+        current_value: f64,
+        rung_step: u64,
+        completed_trials: &[CompletedTrial],
+    ) -> bool {
+        let eta = self.reduction_factor as usize;
+
+        // Collect values at this rung step from all trials that reached it
+        let mut values_at_rung: Vec<f64> = completed_trials
+            .iter()
+            .filter(|t| t.state == TrialState::Complete || t.state == TrialState::Pruned)
+            .filter_map(|t| {
+                t.intermediate_values
+                    .iter()
+                    .find(|(s, _)| *s == rung_step)
+                    .map(|(_, v)| *v)
+            })
+            .collect();
+
+        // Need at least eta trials to make a meaningful comparison
+        // (with fewer trials, we can't determine the top 1/eta fraction)
+        if values_at_rung.len() < eta {
+            return false;
+        }
+
+        // Include the current trial's value for ranking
+        values_at_rung.push(current_value);
+
+        // Sort based on direction: best values first
+        values_at_rung
+            .sort_unstable_by(|a, b| a.partial_cmp(b).unwrap_or(core::cmp::Ordering::Equal));
+        if self.direction == Direction::Maximize {
+            values_at_rung.reverse();
+        }
+
+        // Keep top 1/eta fraction
+        let n_keep = (values_at_rung.len() as f64 / eta as f64).ceil() as usize;
+        let threshold_idx = n_keep.max(1) - 1;
+        let threshold = values_at_rung[threshold_idx];
+
+        // Prune if current value is worse than the threshold
+        match self.direction {
+            Direction::Minimize => current_value > threshold,
+            Direction::Maximize => current_value < threshold,
+        }
+    }
+}
+
+#[cfg(test)]
+#[allow(clippy::cast_precision_loss)]
+mod tests {
+    use super::*;
+
+    fn make_trial(id: u64, values: &[(u64, f64)]) -> CompletedTrial {
+        use std::collections::HashMap;
+
+        use crate::parameter::ParamId;
+
+        CompletedTrial::with_intermediate_values(
+            id,
+            HashMap::<ParamId, crate::ParamValue>::new(),
+            HashMap::new(),
+            HashMap::new(),
+            0.0,
+            values.to_vec(),
+            HashMap::new(),
+        )
+    }
+
+    fn make_pruned_trial(id: u64, values: &[(u64, f64)]) -> CompletedTrial {
+        let mut t = make_trial(id, values);
+        t.state = TrialState::Pruned;
+        t
+    }
+
+    #[test]
+    fn rung_steps_default() {
+        let pruner = SuccessiveHalvingPruner::new();
+        let rungs = pruner.rung_steps();
+        // min=1, max=81, eta=3 → 1, 3, 9, 27, 81
+        assert_eq!(rungs, vec![1, 3, 9, 27, 81]);
+    }
+
+    #[test]
+    fn rung_steps_custom() {
+        let pruner = SuccessiveHalvingPruner::new()
+            .min_resource(2)
+            .max_resource(32)
+            .reduction_factor(2);
+        let rungs = pruner.rung_steps();
+        // 2, 4, 8, 16, 32
+        assert_eq!(rungs, vec![2, 4, 8, 16, 32]);
+    }
+
+    #[test]
+    fn rung_steps_with_early_stopping_rate() {
+        let pruner = SuccessiveHalvingPruner::new().min_early_stopping_rate(2);
+        let rungs = pruner.rung_steps();
+        // Skip rung 0 (step=1) and rung 1 (step=3), keep rung 2+ (9, 27, 81)
+        assert_eq!(rungs, vec![9, 27, 81]);
+    }
+
+    #[test]
+    fn no_prune_before_first_rung() {
+        let pruner = SuccessiveHalvingPruner::new()
+            .min_resource(10)
+            .max_resource(100)
+            .reduction_factor(3);
+        let completed = vec![
+            make_trial(0, &[(5, 1.0)]),
+            make_trial(1, &[(5, 2.0)]),
+            make_trial(2, &[(5, 3.0)]),
+        ];
+        // Step 5 is before the first rung (10)
+        assert!(!pruner.should_prune(3, 5, &[(5, 100.0)], &completed));
+    }
+
+    #[test]
+    fn no_prune_with_single_trial() {
+        let pruner = SuccessiveHalvingPruner::new();
+        let completed = vec![make_trial(0, &[(1, 5.0)]), make_trial(1, &[(1, 3.0)])];
+        // Only 2 completed trials at rung + 1 current = 3 total, threshold = ceil(3/3) = 1
+        // With eta=3, we need at least 3 completed trials
+        assert!(!pruner.should_prune(2, 1, &[(1, 10.0)], &completed));
+    }
+
+    #[test]
+    fn prune_worst_trials_at_rung() {
+        let pruner = SuccessiveHalvingPruner::new().direction(Direction::Minimize);
+
+        // 9 completed trials at rung step=1, with values 1..=9
+        let completed: Vec<_> = (0..9)
+            .map(|i| make_trial(i, &[(1, (i + 1) as f64)]))
+            .collect();
+
+        // With eta=3, keep top 1/3. 10 total values → ceil(10/3) = 4 kept
+        // Best 4 values: 1, 2, 3, 4. Threshold = 4.0
+        // Value 3.0 → keep (in top 1/3)
+        assert!(!pruner.should_prune(9, 1, &[(1, 3.0)], &completed));
+        // Value 5.0 → prune (not in top 1/3)
+        assert!(pruner.should_prune(9, 1, &[(1, 5.0)], &completed));
+    }
+
+    #[test]
+    fn top_fraction_survives() {
+        let pruner = SuccessiveHalvingPruner::new().direction(Direction::Minimize);
+
+        // 6 completed trials at step=1
+        let completed: Vec<_> = (0..6)
+            .map(|i| make_trial(i, &[(1, (i + 1) as f64)]))
+            .collect();
+
+        // 7 total (6 + current). ceil(7/3) = 3 keep. Threshold = 3.0
+        // Value 2.0 → keep
+        assert!(!pruner.should_prune(6, 1, &[(1, 2.0)], &completed));
+        // Value 3.0 → keep (at threshold)
+        assert!(!pruner.should_prune(6, 1, &[(1, 3.0)], &completed));
+        // Value 4.0 → prune
+        assert!(pruner.should_prune(6, 1, &[(1, 4.0)], &completed));
+    }
+
+    #[test]
+    fn maximize_direction() {
+        let pruner = SuccessiveHalvingPruner::new().direction(Direction::Maximize);
+
+        let completed: Vec<_> = (0..6)
+            .map(|i| make_trial(i, &[(1, (i + 1) as f64)]))
+            .collect();
+
+        // 7 total. For maximize, best = highest. ceil(7/3)=3. Top 3: 6,5,4. Threshold=4.0
+        // Value 5.0 → keep
+        assert!(!pruner.should_prune(6, 1, &[(1, 5.0)], &completed));
+        // Value 4.0 → keep (at threshold)
+        assert!(!pruner.should_prune(6, 1, &[(1, 4.0)], &completed));
+        // Value 3.0 → prune
+        assert!(pruner.should_prune(6, 1, &[(1, 3.0)], &completed));
+    }
+
+    #[test]
+    fn reduction_factor_2() {
+        let pruner = SuccessiveHalvingPruner::new()
+            .reduction_factor(2)
+            .min_resource(1)
+            .max_resource(16)
+            .direction(Direction::Minimize);
+
+        // Rungs: 1, 2, 4, 8, 16
+        assert_eq!(pruner.rung_steps(), vec![1, 2, 4, 8, 16]);
+
+        // 4 completed trials at rung step=1
+        let completed: Vec<_> = (0..4)
+            .map(|i| make_trial(i, &[(1, (i + 1) as f64)]))
+            .collect();
+
+        // With eta=2, 5 total. ceil(5/2) = 3 keep. Threshold = 3.0
+        // Value 3.0 → keep
+        assert!(!pruner.should_prune(4, 1, &[(1, 3.0)], &completed));
+        // Value 4.0 → prune
+        assert!(pruner.should_prune(4, 1, &[(1, 4.0)], &completed));
+    }
+
+    #[test]
+    fn reduction_factor_4() {
+        let pruner = SuccessiveHalvingPruner::new()
+            .reduction_factor(4)
+            .min_resource(1)
+            .max_resource(64)
+            .direction(Direction::Minimize);
+
+        // Rungs: 1, 4, 16, 64
+        assert_eq!(pruner.rung_steps(), vec![1, 4, 16, 64]);
+
+        // 12 completed trials at rung step=1
+        let completed: Vec<_> = (0..12)
+            .map(|i| make_trial(i, &[(1, (i + 1) as f64)]))
+            .collect();
+
+        // With eta=4, 13 total. ceil(13/4) = 4 keep. Threshold = 4.0
+        // Value 4.0 → keep
+        assert!(!pruner.should_prune(12, 1, &[(1, 4.0)], &completed));
+        // Value 5.0 → prune
+        assert!(pruner.should_prune(12, 1, &[(1, 5.0)], &completed));
+    }
+
+    #[test]
+    fn non_contiguous_steps() {
+        let pruner = SuccessiveHalvingPruner::new().direction(Direction::Minimize);
+
+        // Trials reporting at rung step=3 (not step=1)
+        let completed: Vec<_> = (0..6)
+            .map(|i| make_trial(i, &[(3, (i + 1) as f64)]))
+            .collect();
+
+        // Current trial reports at step 5 (between rung 3 and rung 9)
+        // Highest rung <= 5 is 3. Use value at rung step 3.
+        // Trial has value at step 3 → use it
+        assert!(!pruner.should_prune(6, 5, &[(3, 2.0)], &completed));
+        assert!(pruner.should_prune(6, 5, &[(3, 5.0)], &completed));
+    }
+
+    #[test]
+    fn no_prune_at_max_resource() {
+        let pruner = SuccessiveHalvingPruner::new();
+        let completed: Vec<_> = (0..9)
+            .map(|i| make_trial(i, &[(81, (i + 1) as f64)]))
+            .collect();
+
+        // At the max resource rung, never prune (trial should complete)
+        assert!(!pruner.should_prune(9, 81, &[(81, 100.0)], &completed));
+    }
+
+    #[test]
+    fn includes_pruned_trials_in_comparison() {
+        let pruner = SuccessiveHalvingPruner::new().direction(Direction::Minimize);
+
+        // Mix of completed and pruned trials at rung step=1
+        let completed = vec![
+            make_trial(0, &[(1, 1.0)]),
+            make_trial(1, &[(1, 2.0)]),
+            make_pruned_trial(2, &[(1, 8.0)]),
+            make_pruned_trial(3, &[(1, 9.0)]),
+            make_pruned_trial(4, &[(1, 10.0)]),
+        ];
+
+        // 6 total. ceil(6/3) = 2 keep. Threshold = 2.0
+        // Value 2.0 → keep
+        assert!(!pruner.should_prune(5, 1, &[(1, 2.0)], &completed));
+        // Value 3.0 → prune
+        assert!(pruner.should_prune(5, 1, &[(1, 3.0)], &completed));
+    }
+
+    #[test]
+    #[should_panic(expected = "min_resource must be > 0")]
+    fn rejects_zero_min_resource() {
+        let _ = SuccessiveHalvingPruner::new().min_resource(0);
+    }
+
+    #[test]
+    #[should_panic(expected = "max_resource must be > 0")]
+    fn rejects_zero_max_resource() {
+        let _ = SuccessiveHalvingPruner::new().max_resource(0);
+    }
+
+    #[test]
+    #[should_panic(expected = "reduction_factor must be >= 2")]
+    fn rejects_reduction_factor_one() {
+        let _ = SuccessiveHalvingPruner::new().reduction_factor(1);
+    }
+}

src/pruner/threshold.rs

@@ -0,0 +1,83 @@
+use super::Pruner;
+use crate::sampler::CompletedTrial;
+
+/// Prune trials whose intermediate values exceed fixed thresholds.
+///
+/// Useful for cutting off trials that are clearly diverging or stuck
+/// at bad values early in training.
+///
+/// # Examples
+///
+/// ```
+/// use optimizer::pruner::ThresholdPruner;
+///
+/// // Prune if the intermediate value exceeds 100.0 or falls below 0.0
+/// let pruner = ThresholdPruner::new().upper(100.0).lower(0.0);
+/// ```
+pub struct ThresholdPruner {
+    /// Prune if intermediate value is greater than this. `None` = no upper bound.
+    upper: Option<f64>,
+    /// Prune if intermediate value is less than this. `None` = no lower bound.
+    lower: Option<f64>,
+}
+
+impl ThresholdPruner {
+    /// Create a new `ThresholdPruner` with no thresholds set.
+    ///
+    /// By default, no pruning occurs. Use [`upper`](Self::upper) and
+    /// [`lower`](Self::lower) to set bounds.
+    #[must_use]
+    pub fn new() -> Self {
+        Self {
+            upper: None,
+            lower: None,
+        }
+    }
+
+    /// Set the upper threshold. Trials with intermediate values above this
+    /// will be pruned.
+    #[must_use]
+    pub fn upper(mut self, threshold: f64) -> Self {
+        self.upper = Some(threshold);
+        self
+    }
+
+    /// Set the lower threshold. Trials with intermediate values below this
+    /// will be pruned.
+    #[must_use]
+    pub fn lower(mut self, threshold: f64) -> Self {
+        self.lower = Some(threshold);
+        self
+    }
+}
+
+impl Default for ThresholdPruner {
+    fn default() -> Self {
+        Self::new()
+    }
+}
+
+impl Pruner for ThresholdPruner {
+    fn should_prune(
+        &self,
+        _trial_id: u64,
+        _step: u64,
+        intermediate_values: &[(u64, f64)],
+        _completed_trials: &[CompletedTrial],
+    ) -> bool {
+        let Some(&(_, latest_value)) = intermediate_values.last() else {
+            return false;
+        };
+        if let Some(upper) = self.upper
+            && latest_value > upper
+        {
+            return true;
+        }
+        if let Some(lower) = self.lower
+            && latest_value < lower
+        {
+            return true;
+        }
+        false
+    }
+}

src/pruner/wilcoxon.rs

@@ -0,0 +1,531 @@
+use core::cmp::Ordering;
+
+use super::Pruner;
+use crate::sampler::CompletedTrial;
+use crate::types::{Direction, TrialState};
+
+/// Prune trials using a Wilcoxon signed-rank test comparing intermediate
+/// values against the best completed trial.
+///
+/// More principled than `MedianPruner` for noisy objectives — it accounts
+/// for the paired nature of step-aligned comparisons and doesn't prune
+/// on random fluctuations.
+///
+/// The test compares intermediate values at matching steps between the
+/// current trial and the best completed trial. If the current trial is
+/// statistically significantly worse (p < threshold), it is pruned.
+///
+/// # Examples
+///
+/// ```
+/// use optimizer::Direction;
+/// use optimizer::pruner::WilcoxonPruner;
+///
+/// let pruner = WilcoxonPruner::new(Direction::Minimize)
+///     .p_value_threshold(0.05)
+///     .n_warmup_steps(5)
+///     .n_min_trials(1);
+/// ```
+pub struct WilcoxonPruner {
+    /// Significance level (default 0.05). Lower = more conservative.
+    p_value_threshold: f64,
+    /// Don't prune in the first N steps (let the trial warm up).
+    n_warmup_steps: u64,
+    /// Require at least N completed trials before pruning.
+    n_min_trials: usize,
+    /// The optimization direction.
+    direction: Direction,
+}
+
+impl WilcoxonPruner {
+    /// Create a new `WilcoxonPruner` for the given optimization direction.
+    ///
+    /// By default, `p_value_threshold` is 0.05, `n_warmup_steps` is 0,
+    /// and `n_min_trials` is 1.
+    #[must_use]
+    pub fn new(direction: Direction) -> Self {
+        Self {
+            p_value_threshold: 0.05,
+            n_warmup_steps: 0,
+            n_min_trials: 1,
+            direction,
+        }
+    }
+
+    /// Set the p-value threshold for significance.
+    ///
+    /// Must be in (0.0, 1.0). Lower values are more conservative (harder to prune).
+    ///
+    /// # Panics
+    ///
+    /// Panics if `p` is not in the open interval (0.0, 1.0).
+    #[must_use]
+    pub fn p_value_threshold(mut self, p: f64) -> Self {
+        assert!(
+            p > 0.0 && p < 1.0,
+            "p_value_threshold must be in (0.0, 1.0)"
+        );
+        self.p_value_threshold = p;
+        self
+    }
+
+    /// Set the number of warmup steps. No pruning occurs before this step.
+    #[must_use]
+    pub fn n_warmup_steps(mut self, n: u64) -> Self {
+        self.n_warmup_steps = n;
+        self
+    }
+
+    /// Set the minimum number of completed trials required before pruning.
+    #[must_use]
+    pub fn n_min_trials(mut self, n: usize) -> Self {
+        self.n_min_trials = n;
+        self
+    }
+}
+
+impl Pruner for WilcoxonPruner {
+    fn should_prune(
+        &self,
+        _trial_id: u64,
+        step: u64,
+        intermediate_values: &[(u64, f64)],
+        completed_trials: &[CompletedTrial],
+    ) -> bool {
+        if step < self.n_warmup_steps {
+            return false;
+        }
+
+        let completed: Vec<&CompletedTrial> = completed_trials
+            .iter()
+            .filter(|t| t.state == TrialState::Complete)
+            .collect();
+
+        if completed.len() < self.n_min_trials {
+            return false;
+        }
+
+        // Find the best completed trial by final objective value.
+        let best = match self.direction {
+            Direction::Minimize => completed
+                .iter()
+                .min_by(|a, b| a.value.partial_cmp(&b.value).unwrap_or(Ordering::Equal)),
+            Direction::Maximize => completed
+                .iter()
+                .max_by(|a, b| a.value.partial_cmp(&b.value).unwrap_or(Ordering::Equal)),
+        };
+
+        let Some(best) = best else {
+            return false;
+        };
+
+        // Pair intermediate values at matching steps.
+        let pairs: Vec<(f64, f64)> = intermediate_values
+            .iter()
+            .filter_map(|&(s, current_v)| {
+                best.intermediate_values
+                    .iter()
+                    .find(|(bs, _)| *bs == s)
+                    .map(|&(_, best_v)| (current_v, best_v))
+            })
+            .collect();
+
+        // Need at least 6 pairs for a meaningful test.
+        if pairs.len() < 6 {
+            return false;
+        }
+
+        // Compute signed differences: current - best.
+        // For minimization: positive diff means current is worse.
+        // For maximization: negative diff means current is worse.
+        let differences: Vec<f64> = pairs
+            .iter()
+            .map(|&(current, best_v)| current - best_v)
+            .collect();
+
+        // Run the Wilcoxon signed-rank test.
+        let p_value = wilcoxon_signed_rank_test(&differences, self.direction);
+
+        p_value < self.p_value_threshold
+    }
+}
+
+/// Perform a one-sided Wilcoxon signed-rank test.
+///
+/// Tests whether the values tend to be worse than zero (positive for
+/// minimization, negative for maximization).
+///
+/// Returns a p-value. Small p-values indicate the current trial is
+/// significantly worse.
+fn wilcoxon_signed_rank_test(differences: &[f64], direction: Direction) -> f64 {
+    // 1. Remove zero differences.
+    let nonzero: Vec<f64> = differences.iter().copied().filter(|d| *d != 0.0).collect();
+    let n = nonzero.len();
+
+    if n < 6 {
+        return 1.0; // Not enough data
+    }
+
+    // 2. Rank by absolute value.
+    let mut abs_ranked: Vec<(usize, f64, f64)> = nonzero
+        .iter()
+        .enumerate()
+        .map(|(i, &d)| (i, d.abs(), d))
+        .collect();
+    abs_ranked.sort_by(|a, b| a.1.partial_cmp(&b.1).unwrap_or(Ordering::Equal));
+
+    // 3. Assign ranks with tie correction.
+    let ranks = assign_ranks(&abs_ranked);
+
+    // 4. Compute W+ (sum of ranks for positive differences) and
+    //    W- (sum of ranks for negative differences).
+    let mut w_plus = 0.0;
+    let mut w_minus = 0.0;
+    for (i, &(_, _, orig)) in abs_ranked.iter().enumerate() {
+        if orig > 0.0 {
+            w_plus += ranks[i];
+        } else {
+            w_minus += ranks[i];
+        }
+    }
+
+    // For one-sided test:
+    // - Minimization: we want to detect positive diffs (current worse).
+    //   Large W+ means significantly worse. Test statistic = W-.
+    // - Maximization: we want to detect negative diffs (current worse).
+    //   Large W- means significantly worse. Test statistic = W+.
+    let w = match direction {
+        Direction::Minimize => w_minus,
+        Direction::Maximize => w_plus,
+    };
+
+    // 5. Normal approximation for the p-value.
+    #[allow(clippy::cast_precision_loss)]
+    let n_f = n as f64;
+    let mean = n_f * (n_f + 1.0) / 4.0;
+    let variance = n_f * (n_f + 1.0) * (2.0 * n_f + 1.0) / 24.0;
+
+    // Tie correction for variance.
+    let tie_correction = compute_tie_correction(&ranks);
+    let adjusted_variance = variance - tie_correction;
+
+    if adjusted_variance <= 0.0 {
+        return 1.0;
+    }
+
+    let std_dev = adjusted_variance.sqrt();
+
+    // Continuity correction: shift W by 0.5 towards mean.
+    let continuity = if w < mean { 0.5 } else { -0.5 };
+    let z = (w + continuity - mean) / std_dev;
+
+    // One-sided p-value (lower tail): probability that the test statistic
+    // is this small or smaller under H0.
+    normal_cdf(z)
+}
+
+/// Assign average ranks, handling ties.
+fn assign_ranks(sorted: &[(usize, f64, f64)]) -> Vec<f64> {
+    let n = sorted.len();
+    let mut ranks = vec![0.0; n];
+    let mut i = 0;
+
+    while i < n {
+        let mut j = i;
+        // Find all items tied with sorted[i].
+        while j < n
+            && (sorted[j].1 - sorted[i].1).abs() < f64::EPSILON * sorted[i].1.max(1.0) * 100.0
+        {
+            j += 1;
+        }
+        // Average rank for the tie group. Ranks are 1-based.
+        #[allow(clippy::cast_precision_loss)]
+        let avg_rank = (i + 1 + j) as f64 / 2.0;
+        for rank in ranks.iter_mut().take(j).skip(i) {
+            *rank = avg_rank;
+        }
+        i = j;
+    }
+
+    ranks
+}
+
+/// Compute the tie correction term for the variance.
+/// For each tie group of size t, subtract t^3 - t from the sum,
+/// then divide by 48.
+fn compute_tie_correction(ranks: &[f64]) -> f64 {
+    let mut correction = 0.0;
+    let mut i = 0;
+
+    while i < ranks.len() {
+        let mut j = i;
+        while j < ranks.len() && (ranks[j] - ranks[i]).abs() < f64::EPSILON {
+            j += 1;
+        }
+        #[allow(clippy::cast_precision_loss)]
+        let t = (j - i) as f64;
+        if t > 1.0 {
+            correction += t * t * t - t;
+        }
+        i = j;
+    }
+
+    correction / 48.0
+}
+
+/// Standard normal CDF using an approximation (Abramowitz & Stegun).
+fn normal_cdf(x: f64) -> f64 {
+    // Use the complementary error function relationship:
+    // Φ(x) = 0.5 * erfc(-x / √2)
+    0.5 * erfc(-x / core::f64::consts::SQRT_2)
+}
+
+/// Complementary error function approximation.
+/// Maximum error: 1.5 × 10⁻⁷ (Abramowitz & Stegun formula 7.1.26).
+fn erfc(x: f64) -> f64 {
+    let t = 1.0 / (1.0 + 0.327_591_1 * x.abs());
+    let poly = t
+        * (0.254_829_592
+            + t * (-0.284_496_736
+                + t * (1.421_413_741 + t * (-1.453_152_027 + t * 1.061_405_429))));
+    let result = poly * (-x * x).exp();
+
+    if x >= 0.0 { result } else { 2.0 - result }
+}
+
+#[cfg(test)]
+mod tests {
+    use std::collections::HashMap;
+
+    use super::*;
+
+    fn trial_with_values(
+        id: u64,
+        value: f64,
+        intermediate_values: Vec<(u64, f64)>,
+    ) -> CompletedTrial {
+        CompletedTrial::with_intermediate_values(
+            id,
+            HashMap::new(),
+            HashMap::new(),
+            HashMap::new(),
+            value,
+            intermediate_values,
+            HashMap::new(),
+        )
+    }
+
+    #[test]
+    fn no_prune_during_warmup() {
+        let pruner = WilcoxonPruner::new(Direction::Minimize).n_warmup_steps(10);
+        let completed = vec![trial_with_values(
+            0,
+            0.1,
+            (0..20).map(|s| (s, 0.1)).collect(),
+        )];
+        let current: Vec<(u64, f64)> = (0..8).map(|s| (s, 100.0)).collect();
+        assert!(!pruner.should_prune(1, 7, &current, &completed));
+    }
+
+    #[test]
+    fn no_prune_with_insufficient_trials() {
+        let pruner = WilcoxonPruner::new(Direction::Minimize).n_min_trials(5);
+        let completed = vec![trial_with_values(
+            0,
+            0.1,
+            (0..20).map(|s| (s, 0.1)).collect(),
+        )];
+        let current: Vec<(u64, f64)> = (0..10).map(|s| (s, 100.0)).collect();
+        assert!(!pruner.should_prune(1, 9, &current, &completed));
+    }
+
+    #[test]
+    fn no_prune_with_fewer_than_6_pairs() {
+        let pruner = WilcoxonPruner::new(Direction::Minimize);
+        let completed = vec![trial_with_values(
+            0,
+            0.1,
+            (0..5).map(|s| (s, 0.1)).collect(),
+        )];
+        // Only 5 matching steps
+        let current: Vec<(u64, f64)> = (0..5).map(|s| (s, 100.0)).collect();
+        assert!(!pruner.should_prune(1, 4, &current, &completed));
+    }
+
+    #[test]
+    fn prune_when_consistently_worse_minimize() {
+        let pruner = WilcoxonPruner::new(Direction::Minimize);
+
+        // Best trial has low values.
+        let best_values: Vec<(u64, f64)> = (0..20).map(|s| (s, 0.1)).collect();
+        let completed = vec![trial_with_values(0, 0.1, best_values)];
+
+        // Current trial is consistently much worse.
+        let current: Vec<(u64, f64)> = (0..20).map(|s| (s, 10.0)).collect();
+        assert!(pruner.should_prune(1, 19, &current, &completed));
+    }
+
+    #[test]
+    fn prune_when_consistently_worse_maximize() {
+        let pruner = WilcoxonPruner::new(Direction::Maximize);
+
+        // Best trial has high values.
+        let best_values: Vec<(u64, f64)> = (0..20).map(|s| (s, 10.0)).collect();
+        let completed = vec![trial_with_values(0, 10.0, best_values)];
+
+        // Current trial is consistently much worse.
+        let current: Vec<(u64, f64)> = (0..20).map(|s| (s, 0.1)).collect();
+        assert!(pruner.should_prune(1, 19, &current, &completed));
+    }
+
+    #[test]
+    fn no_prune_when_statistically_similar() {
+        let pruner = WilcoxonPruner::new(Direction::Minimize);
+
+        // Best trial and current trial have very similar values with noise.
+        let best_values: Vec<(u64, f64)> = (0..20_u64)
+            .map(|s| {
+                let noise = if s.is_multiple_of(2) { 0.01 } else { -0.01 };
+                (s, 1.0 + noise)
+            })
+            .collect();
+        let completed = vec![trial_with_values(0, 1.0, best_values)];
+
+        // Current trial is similar — alternating above/below.
+        let current: Vec<(u64, f64)> = (0..20_u64)
+            .map(|s| {
+                let noise = if s.is_multiple_of(2) { -0.01 } else { 0.01 };
+                (s, 1.0 + noise)
+            })
+            .collect();
+        assert!(!pruner.should_prune(1, 19, &current, &completed));
+    }
+
+    #[test]
+    fn selects_best_trial_minimize() {
+        let pruner = WilcoxonPruner::new(Direction::Minimize);
+
+        // Two completed trials: trial 0 is better (lower).
+        let completed = vec![
+            trial_with_values(0, 0.1, (0..20).map(|s| (s, 0.1)).collect()),
+            trial_with_values(1, 5.0, (0..20).map(|s| (s, 5.0)).collect()),
+        ];
+
+        // Current trial is worse than the best but similar to the second.
+        let current: Vec<(u64, f64)> = (0..20).map(|s| (s, 5.0)).collect();
+        assert!(pruner.should_prune(2, 19, &current, &completed));
+    }
+
+    #[test]
+    fn selects_best_trial_maximize() {
+        let pruner = WilcoxonPruner::new(Direction::Maximize);
+
+        // Two completed trials: trial 1 is better (higher).
+        let completed = vec![
+            trial_with_values(0, 0.1, (0..20).map(|s| (s, 0.1)).collect()),
+            trial_with_values(1, 10.0, (0..20).map(|s| (s, 10.0)).collect()),
+        ];
+
+        // Current trial is worse than the best.
+        let current: Vec<(u64, f64)> = (0..20).map(|s| (s, 0.1)).collect();
+        assert!(pruner.should_prune(2, 19, &current, &completed));
+    }
+
+    #[test]
+    fn ignores_pruned_trials() {
+        let pruner = WilcoxonPruner::new(Direction::Minimize);
+
+        // Only a pruned trial — no complete trials.
+        let mut trial = trial_with_values(0, 0.1, (0..20).map(|s| (s, 0.1)).collect());
+        trial.state = TrialState::Pruned;
+        let completed = vec![trial];
+
+        let current: Vec<(u64, f64)> = (0..20).map(|s| (s, 100.0)).collect();
+        assert!(!pruner.should_prune(1, 19, &current, &completed));
+    }
+
+    #[test]
+    fn lower_p_value_is_more_conservative() {
+        let strict = WilcoxonPruner::new(Direction::Minimize).p_value_threshold(0.001);
+        let lenient = WilcoxonPruner::new(Direction::Minimize).p_value_threshold(0.1);
+
+        let completed = vec![trial_with_values(
+            0,
+            0.1,
+            (0..20).map(|s| (s, 0.1)).collect(),
+        )];
+
+        // Moderately worse — should pass lenient but maybe not strict.
+        let current: Vec<(u64, f64)> = (0..20)
+            .map(|s| if s < 15 { (s, 0.2) } else { (s, 0.15) })
+            .collect();
+
+        let lenient_prunes = lenient.should_prune(1, 19, &current, &completed);
+        let strict_prunes = strict.should_prune(1, 19, &current, &completed);
+
+        // A stricter threshold should never prune when a lenient one doesn't.
+        if !lenient_prunes {
+            assert!(!strict_prunes);
+        }
+    }
+
+    #[test]
+    #[should_panic(expected = "p_value_threshold must be in (0.0, 1.0)")]
+    fn panics_on_zero_p_value() {
+        let _ = WilcoxonPruner::new(Direction::Minimize).p_value_threshold(0.0);
+    }
+
+    #[test]
+    #[should_panic(expected = "p_value_threshold must be in (0.0, 1.0)")]
+    fn panics_on_one_p_value() {
+        let _ = WilcoxonPruner::new(Direction::Minimize).p_value_threshold(1.0);
+    }
+
+    #[test]
+    fn correct_signed_rank_statistic() {
+        // Known example: differences [1, 2, 3, 4, 5, 6] (all positive).
+        // Ranks: 1, 2, 3, 4, 5, 6. W+ = 21, W- = 0.
+        // For minimization (testing if positive = worse), W- = 0.
+        // This should give a very small p-value.
+        let diffs = vec![1.0, 2.0, 3.0, 4.0, 5.0, 6.0];
+        let p = wilcoxon_signed_rank_test(&diffs, Direction::Minimize);
+        assert!(
+            p < 0.05,
+            "p-value {p} should be < 0.05 for all-positive diffs"
+        );
+    }
+
+    #[test]
+    fn symmetric_differences_not_significant() {
+        // Balanced differences: half positive, half negative.
+        let diffs = vec![1.0, -1.0, 2.0, -2.0, 3.0, -3.0, 4.0, -4.0];
+        let p = wilcoxon_signed_rank_test(&diffs, Direction::Minimize);
+        assert!(p > 0.05, "p-value {p} should be > 0.05 for symmetric diffs");
+    }
+
+    #[test]
+    fn normal_cdf_known_values() {
+        assert!((normal_cdf(0.0) - 0.5).abs() < 1e-6);
+        assert!(normal_cdf(-10.0) < 1e-6);
+        assert!((normal_cdf(10.0) - 1.0).abs() < 1e-6);
+        assert!((normal_cdf(-1.96) - 0.025).abs() < 0.001);
+    }
+
+    #[test]
+    fn no_intermediate_values() {
+        let pruner = WilcoxonPruner::new(Direction::Minimize);
+        let completed = vec![trial_with_values(
+            0,
+            0.1,
+            (0..20).map(|s| (s, 0.1)).collect(),
+        )];
+        assert!(!pruner.should_prune(1, 0, &[], &completed));
+    }
+
+    #[test]
+    fn no_completed_trials() {
+        let pruner = WilcoxonPruner::new(Direction::Minimize);
+        let current: Vec<(u64, f64)> = (0..20).map(|s| (s, 1.0)).collect();
+        assert!(!pruner.should_prune(1, 19, &current, &[]));
+    }
+}

src/rng_util.rs

@@ -0,0 +1,5 @@
+/// Generate a random `f64` in the range `[low, high)`.
+#[inline]
+pub(crate) fn f64_range(rng: &mut fastrand::Rng, low: f64, high: f64) -> f64 {
+    low + rng.f64() * (high - low)
+}

src/sampler/bohb.rs

@@ -0,0 +1,691 @@
+//! BOHB (Bayesian Optimization + `HyperBand`) sampler.
+//!
+//! BOHB combines TPE's model-guided sampling with Hyperband's budget-aware
+//! evaluation. Instead of building one global TPE model, BOHB conditions
+//! its TPE model on trials evaluated at a specific budget level, giving
+//! better-calibrated proposals for each rung of the Hyperband schedule.
+//!
+//! # How it works
+//!
+//! 1. Compute all Hyperband rung steps (budget levels) from the config.
+//! 2. On each `sample()` call, scan the history's `intermediate_values`
+//!    to find the **largest budget level** with enough observations
+//!    (`>= min_points_in_model`).
+//! 3. Build a filtered history where each trial's `value` is replaced
+//!    with its intermediate value at that budget level.
+//! 4. Delegate to an internal [`TpeSampler`] for the actual sampling.
+//! 5. Fall back to random sampling if no budget level has enough data.
+//!
+//! # Examples
+//!
+//! ```
+//! use optimizer::sampler::bohb::BohbSampler;
+//! use optimizer::{Direction, Study};
+//!
+//! let bohb = BohbSampler::new();
+//! let pruner = bohb.matching_pruner(Direction::Minimize);
+//! let study: Study<f64> = Study::with_sampler_and_pruner(Direction::Minimize, bohb, pruner);
+//! ```
+//!
+//! Using the builder for custom configuration:
+//!
+//! ```
+//! use optimizer::sampler::bohb::BohbSampler;
+//!
+//! let bohb = BohbSampler::builder()
+//!     .min_resource(1)
+//!     .max_resource(81)
+//!     .reduction_factor(3)
+//!     .min_points_in_model(10)
+//!     .seed(42)
+//!     .build()
+//!     .unwrap();
+//! ```
+
+use crate::distribution::Distribution;
+use crate::error::Result;
+use crate::param::ParamValue;
+use crate::pruner::HyperbandPruner;
+use crate::sampler::tpe::TpeSampler;
+use crate::sampler::{CompletedTrial, Sampler};
+use crate::types::Direction;
+
+/// A BOHB sampler that combines TPE with Hyperband budget awareness.
+///
+/// BOHB filters trial history by budget level before delegating to TPE,
+/// so the surrogate model is conditioned on trials evaluated at the same
+/// resource level. This produces better-calibrated parameter proposals
+/// than using a single global model across all budgets.
+///
+/// Use [`BohbSampler::matching_pruner`] to create a [`HyperbandPruner`]
+/// with matching parameters.
+pub struct BohbSampler {
+    min_resource: u64,
+    max_resource: u64,
+    reduction_factor: u64,
+    min_points_in_model: usize,
+    tpe: TpeSampler,
+}
+
+impl BohbSampler {
+    /// Creates a new BOHB sampler with default settings.
+    ///
+    /// Defaults:
+    /// - `min_resource`: 1
+    /// - `max_resource`: 81
+    /// - `reduction_factor`: 3
+    /// - `min_points_in_model`: 10
+    /// - TPE: default settings
+    #[must_use]
+    pub fn new() -> Self {
+        Self {
+            min_resource: 1,
+            max_resource: 81,
+            reduction_factor: 3,
+            min_points_in_model: 10,
+            tpe: TpeSampler::new(),
+        }
+    }
+
+    /// Creates a builder for configuring a BOHB sampler.
+    ///
+    /// # Examples
+    ///
+    /// ```
+    /// use optimizer::sampler::bohb::BohbSampler;
+    ///
+    /// let sampler = BohbSampler::builder()
+    ///     .min_resource(1)
+    ///     .max_resource(27)
+    ///     .reduction_factor(3)
+    ///     .min_points_in_model(5)
+    ///     .seed(42)
+    ///     .build()
+    ///     .unwrap();
+    /// ```
+    #[must_use]
+    pub fn builder() -> BohbSamplerBuilder {
+        BohbSamplerBuilder::new()
+    }
+
+    /// Creates a [`HyperbandPruner`] with matching Hyperband parameters.
+    ///
+    /// This ensures the pruner's budget schedule is consistent with the
+    /// budget levels used by BOHB for model conditioning.
+    #[must_use]
+    pub fn matching_pruner(&self, direction: Direction) -> HyperbandPruner {
+        HyperbandPruner::new()
+            .min_resource(self.min_resource)
+            .max_resource(self.max_resource)
+            .reduction_factor(self.reduction_factor)
+            .direction(direction)
+    }
+
+    /// Compute all unique budget levels (rung steps) across all Hyperband brackets.
+    ///
+    /// Returns sorted ascending.
+    #[allow(
+        clippy::cast_precision_loss,
+        clippy::cast_possible_truncation,
+        clippy::cast_sign_loss
+    )]
+    fn all_budget_levels(&self) -> Vec<u64> {
+        let eta = self.reduction_factor as f64;
+        let ratio = self.max_resource as f64 / self.min_resource as f64;
+        let s_max = (ratio.ln() / eta.ln()).floor() as u64;
+
+        let mut levels = Vec::new();
+        for bracket in 0..=s_max {
+            let exponent = s_max.saturating_sub(bracket);
+            let min_resource_bracket =
+                (self.max_resource as f64 / eta.powi(exponent as i32)).ceil() as u64;
+
+            let mut rung: u32 = 0;
+            while let Some(power) = self.reduction_factor.checked_pow(rung) {
+                let step = min_resource_bracket.saturating_mul(power);
+                if step > self.max_resource {
+                    break;
+                }
+                levels.push(step);
+                rung += 1;
+            }
+        }
+
+        levels.sort_unstable();
+        levels.dedup();
+        levels
+    }
+
+    /// Build a filtered history for a specific budget level.
+    ///
+    /// For each trial that has an intermediate value at the given budget step,
+    /// creates a new `CompletedTrial` with `value` replaced by the intermediate
+    /// value at that step.
+    fn filter_history_for_budget(history: &[CompletedTrial], budget: u64) -> Vec<CompletedTrial> {
+        history
+            .iter()
+            .filter_map(|trial| {
+                trial
+                    .intermediate_values
+                    .iter()
+                    .find(|(step, _)| *step == budget)
+                    .map(|(_, iv)| CompletedTrial {
+                        id: trial.id,
+                        params: trial.params.clone(),
+                        distributions: trial.distributions.clone(),
+                        param_labels: trial.param_labels.clone(),
+                        value: *iv,
+                        intermediate_values: trial.intermediate_values.clone(),
+                        state: trial.state,
+                        user_attrs: trial.user_attrs.clone(),
+                        constraints: trial.constraints.clone(),
+                    })
+            })
+            .collect()
+    }
+}
+
+impl Default for BohbSampler {
+    fn default() -> Self {
+        Self::new()
+    }
+}
+
+impl Sampler for BohbSampler {
+    fn sample(
+        &self,
+        distribution: &Distribution,
+        trial_id: u64,
+        history: &[CompletedTrial],
+    ) -> ParamValue {
+        // Find the largest budget level with enough observations
+        let levels = self.all_budget_levels();
+
+        for &budget in levels.iter().rev() {
+            let count = history
+                .iter()
+                .filter(|t| {
+                    t.intermediate_values
+                        .iter()
+                        .any(|(step, _)| *step == budget)
+                })
+                .count();
+
+            if count >= self.min_points_in_model {
+                let filtered = Self::filter_history_for_budget(history, budget);
+                return self.tpe.sample(distribution, trial_id, &filtered);
+            }
+        }
+
+        // Not enough data at any budget level: delegate to TPE with empty history
+        // which triggers its uniform-random startup behavior.
+        self.tpe.sample(distribution, trial_id, &[])
+    }
+}
+
+/// Builder for configuring a [`BohbSampler`].
+///
+/// # Examples
+///
+/// ```
+/// use optimizer::sampler::bohb::BohbSamplerBuilder;
+///
+/// let sampler = BohbSamplerBuilder::new()
+///     .min_resource(1)
+///     .max_resource(81)
+///     .reduction_factor(3)
+///     .gamma(0.15)
+///     .seed(42)
+///     .build()
+///     .unwrap();
+/// ```
+pub struct BohbSamplerBuilder {
+    min_resource: u64,
+    max_resource: u64,
+    reduction_factor: u64,
+    min_points_in_model: usize,
+    tpe_builder: crate::sampler::tpe::TpeSamplerBuilder,
+}
+
+impl BohbSamplerBuilder {
+    /// Creates a new builder with default settings.
+    #[must_use]
+    pub fn new() -> Self {
+        Self {
+            min_resource: 1,
+            max_resource: 81,
+            reduction_factor: 3,
+            min_points_in_model: 10,
+            tpe_builder: crate::sampler::tpe::TpeSamplerBuilder::new(),
+        }
+    }
+
+    /// Sets the minimum resource (budget) per trial.
+    ///
+    /// # Panics
+    ///
+    /// Panics if `r` is 0.
+    #[must_use]
+    pub fn min_resource(mut self, r: u64) -> Self {
+        assert!(r > 0, "min_resource must be > 0, got {r}");
+        self.min_resource = r;
+        self
+    }
+
+    /// Sets the maximum resource (budget) per trial.
+    ///
+    /// # Panics
+    ///
+    /// Panics if `r` is 0.
+    #[must_use]
+    pub fn max_resource(mut self, r: u64) -> Self {
+        assert!(r > 0, "max_resource must be > 0, got {r}");
+        self.max_resource = r;
+        self
+    }
+
+    /// Sets the reduction factor (eta).
+    ///
+    /// # Panics
+    ///
+    /// Panics if `eta` is less than 2.
+    #[must_use]
+    pub fn reduction_factor(mut self, eta: u64) -> Self {
+        assert!(eta >= 2, "reduction_factor must be >= 2, got {eta}");
+        self.reduction_factor = eta;
+        self
+    }
+
+    /// Sets the minimum number of observations at a budget level before
+    /// BOHB uses TPE instead of random sampling.
+    #[must_use]
+    pub fn min_points_in_model(mut self, n: usize) -> Self {
+        self.min_points_in_model = n;
+        self
+    }
+
+    /// Sets a fixed gamma value for the internal TPE sampler.
+    #[must_use]
+    pub fn gamma(mut self, gamma: f64) -> Self {
+        self.tpe_builder = self.tpe_builder.gamma(gamma);
+        self
+    }
+
+    /// Sets a custom gamma strategy for the internal TPE sampler.
+    #[must_use]
+    pub fn gamma_strategy<G: crate::sampler::tpe::GammaStrategy + 'static>(
+        mut self,
+        strategy: G,
+    ) -> Self {
+        self.tpe_builder = self.tpe_builder.gamma_strategy(strategy);
+        self
+    }
+
+    /// Sets the number of EI candidates for the internal TPE sampler.
+    #[must_use]
+    pub fn n_ei_candidates(mut self, n: usize) -> Self {
+        self.tpe_builder = self.tpe_builder.n_ei_candidates(n);
+        self
+    }
+
+    /// Sets a fixed KDE bandwidth for the internal TPE sampler.
+    #[must_use]
+    pub fn kde_bandwidth(mut self, bandwidth: f64) -> Self {
+        self.tpe_builder = self.tpe_builder.kde_bandwidth(bandwidth);
+        self
+    }
+
+    /// Sets a seed for reproducible sampling.
+    #[must_use]
+    pub fn seed(mut self, seed: u64) -> Self {
+        self.tpe_builder = self.tpe_builder.seed(seed);
+        self
+    }
+
+    /// Builds the configured [`BohbSampler`].
+    ///
+    /// # Errors
+    ///
+    /// Returns an error if the TPE configuration is invalid (e.g. gamma
+    /// not in (0, 1) or bandwidth not positive).
+    pub fn build(self) -> Result<BohbSampler> {
+        let tpe = self.tpe_builder.build()?;
+        Ok(BohbSampler {
+            min_resource: self.min_resource,
+            max_resource: self.max_resource,
+            reduction_factor: self.reduction_factor,
+            min_points_in_model: self.min_points_in_model,
+            tpe,
+        })
+    }
+}
+
+impl Default for BohbSamplerBuilder {
+    fn default() -> Self {
+        Self::new()
+    }
+}
+
+#[cfg(test)]
+#[allow(clippy::cast_precision_loss)]
+mod tests {
+    use std::collections::HashMap;
+
+    use super::*;
+    use crate::distribution::{FloatDistribution, IntDistribution};
+    use crate::parameter::ParamId;
+    use crate::types::TrialState;
+
+    fn make_trial_with_intermediates(
+        id: u64,
+        value: f64,
+        params: Vec<(ParamId, ParamValue, Distribution)>,
+        intermediate_values: Vec<(u64, f64)>,
+    ) -> CompletedTrial {
+        let mut param_map = HashMap::new();
+        let mut dist_map = HashMap::new();
+        for (param_id, pv, dist) in params {
+            param_map.insert(param_id, pv);
+            dist_map.insert(param_id, dist);
+        }
+        CompletedTrial {
+            id,
+            params: param_map,
+            distributions: dist_map,
+            param_labels: HashMap::new(),
+            value,
+            intermediate_values,
+            state: TrialState::Complete,
+            user_attrs: HashMap::new(),
+            constraints: Vec::new(),
+        }
+    }
+
+    #[test]
+    fn budget_levels_default() {
+        let bohb = BohbSampler::new();
+        let levels = bohb.all_budget_levels();
+        // With min=1, max=81, eta=3:
+        // bracket 0: 1, 3, 9, 27, 81
+        // bracket 1: 3, 9, 27, 81
+        // bracket 2: 9, 27, 81
+        // bracket 3: 27, 81
+        // bracket 4: 81
+        // Unique sorted: [1, 3, 9, 27, 81]
+        assert_eq!(levels, vec![1, 3, 9, 27, 81]);
+    }
+
+    #[test]
+    fn budget_levels_eta2() {
+        let bohb = BohbSampler::builder()
+            .min_resource(1)
+            .max_resource(16)
+            .reduction_factor(2)
+            .build()
+            .unwrap();
+        let levels = bohb.all_budget_levels();
+        // s_max = floor(ln(16)/ln(2)) = 4
+        // bracket 0: 1, 2, 4, 8, 16
+        // bracket 1: 2, 4, 8, 16
+        // bracket 2: 4, 8, 16
+        // bracket 3: 8, 16
+        // bracket 4: 16
+        // Unique sorted: [1, 2, 4, 8, 16]
+        assert_eq!(levels, vec![1, 2, 4, 8, 16]);
+    }
+
+    #[test]
+    fn filter_history_selects_correct_budget() {
+        let x_id = ParamId::new();
+        let dist = Distribution::Float(FloatDistribution {
+            low: 0.0,
+            high: 1.0,
+            log_scale: false,
+            step: None,
+        });
+
+        let history = vec![
+            make_trial_with_intermediates(
+                0,
+                0.5,
+                vec![(x_id, ParamValue::Float(0.3), dist.clone())],
+                vec![(1, 0.9), (3, 0.7), (9, 0.5)],
+            ),
+            make_trial_with_intermediates(
+                1,
+                0.4,
+                vec![(x_id, ParamValue::Float(0.6), dist.clone())],
+                vec![(1, 0.8), (3, 0.4)],
+            ),
+            make_trial_with_intermediates(
+                2,
+                0.3,
+                vec![(x_id, ParamValue::Float(0.1), dist.clone())],
+                vec![(1, 0.7)],
+            ),
+        ];
+
+        // Budget 3: trials 0 and 1 have intermediate values at step 3
+        let filtered = BohbSampler::filter_history_for_budget(&history, 3);
+        assert_eq!(filtered.len(), 2);
+        assert!((filtered[0].value - 0.7).abs() < f64::EPSILON);
+        assert!((filtered[1].value - 0.4).abs() < f64::EPSILON);
+
+        // Budget 9: only trial 0
+        let filtered = BohbSampler::filter_history_for_budget(&history, 9);
+        assert_eq!(filtered.len(), 1);
+        assert!((filtered[0].value - 0.5).abs() < f64::EPSILON);
+
+        // Budget 27: nobody
+        let filtered = BohbSampler::filter_history_for_budget(&history, 27);
+        assert!(filtered.is_empty());
+    }
+
+    #[test]
+    fn matching_pruner_has_same_params() {
+        let bohb = BohbSampler::builder()
+            .min_resource(2)
+            .max_resource(64)
+            .reduction_factor(4)
+            .build()
+            .unwrap();
+        let pruner = bohb.matching_pruner(Direction::Minimize);
+
+        // We can't directly inspect HyperbandPruner fields, but we can
+        // verify it was created without panicking with the same params.
+        // The pruner's rung steps should match BOHB's budget levels.
+        // Just verify it doesn't panic.
+        drop(pruner);
+    }
+
+    #[test]
+    fn fallback_to_random_when_insufficient_data() {
+        let bohb = BohbSampler::builder()
+            .min_points_in_model(10)
+            .seed(42)
+            .build()
+            .unwrap();
+
+        let dist = Distribution::Float(FloatDistribution {
+            low: 0.0,
+            high: 1.0,
+            log_scale: false,
+            step: None,
+        });
+
+        // Only 3 trials with intermediate values (< min_points_in_model=10)
+        let x_id = ParamId::new();
+        let history: Vec<CompletedTrial> = (0..3)
+            .map(|i| {
+                make_trial_with_intermediates(
+                    i,
+                    i as f64,
+                    vec![(x_id, ParamValue::Float(i as f64 / 3.0), dist.clone())],
+                    vec![(1, i as f64)],
+                )
+            })
+            .collect();
+
+        // Should not panic, should sample within bounds
+        for trial_id in 0..20 {
+            let val = bohb.sample(&dist, trial_id, &history);
+            if let ParamValue::Float(v) = val {
+                assert!((0.0..=1.0).contains(&v));
+            } else {
+                panic!("Expected Float");
+            }
+        }
+    }
+
+    #[test]
+    fn uses_budget_level_when_enough_data() {
+        let bohb = BohbSampler::builder()
+            .min_points_in_model(5)
+            .seed(42)
+            .build()
+            .unwrap();
+
+        let dist = Distribution::Float(FloatDistribution {
+            low: 0.0,
+            high: 10.0,
+            log_scale: false,
+            step: None,
+        });
+
+        // Create 20 trials with intermediate values at budget 1.
+        // Good trials have x near 2.0, bad trials have x far from 2.0.
+        let x_id = ParamId::new();
+        let history: Vec<CompletedTrial> = (0..20)
+            .map(|i| {
+                let x = i as f64 / 2.0;
+                let iv_at_1 = (x - 2.0).powi(2);
+                make_trial_with_intermediates(
+                    i,
+                    iv_at_1, // final value same as intermediate for simplicity
+                    vec![(x_id, ParamValue::Float(x), dist.clone())],
+                    vec![(1, iv_at_1)],
+                )
+            })
+            .collect();
+
+        // Should use TPE on filtered history at budget 1
+        let val = bohb.sample(&dist, 100, &history);
+        if let ParamValue::Float(v) = val {
+            assert!((0.0..=10.0).contains(&v), "Value {v} out of bounds");
+        } else {
+            panic!("Expected Float");
+        }
+    }
+
+    #[test]
+    fn prefers_largest_budget_level() {
+        let bohb = BohbSampler::builder()
+            .min_resource(1)
+            .max_resource(9)
+            .reduction_factor(3)
+            .min_points_in_model(3)
+            .seed(42)
+            .build()
+            .unwrap();
+
+        let dist = Distribution::Float(FloatDistribution {
+            low: 0.0,
+            high: 10.0,
+            log_scale: false,
+            step: None,
+        });
+
+        // Budget levels: [1, 3, 9]
+        assert_eq!(bohb.all_budget_levels(), vec![1, 3, 9]);
+
+        // Create 5 trials with intermediates at budget 1 and 3
+        let x_id = ParamId::new();
+        let history: Vec<CompletedTrial> = (0..5)
+            .map(|i| {
+                let x = i as f64;
+                make_trial_with_intermediates(
+                    i,
+                    x,
+                    vec![(x_id, ParamValue::Float(x), dist.clone())],
+                    vec![(1, x * 2.0), (3, x)],
+                )
+            })
+            .collect();
+
+        // Budget 3 has 5 observations (>= 3), budget 9 has 0.
+        // BOHB should pick budget 3 (largest with enough data).
+        // The filtered history at budget 3 has values [0, 1, 2, 3, 4].
+        let filtered_3 = BohbSampler::filter_history_for_budget(&history, 3);
+        assert_eq!(filtered_3.len(), 5);
+        let filtered_9 = BohbSampler::filter_history_for_budget(&history, 9);
+        assert_eq!(filtered_9.len(), 0);
+
+        // Should sample successfully
+        let val = bohb.sample(&dist, 100, &history);
+        assert!(matches!(val, ParamValue::Float(_)));
+    }
+
+    #[test]
+    fn builder_validates_tpe_params() {
+        // Invalid gamma
+        let result = BohbSampler::builder().gamma(1.5).build();
+        assert!(result.is_err());
+
+        // Invalid bandwidth
+        let result = BohbSampler::builder().kde_bandwidth(-1.0).build();
+        assert!(result.is_err());
+    }
+
+    #[test]
+    #[should_panic(expected = "min_resource must be > 0")]
+    fn builder_rejects_zero_min_resource() {
+        let _ = BohbSampler::builder().min_resource(0);
+    }
+
+    #[test]
+    #[should_panic(expected = "max_resource must be > 0")]
+    fn builder_rejects_zero_max_resource() {
+        let _ = BohbSampler::builder().max_resource(0);
+    }
+
+    #[test]
+    #[should_panic(expected = "reduction_factor must be >= 2")]
+    fn builder_rejects_small_reduction_factor() {
+        let _ = BohbSampler::builder().reduction_factor(1);
+    }
+
+    #[test]
+    fn int_distribution_works() {
+        let bohb = BohbSampler::builder()
+            .min_points_in_model(3)
+            .seed(42)
+            .build()
+            .unwrap();
+
+        let dist = Distribution::Int(IntDistribution {
+            low: 0,
+            high: 100,
+            log_scale: false,
+            step: None,
+        });
+
+        let x_id = ParamId::new();
+        let history: Vec<CompletedTrial> = (0..10)
+            .map(|i| {
+                make_trial_with_intermediates(
+                    i,
+                    i as f64,
+                    vec![(x_id, ParamValue::Int(i.cast_signed() * 10), dist.clone())],
+                    vec![(1, i as f64)],
+                )
+            })
+            .collect();
+
+        let val = bohb.sample(&dist, 100, &history);
+        if let ParamValue::Int(v) = val {
+            assert!((0..=100).contains(&v));
+        } else {
+            panic!("Expected Int");
+        }
+    }
+}

src/sampler/differential_evolution.rs

@@ -0,0 +1,995 @@
+//! Differential Evolution (DE) sampler.
+//!
+//! DE is a population-based metaheuristic that maintains a population of
+//! candidate solutions and creates new candidates by combining (mutating +
+//! crossing over) existing ones. It is competitive with CMA-ES on many
+//! problems and simpler to implement.
+//!
+//! Categorical parameters are sampled uniformly at random (not part of the
+//! DE vector). If all parameters are categorical, the sampler falls back to
+//! pure random sampling.
+//!
+//! # Examples
+//!
+//! ```
+//! use optimizer::sampler::differential_evolution::DifferentialEvolutionSampler;
+//! use optimizer::{Direction, Study};
+//!
+//! let sampler = DifferentialEvolutionSampler::with_seed(42);
+//! let study: Study<f64> = Study::with_sampler(Direction::Minimize, sampler);
+//! ```
+
+use std::collections::HashMap;
+
+use parking_lot::Mutex;
+
+use crate::distribution::Distribution;
+use crate::param::ParamValue;
+use crate::rng_util;
+use crate::sampler::{CompletedTrial, Sampler};
+
+/// Differential Evolution mutation strategy.
+///
+/// Controls how mutant vectors are created from the current population.
+#[derive(Clone, Copy, Debug, Default)]
+pub enum DifferentialEvolutionStrategy {
+    /// DE/rand/1: `v = x_r1 + F * (x_r2 - x_r3)`
+    ///
+    /// The most robust strategy. Uses three random population members.
+    #[default]
+    Rand1,
+    /// DE/best/1: `v = x_best + F * (x_r1 - x_r2)`
+    ///
+    /// Greedier strategy that biases toward the current best solution.
+    Best1,
+    /// DE/current-to-best/1: `v = x_i + F * (x_best - x_i) + F * (x_r1 - x_r2)`
+    ///
+    /// Balances exploration and exploitation by blending the current
+    /// individual with the best.
+    CurrentToBest1,
+}
+
+/// Differential Evolution sampler for continuous global optimization.
+///
+/// Maintains a population of candidate solutions. New candidates are
+/// created by combining (mutating + crossing over) existing members.
+///
+/// # Examples
+///
+/// ```
+/// use optimizer::sampler::differential_evolution::DifferentialEvolutionSampler;
+/// use optimizer::{Direction, Study};
+///
+/// // Default configuration
+/// let study: Study<f64> =
+///     Study::with_sampler(Direction::Minimize, DifferentialEvolutionSampler::new());
+///
+/// // With seed for reproducibility
+/// let study: Study<f64> = Study::with_sampler(
+///     Direction::Minimize,
+///     DifferentialEvolutionSampler::with_seed(42),
+/// );
+///
+/// // Custom configuration via builder
+/// use optimizer::sampler::differential_evolution::DifferentialEvolutionStrategy;
+/// let sampler = DifferentialEvolutionSampler::builder()
+///     .mutation_factor(0.8)
+///     .crossover_rate(0.9)
+///     .strategy(DifferentialEvolutionStrategy::Best1)
+///     .population_size(30)
+///     .seed(42)
+///     .build();
+/// let study: Study<f64> = Study::with_sampler(Direction::Minimize, sampler);
+/// ```
+pub struct DifferentialEvolutionSampler {
+    state: Mutex<State>,
+}
+
+impl DifferentialEvolutionSampler {
+    /// Creates a new DE sampler with default settings and a random seed.
+    #[must_use]
+    pub fn new() -> Self {
+        Self {
+            state: Mutex::new(State::new(
+                None,
+                0.8,
+                0.9,
+                DifferentialEvolutionStrategy::Rand1,
+                None,
+            )),
+        }
+    }
+
+    /// Creates a new DE sampler with a fixed seed for reproducibility.
+    #[must_use]
+    pub fn with_seed(seed: u64) -> Self {
+        Self {
+            state: Mutex::new(State::new(
+                None,
+                0.8,
+                0.9,
+                DifferentialEvolutionStrategy::Rand1,
+                Some(seed),
+            )),
+        }
+    }
+
+    /// Creates a builder for configuring a `DifferentialEvolutionSampler`.
+    #[must_use]
+    pub fn builder() -> DifferentialEvolutionSamplerBuilder {
+        DifferentialEvolutionSamplerBuilder::new()
+    }
+}
+
+impl Default for DifferentialEvolutionSampler {
+    fn default() -> Self {
+        Self::new()
+    }
+}
+
+/// Builder for configuring a [`DifferentialEvolutionSampler`].
+///
+/// All options have sensible defaults:
+/// - `population_size`: `max(10 * n_dims, 15)` (auto-computed from parameter count)
+/// - `mutation_factor` (F): 0.8
+/// - `crossover_rate` (CR): 0.9
+/// - `strategy`: `Rand1`
+/// - `seed`: random
+///
+/// # Examples
+///
+/// ```
+/// use optimizer::sampler::differential_evolution::{
+///     DifferentialEvolutionSamplerBuilder, DifferentialEvolutionStrategy,
+/// };
+///
+/// let sampler = DifferentialEvolutionSamplerBuilder::new()
+///     .mutation_factor(0.5)
+///     .crossover_rate(0.7)
+///     .strategy(DifferentialEvolutionStrategy::CurrentToBest1)
+///     .population_size(20)
+///     .seed(42)
+///     .build();
+/// ```
+#[derive(Debug, Clone)]
+pub struct DifferentialEvolutionSamplerBuilder {
+    population_size: Option<usize>,
+    mutation_factor: f64,
+    crossover_rate: f64,
+    strategy: DifferentialEvolutionStrategy,
+    seed: Option<u64>,
+}
+
+impl Default for DifferentialEvolutionSamplerBuilder {
+    fn default() -> Self {
+        Self::new()
+    }
+}
+
+impl DifferentialEvolutionSamplerBuilder {
+    /// Creates a new builder with default settings.
+    #[must_use]
+    pub fn new() -> Self {
+        Self {
+            population_size: None,
+            mutation_factor: 0.8,
+            crossover_rate: 0.9,
+            strategy: DifferentialEvolutionStrategy::Rand1,
+            seed: None,
+        }
+    }
+
+    /// Sets the population size.
+    ///
+    /// Number of candidate solutions maintained across generations.
+    /// Larger populations improve robustness but require more evaluations
+    /// per generation.
+    ///
+    /// Default: `max(10 * n_continuous_dims, 15)`.
+    #[must_use]
+    pub fn population_size(mut self, size: usize) -> Self {
+        self.population_size = Some(size);
+        self
+    }
+
+    /// Sets the mutation factor (F).
+    ///
+    /// Controls the amplification of differential variation.
+    /// Typical values are in `[0.5, 1.0]`. Higher values increase
+    /// exploration; lower values favor exploitation.
+    ///
+    /// Default: 0.8.
+    #[must_use]
+    pub fn mutation_factor(mut self, f: f64) -> Self {
+        self.mutation_factor = f;
+        self
+    }
+
+    /// Sets the crossover rate (CR).
+    ///
+    /// Probability of each dimension being taken from the mutant vector
+    /// rather than the parent. Typical values are in `[0.7, 1.0]`.
+    ///
+    /// Default: 0.9.
+    #[must_use]
+    pub fn crossover_rate(mut self, cr: f64) -> Self {
+        self.crossover_rate = cr;
+        self
+    }
+
+    /// Sets the mutation strategy.
+    ///
+    /// Default: [`DifferentialEvolutionStrategy::Rand1`].
+    #[must_use]
+    pub fn strategy(mut self, strategy: DifferentialEvolutionStrategy) -> Self {
+        self.strategy = strategy;
+        self
+    }
+
+    /// Sets the random seed for reproducibility.
+    #[must_use]
+    pub fn seed(mut self, seed: u64) -> Self {
+        self.seed = Some(seed);
+        self
+    }
+
+    /// Builds the configured [`DifferentialEvolutionSampler`].
+    #[must_use]
+    pub fn build(self) -> DifferentialEvolutionSampler {
+        DifferentialEvolutionSampler {
+            state: Mutex::new(State::new(
+                self.population_size,
+                self.mutation_factor,
+                self.crossover_rate,
+                self.strategy,
+                self.seed,
+            )),
+        }
+    }
+}
+
+// ---------------------------------------------------------------------------
+// Internal types
+// ---------------------------------------------------------------------------
+
+/// Describes how a parameter dimension maps into the DE internal vector.
+#[derive(Clone, Debug)]
+struct DimensionInfo {
+    /// The distribution for this dimension (stored for decoding).
+    distribution: Distribution,
+    /// Whether this dimension participates in DE (Float/Int = true, Categorical = false).
+    is_continuous: bool,
+    /// Internal-space bounds for continuous dimensions: `(low, high)`.
+    /// For log-scale parameters these are in log-space.
+    bounds: Option<(f64, f64)>,
+}
+
+/// A candidate solution produced by mutation + crossover.
+#[derive(Clone, Debug)]
+struct Candidate {
+    /// Internal-space vector (only continuous dimensions).
+    x: Vec<f64>,
+    /// Values for categorical dimensions (index in `dimensions` -> categorical index).
+    categorical_values: HashMap<usize, usize>,
+    /// Index of the population member this candidate competes against.
+    target_idx: usize,
+}
+
+/// Tracks per-trial sampling progress.
+#[derive(Clone, Debug)]
+struct TrialProgress {
+    /// Index of the candidate assigned to this trial.
+    candidate_idx: usize,
+    /// Next dimension to return for this trial.
+    next_dim: usize,
+}
+
+/// Phase of the DE state machine.
+enum Phase {
+    /// Discovering the search space structure (first trial).
+    Discovery,
+    /// Active sampling and evolving.
+    Active,
+}
+
+/// Top-level mutable state behind the `Mutex`.
+struct State {
+    /// The RNG used for sampling.
+    rng: fastrand::Rng,
+    /// User-provided population size (None = auto).
+    user_population_size: Option<usize>,
+    /// Mutation factor (F).
+    mutation_factor: f64,
+    /// Crossover rate (CR).
+    crossover_rate: f64,
+    /// Mutation strategy.
+    strategy: DifferentialEvolutionStrategy,
+    /// Current phase.
+    phase: Phase,
+    /// Discovered dimension info (populated during discovery).
+    dimensions: Vec<DimensionInfo>,
+    /// Last `trial_id` seen during discovery.
+    discovery_trial_id: Option<u64>,
+
+    // --- Population state ---
+    /// Current population (internal-space vectors, continuous dims only).
+    population: Vec<Vec<f64>>,
+    /// Categorical values for each population member.
+    population_categorical: Vec<HashMap<usize, usize>>,
+    /// Objective values for the current population.
+    population_values: Vec<f64>,
+    /// Index of the best population member.
+    best_idx: usize,
+    /// Whether the initial population has been evaluated.
+    initialized: bool,
+    /// Effective population size (resolved after discovery).
+    population_size: usize,
+
+    // --- Current generation ---
+    /// Current generation's candidates.
+    candidates: Vec<Candidate>,
+    /// Mapping from `trial_id` to its progress.
+    trial_progress: HashMap<u64, TrialProgress>,
+    /// Number of candidates assigned so far in the current generation.
+    assigned_count: usize,
+    /// Trial IDs assigned in the current generation.
+    generation_trial_ids: Vec<u64>,
+}
+
+impl State {
+    fn new(
+        user_population_size: Option<usize>,
+        mutation_factor: f64,
+        crossover_rate: f64,
+        strategy: DifferentialEvolutionStrategy,
+        seed: Option<u64>,
+    ) -> Self {
+        let rng = seed.map_or_else(fastrand::Rng::new, fastrand::Rng::with_seed);
+        Self {
+            rng,
+            user_population_size,
+            mutation_factor,
+            crossover_rate,
+            strategy,
+            phase: Phase::Discovery,
+            dimensions: Vec::new(),
+            discovery_trial_id: None,
+            population: Vec::new(),
+            population_categorical: Vec::new(),
+            population_values: Vec::new(),
+            best_idx: 0,
+            initialized: false,
+            population_size: 0,
+            candidates: Vec::new(),
+            trial_progress: HashMap::new(),
+            assigned_count: 0,
+            generation_trial_ids: Vec::new(),
+        }
+    }
+}
+
+// ---------------------------------------------------------------------------
+// Helpers
+// ---------------------------------------------------------------------------
+
+/// Compute internal-space bounds for a distribution.
+#[allow(clippy::cast_precision_loss)]
+fn internal_bounds(distribution: &Distribution) -> Option<(f64, f64)> {
+    match distribution {
+        Distribution::Float(d) => {
+            if d.log_scale {
+                Some((d.low.ln(), d.high.ln()))
+            } else {
+                Some((d.low, d.high))
+            }
+        }
+        Distribution::Int(d) => {
+            if d.log_scale {
+                Some(((d.low as f64).ln(), (d.high as f64).ln()))
+            } else {
+                Some((d.low as f64, d.high as f64))
+            }
+        }
+        Distribution::Categorical(_) => None,
+    }
+}
+
+/// Convert an internal-space value back to a `ParamValue`.
+#[allow(clippy::cast_possible_truncation, clippy::cast_precision_loss)]
+fn from_internal(value: f64, distribution: &Distribution) -> ParamValue {
+    match distribution {
+        Distribution::Float(d) => {
+            let v = if d.log_scale { value.exp() } else { value };
+            let v = if let Some(step) = d.step {
+                let k = ((v - d.low) / step).round();
+                d.low + k * step
+            } else {
+                v
+            };
+            ParamValue::Float(v.clamp(d.low, d.high))
+        }
+        Distribution::Int(d) => {
+            let v = if d.log_scale { value.exp() } else { value };
+            let v = if let Some(step) = d.step {
+                let k = ((v - d.low as f64) / step as f64).round() as i64;
+                d.low + k * step
+            } else {
+                v.round() as i64
+            };
+            ParamValue::Int(v.clamp(d.low, d.high))
+        }
+        Distribution::Categorical(_) => {
+            unreachable!("from_internal should not be called for categorical distributions")
+        }
+    }
+}
+
+/// Sample a random value for any distribution.
+#[allow(clippy::cast_possible_truncation, clippy::cast_precision_loss)]
+fn sample_random(rng: &mut fastrand::Rng, distribution: &Distribution) -> ParamValue {
+    match distribution {
+        Distribution::Float(d) => {
+            let value = if d.log_scale {
+                let log_low = d.low.ln();
+                let log_high = d.high.ln();
+                rng_util::f64_range(rng, log_low, log_high).exp()
+            } else if let Some(step) = d.step {
+                let n_steps = ((d.high - d.low) / step).floor() as i64;
+                let k = rng.i64(0..=n_steps);
+                d.low + (k as f64) * step
+            } else {
+                rng_util::f64_range(rng, d.low, d.high)
+            };
+            ParamValue::Float(value)
+        }
+        Distribution::Int(d) => {
+            let value = if d.log_scale {
+                let log_low = (d.low as f64).ln();
+                let log_high = (d.high as f64).ln();
+                let raw = rng_util::f64_range(rng, log_low, log_high).exp().round() as i64;
+                raw.clamp(d.low, d.high)
+            } else if let Some(step) = d.step {
+                let n_steps = (d.high - d.low) / step;
+                let k = rng.i64(0..=n_steps);
+                d.low + k * step
+            } else {
+                rng.i64(d.low..=d.high)
+            };
+            ParamValue::Int(value)
+        }
+        Distribution::Categorical(d) => ParamValue::Categorical(rng.usize(0..d.n_choices)),
+    }
+}
+
+/// Sample a random value in internal space for a continuous dimension.
+fn sample_random_internal(rng: &mut fastrand::Rng, bounds: (f64, f64)) -> f64 {
+    rng_util::f64_range(rng, bounds.0, bounds.1)
+}
+
+/// Clamp a value to the given bounds.
+fn clamp_to_bounds(value: f64, bounds: Option<(f64, f64)>) -> f64 {
+    if let Some((lo, hi)) = bounds {
+        value.clamp(lo, hi)
+    } else {
+        value
+    }
+}
+
+// ---------------------------------------------------------------------------
+// DE algorithm
+// ---------------------------------------------------------------------------
+
+/// Select `count` distinct random indices from `0..n`, all different from `exclude`.
+fn select_random_indices(
+    rng: &mut fastrand::Rng,
+    n: usize,
+    count: usize,
+    exclude: &[usize],
+) -> Vec<usize> {
+    let mut selected = Vec::with_capacity(count);
+    while selected.len() < count {
+        let idx = rng.usize(0..n);
+        if !exclude.contains(&idx) && !selected.contains(&idx) {
+            selected.push(idx);
+        }
+    }
+    selected
+}
+
+/// Generate trial vectors (mutation + crossover) for the current population.
+fn generate_trial_vectors(state: &mut State) -> Vec<Candidate> {
+    let n_continuous = state.dimensions.iter().filter(|d| d.is_continuous).count();
+    let pop_size = state.population_size;
+
+    let mut candidates = Vec::with_capacity(pop_size);
+
+    for i in 0..pop_size {
+        // Mutation
+        let mutant = create_mutant_with_rng(state, i, n_continuous);
+
+        // Crossover (binomial)
+        let j_rand = state.rng.usize(0..n_continuous.max(1));
+        let trial_x: Vec<f64> = if n_continuous > 0 {
+            (0..n_continuous)
+                .map(|j| {
+                    let use_mutant = j == j_rand || state.rng.f64() < state.crossover_rate;
+                    let val = if use_mutant {
+                        mutant[j]
+                    } else {
+                        state.population[i][j]
+                    };
+                    // Clamp to bounds
+                    let dim_bounds = continuous_dim_bounds(&state.dimensions, j);
+                    clamp_to_bounds(val, dim_bounds)
+                })
+                .collect()
+        } else {
+            Vec::new()
+        };
+
+        // Categorical: randomly sample (DE doesn't optimize categoricals)
+        let mut categorical_values = HashMap::new();
+        for (dim_idx, dim) in state.dimensions.iter().enumerate() {
+            if !dim.is_continuous
+                && let Distribution::Categorical(cat) = &dim.distribution
+            {
+                categorical_values.insert(dim_idx, state.rng.usize(0..cat.n_choices));
+            }
+        }
+
+        candidates.push(Candidate {
+            x: trial_x,
+            categorical_values,
+            target_idx: i,
+        });
+    }
+
+    candidates
+}
+
+/// Create a mutant vector, consuming RNG from state.
+fn create_mutant_with_rng(state: &mut State, target_idx: usize, n_continuous: usize) -> Vec<f64> {
+    if n_continuous == 0 {
+        return Vec::new();
+    }
+
+    let pop = &state.population;
+    let best_idx = state.best_idx;
+    let f = state.mutation_factor;
+    let pop_size = state.population_size;
+
+    match state.strategy {
+        DifferentialEvolutionStrategy::Rand1 => {
+            let indices = select_random_indices(&mut state.rng, pop_size, 3, &[target_idx]);
+            let (r1, r2, r3) = (indices[0], indices[1], indices[2]);
+            (0..n_continuous)
+                .map(|j| pop[r1][j] + f * (pop[r2][j] - pop[r3][j]))
+                .collect()
+        }
+        DifferentialEvolutionStrategy::Best1 => {
+            let indices = select_random_indices(&mut state.rng, pop_size, 2, &[target_idx]);
+            let (r1, r2) = (indices[0], indices[1]);
+            (0..n_continuous)
+                .map(|j| pop[best_idx][j] + f * (pop[r1][j] - pop[r2][j]))
+                .collect()
+        }
+        DifferentialEvolutionStrategy::CurrentToBest1 => {
+            let indices = select_random_indices(&mut state.rng, pop_size, 2, &[target_idx]);
+            let (r1, r2) = (indices[0], indices[1]);
+            (0..n_continuous)
+                .map(|j| {
+                    pop[target_idx][j]
+                        + f * (pop[best_idx][j] - pop[target_idx][j])
+                        + f * (pop[r1][j] - pop[r2][j])
+                })
+                .collect()
+        }
+    }
+}
+
+/// Get the bounds for the j-th continuous dimension.
+fn continuous_dim_bounds(
+    dimensions: &[DimensionInfo],
+    continuous_idx: usize,
+) -> Option<(f64, f64)> {
+    let mut ci = 0;
+    for dim in dimensions {
+        if dim.is_continuous {
+            if ci == continuous_idx {
+                return dim.bounds;
+            }
+            ci += 1;
+        }
+    }
+    None
+}
+
+/// Generate the initial random population.
+fn generate_initial_population(state: &mut State) -> Vec<Candidate> {
+    let n_continuous = state.dimensions.iter().filter(|d| d.is_continuous).count();
+
+    let mut candidates = Vec::with_capacity(state.population_size);
+
+    for i in 0..state.population_size {
+        let x: Vec<f64> = if n_continuous > 0 {
+            let mut v = Vec::with_capacity(n_continuous);
+            for dim in &state.dimensions {
+                if dim.is_continuous {
+                    let val = if let Some(bounds) = dim.bounds {
+                        sample_random_internal(&mut state.rng, bounds)
+                    } else {
+                        0.0
+                    };
+                    v.push(val);
+                }
+            }
+            v
+        } else {
+            Vec::new()
+        };
+
+        let mut categorical_values = HashMap::new();
+        for (dim_idx, dim) in state.dimensions.iter().enumerate() {
+            if !dim.is_continuous
+                && let Distribution::Categorical(cat) = &dim.distribution
+            {
+                categorical_values.insert(dim_idx, state.rng.usize(0..cat.n_choices));
+            }
+        }
+
+        candidates.push(Candidate {
+            x,
+            categorical_values,
+            target_idx: i,
+        });
+    }
+
+    candidates
+}
+
+// ---------------------------------------------------------------------------
+// Sampler trait implementation
+// ---------------------------------------------------------------------------
+
+impl Sampler for DifferentialEvolutionSampler {
+    #[allow(clippy::cast_precision_loss)]
+    fn sample(
+        &self,
+        distribution: &Distribution,
+        trial_id: u64,
+        history: &[CompletedTrial],
+    ) -> ParamValue {
+        let mut state = self.state.lock();
+
+        match &state.phase {
+            Phase::Discovery => sample_discovery(&mut state, distribution, trial_id),
+            Phase::Active => sample_active(&mut state, distribution, trial_id, history),
+        }
+    }
+}
+
+/// Handle sampling during the discovery phase.
+fn sample_discovery(state: &mut State, distribution: &Distribution, trial_id: u64) -> ParamValue {
+    // Check if this is a new trial (discovery phase ended for previous trial)
+    if let Some(prev_id) = state.discovery_trial_id
+        && trial_id != prev_id
+    {
+        // First trial is done; we know the search space. Initialize DE.
+        finalize_discovery(state);
+        return sample_active(state, distribution, trial_id, &[]);
+    }
+
+    // Record this trial_id
+    state.discovery_trial_id = Some(trial_id);
+
+    // Record this dimension
+    let is_continuous = !matches!(distribution, Distribution::Categorical(_));
+    let bounds = internal_bounds(distribution);
+    state.dimensions.push(DimensionInfo {
+        distribution: distribution.clone(),
+        is_continuous,
+        bounds,
+    });
+
+    // Sample randomly for the discovery trial
+    sample_random(&mut state.rng, distribution)
+}
+
+/// Finalize discovery and transition to the active phase.
+#[allow(clippy::cast_precision_loss)]
+fn finalize_discovery(state: &mut State) {
+    let n_continuous = state.dimensions.iter().filter(|d| d.is_continuous).count();
+
+    // Resolve population size
+    state.population_size = state
+        .user_population_size
+        .unwrap_or_else(|| (10 * n_continuous).max(15));
+
+    // Ensure population size is at least 4 (DE needs distinct random indices)
+    state.population_size = state.population_size.max(4);
+
+    // Generate initial random population
+    state.candidates = generate_initial_population(state);
+    state.assigned_count = 0;
+    state.generation_trial_ids.clear();
+    state.trial_progress.clear();
+    state.phase = Phase::Active;
+}
+
+/// Handle sampling during the active phase.
+fn sample_active(
+    state: &mut State,
+    distribution: &Distribution,
+    trial_id: u64,
+    history: &[CompletedTrial],
+) -> ParamValue {
+    // Check if we need to process completed trials and start a new generation
+    maybe_update_generation(state, history);
+
+    // Assign a candidate to this trial if not yet done
+    if !state.trial_progress.contains_key(&trial_id) {
+        assign_candidate(state, trial_id);
+    }
+
+    let progress = state.trial_progress.get_mut(&trial_id).unwrap();
+    let dim_idx = progress.next_dim;
+    progress.next_dim += 1;
+
+    // Safety check
+    if dim_idx >= state.dimensions.len() {
+        return sample_random(&mut state.rng, distribution);
+    }
+
+    let candidate = &state.candidates[progress.candidate_idx];
+    let dim_info = &state.dimensions[dim_idx];
+
+    if dim_info.is_continuous {
+        // Map from overall dimension index to continuous index
+        let ci = state.dimensions[..dim_idx]
+            .iter()
+            .filter(|d| d.is_continuous)
+            .count();
+        if ci < candidate.x.len() {
+            from_internal(candidate.x[ci], &dim_info.distribution)
+        } else {
+            sample_random(&mut state.rng, distribution)
+        }
+    } else {
+        // Categorical: use pre-sampled value
+        if let Some(&cat_idx) = candidate.categorical_values.get(&dim_idx) {
+            ParamValue::Categorical(cat_idx)
+        } else {
+            sample_random(&mut state.rng, distribution)
+        }
+    }
+}
+
+/// Assign a candidate to a trial.
+fn assign_candidate(state: &mut State, trial_id: u64) {
+    let candidate_idx = if state.assigned_count < state.candidates.len() {
+        let idx = state.assigned_count;
+        state.assigned_count += 1;
+        idx
+    } else {
+        // Overflow: generate an extra random candidate
+        let n_continuous = state.dimensions.iter().filter(|d| d.is_continuous).count();
+        let x: Vec<f64> = (0..n_continuous)
+            .map(|j| {
+                let bounds = continuous_dim_bounds(&state.dimensions, j);
+                if let Some(b) = bounds {
+                    sample_random_internal(&mut state.rng, b)
+                } else {
+                    0.0
+                }
+            })
+            .collect();
+        let mut categorical_values = HashMap::new();
+        for (dim_idx, dim) in state.dimensions.iter().enumerate() {
+            if !dim.is_continuous
+                && let Distribution::Categorical(cat) = &dim.distribution
+            {
+                categorical_values.insert(dim_idx, state.rng.usize(0..cat.n_choices));
+            }
+        }
+        state.candidates.push(Candidate {
+            x,
+            categorical_values,
+            target_idx: 0, // overflow candidates don't compete
+        });
+        let idx = state.candidates.len() - 1;
+        state.assigned_count = state.candidates.len();
+        idx
+    };
+
+    state.trial_progress.insert(
+        trial_id,
+        TrialProgress {
+            candidate_idx,
+            next_dim: 0,
+        },
+    );
+    state.generation_trial_ids.push(trial_id);
+}
+
+/// Check if we should process completed trials and start a new generation.
+fn maybe_update_generation(state: &mut State, history: &[CompletedTrial]) {
+    let pop_size = state.population_size;
+
+    // Only update when at least pop_size candidates have been assigned
+    if state.generation_trial_ids.len() < pop_size {
+        return;
+    }
+
+    // Check if the first pop_size trial IDs are all completed
+    let trial_ids: Vec<u64> = state
+        .generation_trial_ids
+        .iter()
+        .take(pop_size)
+        .copied()
+        .collect();
+    let history_map: HashMap<u64, f64> = history.iter().map(|t| (t.id, t.value)).collect();
+
+    let all_completed = trial_ids.iter().all(|id| history_map.contains_key(id));
+    if !all_completed {
+        return;
+    }
+
+    let n_continuous = state.dimensions.iter().filter(|d| d.is_continuous).count();
+
+    if state.initialized {
+        // Subsequent generations: selection
+        perform_selection(state, &trial_ids, &history_map);
+    } else {
+        // First generation: store as initial population
+        initialize_population(state, &trial_ids, &history_map, n_continuous);
+    }
+
+    // Generate next generation's trial vectors
+    state.candidates = if state.initialized && n_continuous > 0 {
+        generate_trial_vectors(state)
+    } else {
+        generate_initial_population(state)
+    };
+    state.assigned_count = 0;
+    state.generation_trial_ids.clear();
+    state.trial_progress.clear();
+}
+
+/// Initialize the population from the first generation's results.
+fn initialize_population(
+    state: &mut State,
+    trial_ids: &[u64],
+    history_map: &HashMap<u64, f64>,
+    _n_continuous: usize,
+) {
+    state.population.clear();
+    state.population_categorical.clear();
+    state.population_values.clear();
+
+    let mut best_value = f64::INFINITY;
+    let mut best_idx = 0;
+
+    for (i, &trial_id) in trial_ids.iter().enumerate() {
+        let progress = &state.trial_progress[&trial_id];
+        let candidate = &state.candidates[progress.candidate_idx];
+        let value = history_map[&trial_id];
+
+        state.population.push(candidate.x.clone());
+        state
+            .population_categorical
+            .push(candidate.categorical_values.clone());
+        state.population_values.push(value);
+
+        if value < best_value {
+            best_value = value;
+            best_idx = i;
+        }
+    }
+
+    state.best_idx = best_idx;
+    state.initialized = true;
+}
+
+/// Perform DE selection: replace parent if trial vector is better.
+fn perform_selection(state: &mut State, trial_ids: &[u64], history_map: &HashMap<u64, f64>) {
+    for &trial_id in trial_ids {
+        let progress = &state.trial_progress[&trial_id];
+        let candidate = &state.candidates[progress.candidate_idx];
+        let trial_value = history_map[&trial_id];
+        let target_idx = candidate.target_idx;
+
+        if target_idx < state.population_size && trial_value <= state.population_values[target_idx]
+        {
+            state.population[target_idx] = candidate.x.clone();
+            state.population_categorical[target_idx] = candidate.categorical_values.clone();
+            state.population_values[target_idx] = trial_value;
+        }
+    }
+
+    // Update best index
+    let mut best_value = f64::INFINITY;
+    let mut best_idx = 0;
+    for (i, &val) in state.population_values.iter().enumerate() {
+        if val < best_value {
+            best_value = val;
+            best_idx = i;
+        }
+    }
+    state.best_idx = best_idx;
+}
+
+#[cfg(test)]
+#[allow(clippy::cast_possible_truncation, clippy::cast_precision_loss)]
+mod tests {
+    use super::*;
+    use crate::distribution::FloatDistribution;
+
+    #[test]
+    fn test_de_sampler_basic_float() {
+        let sampler = DifferentialEvolutionSampler::with_seed(42);
+        let dist = Distribution::Float(FloatDistribution {
+            low: -5.0,
+            high: 5.0,
+            log_scale: false,
+            step: None,
+        });
+
+        // Sample many values and check bounds
+        for i in 0..100 {
+            let value = sampler.sample(&dist, i, &[]);
+            if let ParamValue::Float(v) = value {
+                assert!(
+                    (-5.0..=5.0).contains(&v),
+                    "value {v} out of bounds at trial {i}"
+                );
+            } else {
+                panic!("Expected Float value");
+            }
+        }
+    }
+
+    #[test]
+    fn test_de_sampler_reproducibility() {
+        let dist = Distribution::Float(FloatDistribution {
+            low: 0.0,
+            high: 1.0,
+            log_scale: false,
+            step: None,
+        });
+
+        let sample_values = |seed: u64| {
+            let sampler = DifferentialEvolutionSampler::with_seed(seed);
+            (0..20)
+                .map(|i| sampler.sample(&dist, i, &[]))
+                .collect::<Vec<_>>()
+        };
+
+        let v1 = sample_values(42);
+        let v2 = sample_values(42);
+        assert_eq!(v1, v2, "same seed should produce same results");
+
+        let v3 = sample_values(99);
+        assert_ne!(v1, v3, "different seeds should produce different results");
+    }
+
+    #[test]
+    fn test_de_strategy_default() {
+        assert!(matches!(
+            DifferentialEvolutionStrategy::default(),
+            DifferentialEvolutionStrategy::Rand1
+        ));
+    }
+
+    #[test]
+    fn test_builder_defaults() {
+        let builder = DifferentialEvolutionSamplerBuilder::new();
+        assert!(builder.population_size.is_none());
+        assert!((builder.mutation_factor - 0.8).abs() < f64::EPSILON);
+        assert!((builder.crossover_rate - 0.9).abs() < f64::EPSILON);
+        assert!(matches!(
+            builder.strategy,
+            DifferentialEvolutionStrategy::Rand1
+        ));
+        assert!(builder.seed.is_none());
+    }
+}

src/sampler/genetic.rs

@@ -0,0 +1,572 @@
+//! Shared types and genetic operators for evolutionary multi-objective samplers.
+//!
+//! This module extracts common functionality used by NSGA-II, NSGA-III, and MOEA/D:
+//! candidate management, discovery/active phase logic, SBX crossover,
+//! polynomial mutation, and Das-Dennis reference point generation.
+
+use std::collections::HashMap;
+
+use crate::distribution::Distribution;
+use crate::multi_objective::MultiObjectiveTrial;
+use crate::param::ParamValue;
+use crate::rng_util;
+
+/// Describes a parameter dimension discovered during the first trial.
+#[derive(Clone, Debug)]
+pub(crate) struct DimensionInfo {
+    pub distribution: Distribution,
+}
+
+/// A candidate solution: one value per dimension.
+#[derive(Clone, Debug)]
+pub(crate) struct Candidate {
+    pub params: Vec<ParamValue>,
+}
+
+/// Tracks per-trial sampling progress (which candidate, which dimension next).
+#[derive(Clone, Debug)]
+pub(crate) struct TrialProgress {
+    pub candidate_idx: usize,
+    pub next_dim: usize,
+}
+
+/// Phase of an evolutionary sampler.
+pub(crate) enum Phase {
+    /// First trial reveals parameter dimensions.
+    Discovery,
+    /// Evolutionary optimisation.
+    Active,
+}
+
+/// Common state shared by all evolutionary multi-objective samplers.
+pub(crate) struct EvolutionaryState {
+    pub rng: fastrand::Rng,
+    pub phase: Phase,
+    pub dimensions: Vec<DimensionInfo>,
+    pub population_size: usize,
+    pub candidates: Vec<Candidate>,
+    pub trial_progress: HashMap<u64, TrialProgress>,
+    pub assigned_count: usize,
+    pub generation_trial_ids: Vec<u64>,
+    pub discovery_trial_id: Option<u64>,
+    pub generation: usize,
+}
+
+impl EvolutionaryState {
+    pub(crate) fn new(seed: Option<u64>) -> Self {
+        let rng = seed.map_or_else(fastrand::Rng::new, fastrand::Rng::with_seed);
+        Self {
+            rng,
+            phase: Phase::Discovery,
+            dimensions: Vec::new(),
+            population_size: 4,
+            candidates: Vec::new(),
+            trial_progress: HashMap::new(),
+            assigned_count: 0,
+            generation_trial_ids: Vec::new(),
+            discovery_trial_id: None,
+            generation: 0,
+        }
+    }
+}
+
+// ---------------------------------------------------------------------------
+// Discovery phase helpers
+// ---------------------------------------------------------------------------
+
+/// Handle sampling during the discovery phase.
+///
+/// Returns `Some(value)` if the discovery phase handled the sample,
+/// or `None` if it transitioned to active phase and the caller should
+/// generate candidates and sample from them.
+pub(crate) fn sample_discovery(
+    evo: &mut EvolutionaryState,
+    distribution: &Distribution,
+    trial_id: u64,
+) -> Option<ParamValue> {
+    if let Some(prev_id) = evo.discovery_trial_id
+        && trial_id != prev_id
+    {
+        // A new trial arrived — transition to active phase
+        return None;
+    }
+
+    evo.discovery_trial_id = Some(trial_id);
+    evo.dimensions.push(DimensionInfo {
+        distribution: distribution.clone(),
+    });
+
+    Some(sample_random(&mut evo.rng, distribution))
+}
+
+/// Compute population size from dimensions and optional user override.
+#[allow(
+    clippy::cast_precision_loss,
+    clippy::cast_possible_truncation,
+    clippy::cast_sign_loss
+)]
+pub(crate) fn compute_population_size(
+    n_dims: usize,
+    user_pop_size: Option<usize>,
+    minimum: usize,
+) -> usize {
+    user_pop_size
+        .unwrap_or_else(|| (4.0 + 3.0 * (n_dims as f64).ln().max(0.0)).floor() as usize)
+        .max(minimum)
+}
+
+/// Transition from discovery to active phase.
+pub(crate) fn finalize_discovery(evo: &mut EvolutionaryState, user_pop_size: Option<usize>) {
+    evo.population_size = compute_population_size(evo.dimensions.len(), user_pop_size, 4);
+    evo.phase = Phase::Active;
+}
+
+/// Generate `population_size` random candidates.
+pub(crate) fn generate_random_candidates(evo: &mut EvolutionaryState) {
+    let pop = evo.population_size;
+    evo.candidates = (0..pop)
+        .map(|_| {
+            let params: Vec<ParamValue> = evo
+                .dimensions
+                .iter()
+                .map(|d| sample_random(&mut evo.rng, &d.distribution))
+                .collect();
+            Candidate { params }
+        })
+        .collect();
+    evo.assigned_count = 0;
+    evo.generation_trial_ids.clear();
+    evo.trial_progress.clear();
+}
+
+/// Assign a candidate to a trial and return the next dimension value.
+pub(crate) fn sample_from_candidate(evo: &mut EvolutionaryState, trial_id: u64) -> ParamValue {
+    if !evo.trial_progress.contains_key(&trial_id) {
+        let candidate_idx = if evo.assigned_count < evo.candidates.len() {
+            let idx = evo.assigned_count;
+            evo.assigned_count += 1;
+            idx
+        } else {
+            // Overflow: generate a random candidate
+            let params: Vec<ParamValue> = evo
+                .dimensions
+                .iter()
+                .map(|d| sample_random(&mut evo.rng, &d.distribution))
+                .collect();
+            evo.candidates.push(Candidate { params });
+            let idx = evo.candidates.len() - 1;
+            evo.assigned_count = evo.candidates.len();
+            idx
+        };
+
+        evo.trial_progress.insert(
+            trial_id,
+            TrialProgress {
+                candidate_idx,
+                next_dim: 0,
+            },
+        );
+        evo.generation_trial_ids.push(trial_id);
+    }
+
+    let progress = evo.trial_progress.get_mut(&trial_id).unwrap();
+    let dim_idx = progress.next_dim;
+    progress.next_dim += 1;
+
+    if dim_idx >= evo.dimensions.len() {
+        return sample_random(&mut evo.rng, &evo.dimensions.last().unwrap().distribution);
+    }
+
+    evo.candidates[progress.candidate_idx].params[dim_idx].clone()
+}
+
+/// Extract parameter values from a trial, ordered by dimension index.
+pub(crate) fn extract_trial_params(
+    trial: &MultiObjectiveTrial,
+    dimensions: &[DimensionInfo],
+    rng: &mut fastrand::Rng,
+) -> Vec<ParamValue> {
+    let mut param_pairs: Vec<_> = trial.params.iter().collect();
+    param_pairs.sort_by_key(|(id, _)| *id);
+
+    dimensions
+        .iter()
+        .enumerate()
+        .map(|(dim_idx, dim_info)| {
+            if dim_idx < param_pairs.len() {
+                param_pairs[dim_idx].1.clone()
+            } else {
+                sample_random(rng, &dim_info.distribution)
+            }
+        })
+        .collect()
+}
+
+/// Install new offspring as the next generation's candidates.
+pub(crate) fn advance_generation(evo: &mut EvolutionaryState, offspring: Vec<Candidate>) {
+    evo.candidates = offspring;
+    evo.assigned_count = 0;
+    evo.generation_trial_ids.clear();
+    evo.trial_progress.clear();
+    evo.generation += 1;
+}
+
+/// Check if the current generation is fully evaluated and return the
+/// evaluated trials if so.
+pub(crate) fn collect_evaluated_generation<'a>(
+    evo: &EvolutionaryState,
+    history: &'a [MultiObjectiveTrial],
+) -> Option<Vec<&'a MultiObjectiveTrial>> {
+    let pop_size = evo.population_size;
+
+    if evo.generation_trial_ids.len() < pop_size {
+        return None;
+    }
+
+    let gen_ids: Vec<u64> = evo
+        .generation_trial_ids
+        .iter()
+        .take(pop_size)
+        .copied()
+        .collect();
+    let history_map: HashMap<u64, &MultiObjectiveTrial> =
+        history.iter().map(|t| (t.id, t)).collect();
+
+    if !gen_ids.iter().all(|id| history_map.contains_key(id)) {
+        return None;
+    }
+
+    Some(
+        gen_ids
+            .iter()
+            .filter_map(|id| history_map.get(id).copied())
+            .collect(),
+    )
+}
+
+// ---------------------------------------------------------------------------
+// Genetic operators
+// ---------------------------------------------------------------------------
+
+/// SBX crossover for continuous params, uniform crossover for categorical.
+pub(crate) fn crossover(
+    rng: &mut fastrand::Rng,
+    parent1: &[ParamValue],
+    parent2: &[ParamValue],
+    dimensions: &[DimensionInfo],
+    crossover_prob: f64,
+    eta: f64,
+) -> (Vec<ParamValue>, Vec<ParamValue>) {
+    let n = parent1.len();
+    let mut child1 = parent1.to_vec();
+    let mut child2 = parent2.to_vec();
+
+    let u: f64 = rng_util::f64_range(rng, 0.0, 1.0);
+    if u > crossover_prob {
+        return (child1, child2);
+    }
+
+    for i in 0..n {
+        match (&parent1[i], &parent2[i], &dimensions[i].distribution) {
+            (ParamValue::Float(p1), ParamValue::Float(p2), Distribution::Float(d)) => {
+                if (p1 - p2).abs() < 1e-14 {
+                    continue;
+                }
+                let (c1, c2) = sbx_crossover_f64(rng, *p1, *p2, d.low, d.high, eta);
+                child1[i] = ParamValue::Float(c1);
+                child2[i] = ParamValue::Float(c2);
+            }
+            (ParamValue::Int(p1), ParamValue::Int(p2), Distribution::Int(d)) => {
+                if p1 == p2 {
+                    continue;
+                }
+                #[allow(clippy::cast_precision_loss)]
+                let (c1, c2) = sbx_crossover_f64(
+                    rng,
+                    *p1 as f64,
+                    *p2 as f64,
+                    d.low as f64,
+                    d.high as f64,
+                    eta,
+                );
+                #[allow(clippy::cast_possible_truncation)]
+                {
+                    child1[i] = ParamValue::Int((c1.round() as i64).clamp(d.low, d.high));
+                    child2[i] = ParamValue::Int((c2.round() as i64).clamp(d.low, d.high));
+                }
+            }
+            (ParamValue::Categorical(_), ParamValue::Categorical(_), _) => {
+                if rng_util::f64_range(rng, 0.0, 1.0) < 0.5 {
+                    core::mem::swap(&mut child1[i], &mut child2[i]);
+                }
+            }
+            _ => {}
+        }
+    }
+
+    (child1, child2)
+}
+
+/// SBX crossover for a single float dimension.
+pub(crate) fn sbx_crossover_f64(
+    rng: &mut fastrand::Rng,
+    p1: f64,
+    p2: f64,
+    low: f64,
+    high: f64,
+    eta: f64,
+) -> (f64, f64) {
+    let u: f64 = rng_util::f64_range(rng, 0.0, 1.0);
+
+    let beta = if u <= 0.5 {
+        (2.0 * u).powf(1.0 / (eta + 1.0))
+    } else {
+        (1.0 / (2.0 * (1.0 - u))).powf(1.0 / (eta + 1.0))
+    };
+
+    let c1 = 0.5 * ((1.0 + beta) * p1 + (1.0 - beta) * p2);
+    let c2 = 0.5 * ((1.0 - beta) * p1 + (1.0 + beta) * p2);
+
+    (c1.clamp(low, high), c2.clamp(low, high))
+}
+
+/// Polynomial mutation for each dimension.
+#[allow(clippy::cast_precision_loss)]
+pub(crate) fn mutate(
+    rng: &mut fastrand::Rng,
+    individual: &mut [ParamValue],
+    dimensions: &[DimensionInfo],
+    eta: f64,
+) {
+    let n = individual.len();
+    if n == 0 {
+        return;
+    }
+    let mutation_prob = 1.0 / n as f64;
+
+    for (i, value) in individual.iter_mut().enumerate() {
+        if rng_util::f64_range(rng, 0.0, 1.0) >= mutation_prob {
+            continue;
+        }
+
+        match (value, &dimensions[i].distribution) {
+            (v @ ParamValue::Float(_), Distribution::Float(d)) => {
+                let ParamValue::Float(x) = *v else {
+                    unreachable!();
+                };
+                let mutated = polynomial_mutation_f64(rng, x, d.low, d.high, eta);
+                *v = ParamValue::Float(mutated);
+            }
+            (v @ ParamValue::Int(_), Distribution::Int(d)) => {
+                let ParamValue::Int(x) = *v else {
+                    unreachable!();
+                };
+                #[allow(clippy::cast_possible_truncation)]
+                {
+                    let mutated =
+                        polynomial_mutation_f64(rng, x as f64, d.low as f64, d.high as f64, eta);
+                    *v = ParamValue::Int((mutated.round() as i64).clamp(d.low, d.high));
+                }
+            }
+            (v @ ParamValue::Categorical(_), Distribution::Categorical(d)) => {
+                *v = ParamValue::Categorical(rng.usize(0..d.n_choices));
+            }
+            _ => {}
+        }
+    }
+}
+
+/// Polynomial mutation for a single float value.
+pub(crate) fn polynomial_mutation_f64(
+    rng: &mut fastrand::Rng,
+    x: f64,
+    low: f64,
+    high: f64,
+    eta: f64,
+) -> f64 {
+    let u: f64 = rng_util::f64_range(rng, 0.0, 1.0);
+    let range = high - low;
+    if range <= 0.0 {
+        return x;
+    }
+
+    let delta1 = (x - low) / range;
+    let delta2 = (high - x) / range;
+
+    let delta_q = if u < 0.5 {
+        let xy = 1.0 - delta1;
+        let val = 2.0 * u + (1.0 - 2.0 * u) * xy.powf(eta + 1.0);
+        val.powf(1.0 / (eta + 1.0)) - 1.0
+    } else {
+        let xy = 1.0 - delta2;
+        let val = 2.0 * (1.0 - u) + 2.0 * (u - 0.5) * xy.powf(eta + 1.0);
+        1.0 - val.powf(1.0 / (eta + 1.0))
+    };
+
+    (x + delta_q * range).clamp(low, high)
+}
+
+/// Random sampling for a single distribution.
+#[allow(clippy::cast_possible_truncation, clippy::cast_precision_loss)]
+pub(crate) fn sample_random(rng: &mut fastrand::Rng, distribution: &Distribution) -> ParamValue {
+    match distribution {
+        Distribution::Float(d) => {
+            let value = if d.log_scale {
+                let log_low = d.low.ln();
+                let log_high = d.high.ln();
+                rng_util::f64_range(rng, log_low, log_high).exp()
+            } else if let Some(step) = d.step {
+                let n_steps = ((d.high - d.low) / step).floor() as i64;
+                let k = rng.i64(0..=n_steps);
+                d.low + (k as f64) * step
+            } else {
+                rng_util::f64_range(rng, d.low, d.high)
+            };
+            ParamValue::Float(value)
+        }
+        Distribution::Int(d) => {
+            let value = if d.log_scale {
+                let log_low = (d.low as f64).ln();
+                let log_high = (d.high as f64).ln();
+                let raw = rng_util::f64_range(rng, log_low, log_high).exp().round() as i64;
+                raw.clamp(d.low, d.high)
+            } else if let Some(step) = d.step {
+                let n_steps = (d.high - d.low) / step;
+                let k = rng.i64(0..=n_steps);
+                d.low + k * step
+            } else {
+                rng.i64(d.low..=d.high)
+            };
+            ParamValue::Int(value)
+        }
+        Distribution::Categorical(d) => ParamValue::Categorical(rng.usize(0..d.n_choices)),
+    }
+}
+
+// ---------------------------------------------------------------------------
+// Das-Dennis reference point generation
+// ---------------------------------------------------------------------------
+
+/// Generate Das-Dennis (simplex-lattice) reference points.
+///
+/// Returns `C(H + M - 1, M - 1)` uniformly spaced points on the
+/// `M`-dimensional unit simplex, where `M = n_objectives` and
+/// `H = divisions`.
+pub(crate) fn das_dennis(n_objectives: usize, divisions: usize) -> Vec<Vec<f64>> {
+    let mut points = Vec::new();
+    let mut point = vec![0.0_f64; n_objectives];
+    das_dennis_recursive(
+        n_objectives,
+        divisions,
+        0,
+        divisions,
+        &mut point,
+        &mut points,
+    );
+    points
+}
+
+#[allow(clippy::cast_precision_loss)]
+fn das_dennis_recursive(
+    n_objectives: usize,
+    divisions: usize,
+    depth: usize,
+    remaining: usize,
+    current: &mut Vec<f64>,
+    result: &mut Vec<Vec<f64>>,
+) {
+    if depth == n_objectives - 1 {
+        current[depth] = remaining as f64 / divisions as f64;
+        result.push(current.clone());
+        return;
+    }
+
+    for i in 0..=remaining {
+        current[depth] = i as f64 / divisions as f64;
+        das_dennis_recursive(
+            n_objectives,
+            divisions,
+            depth + 1,
+            remaining - i,
+            current,
+            result,
+        );
+    }
+}
+
+/// Choose the number of divisions for Das-Dennis to get close to a target
+/// population size.
+///
+/// The number of reference points is `C(H + M - 1, M - 1)`. This function
+/// finds the smallest `H` such that the number of points >= `target_pop`.
+pub(crate) fn auto_divisions(n_objectives: usize, target_pop: usize) -> usize {
+    let m = n_objectives;
+    for h in 1..200 {
+        let n_points = n_combinations(h + m - 1, m - 1);
+        if n_points >= target_pop {
+            return h;
+        }
+    }
+    12
+}
+
+/// Compute `C(n, k)` = n! / (k! * (n-k)!).
+fn n_combinations(n: usize, k: usize) -> usize {
+    if k > n {
+        return 0;
+    }
+    let k = k.min(n - k);
+    let mut result: usize = 1;
+    for i in 0..k {
+        result = result.saturating_mul(n - i) / (i + 1);
+    }
+    result
+}
+
+#[cfg(test)]
+mod tests {
+    use super::*;
+
+    #[test]
+    fn test_das_dennis_2d() {
+        let points = das_dennis(2, 4);
+        // C(4+1, 1) = 5 points
+        assert_eq!(points.len(), 5);
+        for p in &points {
+            let sum: f64 = p.iter().sum();
+            assert!((sum - 1.0).abs() < 1e-10, "point {p:?} doesn't sum to 1");
+        }
+    }
+
+    #[test]
+    fn test_das_dennis_3d() {
+        let points = das_dennis(3, 4);
+        // C(4+2, 2) = 15 points
+        assert_eq!(points.len(), 15);
+        for p in &points {
+            let sum: f64 = p.iter().sum();
+            assert!((sum - 1.0).abs() < 1e-10);
+        }
+    }
+
+    #[test]
+    fn test_auto_divisions() {
+        // For 2 objectives targeting 10 points: H=9 gives C(10,1)=10
+        let h = auto_divisions(2, 10);
+        let n = n_combinations(h + 1, 1);
+        assert!(n >= 10);
+
+        // For 3 objectives targeting ~91 points: H=12 gives C(14,2)=91
+        let h3 = auto_divisions(3, 91);
+        let n3 = n_combinations(h3 + 2, 2);
+        assert!(n3 >= 91);
+    }
+
+    #[test]
+    fn test_n_combinations() {
+        assert_eq!(n_combinations(5, 2), 10);
+        assert_eq!(n_combinations(4, 0), 1);
+        assert_eq!(n_combinations(4, 4), 1);
+        assert_eq!(n_combinations(6, 3), 20);
+    }
+}

src/sampler/gp.rs

@@ -0,0 +1,848 @@
+//! Gaussian Process (GP) sampler with Expected Improvement acquisition.
+//!
+//! A classical Bayesian optimization sampler that uses a Gaussian Process
+//! surrogate model with a Matérn 5/2 kernel and Expected Improvement (EI)
+//! acquisition function. Best suited for small, expensive evaluations in
+//! low-dimensional continuous spaces (d ≤ 20).
+//!
+//! Categorical parameters are sampled uniformly at random (not part of
+//! the GP model). If all parameters are categorical, the sampler falls
+//! back to pure random sampling.
+//!
+//! Requires the `gp` feature flag.
+//!
+//! # Examples
+//!
+//! ```
+//! use optimizer::sampler::gp::GpSampler;
+//! use optimizer::{Direction, Study};
+//!
+//! let sampler = GpSampler::with_seed(42);
+//! let study: Study<f64> = Study::with_sampler(Direction::Minimize, sampler);
+//! ```
+
+use std::collections::HashMap;
+
+use nalgebra::DMatrix;
+use parking_lot::Mutex;
+
+use crate::distribution::Distribution;
+use crate::param::ParamValue;
+use crate::rng_util;
+use crate::sampler::{CompletedTrial, Sampler};
+
+// ---------------------------------------------------------------------------
+// Public API
+// ---------------------------------------------------------------------------
+
+/// Gaussian Process sampler for Bayesian optimization.
+///
+/// Uses a GP surrogate with Matérn 5/2 kernel and Expected Improvement
+/// acquisition to guide sampling toward promising regions of the search
+/// space. Best suited for continuous (float/int) parameters in low
+/// dimensions (up to ~20).
+///
+/// # Examples
+///
+/// ```
+/// use optimizer::sampler::gp::GpSampler;
+/// use optimizer::{Direction, Study};
+///
+/// // Default configuration
+/// let study: Study<f64> = Study::with_sampler(Direction::Minimize, GpSampler::new());
+///
+/// // With seed for reproducibility
+/// let study: Study<f64> = Study::with_sampler(Direction::Minimize, GpSampler::with_seed(42));
+///
+/// // Custom configuration via builder
+/// let sampler = GpSampler::builder()
+///     .n_startup_trials(15)
+///     .n_candidates(2000)
+///     .noise_variance(1e-4)
+///     .seed(42)
+///     .build();
+/// let study: Study<f64> = Study::with_sampler(Direction::Minimize, sampler);
+/// ```
+pub struct GpSampler {
+    state: Mutex<GpState>,
+}
+
+impl GpSampler {
+    /// Creates a new GP sampler with a random seed.
+    #[must_use]
+    pub fn new() -> Self {
+        Self {
+            state: Mutex::new(GpState::new(None, None, None, None)),
+        }
+    }
+
+    /// Creates a new GP sampler with a fixed seed for reproducibility.
+    #[must_use]
+    pub fn with_seed(seed: u64) -> Self {
+        Self {
+            state: Mutex::new(GpState::new(None, None, None, Some(seed))),
+        }
+    }
+
+    /// Creates a builder for configuring a `GpSampler`.
+    #[must_use]
+    pub fn builder() -> GpSamplerBuilder {
+        GpSamplerBuilder::new()
+    }
+}
+
+impl Default for GpSampler {
+    fn default() -> Self {
+        Self::new()
+    }
+}
+
+/// Builder for configuring a [`GpSampler`].
+///
+/// All options have sensible defaults:
+/// - `n_startup_trials`: 10
+/// - `n_candidates`: 1000
+/// - `noise_variance`: 1e-6
+/// - `seed`: random
+///
+/// # Examples
+///
+/// ```
+/// use optimizer::sampler::gp::GpSamplerBuilder;
+///
+/// let sampler = GpSamplerBuilder::new()
+///     .n_startup_trials(15)
+///     .n_candidates(2000)
+///     .noise_variance(1e-4)
+///     .seed(42)
+///     .build();
+/// ```
+#[derive(Debug, Clone, Default)]
+pub struct GpSamplerBuilder {
+    n_startup_trials: Option<usize>,
+    n_candidates: Option<usize>,
+    noise_variance: Option<f64>,
+    seed: Option<u64>,
+}
+
+impl GpSamplerBuilder {
+    /// Creates a new builder with default settings.
+    #[must_use]
+    pub fn new() -> Self {
+        Self::default()
+    }
+
+    /// Sets the number of random trials before GP-guided sampling begins.
+    ///
+    /// Default: 10.
+    #[must_use]
+    pub fn n_startup_trials(mut self, n: usize) -> Self {
+        self.n_startup_trials = Some(n);
+        self
+    }
+
+    /// Sets the number of random candidate points for acquisition optimization.
+    ///
+    /// More candidates improve the quality of the acquisition maximum
+    /// at the cost of more GP predictions per trial.
+    ///
+    /// Default: 1000.
+    #[must_use]
+    pub fn n_candidates(mut self, n: usize) -> Self {
+        self.n_candidates = Some(n);
+        self
+    }
+
+    /// Sets the observation noise variance added to the kernel diagonal.
+    ///
+    /// Controls the assumed noise level. Larger values make the GP smoother.
+    ///
+    /// Default: 1e-6 (near-noiseless).
+    #[must_use]
+    pub fn noise_variance(mut self, v: f64) -> Self {
+        self.noise_variance = Some(v);
+        self
+    }
+
+    /// Sets the random seed for reproducibility.
+    #[must_use]
+    pub fn seed(mut self, seed: u64) -> Self {
+        self.seed = Some(seed);
+        self
+    }
+
+    /// Builds the configured [`GpSampler`].
+    #[must_use]
+    pub fn build(self) -> GpSampler {
+        GpSampler {
+            state: Mutex::new(GpState::new(
+                self.n_startup_trials,
+                self.n_candidates,
+                self.noise_variance,
+                self.seed,
+            )),
+        }
+    }
+}
+
+// ---------------------------------------------------------------------------
+// Internal types
+// ---------------------------------------------------------------------------
+
+/// Default number of random startup trials before GP kicks in.
+const DEFAULT_N_STARTUP: usize = 10;
+/// Default number of candidate points for EI optimization.
+const DEFAULT_N_CANDIDATES: usize = 1000;
+/// Default observation noise variance.
+const DEFAULT_NOISE_VAR: f64 = 1e-6;
+
+/// Describes how a parameter dimension maps into the GP internal vector.
+#[derive(Clone, Debug)]
+struct DimensionInfo {
+    distribution: Distribution,
+    is_continuous: bool,
+    bounds: Option<(f64, f64)>,
+}
+
+/// Tracks per-trial sampling progress.
+#[derive(Clone, Debug)]
+struct TrialProgress {
+    /// The candidate values for each dimension.
+    values: Vec<ParamValue>,
+    /// Next dimension to return.
+    next_dim: usize,
+}
+
+/// Phase of the GP state machine.
+enum GpPhase {
+    /// Discovering the search space (first trial).
+    Discovery,
+    /// Steady-state sampling.
+    Active,
+}
+
+/// A fitted GP model ready for predictions.
+struct GpModel {
+    /// Cholesky factor L of K + σ²I.
+    cholesky: nalgebra::linalg::Cholesky<f64, nalgebra::Dyn>,
+    /// α = (K + σ²I)^{-1} y.
+    alpha: nalgebra::DVector<f64>,
+    /// Training inputs (each row is a data point, normalized to [0, 1]).
+    x_train: Vec<Vec<f64>>,
+    /// ARD lengthscales per dimension.
+    lengthscales: Vec<f64>,
+    /// Signal variance.
+    signal_var: f64,
+    /// Mean of original y values (for un-standardization, unused but kept for diagnostics).
+    _y_mean: f64,
+    /// Std dev of original y values (unused but kept for diagnostics).
+    _y_std: f64,
+    /// Best observed (standardized) y.
+    f_best: f64,
+}
+
+/// Top-level mutable state behind the `Mutex`.
+struct GpState {
+    rng: fastrand::Rng,
+    n_startup_trials: usize,
+    n_candidates: usize,
+    noise_variance: f64,
+    phase: GpPhase,
+    dimensions: Vec<DimensionInfo>,
+    trial_progress: HashMap<u64, TrialProgress>,
+    discovery_trial_id: Option<u64>,
+}
+
+impl GpState {
+    fn new(
+        n_startup: Option<usize>,
+        n_candidates: Option<usize>,
+        noise_var: Option<f64>,
+        seed: Option<u64>,
+    ) -> Self {
+        let rng = seed.map_or_else(fastrand::Rng::new, fastrand::Rng::with_seed);
+        Self {
+            rng,
+            n_startup_trials: n_startup.unwrap_or(DEFAULT_N_STARTUP),
+            n_candidates: n_candidates.unwrap_or(DEFAULT_N_CANDIDATES),
+            noise_variance: noise_var.unwrap_or(DEFAULT_NOISE_VAR),
+            phase: GpPhase::Discovery,
+            dimensions: Vec::new(),
+            trial_progress: HashMap::new(),
+            discovery_trial_id: None,
+        }
+    }
+}
+
+// ---------------------------------------------------------------------------
+// Matérn 5/2 kernel
+// ---------------------------------------------------------------------------
+
+/// Matérn 5/2 kernel with ARD lengthscales.
+///
+/// `k(x1, x2) = σ² (1 + √5 r + 5/3 r²) exp(-√5 r)`
+/// where `r = sqrt(Σ ((x1_i - x2_i) / l_i)²)`
+fn matern52(x1: &[f64], x2: &[f64], lengthscales: &[f64], signal_var: f64) -> f64 {
+    let mut r_sq = 0.0;
+    for i in 0..x1.len() {
+        let diff = (x1[i] - x2[i]) / lengthscales[i];
+        r_sq += diff * diff;
+    }
+    let r = r_sq.sqrt();
+    let sqrt5_r = SQRT_5 * r;
+    signal_var * (1.0 + sqrt5_r + 5.0 / 3.0 * r_sq) * (-sqrt5_r).exp()
+}
+
+/// Build the kernel matrix `K + σ²I`.
+fn kernel_matrix(
+    x: &[Vec<f64>],
+    lengthscales: &[f64],
+    signal_var: f64,
+    noise_var: f64,
+) -> DMatrix<f64> {
+    let n = x.len();
+    DMatrix::from_fn(n, n, |i, j| {
+        let k = matern52(&x[i], &x[j], lengthscales, signal_var);
+        if i == j { k + noise_var } else { k }
+    })
+}
+
+/// Compute the kernel vector k(x*, X) for a test point.
+fn kernel_vector(
+    x_star: &[f64],
+    x_train: &[Vec<f64>],
+    lengthscales: &[f64],
+    signal_var: f64,
+) -> nalgebra::DVector<f64> {
+    nalgebra::DVector::from_fn(x_train.len(), |i, _| {
+        matern52(x_star, &x_train[i], lengthscales, signal_var)
+    })
+}
+
+/// Precomputed √5 constant.
+const SQRT_5: f64 = 2.236_213_562_373_095;
+
+// ---------------------------------------------------------------------------
+// GP fitting and prediction
+// ---------------------------------------------------------------------------
+
+/// Fit a GP model to the training data.
+///
+/// Returns `None` if fitting fails (e.g. Cholesky decomposition failure).
+#[allow(clippy::cast_precision_loss)]
+fn fit_gp(x_train: &[Vec<f64>], y_train: &[f64], noise_var: f64) -> Option<GpModel> {
+    let n = y_train.len();
+    if n == 0 {
+        return None;
+    }
+
+    // Standardize y
+    let y_mean = y_train.iter().sum::<f64>() / n as f64;
+    let y_var = if n > 1 {
+        y_train.iter().map(|&y| (y - y_mean).powi(2)).sum::<f64>() / (n - 1) as f64
+    } else {
+        1.0
+    };
+    let y_std = y_var.sqrt().max(1e-10);
+    let y_standardized: Vec<f64> = y_train.iter().map(|&y| (y - y_mean) / y_std).collect();
+
+    let f_best = y_standardized.iter().copied().fold(f64::INFINITY, f64::min);
+
+    // ARD lengthscales: per-dimension std dev of training X, clamped
+    let d = if x_train.is_empty() {
+        0
+    } else {
+        x_train[0].len()
+    };
+    let lengthscales: Vec<f64> = (0..d)
+        .map(|j| {
+            let vals: Vec<f64> = x_train.iter().map(|x| x[j]).collect();
+            let mean_j = vals.iter().sum::<f64>() / n as f64;
+            let var_j = vals.iter().map(|&v| (v - mean_j).powi(2)).sum::<f64>() / n as f64;
+            var_j.sqrt().max(0.01)
+        })
+        .collect();
+
+    // Signal variance = 1.0 (data is standardized)
+    let signal_var = 1.0;
+
+    let k = kernel_matrix(x_train, &lengthscales, signal_var, noise_var);
+    let cholesky = nalgebra::linalg::Cholesky::new(k)?;
+
+    // α = (K + σ²I)^{-1} y
+    let y_vec = nalgebra::DVector::from_column_slice(&y_standardized);
+    let alpha = cholesky.solve(&y_vec);
+
+    Some(GpModel {
+        cholesky,
+        alpha,
+        x_train: x_train.to_vec(),
+        lengthscales,
+        signal_var,
+        _y_mean: y_mean,
+        _y_std: y_std,
+        f_best,
+    })
+}
+
+/// Predict mean and standard deviation at a test point.
+fn predict(model: &GpModel, x: &[f64]) -> (f64, f64) {
+    let k_star = kernel_vector(x, &model.x_train, &model.lengthscales, model.signal_var);
+
+    // Mean: k*^T α
+    let mean = k_star.dot(&model.alpha);
+
+    // Variance: k(x*, x*) - k*^T (K + σ²I)^{-1} k*
+    let k_self = model.signal_var;
+    let v = model.cholesky.solve(&k_star);
+    let var = (k_self - k_star.dot(&v)).max(0.0);
+
+    (mean, var.sqrt())
+}
+
+// ---------------------------------------------------------------------------
+// Normal distribution helpers (Abramowitz-Stegun approximation)
+// ---------------------------------------------------------------------------
+
+/// Standard normal PDF.
+fn norm_pdf(x: f64) -> f64 {
+    const INV_SQRT_2PI: f64 = 0.398_942_280_401_432_7;
+    INV_SQRT_2PI * (-0.5 * x * x).exp()
+}
+
+/// Standard normal CDF (Abramowitz-Stegun rational approximation).
+fn norm_cdf(x: f64) -> f64 {
+    // Hart approximation (higher precision than basic A&S)
+    if x < -8.0 {
+        return 0.0;
+    }
+    if x > 8.0 {
+        return 1.0;
+    }
+
+    let abs_x = x.abs();
+    let t = 1.0 / (1.0 + 0.231_641_9 * abs_x);
+    let t2 = t * t;
+    let t3 = t2 * t;
+    let t4 = t3 * t;
+    let t5 = t4 * t;
+
+    let poly = 0.319_381_530 * t - 0.356_563_782 * t2 + 1.781_477_937 * t3 - 1.821_255_978 * t4
+        + 1.330_274_429 * t5;
+    let pdf = norm_pdf(abs_x);
+    let cdf = 1.0 - pdf * poly;
+
+    if x >= 0.0 { cdf } else { 1.0 - cdf }
+}
+
+// ---------------------------------------------------------------------------
+// Expected Improvement
+// ---------------------------------------------------------------------------
+
+/// Compute Expected Improvement at a point.
+///
+/// `EI(x) = (f_best - mean) Φ(z) + std φ(z)`
+/// where `z = (f_best - mean) / std`
+fn expected_improvement(mean: f64, std: f64, f_best: f64) -> f64 {
+    if std < 1e-12 {
+        return (f_best - mean).max(0.0);
+    }
+    let z = (f_best - mean) / std;
+    let improvement = (f_best - mean) * norm_cdf(z) + std * norm_pdf(z);
+    improvement.max(0.0)
+}
+
+// ---------------------------------------------------------------------------
+// Acquisition optimization
+// ---------------------------------------------------------------------------
+
+/// Find the point in [0, 1]^d that maximizes EI via multi-start random search.
+fn optimize_acquisition(
+    model: &GpModel,
+    n_dims: usize,
+    n_candidates: usize,
+    rng: &mut fastrand::Rng,
+) -> Vec<f64> {
+    let mut best_ei = f64::NEG_INFINITY;
+    let mut best_x = vec![0.5; n_dims];
+
+    for _ in 0..n_candidates {
+        let x: Vec<f64> = (0..n_dims)
+            .map(|_| rng_util::f64_range(rng, 0.0, 1.0))
+            .collect();
+        let (mean, std) = predict(model, &x);
+        let ei = expected_improvement(mean, std, model.f_best);
+        if ei > best_ei {
+            best_ei = ei;
+            best_x = x;
+        }
+    }
+
+    best_x
+}
+
+// ---------------------------------------------------------------------------
+// Data preprocessing helpers
+// ---------------------------------------------------------------------------
+
+/// Compute internal-space bounds for a distribution.
+#[allow(clippy::cast_precision_loss)]
+fn internal_bounds(distribution: &Distribution) -> Option<(f64, f64)> {
+    match distribution {
+        Distribution::Float(d) => {
+            if d.log_scale {
+                Some((d.low.ln(), d.high.ln()))
+            } else {
+                Some((d.low, d.high))
+            }
+        }
+        Distribution::Int(d) => {
+            if d.log_scale {
+                Some(((d.low as f64).ln(), (d.high as f64).ln()))
+            } else {
+                Some((d.low as f64, d.high as f64))
+            }
+        }
+        Distribution::Categorical(_) => None,
+    }
+}
+
+/// Convert a value from internal space to a `ParamValue` in original space.
+#[allow(clippy::cast_possible_truncation, clippy::cast_precision_loss)]
+fn from_internal(value: f64, distribution: &Distribution) -> ParamValue {
+    match distribution {
+        Distribution::Float(d) => {
+            let v = if d.log_scale { value.exp() } else { value };
+            let v = if let Some(step) = d.step {
+                let k = ((v - d.low) / step).round();
+                d.low + k * step
+            } else {
+                v
+            };
+            ParamValue::Float(v.clamp(d.low, d.high))
+        }
+        Distribution::Int(d) => {
+            let v = if d.log_scale { value.exp() } else { value };
+            let v = if let Some(step) = d.step {
+                let k = ((v - d.low as f64) / step as f64).round() as i64;
+                d.low + k * step
+            } else {
+                v.round() as i64
+            };
+            ParamValue::Int(v.clamp(d.low, d.high))
+        }
+        Distribution::Categorical(_) => {
+            unreachable!("from_internal should not be called for categorical distributions")
+        }
+    }
+}
+
+/// Convert an internal-space value to normalized [0, 1] using bounds.
+fn to_normalized(value: f64, lo: f64, hi: f64) -> f64 {
+    if (hi - lo).abs() < 1e-15 {
+        0.5
+    } else {
+        (value - lo) / (hi - lo)
+    }
+}
+
+/// Convert a normalized [0, 1] value back to internal space.
+fn from_normalized(value: f64, lo: f64, hi: f64) -> f64 {
+    lo + value * (hi - lo)
+}
+
+/// Convert a `ParamValue` to its internal-space representation.
+#[allow(clippy::cast_precision_loss)]
+fn to_internal(value: &ParamValue, distribution: &Distribution) -> f64 {
+    match (value, distribution) {
+        (ParamValue::Float(v), Distribution::Float(d)) => {
+            if d.log_scale {
+                v.ln()
+            } else {
+                *v
+            }
+        }
+        (ParamValue::Int(v), Distribution::Int(d)) => {
+            if d.log_scale {
+                (*v as f64).ln()
+            } else {
+                *v as f64
+            }
+        }
+        _ => 0.0,
+    }
+}
+
+/// Sample a random value for any distribution.
+#[allow(clippy::cast_possible_truncation, clippy::cast_precision_loss)]
+fn sample_random(rng: &mut fastrand::Rng, distribution: &Distribution) -> ParamValue {
+    match distribution {
+        Distribution::Float(d) => {
+            let value = if d.log_scale {
+                let log_low = d.low.ln();
+                let log_high = d.high.ln();
+                rng_util::f64_range(rng, log_low, log_high).exp()
+            } else if let Some(step) = d.step {
+                let n_steps = ((d.high - d.low) / step).floor() as i64;
+                let k = rng.i64(0..=n_steps);
+                d.low + (k as f64) * step
+            } else {
+                rng_util::f64_range(rng, d.low, d.high)
+            };
+            ParamValue::Float(value)
+        }
+        Distribution::Int(d) => {
+            let value = if d.log_scale {
+                let log_low = (d.low as f64).ln();
+                let log_high = (d.high as f64).ln();
+                let raw = rng_util::f64_range(rng, log_low, log_high).exp().round() as i64;
+                raw.clamp(d.low, d.high)
+            } else if let Some(step) = d.step {
+                let n_steps = (d.high - d.low) / step;
+                let k = rng.i64(0..=n_steps);
+                d.low + k * step
+            } else {
+                rng.i64(d.low..=d.high)
+            };
+            ParamValue::Int(value)
+        }
+        Distribution::Categorical(d) => ParamValue::Categorical(rng.usize(0..d.n_choices)),
+    }
+}
+
+// ---------------------------------------------------------------------------
+// Extract training data from history
+// ---------------------------------------------------------------------------
+
+/// Maximum number of training points to use for the GP.
+/// Caps computational cost at O(`MAX_TRAIN_POINTS`^3) per trial.
+const MAX_TRAIN_POINTS: usize = 100;
+
+/// Establish a deterministic mapping from dimension index to `ParamId`
+/// using the first trial in history.
+///
+/// Matches dimensions to params by distribution equality, consuming
+/// matched params to correctly handle duplicate distributions.
+fn establish_param_mapping(
+    trial: &CompletedTrial,
+    dimensions: &[DimensionInfo],
+) -> Vec<Option<crate::parameter::ParamId>> {
+    use crate::parameter::ParamId;
+
+    let mut available: Vec<(ParamId, &Distribution)> =
+        trial.distributions.iter().map(|(id, d)| (*id, d)).collect();
+    // Sort for deterministic matching order
+    available.sort_by_key(|(id, _)| *id);
+
+    let mut mapping = Vec::with_capacity(dimensions.len());
+    for dim in dimensions {
+        let pos = available.iter().position(|(_, d)| **d == dim.distribution);
+        if let Some(pos) = pos {
+            mapping.push(Some(available.remove(pos).0));
+        } else {
+            mapping.push(None);
+        }
+    }
+    mapping
+}
+
+/// Build normalized training data from completed trials.
+///
+/// Returns `(x_train, y_train)` where x values are normalized to [0, 1]
+/// per dimension using the bounds from `dimensions`. Only continuous
+/// dimensions are included. Uses at most [`MAX_TRAIN_POINTS`] most recent
+/// trials.
+#[allow(clippy::cast_precision_loss)]
+fn build_training_data(
+    history: &[CompletedTrial],
+    dimensions: &[DimensionInfo],
+) -> (Vec<Vec<f64>>, Vec<f64>) {
+    if history.is_empty() {
+        return (Vec::new(), Vec::new());
+    }
+
+    // Use only the most recent trials to cap GP fitting cost
+    let start = history.len().saturating_sub(MAX_TRAIN_POINTS);
+    let recent = &history[start..];
+
+    // Establish dimension → ParamId mapping from the first trial
+    let param_mapping = establish_param_mapping(&recent[0], dimensions);
+
+    let continuous_indices: Vec<usize> = dimensions
+        .iter()
+        .enumerate()
+        .filter(|(_, d)| d.is_continuous)
+        .map(|(i, _)| i)
+        .collect();
+
+    let mut x_train = Vec::with_capacity(recent.len());
+    let mut y_train = Vec::with_capacity(recent.len());
+
+    for trial in recent {
+        let mut x_row = Vec::with_capacity(continuous_indices.len());
+        let mut valid = true;
+
+        for &dim_idx in &continuous_indices {
+            let dim_info = &dimensions[dim_idx];
+            if let Some(param_id) = param_mapping[dim_idx] {
+                if let Some(param_val) = trial.params.get(&param_id) {
+                    let internal = to_internal(param_val, &dim_info.distribution);
+                    let (lo, hi) = dim_info.bounds.unwrap_or((0.0, 1.0));
+                    x_row.push(to_normalized(internal, lo, hi));
+                } else {
+                    valid = false;
+                    break;
+                }
+            } else {
+                valid = false;
+                break;
+            }
+        }
+
+        if valid && x_row.len() == continuous_indices.len() {
+            x_train.push(x_row);
+            y_train.push(trial.value);
+        }
+    }
+
+    (x_train, y_train)
+}
+
+// ---------------------------------------------------------------------------
+// Sampler trait implementation
+// ---------------------------------------------------------------------------
+
+impl Sampler for GpSampler {
+    #[allow(clippy::cast_precision_loss, clippy::cast_possible_truncation)]
+    fn sample(
+        &self,
+        distribution: &Distribution,
+        trial_id: u64,
+        history: &[CompletedTrial],
+    ) -> ParamValue {
+        let mut state = self.state.lock();
+
+        match &state.phase {
+            GpPhase::Discovery => sample_discovery(&mut state, distribution, trial_id),
+            GpPhase::Active => sample_active(&mut state, distribution, trial_id, history),
+        }
+    }
+}
+
+/// Handle sampling during the discovery phase.
+fn sample_discovery(state: &mut GpState, distribution: &Distribution, trial_id: u64) -> ParamValue {
+    // A new trial_id means discovery is done
+    if let Some(prev_id) = state.discovery_trial_id
+        && trial_id != prev_id
+    {
+        finalize_discovery(state);
+        return sample_active(state, distribution, trial_id, &[]);
+    }
+
+    state.discovery_trial_id = Some(trial_id);
+
+    let is_continuous = !matches!(distribution, Distribution::Categorical(_));
+    let bounds = internal_bounds(distribution);
+    state.dimensions.push(DimensionInfo {
+        distribution: distribution.clone(),
+        is_continuous,
+        bounds,
+    });
+
+    sample_random(&mut state.rng, distribution)
+}
+
+/// Finalize discovery and transition to the active phase.
+fn finalize_discovery(state: &mut GpState) {
+    state.phase = GpPhase::Active;
+    state.trial_progress.clear();
+}
+
+/// Handle sampling during the active phase.
+fn sample_active(
+    state: &mut GpState,
+    distribution: &Distribution,
+    trial_id: u64,
+    history: &[CompletedTrial],
+) -> ParamValue {
+    // If this trial already has progress, return the next pre-computed value
+    if let Some(progress) = state.trial_progress.get_mut(&trial_id) {
+        let dim_idx = progress.next_dim;
+        progress.next_dim += 1;
+        if dim_idx < progress.values.len() {
+            return progress.values[dim_idx].clone();
+        }
+        // Extra dimension not seen during discovery
+        return sample_random(&mut state.rng, distribution);
+    }
+
+    // New trial: compute all dimension values at once
+    let n_continuous = state.dimensions.iter().filter(|d| d.is_continuous).count();
+    let use_gp = n_continuous > 0 && history.len() >= state.n_startup_trials;
+
+    let values = if use_gp {
+        compute_gp_candidate(state, history)
+    } else {
+        compute_random_candidate(state)
+    };
+
+    let first_value = values
+        .first()
+        .cloned()
+        .unwrap_or_else(|| sample_random(&mut state.rng, distribution));
+
+    state.trial_progress.insert(
+        trial_id,
+        TrialProgress {
+            values,
+            next_dim: 1,
+        },
+    );
+
+    first_value
+}
+
+/// Compute a candidate using the GP model.
+fn compute_gp_candidate(state: &mut GpState, history: &[CompletedTrial]) -> Vec<ParamValue> {
+    let (x_train, y_train) = build_training_data(history, &state.dimensions);
+
+    // Try to fit GP; fall back to random if it fails
+    let model = fit_gp(&x_train, &y_train, state.noise_variance);
+
+    let n_continuous = state.dimensions.iter().filter(|d| d.is_continuous).count();
+
+    let normalized_candidate = if let Some(ref model) = model {
+        optimize_acquisition(model, n_continuous, state.n_candidates, &mut state.rng)
+    } else {
+        // GP fitting failed; use random
+        (0..n_continuous)
+            .map(|_| rng_util::f64_range(&mut state.rng, 0.0, 1.0))
+            .collect()
+    };
+
+    // Convert normalized candidate back to parameter values
+    let mut values = Vec::with_capacity(state.dimensions.len());
+    let mut ci = 0; // continuous dimension index
+
+    for dim in &state.dimensions {
+        if dim.is_continuous {
+            let (lo, hi) = dim.bounds.unwrap_or((0.0, 1.0));
+            let internal_val = from_normalized(normalized_candidate[ci], lo, hi);
+            values.push(from_internal(internal_val, &dim.distribution));
+            ci += 1;
+        } else {
+            values.push(sample_random(&mut state.rng, &dim.distribution));
+        }
+    }
+
+    values
+}
+
+/// Compute a random candidate for all dimensions.
+fn compute_random_candidate(state: &mut GpState) -> Vec<ParamValue> {
+    state
+        .dimensions
+        .iter()
+        .map(|dim| sample_random(&mut state.rng, &dim.distribution))
+        .collect()
+}

src/sampler/mod.rs

@@ -1,7 +1,20 @@
 //! Sampler trait and implementations for parameter sampling.
 
+pub mod bohb;
+#[cfg(feature = "cma-es")]
+pub mod cma_es;
+pub mod differential_evolution;
+pub(crate) mod genetic;
+#[cfg(feature = "gp")]
+pub mod gp;
 pub mod grid;
+pub mod moead;
+pub mod motpe;
+pub mod nsga2;
+pub mod nsga3;
 pub mod random;
+#[cfg(feature = "sobol")]
+pub mod sobol;
 pub mod tpe;
 
 use std::collections::HashMap;
@@ -9,6 +22,8 @@ use std::collections::HashMap;
 use crate::distribution::Distribution;
 use crate::param::ParamValue;
 use crate::parameter::{ParamId, Parameter};
+use crate::trial::AttrValue;
+use crate::types::TrialState;
 
 /// A completed trial with its parameters, distributions, and objective value.
 ///
@@ -16,6 +31,7 @@ use crate::parameter::{ParamId, Parameter};
 /// parameter values, their distributions, and the objective value returned
 /// by the objective function.
 #[derive(Clone, Debug)]
+#[cfg_attr(feature = "serde", derive(serde::Serialize, serde::Deserialize))]
 pub struct CompletedTrial<V = f64> {
     /// The unique identifier for this trial.
     pub id: u64,
@@ -27,6 +43,15 @@ pub struct CompletedTrial<V = f64> {
     pub param_labels: HashMap<ParamId, String>,
     /// The objective value returned by the objective function.
     pub value: V,
+    /// Intermediate objective values reported during the trial.
+    pub intermediate_values: Vec<(u64, f64)>,
+    /// The state of the trial (Complete, Pruned, or Failed).
+    pub state: TrialState,
+    /// User-defined attributes stored during the trial.
+    pub user_attrs: HashMap<String, AttrValue>,
+    /// Constraint values for this trial (<=0.0 means feasible).
+    #[cfg_attr(feature = "serde", serde(default))]
+    pub constraints: Vec<f64>,
 }
 
 impl<V> CompletedTrial<V> {
@@ -44,6 +69,33 @@ impl<V> CompletedTrial<V> {
             distributions,
             param_labels,
             value,
+            intermediate_values: Vec::new(),
+            state: TrialState::Complete,
+            user_attrs: HashMap::new(),
+            constraints: Vec::new(),
+        }
+    }
+
+    /// Creates a new completed trial with intermediate values and user attributes.
+    pub fn with_intermediate_values(
+        id: u64,
+        params: HashMap<ParamId, ParamValue>,
+        distributions: HashMap<ParamId, Distribution>,
+        param_labels: HashMap<ParamId, String>,
+        value: V,
+        intermediate_values: Vec<(u64, f64)>,
+        user_attrs: HashMap<String, AttrValue>,
+    ) -> Self {
+        Self {
+            id,
+            params,
+            distributions,
+            param_labels,
+            value,
+            intermediate_values,
+            state: TrialState::Complete,
+            user_attrs,
+            constraints: Vec::new(),
         }
     }
 
@@ -87,6 +139,26 @@ impl<V> CompletedTrial<V> {
                 .expect("parameter type mismatch: stored value incompatible with parameter")
         })
     }
+
+    /// Returns `true` if all constraints are satisfied (values <= 0.0).
+    ///
+    /// A trial with no constraints is considered feasible.
+    #[must_use]
+    pub fn is_feasible(&self) -> bool {
+        self.constraints.iter().all(|&c| c <= 0.0)
+    }
+
+    /// Gets a user attribute by key.
+    #[must_use]
+    pub fn user_attr(&self, key: &str) -> Option<&AttrValue> {
+        self.user_attrs.get(key)
+    }
+
+    /// Returns all user attributes.
+    #[must_use]
+    pub fn user_attrs(&self) -> &HashMap<String, AttrValue> {
+        &self.user_attrs
+    }
 }
 
 /// A pending (running) trial with its parameters and distributions, but no objective value yet.

src/sampler/moead.rs

@@ -0,0 +1,603 @@
+//! MOEA/D (Multi-Objective Evolutionary Algorithm based on Decomposition) sampler.
+//!
+//! Decomposes a multi-objective problem into scalar subproblems using
+//! weight vectors and solves them collaboratively. Supports Weighted Sum,
+//! Tchebycheff, and Penalty-based Boundary Intersection (PBI) scalarization.
+//!
+//! # Examples
+//!
+//! ```
+//! use optimizer::Direction;
+//! use optimizer::multi_objective::MultiObjectiveStudy;
+//! use optimizer::parameter::{FloatParam, Parameter};
+//! use optimizer::sampler::moead::MoeadSampler;
+//!
+//! let sampler = MoeadSampler::with_seed(42);
+//! let study =
+//!     MultiObjectiveStudy::with_sampler(vec![Direction::Minimize, Direction::Minimize], sampler);
+//!
+//! let x = FloatParam::new(0.0, 1.0);
+//! study
+//!     .optimize(100, |trial| {
+//!         let xv = x.suggest(trial)?;
+//!         Ok::<_, optimizer::Error>(vec![xv, 1.0 - xv])
+//!     })
+//!     .unwrap();
+//! ```
+
+use parking_lot::Mutex;
+
+use super::genetic::{
+    self, Candidate, EvolutionaryState, Phase, advance_generation, auto_divisions,
+    collect_evaluated_generation, crossover, das_dennis, extract_trial_params,
+    generate_random_candidates, mutate, sample_from_candidate, sample_random,
+};
+use crate::distribution::Distribution;
+use crate::multi_objective::MultiObjectiveTrial;
+use crate::param::ParamValue;
+use crate::types::Direction;
+
+/// Decomposition (scalarization) method for MOEA/D.
+#[derive(Debug, Clone, Default)]
+pub enum Decomposition {
+    /// Weighted sum: `sum(w_i * f_i)`.
+    WeightedSum,
+    /// Tchebycheff: `max(w_i * |f_i - z_i*|)`.
+    #[default]
+    Tchebycheff,
+    /// Penalty-based Boundary Intersection with parameter theta.
+    Pbi {
+        /// Penalty parameter controlling the balance between convergence
+        /// and diversity. Default: 5.0.
+        theta: f64,
+    },
+}
+
+/// MOEA/D sampler for multi-objective optimization.
+///
+/// Decomposes the multi-objective problem into scalar subproblems
+/// using weight vectors, solving them collaboratively via
+/// neighborhood-based mating and replacement.
+pub struct MoeadSampler {
+    state: Mutex<MoeadState>,
+}
+
+impl MoeadSampler {
+    /// Creates a new MOEA/D sampler with a random seed.
+    #[must_use]
+    pub fn new() -> Self {
+        Self {
+            state: Mutex::new(MoeadState::new(MoeadConfig::default(), None)),
+        }
+    }
+
+    /// Creates a new MOEA/D sampler with a fixed seed.
+    #[must_use]
+    pub fn with_seed(seed: u64) -> Self {
+        Self {
+            state: Mutex::new(MoeadState::new(MoeadConfig::default(), Some(seed))),
+        }
+    }
+
+    /// Creates a builder for configuring a `MoeadSampler`.
+    #[must_use]
+    pub fn builder() -> MoeadSamplerBuilder {
+        MoeadSamplerBuilder::default()
+    }
+}
+
+impl Default for MoeadSampler {
+    fn default() -> Self {
+        Self::new()
+    }
+}
+
+/// Builder for [`MoeadSampler`].
+#[derive(Debug, Clone, Default)]
+pub struct MoeadSamplerBuilder {
+    population_size: Option<usize>,
+    neighborhood_size: Option<usize>,
+    decomposition: Decomposition,
+    crossover_prob: Option<f64>,
+    crossover_eta: Option<f64>,
+    mutation_eta: Option<f64>,
+    seed: Option<u64>,
+}
+
+impl MoeadSamplerBuilder {
+    /// Sets the population size. If unset, equals the number of
+    /// Das-Dennis weight vectors.
+    #[must_use]
+    pub fn population_size(mut self, size: usize) -> Self {
+        self.population_size = Some(size);
+        self
+    }
+
+    /// Sets the neighborhood size (T). Default: `min(20, pop_size)`.
+    #[must_use]
+    pub fn neighborhood_size(mut self, size: usize) -> Self {
+        self.neighborhood_size = Some(size);
+        self
+    }
+
+    /// Sets the decomposition method. Default: Tchebycheff.
+    #[must_use]
+    pub fn decomposition(mut self, decomp: Decomposition) -> Self {
+        self.decomposition = decomp;
+        self
+    }
+
+    /// Sets the crossover probability. Default: 1.0.
+    #[must_use]
+    pub fn crossover_prob(mut self, prob: f64) -> Self {
+        self.crossover_prob = Some(prob);
+        self
+    }
+
+    /// Sets the SBX distribution index. Default: 20.0.
+    #[must_use]
+    pub fn crossover_eta(mut self, eta: f64) -> Self {
+        self.crossover_eta = Some(eta);
+        self
+    }
+
+    /// Sets the polynomial mutation distribution index. Default: 20.0.
+    #[must_use]
+    pub fn mutation_eta(mut self, eta: f64) -> Self {
+        self.mutation_eta = Some(eta);
+        self
+    }
+
+    /// Sets the random seed for reproducibility.
+    #[must_use]
+    pub fn seed(mut self, seed: u64) -> Self {
+        self.seed = Some(seed);
+        self
+    }
+
+    /// Builds the configured [`MoeadSampler`].
+    #[must_use]
+    pub fn build(self) -> MoeadSampler {
+        let config = MoeadConfig {
+            user_population_size: self.population_size,
+            neighborhood_size: self.neighborhood_size,
+            decomposition: self.decomposition,
+            crossover_prob: self.crossover_prob.unwrap_or(1.0),
+            crossover_eta: self.crossover_eta.unwrap_or(20.0),
+            mutation_eta: self.mutation_eta.unwrap_or(20.0),
+        };
+        MoeadSampler {
+            state: Mutex::new(MoeadState::new(config, self.seed)),
+        }
+    }
+}
+
+// ---------------------------------------------------------------------------
+// Internal types
+// ---------------------------------------------------------------------------
+
+#[derive(Debug, Clone)]
+struct MoeadConfig {
+    user_population_size: Option<usize>,
+    neighborhood_size: Option<usize>,
+    decomposition: Decomposition,
+    crossover_prob: f64,
+    crossover_eta: f64,
+    mutation_eta: f64,
+}
+
+impl Default for MoeadConfig {
+    fn default() -> Self {
+        Self {
+            user_population_size: None,
+            neighborhood_size: None,
+            decomposition: Decomposition::default(),
+            crossover_prob: 1.0,
+            crossover_eta: 20.0,
+            mutation_eta: 20.0,
+        }
+    }
+}
+
+struct MoeadState {
+    evo: EvolutionaryState,
+    config: MoeadConfig,
+    /// Weight vectors (Das-Dennis), one per subproblem.
+    weight_vectors: Vec<Vec<f64>>,
+    /// Neighborhoods: for each subproblem, indices of T nearest weight vectors.
+    neighborhoods: Vec<Vec<usize>>,
+    /// Ideal point z* (best per-objective in minimize-space).
+    ideal_point: Vec<f64>,
+    /// Current population's objective values in minimize-space (one per subproblem).
+    population_values: Vec<Vec<f64>>,
+    /// Current population's parameter vectors (one per subproblem).
+    population_params: Vec<Vec<ParamValue>>,
+    /// Whether the MOEA/D state has been initialized.
+    initialized: bool,
+}
+
+impl MoeadState {
+    fn new(config: MoeadConfig, seed: Option<u64>) -> Self {
+        Self {
+            evo: EvolutionaryState::new(seed),
+            config,
+            weight_vectors: Vec::new(),
+            neighborhoods: Vec::new(),
+            ideal_point: Vec::new(),
+            population_values: Vec::new(),
+            population_params: Vec::new(),
+            initialized: false,
+        }
+    }
+}
+
+// ---------------------------------------------------------------------------
+// MultiObjectiveSampler implementation
+// ---------------------------------------------------------------------------
+
+impl crate::multi_objective::MultiObjectiveSampler for MoeadSampler {
+    fn sample(
+        &self,
+        distribution: &Distribution,
+        trial_id: u64,
+        history: &[MultiObjectiveTrial],
+        directions: &[Direction],
+    ) -> ParamValue {
+        let mut state = self.state.lock();
+
+        match &state.evo.phase {
+            Phase::Discovery => {
+                if let Some(value) =
+                    genetic::sample_discovery(&mut state.evo, distribution, trial_id)
+                {
+                    return value;
+                }
+                // Transitioned to active phase
+                initialize_moead(&mut state, directions);
+                generate_random_candidates(&mut state.evo);
+                sample_from_candidate(&mut state.evo, trial_id)
+            }
+            Phase::Active => {
+                maybe_generate_new_generation(&mut state, history, directions);
+                sample_from_candidate(&mut state.evo, trial_id)
+            }
+        }
+    }
+}
+
+/// Initialize MOEA/D: weight vectors, neighborhoods, ideal point.
+fn initialize_moead(state: &mut MoeadState, directions: &[Direction]) {
+    let n_obj = directions.len();
+
+    // Generate weight vectors
+    let divisions = auto_divisions(n_obj, state.config.user_population_size.unwrap_or(100));
+    state.weight_vectors = das_dennis(n_obj, divisions);
+
+    let pop_size = state
+        .config
+        .user_population_size
+        .unwrap_or(state.weight_vectors.len())
+        .max(4);
+
+    // Trim or pad weight vectors to match population size
+    state.weight_vectors.truncate(pop_size);
+    while state.weight_vectors.len() < pop_size {
+        // Duplicate random existing weight vectors
+        let idx = state.evo.rng.usize(0..state.weight_vectors.len());
+        let w = state.weight_vectors[idx].clone();
+        state.weight_vectors.push(w);
+    }
+
+    // Compute neighborhoods
+    let t = state
+        .config
+        .neighborhood_size
+        .unwrap_or_else(|| 20.min(pop_size));
+    let t = t.min(pop_size);
+    state.neighborhoods = compute_neighborhoods(&state.weight_vectors, t);
+
+    state.evo.population_size = pop_size;
+    state.evo.phase = Phase::Active;
+    state.ideal_point = vec![f64::INFINITY; n_obj];
+    state.initialized = true;
+}
+
+/// Compute T-nearest neighborhoods by Euclidean distance between weight vectors.
+fn compute_neighborhoods(weights: &[Vec<f64>], t: usize) -> Vec<Vec<usize>> {
+    let n = weights.len();
+    weights
+        .iter()
+        .map(|wi| {
+            let mut distances: Vec<(usize, f64)> = (0..n)
+                .map(|j| {
+                    let d: f64 = wi
+                        .iter()
+                        .zip(&weights[j])
+                        .map(|(&a, &b)| (a - b).powi(2))
+                        .sum::<f64>()
+                        .sqrt();
+                    (j, d)
+                })
+                .collect();
+            distances.sort_by(|a, b| a.1.partial_cmp(&b.1).unwrap_or(core::cmp::Ordering::Equal));
+            distances.into_iter().take(t).map(|(idx, _)| idx).collect()
+        })
+        .collect()
+}
+
+/// Convert values to minimize-space.
+fn to_minimize_space(values: &[f64], directions: &[Direction]) -> Vec<f64> {
+    values
+        .iter()
+        .zip(directions)
+        .map(|(&v, d)| match d {
+            Direction::Minimize => v,
+            Direction::Maximize => -v,
+        })
+        .collect()
+}
+
+fn maybe_generate_new_generation(
+    state: &mut MoeadState,
+    history: &[MultiObjectiveTrial],
+    directions: &[Direction],
+) {
+    if state.evo.candidates.is_empty() {
+        generate_random_candidates(&mut state.evo);
+        return;
+    }
+
+    if let Some(evaluated) = collect_evaluated_generation(&state.evo, history) {
+        let offspring = moead_generate_offspring(state, &evaluated, directions);
+        advance_generation(&mut state.evo, offspring);
+    }
+}
+
+// ---------------------------------------------------------------------------
+// Scalarization functions
+// ---------------------------------------------------------------------------
+
+/// Weighted sum scalarization: `sum(w_i * f_i)`.
+fn scalarize_weighted_sum(values: &[f64], weight: &[f64]) -> f64 {
+    values.iter().zip(weight).map(|(&v, &w)| w * v).sum()
+}
+
+/// Tchebycheff scalarization: `max(w_i * |f_i - z_i*|)`.
+fn scalarize_tchebycheff(values: &[f64], weight: &[f64], ideal: &[f64]) -> f64 {
+    values
+        .iter()
+        .zip(weight)
+        .zip(ideal)
+        .map(|((&v, &w), &z)| {
+            let w = if w < 1e-6 { 1e-6 } else { w };
+            w * (v - z).abs()
+        })
+        .fold(f64::NEG_INFINITY, f64::max)
+}
+
+/// PBI scalarization: `d1 + theta * d2`.
+///
+/// d1 = projection onto weight direction, d2 = perpendicular distance.
+fn scalarize_pbi(values: &[f64], weight: &[f64], ideal: &[f64], theta: f64) -> f64 {
+    let n = values.len();
+
+    // Direction from ideal to the point
+    let diff: Vec<f64> = values.iter().zip(ideal).map(|(&v, &z)| v - z).collect();
+
+    // Normalize weight vector
+    let w_norm: f64 = weight.iter().map(|&w| w * w).sum::<f64>().sqrt();
+    if w_norm < 1e-30 {
+        return f64::INFINITY;
+    }
+    let w_unit: Vec<f64> = weight.iter().map(|&w| w / w_norm).collect();
+
+    // d1 = projection of diff onto weight direction
+    let d1: f64 = diff.iter().zip(&w_unit).map(|(&d, &w)| d * w).sum();
+
+    // d2 = perpendicular distance
+    let d2_sq: f64 = (0..n)
+        .map(|i| {
+            let proj = d1 * w_unit[i];
+            (diff[i] - proj).powi(2)
+        })
+        .sum::<f64>();
+
+    d1 + theta * d2_sq.sqrt()
+}
+
+/// Evaluate scalarization for a given decomposition method.
+fn scalarize(values: &[f64], weight: &[f64], ideal: &[f64], decomposition: &Decomposition) -> f64 {
+    match decomposition {
+        Decomposition::WeightedSum => scalarize_weighted_sum(values, weight),
+        Decomposition::Tchebycheff => scalarize_tchebycheff(values, weight, ideal),
+        Decomposition::Pbi { theta } => scalarize_pbi(values, weight, ideal, *theta),
+    }
+}
+
+// ---------------------------------------------------------------------------
+// MOEA/D generation algorithm
+// ---------------------------------------------------------------------------
+
+fn moead_generate_offspring(
+    state: &mut MoeadState,
+    population: &[&MultiObjectiveTrial],
+    directions: &[Direction],
+) -> Vec<Candidate> {
+    let pop_size = state.evo.population_size;
+
+    if population.len() < 2 {
+        return (0..pop_size)
+            .map(|_| {
+                let params = state
+                    .evo
+                    .dimensions
+                    .iter()
+                    .map(|d| sample_random(&mut state.evo.rng, &d.distribution))
+                    .collect();
+                Candidate { params }
+            })
+            .collect();
+    }
+
+    // Extract current population parameters and objective values
+    let current_params: Vec<Vec<ParamValue>> = population
+        .iter()
+        .map(|t| extract_trial_params(t, &state.evo.dimensions, &mut state.evo.rng))
+        .collect();
+
+    let current_values: Vec<Vec<f64>> = population
+        .iter()
+        .map(|t| to_minimize_space(&t.values, directions))
+        .collect();
+
+    // Update ideal point
+    for vals in &current_values {
+        for (i, &v) in vals.iter().enumerate() {
+            if i < state.ideal_point.len() && v < state.ideal_point[i] {
+                state.ideal_point[i] = v;
+            }
+        }
+    }
+
+    // Assign each solution to its best subproblem via scalarization
+    // and select the best solution for each subproblem as its representative
+    let n_weights = state.weight_vectors.len();
+    let mut best_for_subproblem: Vec<usize> = Vec::with_capacity(n_weights);
+
+    for j in 0..n_weights {
+        let mut best_idx = 0;
+        let mut best_val = f64::INFINITY;
+        for (k, vals) in current_values.iter().enumerate() {
+            let s = scalarize(
+                vals,
+                &state.weight_vectors[j],
+                &state.ideal_point,
+                &state.config.decomposition,
+            );
+            if s < best_val {
+                best_val = s;
+                best_idx = k;
+            }
+        }
+        best_for_subproblem.push(best_idx);
+    }
+
+    // Store current population state
+    state.population_values = current_values;
+    state.population_params = current_params;
+
+    // Generate offspring: for each subproblem, mate from neighborhood
+    let mut offspring = Vec::with_capacity(pop_size);
+
+    for i in 0..pop_size.min(state.neighborhoods.len()) {
+        let neighborhood = &state.neighborhoods[i];
+
+        // Pick two parents from the neighborhood using subproblem assignments
+        let n1 = neighborhood[state.evo.rng.usize(0..neighborhood.len())];
+        let n2 = neighborhood[state.evo.rng.usize(0..neighborhood.len())];
+
+        let p1_idx = best_for_subproblem[n1 % best_for_subproblem.len()];
+        let p2_idx = best_for_subproblem[n2 % best_for_subproblem.len()];
+
+        let p1 = &state.population_params[p1_idx];
+        let p2 = &state.population_params[p2_idx];
+
+        let (mut child1, _child2) = crossover(
+            &mut state.evo.rng,
+            p1,
+            p2,
+            &state.evo.dimensions,
+            state.config.crossover_prob,
+            state.config.crossover_eta,
+        );
+
+        mutate(
+            &mut state.evo.rng,
+            &mut child1,
+            &state.evo.dimensions,
+            state.config.mutation_eta,
+        );
+
+        offspring.push(Candidate { params: child1 });
+    }
+
+    // If pop_size > neighborhoods, fill remaining with random neighborhood crossover
+    while offspring.len() < pop_size {
+        let i = state.evo.rng.usize(0..state.neighborhoods.len());
+        let neighborhood = &state.neighborhoods[i];
+        let n1 = neighborhood[state.evo.rng.usize(0..neighborhood.len())];
+        let n2 = neighborhood[state.evo.rng.usize(0..neighborhood.len())];
+
+        let p1_idx = best_for_subproblem[n1 % best_for_subproblem.len()];
+        let p2_idx = best_for_subproblem[n2 % best_for_subproblem.len()];
+
+        let (mut child1, _) = crossover(
+            &mut state.evo.rng,
+            &state.population_params[p1_idx],
+            &state.population_params[p2_idx],
+            &state.evo.dimensions,
+            state.config.crossover_prob,
+            state.config.crossover_eta,
+        );
+
+        mutate(
+            &mut state.evo.rng,
+            &mut child1,
+            &state.evo.dimensions,
+            state.config.mutation_eta,
+        );
+
+        offspring.push(Candidate { params: child1 });
+    }
+
+    offspring
+}
+
+#[cfg(test)]
+mod tests {
+    use super::*;
+
+    #[test]
+    fn test_scalarize_weighted_sum() {
+        let values = [1.0, 2.0, 3.0];
+        let weight = [0.5, 0.3, 0.2];
+        let result = scalarize_weighted_sum(&values, &weight);
+        assert!((result - (0.5 + 0.6 + 0.6)).abs() < 1e-10);
+    }
+
+    #[test]
+    fn test_scalarize_tchebycheff() {
+        let values = [3.0, 2.0];
+        let weight = [0.5, 0.5];
+        let ideal = [1.0, 1.0];
+        let result = scalarize_tchebycheff(&values, &weight, &ideal);
+        // max(0.5 * |3-1|, 0.5 * |2-1|) = max(1.0, 0.5) = 1.0
+        assert!((result - 1.0).abs() < 1e-10);
+    }
+
+    #[test]
+    fn test_scalarize_pbi() {
+        let values = [2.0, 2.0];
+        let weight = [1.0, 1.0];
+        let ideal = [0.0, 0.0];
+        let result = scalarize_pbi(&values, &weight, &ideal, 5.0);
+        // d1 = projection of (2,2) onto (1/√2, 1/√2) = 2*√2
+        // d2 = 0 (point is on the weight direction)
+        let expected_d1 = 2.0 * (2.0_f64).sqrt();
+        assert!((result - expected_d1).abs() < 1e-10);
+    }
+
+    #[test]
+    fn test_compute_neighborhoods() {
+        let weights = vec![vec![1.0, 0.0], vec![0.5, 0.5], vec![0.0, 1.0]];
+        let neighborhoods = compute_neighborhoods(&weights, 2);
+        assert_eq!(neighborhoods.len(), 3);
+        // Each neighborhood should have 2 entries
+        for n in &neighborhoods {
+            assert_eq!(n.len(), 2);
+        }
+        // First weight [1,0] should be closest to itself and [0.5,0.5]
+        assert_eq!(neighborhoods[0][0], 0); // itself
+        assert_eq!(neighborhoods[0][1], 1); // nearest neighbor
+    }
+}

src/sampler/motpe.rs

@@ -0,0 +1,928 @@
+//! Multi-Objective Tree-Parzen Estimator (MOTPE) sampler.
+//!
+//! Extends TPE to handle multi-objective optimization by using Pareto
+//! non-dominated sorting to define "good" vs "bad" trial regions for
+//! the KDE models, replacing the single-objective gamma-based split.
+//!
+//! # Algorithm
+//!
+//! In single-objective TPE, trials are sorted by value and split at a
+//! gamma percentile into good/bad groups. MOTPE replaces this with:
+//!
+//! 1. Compute non-dominated sorting on all completed trials
+//! 2. Use the Pareto front (rank 0) as "good" trials
+//! 3. Use dominated trials as "bad" trials
+//! 4. Build KDE l(x) from good, g(x) from bad
+//! 5. Sample candidates and score by l(x)/g(x)
+//!
+//! # Examples
+//!
+//! ```
+//! use optimizer::Direction;
+//! use optimizer::multi_objective::MultiObjectiveStudy;
+//! use optimizer::parameter::{FloatParam, Parameter};
+//! use optimizer::sampler::motpe::MotpeSampler;
+//!
+//! let sampler = MotpeSampler::builder().seed(42).build();
+//! let study =
+//!     MultiObjectiveStudy::with_sampler(vec![Direction::Minimize, Direction::Minimize], sampler);
+//!
+//! let x = FloatParam::new(0.0, 1.0);
+//! study
+//!     .optimize(30, |trial| {
+//!         let xv = x.suggest(trial)?;
+//!         Ok::<_, optimizer::Error>(vec![xv, 1.0 - xv])
+//!     })
+//!     .unwrap();
+//!
+//! let front = study.pareto_front();
+//! assert!(!front.is_empty());
+//! ```
+
+use parking_lot::Mutex;
+
+use crate::distribution::Distribution;
+use crate::kde::KernelDensityEstimator;
+use crate::multi_objective::{MultiObjectiveSampler, MultiObjectiveTrial};
+use crate::param::ParamValue;
+use crate::types::{Direction, TrialState};
+use crate::{pareto, rng_util};
+
+/// Multi-Objective TPE (MOTPE) sampler for multi-objective Bayesian optimization.
+///
+/// Uses Pareto non-dominated sorting to split completed trials into
+/// "good" (non-dominated, rank 0) and "bad" (dominated) groups, then
+/// fits kernel density estimators to each group and samples new points
+/// that maximize l(x)/g(x).
+///
+/// During the startup phase (fewer than `n_startup_trials` completed),
+/// MOTPE falls back to random sampling.
+///
+/// # Examples
+///
+/// ```
+/// use optimizer::Direction;
+/// use optimizer::multi_objective::MultiObjectiveStudy;
+/// use optimizer::parameter::{FloatParam, Parameter};
+/// use optimizer::sampler::motpe::MotpeSampler;
+///
+/// let sampler = MotpeSampler::builder()
+///     .n_startup_trials(10)
+///     .n_ei_candidates(24)
+///     .seed(42)
+///     .build();
+///
+/// let study =
+///     MultiObjectiveStudy::with_sampler(vec![Direction::Minimize, Direction::Minimize], sampler);
+/// ```
+pub struct MotpeSampler {
+    /// Number of trials before MOTPE kicks in (uses random sampling before this).
+    n_startup_trials: usize,
+    /// Number of candidate samples to evaluate when selecting the next point.
+    n_ei_candidates: usize,
+    /// Optional fixed bandwidth for KDE. If None, uses Scott's rule.
+    kde_bandwidth: Option<f64>,
+    /// Thread-safe RNG for sampling.
+    rng: Mutex<fastrand::Rng>,
+}
+
+impl MotpeSampler {
+    /// Creates a new MOTPE sampler with default settings.
+    ///
+    /// Defaults:
+    /// - `n_startup_trials`: 11
+    /// - `n_ei_candidates`: 24
+    /// - `kde_bandwidth`: None (Scott's rule)
+    #[must_use]
+    pub fn new() -> Self {
+        Self {
+            n_startup_trials: 11,
+            n_ei_candidates: 24,
+            kde_bandwidth: None,
+            rng: Mutex::new(fastrand::Rng::new()),
+        }
+    }
+
+    /// Creates a new MOTPE sampler with a fixed seed.
+    #[must_use]
+    pub fn with_seed(seed: u64) -> Self {
+        Self {
+            n_startup_trials: 11,
+            n_ei_candidates: 24,
+            kde_bandwidth: None,
+            rng: Mutex::new(fastrand::Rng::with_seed(seed)),
+        }
+    }
+
+    /// Creates a builder for configuring a MOTPE sampler.
+    #[must_use]
+    pub fn builder() -> MotpeSamplerBuilder {
+        MotpeSamplerBuilder::new()
+    }
+
+    /// Splits trials into good (non-dominated) and bad (dominated) groups
+    /// using Pareto non-dominated sorting.
+    fn split_trials<'a>(
+        history: &'a [MultiObjectiveTrial],
+        directions: &[Direction],
+    ) -> (Vec<&'a MultiObjectiveTrial>, Vec<&'a MultiObjectiveTrial>) {
+        let complete: Vec<(usize, &MultiObjectiveTrial)> = history
+            .iter()
+            .enumerate()
+            .filter(|(_, t)| t.state == TrialState::Complete)
+            .collect();
+
+        if complete.is_empty() {
+            return (vec![], vec![]);
+        }
+
+        let values: Vec<Vec<f64>> = complete.iter().map(|(_, t)| t.values.clone()).collect();
+        let constraints: Vec<Vec<f64>> = complete
+            .iter()
+            .map(|(_, t)| t.constraints.clone())
+            .collect();
+        let has_constraints = constraints.iter().any(|c| !c.is_empty());
+
+        let fronts = if has_constraints {
+            pareto::fast_non_dominated_sort_constrained(&values, directions, &constraints)
+        } else {
+            pareto::fast_non_dominated_sort(&values, directions)
+        };
+
+        if fronts.is_empty() {
+            return (vec![], vec![]);
+        }
+
+        // Front 0 = good (non-dominated), everything else = bad
+        let good: Vec<&MultiObjectiveTrial> = fronts[0].iter().map(|&i| complete[i].1).collect();
+
+        let bad: Vec<&MultiObjectiveTrial> = fronts[1..]
+            .iter()
+            .flatten()
+            .map(|&i| complete[i].1)
+            .collect();
+
+        (good, bad)
+    }
+
+    /// Samples uniformly from a distribution (used during startup phase).
+    #[allow(
+        clippy::cast_possible_truncation,
+        clippy::cast_precision_loss,
+        clippy::unused_self
+    )]
+    fn sample_uniform(distribution: &Distribution, rng: &mut fastrand::Rng) -> ParamValue {
+        match distribution {
+            Distribution::Float(d) => {
+                let value = if d.log_scale {
+                    let log_low = d.low.ln();
+                    let log_high = d.high.ln();
+                    rng_util::f64_range(rng, log_low, log_high).exp()
+                } else if let Some(step) = d.step {
+                    let n_steps = ((d.high - d.low) / step).floor() as i64;
+                    let k = rng.i64(0..=n_steps);
+                    d.low + (k as f64) * step
+                } else {
+                    rng_util::f64_range(rng, d.low, d.high)
+                };
+                ParamValue::Float(value)
+            }
+            Distribution::Int(d) => {
+                let value = if d.log_scale {
+                    let log_low = (d.low as f64).ln();
+                    let log_high = (d.high as f64).ln();
+                    let raw = rng_util::f64_range(rng, log_low, log_high).exp().round() as i64;
+                    raw.clamp(d.low, d.high)
+                } else if let Some(step) = d.step {
+                    let n_steps = (d.high - d.low) / step;
+                    let k = rng.i64(0..=n_steps);
+                    d.low + k * step
+                } else {
+                    rng.i64(d.low..=d.high)
+                };
+                ParamValue::Int(value)
+            }
+            Distribution::Categorical(d) => ParamValue::Categorical(rng.usize(0..d.n_choices)),
+        }
+    }
+
+    /// Samples using TPE for float distributions.
+    #[allow(clippy::too_many_arguments)]
+    fn sample_tpe_float(
+        &self,
+        low: f64,
+        high: f64,
+        log_scale: bool,
+        step: Option<f64>,
+        good_values: Vec<f64>,
+        bad_values: Vec<f64>,
+        rng: &mut fastrand::Rng,
+    ) -> f64 {
+        // Transform to internal space (log space if needed)
+        let (internal_low, internal_high, good_internal, bad_internal) = if log_scale {
+            let i_low = low.ln();
+            let i_high = high.ln();
+            let g: Vec<f64> = good_values.iter().map(|&v| v.ln()).collect();
+            let b: Vec<f64> = bad_values.iter().map(|&v| v.ln()).collect();
+            (i_low, i_high, g, b)
+        } else {
+            (low, high, good_values, bad_values)
+        };
+
+        // Fit KDEs to good and bad groups
+        let l_kde = match self.kde_bandwidth {
+            Some(bw) => KernelDensityEstimator::with_bandwidth(good_internal, bw),
+            None => KernelDensityEstimator::new(good_internal),
+        };
+        let g_kde = match self.kde_bandwidth {
+            Some(bw) => KernelDensityEstimator::with_bandwidth(bad_internal, bw),
+            None => KernelDensityEstimator::new(bad_internal),
+        };
+
+        // If KDE construction fails, fall back to uniform sampling
+        let (Ok(l_kde), Ok(g_kde)) = (l_kde, g_kde) else {
+            return rng_util::f64_range(rng, low, high);
+        };
+
+        // Generate candidates from l(x) and select the one with best l(x)/g(x)
+        let mut best_candidate = internal_low;
+        let mut best_ratio = f64::NEG_INFINITY;
+
+        for _ in 0..self.n_ei_candidates {
+            let candidate = l_kde.sample(rng).clamp(internal_low, internal_high);
+
+            let l_density = l_kde.pdf(candidate);
+            let g_density = g_kde.pdf(candidate);
+
+            let ratio = if g_density < f64::EPSILON {
+                if l_density > f64::EPSILON {
+                    f64::INFINITY
+                } else {
+                    0.0
+                }
+            } else {
+                l_density / g_density
+            };
+
+            if ratio > best_ratio {
+                best_ratio = ratio;
+                best_candidate = candidate;
+            }
+        }
+
+        // Transform back from internal space
+        let mut value = if log_scale {
+            best_candidate.exp()
+        } else {
+            best_candidate
+        };
+
+        // Apply step constraint if present
+        if let Some(step) = step {
+            let k = ((value - low) / step).round();
+            value = low + k * step;
+        }
+
+        value.clamp(low, high)
+    }
+
+    /// Samples using TPE for integer distributions.
+    #[allow(
+        clippy::too_many_arguments,
+        clippy::cast_precision_loss,
+        clippy::cast_possible_truncation
+    )]
+    fn sample_tpe_int(
+        &self,
+        low: i64,
+        high: i64,
+        log_scale: bool,
+        step: Option<i64>,
+        good_values: &[i64],
+        bad_values: &[i64],
+        rng: &mut fastrand::Rng,
+    ) -> i64 {
+        let good_floats: Vec<f64> = good_values.iter().map(|&v| v as f64).collect();
+        let bad_floats: Vec<f64> = bad_values.iter().map(|&v| v as f64).collect();
+
+        let float_value = self.sample_tpe_float(
+            low as f64,
+            high as f64,
+            log_scale,
+            step.map(|s| s as f64),
+            good_floats,
+            bad_floats,
+            rng,
+        );
+
+        let int_value = float_value.round() as i64;
+        let int_value = if let Some(step) = step {
+            let k = ((int_value - low) as f64 / step as f64).round() as i64;
+            low + k * step
+        } else {
+            int_value
+        };
+
+        int_value.clamp(low, high)
+    }
+
+    /// Samples using TPE for categorical distributions.
+    #[allow(clippy::cast_precision_loss)]
+    fn sample_tpe_categorical(
+        n_choices: usize,
+        good_indices: &[usize],
+        bad_indices: &[usize],
+        rng: &mut fastrand::Rng,
+    ) -> usize {
+        let mut good_counts = vec![0usize; n_choices];
+        let mut bad_counts = vec![0usize; n_choices];
+
+        for &idx in good_indices {
+            if idx < n_choices {
+                good_counts[idx] += 1;
+            }
+        }
+        for &idx in bad_indices {
+            if idx < n_choices {
+                bad_counts[idx] += 1;
+            }
+        }
+
+        // Laplace smoothing
+        let good_total = good_indices.len() as f64 + n_choices as f64;
+        let bad_total = bad_indices.len() as f64 + n_choices as f64;
+
+        let mut weights = vec![0.0f64; n_choices];
+        for i in 0..n_choices {
+            let l_prob = (good_counts[i] as f64 + 1.0) / good_total;
+            let g_prob = (bad_counts[i] as f64 + 1.0) / bad_total;
+            weights[i] = l_prob / g_prob;
+        }
+
+        // Sample proportionally to weights
+        let total_weight: f64 = weights.iter().sum();
+        let threshold = rng.f64() * total_weight;
+
+        let mut cumulative = 0.0;
+        for (i, &w) in weights.iter().enumerate() {
+            cumulative += w;
+            if cumulative >= threshold {
+                return i;
+            }
+        }
+
+        n_choices - 1
+    }
+}
+
+impl Default for MotpeSampler {
+    fn default() -> Self {
+        Self::new()
+    }
+}
+
+impl MultiObjectiveSampler for MotpeSampler {
+    #[allow(clippy::too_many_lines)]
+    fn sample(
+        &self,
+        distribution: &Distribution,
+        _trial_id: u64,
+        history: &[MultiObjectiveTrial],
+        directions: &[Direction],
+    ) -> ParamValue {
+        let mut rng = self.rng.lock();
+
+        // Fall back to random sampling during startup phase
+        let n_complete = history
+            .iter()
+            .filter(|t| t.state == TrialState::Complete)
+            .count();
+        if n_complete < self.n_startup_trials {
+            return Self::sample_uniform(distribution, &mut rng);
+        }
+
+        // Split trials into good (Pareto front) and bad (dominated)
+        let (good_trials, bad_trials) = Self::split_trials(history, directions);
+
+        if good_trials.is_empty() || bad_trials.is_empty() {
+            return Self::sample_uniform(distribution, &mut rng);
+        }
+
+        match distribution {
+            Distribution::Float(d) => {
+                let good_values: Vec<f64> = good_trials
+                    .iter()
+                    .flat_map(|t| t.params.values())
+                    .filter_map(|v| match v {
+                        ParamValue::Float(f) => Some(*f),
+                        _ => None,
+                    })
+                    .filter(|&v| v >= d.low && v <= d.high)
+                    .collect();
+
+                let bad_values: Vec<f64> = bad_trials
+                    .iter()
+                    .flat_map(|t| t.params.values())
+                    .filter_map(|v| match v {
+                        ParamValue::Float(f) => Some(*f),
+                        _ => None,
+                    })
+                    .filter(|&v| v >= d.low && v <= d.high)
+                    .collect();
+
+                if good_values.is_empty() || bad_values.is_empty() {
+                    return Self::sample_uniform(distribution, &mut rng);
+                }
+
+                let value = self.sample_tpe_float(
+                    d.low,
+                    d.high,
+                    d.log_scale,
+                    d.step,
+                    good_values,
+                    bad_values,
+                    &mut rng,
+                );
+                ParamValue::Float(value)
+            }
+            Distribution::Int(d) => {
+                let good_values: Vec<i64> = good_trials
+                    .iter()
+                    .flat_map(|t| t.params.values())
+                    .filter_map(|v| match v {
+                        ParamValue::Int(i) => Some(*i),
+                        _ => None,
+                    })
+                    .filter(|&v| v >= d.low && v <= d.high)
+                    .collect();
+
+                let bad_values: Vec<i64> = bad_trials
+                    .iter()
+                    .flat_map(|t| t.params.values())
+                    .filter_map(|v| match v {
+                        ParamValue::Int(i) => Some(*i),
+                        _ => None,
+                    })
+                    .filter(|&v| v >= d.low && v <= d.high)
+                    .collect();
+
+                if good_values.is_empty() || bad_values.is_empty() {
+                    return Self::sample_uniform(distribution, &mut rng);
+                }
+
+                let value = self.sample_tpe_int(
+                    d.low,
+                    d.high,
+                    d.log_scale,
+                    d.step,
+                    &good_values,
+                    &bad_values,
+                    &mut rng,
+                );
+                ParamValue::Int(value)
+            }
+            Distribution::Categorical(d) => {
+                let good_indices: Vec<usize> = good_trials
+                    .iter()
+                    .flat_map(|t| t.params.values())
+                    .filter_map(|v| match v {
+                        ParamValue::Categorical(i) => Some(*i),
+                        _ => None,
+                    })
+                    .filter(|&i| i < d.n_choices)
+                    .collect();
+
+                let bad_indices: Vec<usize> = bad_trials
+                    .iter()
+                    .flat_map(|t| t.params.values())
+                    .filter_map(|v| match v {
+                        ParamValue::Categorical(i) => Some(*i),
+                        _ => None,
+                    })
+                    .filter(|&i| i < d.n_choices)
+                    .collect();
+
+                if good_indices.is_empty() || bad_indices.is_empty() {
+                    return Self::sample_uniform(distribution, &mut rng);
+                }
+
+                let index = Self::sample_tpe_categorical(
+                    d.n_choices,
+                    &good_indices,
+                    &bad_indices,
+                    &mut rng,
+                );
+                ParamValue::Categorical(index)
+            }
+        }
+    }
+}
+
+/// Builder for configuring a [`MotpeSampler`].
+///
+/// # Examples
+///
+/// ```
+/// use optimizer::sampler::motpe::MotpeSamplerBuilder;
+///
+/// let sampler = MotpeSamplerBuilder::new()
+///     .n_startup_trials(15)
+///     .n_ei_candidates(32)
+///     .seed(42)
+///     .build();
+/// ```
+#[derive(Debug, Clone)]
+pub struct MotpeSamplerBuilder {
+    n_startup_trials: usize,
+    n_ei_candidates: usize,
+    kde_bandwidth: Option<f64>,
+    seed: Option<u64>,
+}
+
+impl MotpeSamplerBuilder {
+    /// Creates a new builder with default settings.
+    #[must_use]
+    pub fn new() -> Self {
+        Self {
+            n_startup_trials: 11,
+            n_ei_candidates: 24,
+            kde_bandwidth: None,
+            seed: None,
+        }
+    }
+
+    /// Sets the number of startup trials before MOTPE sampling begins.
+    #[must_use]
+    pub fn n_startup_trials(mut self, n: usize) -> Self {
+        self.n_startup_trials = n;
+        self
+    }
+
+    /// Sets the number of EI candidates to evaluate per sample.
+    #[must_use]
+    pub fn n_ei_candidates(mut self, n: usize) -> Self {
+        self.n_ei_candidates = n;
+        self
+    }
+
+    /// Sets a fixed bandwidth for the kernel density estimator.
+    ///
+    /// By default, Scott's rule is used for automatic bandwidth selection.
+    #[must_use]
+    pub fn kde_bandwidth(mut self, bandwidth: f64) -> Self {
+        self.kde_bandwidth = Some(bandwidth);
+        self
+    }
+
+    /// Sets a seed for reproducible sampling.
+    #[must_use]
+    pub fn seed(mut self, seed: u64) -> Self {
+        self.seed = Some(seed);
+        self
+    }
+
+    /// Builds the configured [`MotpeSampler`].
+    #[must_use]
+    pub fn build(self) -> MotpeSampler {
+        let rng = match self.seed {
+            Some(s) => fastrand::Rng::with_seed(s),
+            None => fastrand::Rng::new(),
+        };
+
+        MotpeSampler {
+            n_startup_trials: self.n_startup_trials,
+            n_ei_candidates: self.n_ei_candidates,
+            kde_bandwidth: self.kde_bandwidth,
+            rng: Mutex::new(rng),
+        }
+    }
+}
+
+impl Default for MotpeSamplerBuilder {
+    fn default() -> Self {
+        Self::new()
+    }
+}
+
+#[cfg(test)]
+#[allow(
+    clippy::similar_names,
+    clippy::cast_sign_loss,
+    clippy::cast_precision_loss
+)]
+mod tests {
+    use std::collections::HashMap;
+
+    use super::*;
+    use crate::distribution::{CategoricalDistribution, FloatDistribution, IntDistribution};
+    use crate::parameter::ParamId;
+
+    fn create_mo_trial(
+        id: u64,
+        values: Vec<f64>,
+        params: Vec<(ParamId, ParamValue, Distribution)>,
+    ) -> MultiObjectiveTrial {
+        let mut param_map = HashMap::new();
+        let mut dist_map = HashMap::new();
+        let label_map = HashMap::new();
+        for (param_id, pv, dist) in params {
+            param_map.insert(param_id, pv);
+            dist_map.insert(param_id, dist);
+        }
+        MultiObjectiveTrial {
+            id,
+            params: param_map,
+            distributions: dist_map,
+            param_labels: label_map,
+            values,
+            state: TrialState::Complete,
+            user_attrs: HashMap::new(),
+            constraints: Vec::new(),
+        }
+    }
+
+    #[test]
+    fn test_motpe_startup_random_sampling() {
+        let sampler = MotpeSampler::with_seed(42);
+        let dist = Distribution::Float(FloatDistribution {
+            low: 0.0,
+            high: 1.0,
+            log_scale: false,
+            step: None,
+        });
+        let directions = [Direction::Minimize, Direction::Minimize];
+
+        // With no history, should use random sampling
+        let history: Vec<MultiObjectiveTrial> = vec![];
+        for _ in 0..50 {
+            let value = sampler.sample(&dist, 0, &history, &directions);
+            if let ParamValue::Float(v) = value {
+                assert!((0.0..=1.0).contains(&v));
+            } else {
+                panic!("Expected Float value");
+            }
+        }
+    }
+
+    #[test]
+    fn test_motpe_split_pareto() {
+        let dist = Distribution::Float(FloatDistribution {
+            low: 0.0,
+            high: 1.0,
+            log_scale: false,
+            step: None,
+        });
+        let directions = [Direction::Minimize, Direction::Minimize];
+        let x_id = ParamId::new();
+
+        // Create trials: Pareto front = {(0.1, 0.9), (0.5, 0.5), (0.9, 0.1)}
+        // Dominated = {(0.6, 0.8), (0.8, 0.7)}
+        let history = vec![
+            create_mo_trial(
+                0,
+                vec![0.1, 0.9],
+                vec![(x_id, ParamValue::Float(0.1), dist.clone())],
+            ),
+            create_mo_trial(
+                1,
+                vec![0.5, 0.5],
+                vec![(x_id, ParamValue::Float(0.5), dist.clone())],
+            ),
+            create_mo_trial(
+                2,
+                vec![0.9, 0.1],
+                vec![(x_id, ParamValue::Float(0.9), dist.clone())],
+            ),
+            create_mo_trial(
+                3,
+                vec![0.6, 0.8],
+                vec![(x_id, ParamValue::Float(0.6), dist.clone())],
+            ),
+            create_mo_trial(
+                4,
+                vec![0.8, 0.7],
+                vec![(x_id, ParamValue::Float(0.8), dist.clone())],
+            ),
+        ];
+
+        let (good, bad) = MotpeSampler::split_trials(&history, &directions);
+        assert_eq!(good.len(), 3, "Pareto front should have 3 members");
+        assert_eq!(bad.len(), 2, "2 dominated trials");
+    }
+
+    #[test]
+    fn test_motpe_samples_float() {
+        let sampler = MotpeSampler::builder()
+            .n_startup_trials(5)
+            .n_ei_candidates(24)
+            .seed(42)
+            .build();
+
+        let dist = Distribution::Float(FloatDistribution {
+            low: 0.0,
+            high: 1.0,
+            log_scale: false,
+            step: None,
+        });
+        let directions = [Direction::Minimize, Direction::Minimize];
+        let x_id = ParamId::new();
+
+        // Build history where values near 0.3 are on the Pareto front
+        let mut history = Vec::new();
+        for i in 0..20 {
+            let x = f64::from(i) / 20.0;
+            // Pareto front: f1 = (x - 0.3)^2, f2 = (x - 0.3)^2 + 0.1
+            // Best solutions cluster around x = 0.3
+            let f1 = (x - 0.3).powi(2);
+            let f2 = (x - 0.7).powi(2);
+            history.push(create_mo_trial(
+                i as u64,
+                vec![f1, f2],
+                vec![(x_id, ParamValue::Float(x), dist.clone())],
+            ));
+        }
+
+        // MOTPE should produce values within [0, 1]
+        for i in 0..50 {
+            let value = sampler.sample(&dist, 100 + i, &history, &directions);
+            if let ParamValue::Float(v) = value {
+                assert!((0.0..=1.0).contains(&v), "Value {v} out of range");
+            } else {
+                panic!("Expected Float value");
+            }
+        }
+    }
+
+    #[test]
+    fn test_motpe_int_sampling() {
+        let sampler = MotpeSampler::builder().n_startup_trials(5).seed(42).build();
+
+        let dist = Distribution::Int(IntDistribution {
+            low: 0,
+            high: 100,
+            log_scale: false,
+            step: None,
+        });
+        let directions = [Direction::Minimize, Direction::Minimize];
+        let x_id = ParamId::new();
+
+        let mut history = Vec::new();
+        for i in 0..20 {
+            let x = i * 5;
+            let f1 = ((x as f64) - 30.0).powi(2);
+            let f2 = ((x as f64) - 70.0).powi(2);
+            history.push(create_mo_trial(
+                i as u64,
+                vec![f1, f2],
+                vec![(x_id, ParamValue::Int(x), dist.clone())],
+            ));
+        }
+
+        for i in 0..50 {
+            let value = sampler.sample(&dist, 100 + i, &history, &directions);
+            if let ParamValue::Int(v) = value {
+                assert!((0..=100).contains(&v), "Value {v} out of range");
+            } else {
+                panic!("Expected Int value");
+            }
+        }
+    }
+
+    #[test]
+    fn test_motpe_categorical_sampling() {
+        let sampler = MotpeSampler::builder().n_startup_trials(5).seed(42).build();
+
+        let dist = Distribution::Categorical(CategoricalDistribution { n_choices: 3 });
+        let directions = [Direction::Minimize, Direction::Minimize];
+        let cat_id = ParamId::new();
+
+        // Category 1 is on the Pareto front, others are dominated
+        let mut history = Vec::new();
+        for i in 0..15 {
+            let category = i % 3;
+            let (f1, f2) = match category {
+                0 => (0.8, 0.8), // dominated
+                1 => (0.1, 0.9), // Pareto front
+                2 => (0.9, 0.1), // Pareto front
+                _ => unreachable!(),
+            };
+            history.push(create_mo_trial(
+                i as u64,
+                vec![f1, f2],
+                vec![(
+                    cat_id,
+                    ParamValue::Categorical(category as usize),
+                    dist.clone(),
+                )],
+            ));
+        }
+
+        let mut counts = vec![0usize; 3];
+        for i in 0..200 {
+            let value = sampler.sample(&dist, 100 + i, &history, &directions);
+            if let ParamValue::Categorical(idx) = value {
+                assert!(idx < 3, "Category {idx} out of range");
+                counts[idx] += 1;
+            } else {
+                panic!("Expected Categorical value");
+            }
+        }
+
+        // Categories 1 and 2 (on Pareto front) should dominate category 0
+        assert!(
+            counts[1] + counts[2] > counts[0],
+            "Pareto-front categories should be sampled more: {counts:?}"
+        );
+    }
+
+    #[test]
+    fn test_motpe_reproducibility() {
+        let dist = Distribution::Float(FloatDistribution {
+            low: 0.0,
+            high: 1.0,
+            log_scale: false,
+            step: None,
+        });
+        let directions = [Direction::Minimize, Direction::Minimize];
+        let x_id = ParamId::new();
+
+        let history: Vec<MultiObjectiveTrial> = (0..20)
+            .map(|i| {
+                let x = f64::from(i) / 20.0;
+                create_mo_trial(
+                    i as u64,
+                    vec![x, 1.0 - x],
+                    vec![(x_id, ParamValue::Float(x), dist.clone())],
+                )
+            })
+            .collect();
+
+        let sampler1 = MotpeSampler::builder()
+            .seed(12345)
+            .n_startup_trials(5)
+            .build();
+        let sampler2 = MotpeSampler::builder()
+            .seed(12345)
+            .n_startup_trials(5)
+            .build();
+
+        for i in 0..10 {
+            let v1 = sampler1.sample(&dist, i, &history, &directions);
+            let v2 = sampler2.sample(&dist, i, &history, &directions);
+            assert_eq!(v1, v2, "Samples should be identical with same seed");
+        }
+    }
+
+    #[test]
+    fn test_motpe_with_study() {
+        use crate::multi_objective::MultiObjectiveStudy;
+        use crate::parameter::{FloatParam, Parameter};
+
+        let sampler = MotpeSampler::builder().seed(42).build();
+        let study = MultiObjectiveStudy::with_sampler(
+            vec![Direction::Minimize, Direction::Minimize],
+            sampler,
+        );
+
+        let x = FloatParam::new(0.0, 1.0);
+        study
+            .optimize(30, |trial| {
+                let xv = x.suggest(trial)?;
+                Ok::<_, crate::Error>(vec![xv, 1.0 - xv])
+            })
+            .unwrap();
+
+        let front = study.pareto_front();
+        assert!(!front.is_empty(), "Should have Pareto-optimal solutions");
+
+        // All front solutions should have values summing to ~1.0
+        for trial in &front {
+            let sum: f64 = trial.values.iter().sum();
+            assert!(
+                (sum - 1.0).abs() < 0.01,
+                "Pareto front values should sum to ~1.0, got {sum}"
+            );
+        }
+    }
+
+    #[test]
+    fn test_motpe_builder_defaults() {
+        let sampler = MotpeSamplerBuilder::new().build();
+        assert_eq!(sampler.n_startup_trials, 11);
+        assert_eq!(sampler.n_ei_candidates, 24);
+        assert!(sampler.kde_bandwidth.is_none());
+    }
+
+    #[test]
+    fn test_motpe_builder_custom() {
+        let sampler = MotpeSamplerBuilder::new()
+            .n_startup_trials(20)
+            .n_ei_candidates(48)
+            .kde_bandwidth(0.5)
+            .seed(99)
+            .build();
+        assert_eq!(sampler.n_startup_trials, 20);
+        assert_eq!(sampler.n_ei_candidates, 48);
+        assert_eq!(sampler.kde_bandwidth, Some(0.5));
+    }
+}

src/sampler/nsga2.rs

@@ -0,0 +1,378 @@
+//! NSGA-II (Non-dominated Sorting Genetic Algorithm II) sampler.
+//!
+//! Implements multi-objective optimization using non-dominated sorting,
+//! crowding distance, SBX crossover, and polynomial mutation.
+//!
+//! # Examples
+//!
+//! ```
+//! use optimizer::Direction;
+//! use optimizer::multi_objective::MultiObjectiveStudy;
+//! use optimizer::parameter::{FloatParam, Parameter};
+//! use optimizer::sampler::nsga2::Nsga2Sampler;
+//!
+//! let sampler = Nsga2Sampler::with_seed(42);
+//! let study =
+//!     MultiObjectiveStudy::with_sampler(vec![Direction::Minimize, Direction::Minimize], sampler);
+//!
+//! let x = FloatParam::new(0.0, 1.0);
+//! study
+//!     .optimize(50, |trial| {
+//!         let xv = x.suggest(trial)?;
+//!         Ok::<_, optimizer::Error>(vec![xv * xv, (xv - 1.0).powi(2)])
+//!     })
+//!     .unwrap();
+//! ```
+
+use parking_lot::Mutex;
+
+use super::genetic::{
+    self, Candidate, EvolutionaryState, Phase, advance_generation, collect_evaluated_generation,
+    crossover, extract_trial_params, finalize_discovery, generate_random_candidates, mutate,
+    sample_from_candidate, sample_random,
+};
+use crate::distribution::Distribution;
+use crate::multi_objective::MultiObjectiveTrial;
+use crate::param::ParamValue;
+use crate::pareto;
+use crate::types::Direction;
+
+/// NSGA-II sampler for multi-objective optimization.
+///
+/// Provides non-dominated sorting, crowding distance selection,
+/// SBX crossover, and polynomial mutation.
+pub struct Nsga2Sampler {
+    state: Mutex<Nsga2State>,
+}
+
+impl Nsga2Sampler {
+    /// Creates a new NSGA-II sampler with a random seed.
+    #[must_use]
+    pub fn new() -> Self {
+        Self {
+            state: Mutex::new(Nsga2State::new(Nsga2Config::default(), None)),
+        }
+    }
+
+    /// Creates a new NSGA-II sampler with a fixed seed.
+    #[must_use]
+    pub fn with_seed(seed: u64) -> Self {
+        Self {
+            state: Mutex::new(Nsga2State::new(Nsga2Config::default(), Some(seed))),
+        }
+    }
+
+    /// Creates a builder for configuring an `Nsga2Sampler`.
+    #[must_use]
+    pub fn builder() -> Nsga2SamplerBuilder {
+        Nsga2SamplerBuilder::default()
+    }
+}
+
+impl Default for Nsga2Sampler {
+    fn default() -> Self {
+        Self::new()
+    }
+}
+
+/// Builder for [`Nsga2Sampler`].
+#[derive(Debug, Clone, Default)]
+pub struct Nsga2SamplerBuilder {
+    population_size: Option<usize>,
+    crossover_prob: Option<f64>,
+    crossover_eta: Option<f64>,
+    mutation_eta: Option<f64>,
+    seed: Option<u64>,
+}
+
+impl Nsga2SamplerBuilder {
+    /// Sets the population size. Default: `4 + floor(3 * ln(n_params))`, minimum 4.
+    #[must_use]
+    pub fn population_size(mut self, size: usize) -> Self {
+        self.population_size = Some(size);
+        self
+    }
+
+    /// Sets the crossover probability. Default: 0.9.
+    #[must_use]
+    pub fn crossover_prob(mut self, prob: f64) -> Self {
+        self.crossover_prob = Some(prob);
+        self
+    }
+
+    /// Sets the SBX distribution index. Default: 20.0.
+    #[must_use]
+    pub fn crossover_eta(mut self, eta: f64) -> Self {
+        self.crossover_eta = Some(eta);
+        self
+    }
+
+    /// Sets the polynomial mutation distribution index. Default: 20.0.
+    #[must_use]
+    pub fn mutation_eta(mut self, eta: f64) -> Self {
+        self.mutation_eta = Some(eta);
+        self
+    }
+
+    /// Sets the random seed for reproducibility.
+    #[must_use]
+    pub fn seed(mut self, seed: u64) -> Self {
+        self.seed = Some(seed);
+        self
+    }
+
+    /// Builds the configured [`Nsga2Sampler`].
+    #[must_use]
+    pub fn build(self) -> Nsga2Sampler {
+        let config = Nsga2Config {
+            user_population_size: self.population_size,
+            crossover_prob: self.crossover_prob.unwrap_or(0.9),
+            crossover_eta: self.crossover_eta.unwrap_or(20.0),
+            mutation_eta: self.mutation_eta.unwrap_or(20.0),
+        };
+        Nsga2Sampler {
+            state: Mutex::new(Nsga2State::new(config, self.seed)),
+        }
+    }
+}
+
+// ---------------------------------------------------------------------------
+// Internal types
+// ---------------------------------------------------------------------------
+
+#[derive(Clone, Debug)]
+struct Nsga2Config {
+    user_population_size: Option<usize>,
+    crossover_prob: f64,
+    crossover_eta: f64,
+    mutation_eta: f64,
+}
+
+impl Default for Nsga2Config {
+    fn default() -> Self {
+        Self {
+            user_population_size: None,
+            crossover_prob: 0.9,
+            crossover_eta: 20.0,
+            mutation_eta: 20.0,
+        }
+    }
+}
+
+struct Nsga2State {
+    evo: EvolutionaryState,
+    config: Nsga2Config,
+}
+
+impl Nsga2State {
+    fn new(config: Nsga2Config, seed: Option<u64>) -> Self {
+        Self {
+            evo: EvolutionaryState::new(seed),
+            config,
+        }
+    }
+}
+
+// ---------------------------------------------------------------------------
+// MultiObjectiveSampler implementation
+// ---------------------------------------------------------------------------
+
+impl crate::multi_objective::MultiObjectiveSampler for Nsga2Sampler {
+    fn sample(
+        &self,
+        distribution: &Distribution,
+        trial_id: u64,
+        history: &[MultiObjectiveTrial],
+        directions: &[Direction],
+    ) -> ParamValue {
+        let mut state = self.state.lock();
+
+        match &state.evo.phase {
+            Phase::Discovery => {
+                if let Some(value) =
+                    genetic::sample_discovery(&mut state.evo, distribution, trial_id)
+                {
+                    return value;
+                }
+                // Transitioned to active phase
+                let user_pop = state.config.user_population_size;
+                finalize_discovery(&mut state.evo, user_pop);
+                generate_random_candidates(&mut state.evo);
+                sample_from_candidate(&mut state.evo, trial_id)
+            }
+            Phase::Active => {
+                maybe_generate_new_generation(&mut state, history, directions);
+                sample_from_candidate(&mut state.evo, trial_id)
+            }
+        }
+    }
+}
+
+/// Check if all candidates in the current generation have been evaluated;
+/// if so, run NSGA-II selection and generate offspring.
+fn maybe_generate_new_generation(
+    state: &mut Nsga2State,
+    history: &[MultiObjectiveTrial],
+    directions: &[Direction],
+) {
+    if state.evo.candidates.is_empty() {
+        generate_random_candidates(&mut state.evo);
+        return;
+    }
+
+    if let Some(evaluated) = collect_evaluated_generation(&state.evo, history) {
+        let offspring = nsga2_generate_offspring(state, &evaluated, directions);
+        advance_generation(&mut state.evo, offspring);
+    }
+}
+
+// ---------------------------------------------------------------------------
+// NSGA-II generation algorithm
+// ---------------------------------------------------------------------------
+
+/// Performs NSGA-II selection: non-dominated sort + crowding distance,
+/// then selects `pop_size` parents from the population.
+fn nsga2_select(
+    state: &mut Nsga2State,
+    population: &[&MultiObjectiveTrial],
+    directions: &[Direction],
+) -> (Vec<Vec<ParamValue>>, Vec<usize>, Vec<f64>) {
+    let pop_size = state.evo.population_size;
+
+    let values: Vec<Vec<f64>> = population.iter().map(|t| t.values.clone()).collect();
+    let constraints: Vec<Vec<f64>> = population.iter().map(|t| t.constraints.clone()).collect();
+    let has_constraints = constraints.iter().any(|c| !c.is_empty());
+
+    let fronts = if has_constraints {
+        pareto::fast_non_dominated_sort_constrained(&values, directions, &constraints)
+    } else {
+        pareto::fast_non_dominated_sort(&values, directions)
+    };
+
+    let n = population.len();
+    let mut rank = vec![0_usize; n];
+    let mut crowding = vec![0.0_f64; n];
+
+    for (front_rank, front) in fronts.iter().enumerate() {
+        let cd = pareto::crowding_distance_indexed(front, &values);
+        for (i, &idx) in front.iter().enumerate() {
+            rank[idx] = front_rank;
+            crowding[idx] = cd[i];
+        }
+    }
+
+    let mut selected: Vec<usize> = Vec::with_capacity(pop_size);
+    for front in &fronts {
+        if selected.len() + front.len() <= pop_size {
+            selected.extend_from_slice(front);
+        } else {
+            let remaining = pop_size - selected.len();
+            let mut front_sorted: Vec<usize> = front.clone();
+            front_sorted.sort_by(|&a, &b| {
+                crowding[b]
+                    .partial_cmp(&crowding[a])
+                    .unwrap_or(core::cmp::Ordering::Equal)
+            });
+            selected.extend_from_slice(&front_sorted[..remaining]);
+            break;
+        }
+    }
+
+    while selected.len() < pop_size {
+        selected.push(state.evo.rng.usize(0..n));
+    }
+
+    let parents: Vec<Vec<ParamValue>> = selected
+        .iter()
+        .map(|&idx| {
+            extract_trial_params(population[idx], &state.evo.dimensions, &mut state.evo.rng)
+        })
+        .collect();
+
+    let sel_rank: Vec<usize> = selected.iter().map(|&i| rank[i]).collect();
+    let sel_crowding: Vec<f64> = selected.iter().map(|&i| crowding[i]).collect();
+
+    (parents, sel_rank, sel_crowding)
+}
+
+/// Runs NSGA-II selection and generates offspring candidates.
+fn nsga2_generate_offspring(
+    state: &mut Nsga2State,
+    population: &[&MultiObjectiveTrial],
+    directions: &[Direction],
+) -> Vec<Candidate> {
+    let pop_size = state.evo.population_size;
+
+    if population.len() < 2 {
+        return (0..pop_size)
+            .map(|_| {
+                let params = state
+                    .evo
+                    .dimensions
+                    .iter()
+                    .map(|d| sample_random(&mut state.evo.rng, &d.distribution))
+                    .collect();
+                Candidate { params }
+            })
+            .collect();
+    }
+
+    let (parents, sel_rank, sel_crowding) = nsga2_select(state, population, directions);
+
+    let mut offspring = Vec::with_capacity(pop_size);
+    while offspring.len() < pop_size {
+        let p1 = tournament_select(&mut state.evo.rng, &sel_rank, &sel_crowding, parents.len());
+        let p2 = tournament_select(&mut state.evo.rng, &sel_rank, &sel_crowding, parents.len());
+
+        let (mut child1, mut child2) = crossover(
+            &mut state.evo.rng,
+            &parents[p1],
+            &parents[p2],
+            &state.evo.dimensions,
+            state.config.crossover_prob,
+            state.config.crossover_eta,
+        );
+
+        mutate(
+            &mut state.evo.rng,
+            &mut child1,
+            &state.evo.dimensions,
+            state.config.mutation_eta,
+        );
+        mutate(
+            &mut state.evo.rng,
+            &mut child2,
+            &state.evo.dimensions,
+            state.config.mutation_eta,
+        );
+
+        offspring.push(Candidate { params: child1 });
+        if offspring.len() < pop_size {
+            offspring.push(Candidate { params: child2 });
+        }
+    }
+
+    offspring
+}
+
+/// Tournament selection: pick 2 random individuals, return index of winner.
+/// Winner has lower rank; ties broken by higher crowding distance.
+fn tournament_select(
+    rng: &mut fastrand::Rng,
+    ranks: &[usize],
+    crowding: &[f64],
+    n: usize,
+) -> usize {
+    let a = rng.usize(0..n);
+    let b = rng.usize(0..n);
+
+    if ranks[a] < ranks[b] {
+        a
+    } else if ranks[b] < ranks[a] {
+        b
+    } else if crowding[a] >= crowding[b] {
+        a
+    } else {
+        b
+    }
+}

src/sampler/nsga3.rs

@@ -0,0 +1,720 @@
+//! NSGA-III (Non-dominated Sorting Genetic Algorithm III) sampler.
+//!
+//! Uses reference-point-based niching for better diversity in
+//! many-objective (3+) optimization problems. Das-Dennis structured
+//! reference points guide the search toward a well-distributed
+//! Pareto front.
+//!
+//! # Examples
+//!
+//! ```
+//! use optimizer::Direction;
+//! use optimizer::multi_objective::MultiObjectiveStudy;
+//! use optimizer::parameter::{FloatParam, Parameter};
+//! use optimizer::sampler::nsga3::Nsga3Sampler;
+//!
+//! let sampler = Nsga3Sampler::with_seed(42);
+//! let study = MultiObjectiveStudy::with_sampler(
+//!     vec![
+//!         Direction::Minimize,
+//!         Direction::Minimize,
+//!         Direction::Minimize,
+//!     ],
+//!     sampler,
+//! );
+//!
+//! let x = FloatParam::new(0.0, 1.0);
+//! let y = FloatParam::new(0.0, 1.0);
+//! study
+//!     .optimize(100, |trial| {
+//!         let xv = x.suggest(trial)?;
+//!         let yv = y.suggest(trial)?;
+//!         Ok::<_, optimizer::Error>(vec![xv, yv, (1.0 - xv - yv).abs()])
+//!     })
+//!     .unwrap();
+//! ```
+
+use parking_lot::Mutex;
+
+use super::genetic::{
+    self, Candidate, EvolutionaryState, Phase, advance_generation, auto_divisions,
+    collect_evaluated_generation, crossover, das_dennis, extract_trial_params,
+    generate_random_candidates, mutate, sample_from_candidate, sample_random,
+};
+use crate::distribution::Distribution;
+use crate::multi_objective::MultiObjectiveTrial;
+use crate::param::ParamValue;
+use crate::pareto;
+use crate::types::Direction;
+
+/// NSGA-III sampler for multi-objective optimization.
+///
+/// Uses reference-point-based niching to maintain diversity,
+/// especially effective for problems with 3 or more objectives.
+pub struct Nsga3Sampler {
+    state: Mutex<Nsga3State>,
+}
+
+impl Nsga3Sampler {
+    /// Creates a new NSGA-III sampler with a random seed.
+    #[must_use]
+    pub fn new() -> Self {
+        Self {
+            state: Mutex::new(Nsga3State::new(Nsga3Config::default(), None)),
+        }
+    }
+
+    /// Creates a new NSGA-III sampler with a fixed seed.
+    #[must_use]
+    pub fn with_seed(seed: u64) -> Self {
+        Self {
+            state: Mutex::new(Nsga3State::new(Nsga3Config::default(), Some(seed))),
+        }
+    }
+
+    /// Creates a builder for configuring an `Nsga3Sampler`.
+    #[must_use]
+    pub fn builder() -> Nsga3SamplerBuilder {
+        Nsga3SamplerBuilder::default()
+    }
+}
+
+impl Default for Nsga3Sampler {
+    fn default() -> Self {
+        Self::new()
+    }
+}
+
+/// Builder for [`Nsga3Sampler`].
+#[derive(Debug, Clone, Default)]
+pub struct Nsga3SamplerBuilder {
+    population_size: Option<usize>,
+    n_divisions: Option<usize>,
+    crossover_prob: Option<f64>,
+    crossover_eta: Option<f64>,
+    mutation_eta: Option<f64>,
+    seed: Option<u64>,
+}
+
+impl Nsga3SamplerBuilder {
+    /// Sets the population size. If unset, equals the number of
+    /// Das-Dennis reference points.
+    #[must_use]
+    pub fn population_size(mut self, size: usize) -> Self {
+        self.population_size = Some(size);
+        self
+    }
+
+    /// Sets the number of divisions (H) for Das-Dennis reference points.
+    /// If unset, automatically chosen based on population size and number
+    /// of objectives.
+    #[must_use]
+    pub fn n_divisions(mut self, h: usize) -> Self {
+        self.n_divisions = Some(h);
+        self
+    }
+
+    /// Sets the crossover probability. Default: 1.0.
+    #[must_use]
+    pub fn crossover_prob(mut self, prob: f64) -> Self {
+        self.crossover_prob = Some(prob);
+        self
+    }
+
+    /// Sets the SBX distribution index. Default: 30.0.
+    #[must_use]
+    pub fn crossover_eta(mut self, eta: f64) -> Self {
+        self.crossover_eta = Some(eta);
+        self
+    }
+
+    /// Sets the polynomial mutation distribution index. Default: 20.0.
+    #[must_use]
+    pub fn mutation_eta(mut self, eta: f64) -> Self {
+        self.mutation_eta = Some(eta);
+        self
+    }
+
+    /// Sets the random seed for reproducibility.
+    #[must_use]
+    pub fn seed(mut self, seed: u64) -> Self {
+        self.seed = Some(seed);
+        self
+    }
+
+    /// Builds the configured [`Nsga3Sampler`].
+    #[must_use]
+    pub fn build(self) -> Nsga3Sampler {
+        let config = Nsga3Config {
+            user_population_size: self.population_size,
+            n_divisions: self.n_divisions,
+            crossover_prob: self.crossover_prob.unwrap_or(1.0),
+            crossover_eta: self.crossover_eta.unwrap_or(30.0),
+            mutation_eta: self.mutation_eta.unwrap_or(20.0),
+        };
+        Nsga3Sampler {
+            state: Mutex::new(Nsga3State::new(config, self.seed)),
+        }
+    }
+}
+
+// ---------------------------------------------------------------------------
+// Internal types
+// ---------------------------------------------------------------------------
+
+#[derive(Clone, Debug)]
+struct Nsga3Config {
+    user_population_size: Option<usize>,
+    n_divisions: Option<usize>,
+    crossover_prob: f64,
+    crossover_eta: f64,
+    mutation_eta: f64,
+}
+
+impl Default for Nsga3Config {
+    fn default() -> Self {
+        Self {
+            user_population_size: None,
+            n_divisions: None,
+            crossover_prob: 1.0,
+            crossover_eta: 30.0,
+            mutation_eta: 20.0,
+        }
+    }
+}
+
+struct Nsga3State {
+    evo: EvolutionaryState,
+    config: Nsga3Config,
+    /// Das-Dennis reference points (lazily generated once objectives are known).
+    reference_points: Vec<Vec<f64>>,
+    /// Best value seen per objective (minimize-space).
+    ideal_point: Vec<f64>,
+    /// Whether reference points have been initialized.
+    initialized: bool,
+}
+
+impl Nsga3State {
+    fn new(config: Nsga3Config, seed: Option<u64>) -> Self {
+        Self {
+            evo: EvolutionaryState::new(seed),
+            config,
+            reference_points: Vec::new(),
+            ideal_point: Vec::new(),
+            initialized: false,
+        }
+    }
+}
+
+// ---------------------------------------------------------------------------
+// MultiObjectiveSampler implementation
+// ---------------------------------------------------------------------------
+
+impl crate::multi_objective::MultiObjectiveSampler for Nsga3Sampler {
+    fn sample(
+        &self,
+        distribution: &Distribution,
+        trial_id: u64,
+        history: &[MultiObjectiveTrial],
+        directions: &[Direction],
+    ) -> ParamValue {
+        let mut state = self.state.lock();
+
+        match &state.evo.phase {
+            Phase::Discovery => {
+                if let Some(value) =
+                    genetic::sample_discovery(&mut state.evo, distribution, trial_id)
+                {
+                    return value;
+                }
+                // Transitioned to active phase
+                initialize_nsga3(&mut state, directions);
+                generate_random_candidates(&mut state.evo);
+                sample_from_candidate(&mut state.evo, trial_id)
+            }
+            Phase::Active => {
+                maybe_generate_new_generation(&mut state, history, directions);
+                sample_from_candidate(&mut state.evo, trial_id)
+            }
+        }
+    }
+}
+
+/// Initialize NSGA-III: generate reference points and set population size.
+fn initialize_nsga3(state: &mut Nsga3State, directions: &[Direction]) {
+    let n_obj = directions.len();
+
+    // Determine divisions
+    let divisions = state
+        .config
+        .n_divisions
+        .unwrap_or_else(|| auto_divisions(n_obj, state.config.user_population_size.unwrap_or(100)));
+
+    state.reference_points = das_dennis(n_obj, divisions);
+    let n_ref = state.reference_points.len();
+
+    // Population size = number of reference points (or user override, at least n_ref)
+    let pop_size = state.config.user_population_size.unwrap_or(n_ref).max(4);
+    state.evo.population_size = pop_size;
+    state.evo.phase = Phase::Active;
+    state.ideal_point = vec![f64::INFINITY; n_obj];
+    state.initialized = true;
+}
+
+fn maybe_generate_new_generation(
+    state: &mut Nsga3State,
+    history: &[MultiObjectiveTrial],
+    directions: &[Direction],
+) {
+    if state.evo.candidates.is_empty() {
+        generate_random_candidates(&mut state.evo);
+        return;
+    }
+
+    if let Some(evaluated) = collect_evaluated_generation(&state.evo, history) {
+        let offspring = nsga3_generate_offspring(state, &evaluated, directions);
+        advance_generation(&mut state.evo, offspring);
+    }
+}
+
+// ---------------------------------------------------------------------------
+// NSGA-III selection algorithm
+// ---------------------------------------------------------------------------
+
+/// Normalize objectives to minimize-space.
+fn to_minimize_space(values: &[f64], directions: &[Direction]) -> Vec<f64> {
+    values
+        .iter()
+        .zip(directions)
+        .map(|(&v, d)| match d {
+            Direction::Minimize => v,
+            Direction::Maximize => -v,
+        })
+        .collect()
+}
+
+/// Update ideal point with new observations.
+fn update_ideal_point(ideal: &mut [f64], normalized_values: &[Vec<f64>]) {
+    for vals in normalized_values {
+        for (i, &v) in vals.iter().enumerate() {
+            if v < ideal[i] {
+                ideal[i] = v;
+            }
+        }
+    }
+}
+
+/// Compute Achievement Scalarizing Function (ASF) for extreme point finding.
+fn asf(point: &[f64], weight: &[f64], ideal: &[f64]) -> f64 {
+    point
+        .iter()
+        .zip(weight)
+        .zip(ideal)
+        .map(|((&p, &w), &z)| {
+            let w = if w < 1e-6 { 1e-6 } else { w };
+            (p - z) / w
+        })
+        .fold(f64::NEG_INFINITY, f64::max)
+}
+
+/// Find intercepts for normalization via extreme points.
+///
+/// For each objective, find the point with best ASF (using a weight vector
+/// that emphasizes that objective). The intercepts are where the hyperplane
+/// through the extreme points crosses each axis.
+fn find_intercepts(normalized_values: &[Vec<f64>], ideal: &[f64]) -> Vec<f64> {
+    let n_obj = ideal.len();
+    let n = normalized_values.len();
+
+    if n == 0 || n_obj == 0 {
+        return vec![1.0; n_obj];
+    }
+
+    // Find extreme points (one per objective)
+    let mut extreme_indices = Vec::with_capacity(n_obj);
+    for obj in 0..n_obj {
+        let mut weight = vec![1e-6; n_obj];
+        weight[obj] = 1.0;
+
+        let mut best_idx = 0;
+        let mut best_asf = f64::INFINITY;
+        for (i, vals) in normalized_values.iter().enumerate() {
+            let a = asf(vals, &weight, ideal);
+            if a < best_asf {
+                best_asf = a;
+                best_idx = i;
+            }
+        }
+        extreme_indices.push(best_idx);
+    }
+
+    // Try to compute hyperplane intercepts
+    // For stability, if the extreme points are degenerate, fall back to
+    // max - ideal per objective
+    let mut intercepts = Vec::with_capacity(n_obj);
+    for obj in 0..n_obj {
+        let max_val = normalized_values
+            .iter()
+            .map(|v| v[obj])
+            .fold(f64::NEG_INFINITY, f64::max);
+        let intercept = max_val - ideal[obj];
+        intercepts.push(if intercept > 1e-10 { intercept } else { 1.0 });
+    }
+
+    intercepts
+}
+
+/// Normalize objective values: subtract ideal, divide by intercepts.
+fn normalize_objectives(values: &[Vec<f64>], ideal: &[f64], intercepts: &[f64]) -> Vec<Vec<f64>> {
+    values
+        .iter()
+        .map(|v| {
+            v.iter()
+                .zip(ideal)
+                .zip(intercepts)
+                .map(|((&val, &z), &a)| {
+                    let norm = if a > 1e-10 { a } else { 1.0 };
+                    (val - z) / norm
+                })
+                .collect()
+        })
+        .collect()
+}
+
+/// Perpendicular distance from a point to a reference line (direction vector).
+fn perpendicular_distance(point: &[f64], reference: &[f64]) -> f64 {
+    let dot: f64 = point.iter().zip(reference).map(|(&p, &r)| p * r).sum();
+    let ref_norm_sq: f64 = reference.iter().map(|&r| r * r).sum();
+
+    if ref_norm_sq < 1e-30 {
+        return f64::INFINITY;
+    }
+
+    let proj_scalar = dot / ref_norm_sq;
+    let dist_sq: f64 = point
+        .iter()
+        .zip(reference)
+        .map(|(&p, &r)| {
+            let proj = proj_scalar * r;
+            (p - proj).powi(2)
+        })
+        .sum();
+
+    dist_sq.sqrt()
+}
+
+/// Associate each solution with its nearest reference point.
+/// Returns (`closest_ref_idx`, distance) for each solution.
+fn associate_to_reference_points(
+    normalized: &[Vec<f64>],
+    reference_points: &[Vec<f64>],
+) -> Vec<(usize, f64)> {
+    normalized
+        .iter()
+        .map(|point| {
+            let mut best_ref = 0;
+            let mut best_dist = f64::INFINITY;
+            for (j, rp) in reference_points.iter().enumerate() {
+                let d = perpendicular_distance(point, rp);
+                if d < best_dist {
+                    best_dist = d;
+                    best_ref = j;
+                }
+            }
+            (best_ref, best_dist)
+        })
+        .collect()
+}
+
+/// NSGA-III niching-based selection from the last front.
+///
+/// `already_selected` are indices into the combined population that are
+/// already accepted (from fronts 0..L-1). `last_front` contains indices
+/// from front L. We need to pick `remaining` more from `last_front`.
+fn niching_select(
+    rng: &mut fastrand::Rng,
+    associations: &[(usize, f64)],
+    already_selected: &[usize],
+    last_front: &[usize],
+    n_reference_points: usize,
+    remaining: usize,
+) -> Vec<usize> {
+    // Count niche per reference point for already selected
+    let mut niche_count = vec![0_usize; n_reference_points];
+    for &idx in already_selected {
+        niche_count[associations[idx].0] += 1;
+    }
+
+    // Build per-reference-point candidate lists from the last front
+    let mut ref_candidates: Vec<Vec<(usize, f64)>> = vec![Vec::new(); n_reference_points];
+    for &idx in last_front {
+        let (ref_idx, dist) = associations[idx];
+        ref_candidates[ref_idx].push((idx, dist));
+    }
+
+    let mut selected = Vec::with_capacity(remaining);
+    let mut excluded = vec![false; associations.len()];
+
+    for _ in 0..remaining {
+        // Find minimum niche count among reference points that still have candidates
+        let min_count = (0..n_reference_points)
+            .filter(|&j| ref_candidates[j].iter().any(|&(idx, _)| !excluded[idx]))
+            .map(|j| niche_count[j])
+            .min();
+
+        let Some(min_count) = min_count else {
+            break;
+        };
+
+        // Collect reference points with this minimum count that have candidates
+        let min_refs: Vec<usize> = (0..n_reference_points)
+            .filter(|&j| {
+                niche_count[j] == min_count
+                    && ref_candidates[j].iter().any(|&(idx, _)| !excluded[idx])
+            })
+            .collect();
+
+        if min_refs.is_empty() {
+            break;
+        }
+
+        // Pick a random reference point from the minimum set
+        let chosen_ref = min_refs[rng.usize(0..min_refs.len())];
+
+        // Available candidates for this reference point
+        let available: Vec<(usize, f64)> = ref_candidates[chosen_ref]
+            .iter()
+            .filter(|&&(idx, _)| !excluded[idx])
+            .copied()
+            .collect();
+
+        if available.is_empty() {
+            continue;
+        }
+
+        let chosen_idx = if min_count == 0 {
+            // Pick closest to reference line
+            available
+                .iter()
+                .min_by(|a, b| a.1.partial_cmp(&b.1).unwrap_or(core::cmp::Ordering::Equal))
+                .unwrap()
+                .0
+        } else {
+            // Pick random
+            available[rng.usize(0..available.len())].0
+        };
+
+        selected.push(chosen_idx);
+        excluded[chosen_idx] = true;
+        niche_count[chosen_ref] += 1;
+    }
+
+    selected
+}
+
+/// Perform NSGA-III selection: non-dominated sort + reference-point niching.
+fn nsga3_select(
+    state: &mut Nsga3State,
+    population: &[&MultiObjectiveTrial],
+    directions: &[Direction],
+) -> Vec<Vec<ParamValue>> {
+    let pop_size = state.evo.population_size;
+    let n_obj = directions.len();
+
+    // Convert to minimize-space
+    let min_values: Vec<Vec<f64>> = population
+        .iter()
+        .map(|t| to_minimize_space(&t.values, directions))
+        .collect();
+
+    // Non-dominated sort
+    let constraints: Vec<Vec<f64>> = population.iter().map(|t| t.constraints.clone()).collect();
+    let has_constraints = constraints.iter().any(|c| !c.is_empty());
+    let fronts = if has_constraints {
+        pareto::fast_non_dominated_sort_constrained(
+            &min_values,
+            &vec![Direction::Minimize; n_obj],
+            &constraints,
+        )
+    } else {
+        pareto::fast_non_dominated_sort(&min_values, &vec![Direction::Minimize; n_obj])
+    };
+
+    // Fill front-by-front
+    let mut selected: Vec<usize> = Vec::with_capacity(pop_size);
+    let mut last_front_idx = None;
+
+    for (fi, front) in fronts.iter().enumerate() {
+        if selected.len() + front.len() <= pop_size {
+            selected.extend_from_slice(front);
+        } else {
+            last_front_idx = Some(fi);
+            break;
+        }
+    }
+
+    // If we filled exactly or all fronts fit, done
+    if selected.len() < pop_size
+        && let Some(lf_idx) = last_front_idx
+    {
+        // Need niching from the last partial front
+        let remaining = pop_size - selected.len();
+
+        // Update ideal point
+        update_ideal_point(&mut state.ideal_point, &min_values);
+
+        // Find intercepts and normalize
+        let intercepts = find_intercepts(&min_values, &state.ideal_point);
+        let normalized = normalize_objectives(&min_values, &state.ideal_point, &intercepts);
+
+        // Associate all solutions with reference points
+        let associations = associate_to_reference_points(&normalized, &state.reference_points);
+
+        // Select from last front using niching
+        let last_front = &fronts[lf_idx];
+        let additional = niching_select(
+            &mut state.evo.rng,
+            &associations,
+            &selected,
+            last_front,
+            state.reference_points.len(),
+            remaining,
+        );
+        selected.extend(additional);
+    }
+
+    // Pad if needed
+    let n = population.len();
+    while selected.len() < pop_size {
+        selected.push(state.evo.rng.usize(0..n));
+    }
+
+    selected
+        .iter()
+        .map(|&idx| {
+            extract_trial_params(population[idx], &state.evo.dimensions, &mut state.evo.rng)
+        })
+        .collect()
+}
+
+/// Tournament selection based on rank only (no crowding distance in NSGA-III).
+fn tournament_select_rank(rng: &mut fastrand::Rng, ranks: &[usize], n: usize) -> usize {
+    let a = rng.usize(0..n);
+    let b = rng.usize(0..n);
+
+    if ranks[a] <= ranks[b] { a } else { b }
+}
+
+fn nsga3_generate_offspring(
+    state: &mut Nsga3State,
+    population: &[&MultiObjectiveTrial],
+    directions: &[Direction],
+) -> Vec<Candidate> {
+    let pop_size = state.evo.population_size;
+
+    if population.len() < 2 {
+        return (0..pop_size)
+            .map(|_| {
+                let params = state
+                    .evo
+                    .dimensions
+                    .iter()
+                    .map(|d| sample_random(&mut state.evo.rng, &d.distribution))
+                    .collect();
+                Candidate { params }
+            })
+            .collect();
+    }
+
+    // Initialize reference points and ideal on first generation
+    if !state.initialized {
+        initialize_nsga3(state, directions);
+    }
+
+    let parents = nsga3_select(state, population, directions);
+
+    // Assign ranks for tournament selection
+    let n_obj = directions.len();
+    let min_values: Vec<Vec<f64>> = population
+        .iter()
+        .map(|t| to_minimize_space(&t.values, directions))
+        .collect();
+    let fronts = pareto::fast_non_dominated_sort(&min_values, &vec![Direction::Minimize; n_obj]);
+    let mut rank = vec![0_usize; parents.len()];
+    for (front_rank, front) in fronts.iter().enumerate() {
+        for &idx in front {
+            if idx < rank.len() {
+                rank[idx] = front_rank;
+            }
+        }
+    }
+    // Ranks for selected parents (simplified: use index order)
+    let parent_ranks: Vec<usize> = (0..parents.len())
+        .map(|i| i % (fronts.len().max(1)))
+        .collect();
+
+    let mut offspring = Vec::with_capacity(pop_size);
+    while offspring.len() < pop_size {
+        let p1 = tournament_select_rank(&mut state.evo.rng, &parent_ranks, parents.len());
+        let p2 = tournament_select_rank(&mut state.evo.rng, &parent_ranks, parents.len());
+
+        let (mut child1, mut child2) = crossover(
+            &mut state.evo.rng,
+            &parents[p1],
+            &parents[p2],
+            &state.evo.dimensions,
+            state.config.crossover_prob,
+            state.config.crossover_eta,
+        );
+
+        mutate(
+            &mut state.evo.rng,
+            &mut child1,
+            &state.evo.dimensions,
+            state.config.mutation_eta,
+        );
+        mutate(
+            &mut state.evo.rng,
+            &mut child2,
+            &state.evo.dimensions,
+            state.config.mutation_eta,
+        );
+
+        offspring.push(Candidate { params: child1 });
+        if offspring.len() < pop_size {
+            offspring.push(Candidate { params: child2 });
+        }
+    }
+
+    offspring
+}
+
+#[cfg(test)]
+mod tests {
+    use super::*;
+
+    #[test]
+    fn test_perpendicular_distance() {
+        // Point (1, 0) to reference line (1, 1) (45-degree line)
+        let d = perpendicular_distance(&[1.0, 0.0], &[1.0, 1.0]);
+        // Projection is (0.5, 0.5), distance = sqrt(0.25 + 0.25) = sqrt(0.5)
+        assert!((d - (0.5_f64).sqrt()).abs() < 1e-10);
+    }
+
+    #[test]
+    fn test_perpendicular_distance_on_line() {
+        // Point on the reference line
+        let d = perpendicular_distance(&[2.0, 2.0], &[1.0, 1.0]);
+        assert!(d < 1e-10);
+    }
+
+    #[test]
+    fn test_normalize_objectives() {
+        let values = vec![vec![2.0, 4.0], vec![4.0, 2.0]];
+        let ideal = vec![1.0, 1.0];
+        let intercepts = vec![3.0, 3.0];
+        let normalized = normalize_objectives(&values, &ideal, &intercepts);
+        assert!((normalized[0][0] - 1.0 / 3.0).abs() < 1e-10);
+        assert!((normalized[0][1] - 1.0).abs() < 1e-10);
+    }
+}

src/sampler/random.rs

@@ -1,11 +1,10 @@
 //! Random sampler implementation.
 
 use parking_lot::Mutex;
-use rand::rngs::StdRng;
-use rand::{RngExt, SeedableRng};
 
 use crate::distribution::Distribution;
 use crate::param::ParamValue;
+use crate::rng_util;
 use crate::sampler::{CompletedTrial, Sampler};
 
 /// A simple random sampler that samples uniformly from distributions.
@@ -26,7 +25,7 @@ use crate::sampler::{CompletedTrial, Sampler};
 /// let sampler = RandomSampler::with_seed(42);
 /// ```
 pub struct RandomSampler {
-    rng: Mutex<StdRng>,
+    rng: Mutex<fastrand::Rng>,
 }
 
 impl RandomSampler {
@@ -34,7 +33,7 @@ impl RandomSampler {
     #[must_use]
     pub fn new() -> Self {
         Self {
-            rng: Mutex::new(rand::make_rng()),
+            rng: Mutex::new(fastrand::Rng::new()),
         }
     }
 
@@ -44,7 +43,7 @@ impl RandomSampler {
     #[must_use]
     pub fn with_seed(seed: u64) -> Self {
         Self {
-            rng: Mutex::new(StdRng::seed_from_u64(seed)),
+            rng: Mutex::new(fastrand::Rng::with_seed(seed)),
         }
     }
 }
@@ -71,16 +70,16 @@ impl Sampler for RandomSampler {
                     // Sample uniformly in log space
                     let log_low = d.low.ln();
                     let log_high = d.high.ln();
-                    let log_value = rng.random_range(log_low..=log_high);
+                    let log_value = rng_util::f64_range(&mut rng, log_low, log_high);
                     log_value.exp()
                 } else if let Some(step) = d.step {
                     // Sample from step grid
                     let n_steps = ((d.high - d.low) / step).floor() as i64;
-                    let k = rng.random_range(0..=n_steps);
+                    let k = rng.i64(0..=n_steps);
                     d.low + (k as f64) * step
                 } else {
                     // Uniform sampling
-                    rng.random_range(d.low..=d.high)
+                    rng_util::f64_range(&mut rng, d.low, d.high)
                 };
                 ParamValue::Float(value)
             }
@@ -89,23 +88,23 @@ impl Sampler for RandomSampler {
                     // Sample uniformly in log space, then round
                     let log_low = (d.low as f64).ln();
                     let log_high = (d.high as f64).ln();
-                    let log_value = rng.random_range(log_low..=log_high);
+                    let log_value = rng_util::f64_range(&mut rng, log_low, log_high);
                     let raw = log_value.exp().round() as i64;
                     // Clamp to bounds since rounding might push outside
                     raw.clamp(d.low, d.high)
                 } else if let Some(step) = d.step {
                     // Sample from step grid
                     let n_steps = (d.high - d.low) / step;
-                    let k = rng.random_range(0..=n_steps);
+                    let k = rng.i64(0..=n_steps);
                     d.low + k * step
                 } else {
                     // Uniform sampling
-                    rng.random_range(d.low..=d.high)
+                    rng.i64(d.low..=d.high)
                 };
                 ParamValue::Int(value)
             }
             Distribution::Categorical(d) => {
-                let index = rng.random_range(0..d.n_choices);
+                let index = rng.usize(0..d.n_choices);
                 ParamValue::Categorical(index)
             }
         }

src/sampler/sobol.rs

@@ -0,0 +1,420 @@
+//! Quasi-random sampler using Sobol low-discrepancy sequences.
+
+use parking_lot::Mutex;
+use sobol_burley::sample;
+
+use crate::distribution::Distribution;
+use crate::param::ParamValue;
+use crate::sampler::{CompletedTrial, Sampler};
+
+/// Internal state for tracking the dimension counter within a trial.
+struct SobolState {
+    /// The `trial_id` of the current trial (used to reset dimension counter).
+    current_trial: u64,
+    /// Next Sobol dimension to use for the current trial.
+    next_dimension: u32,
+}
+
+/// Quasi-random sampler using Sobol low-discrepancy sequences.
+///
+/// Provides better uniform coverage of the parameter space than
+/// [`RandomSampler`](super::random::RandomSampler). Useful as a baseline or
+/// for the startup phase of model-based samplers.
+///
+/// Unlike random sampling, Sobol sequences are deterministic and fill the
+/// space more evenly, reducing the number of trials needed to adequately
+/// cover the search space.
+///
+/// Each trial uses a different Sobol sequence index, and each parameter
+/// within a trial maps to a different Sobol dimension. Parameters must be
+/// suggested in the same order across trials for consistent dimension
+/// assignment.
+///
+/// Sobol sequences are most effective in moderate dimensions (up to ~20).
+/// For very high dimensions, the uniformity advantage diminishes.
+///
+/// Requires the `sobol` feature flag.
+///
+/// # Examples
+///
+/// ```
+/// use optimizer::sampler::sobol::SobolSampler;
+/// use optimizer::{Direction, Study};
+///
+/// let study: Study<f64> = Study::with_sampler(Direction::Minimize, SobolSampler::new());
+/// ```
+pub struct SobolSampler {
+    seed: u32,
+    state: Mutex<SobolState>,
+}
+
+impl SobolSampler {
+    /// Creates a new Sobol sampler with a default seed of 0.
+    #[must_use]
+    pub fn new() -> Self {
+        Self::with_seed(0)
+    }
+
+    /// Creates a new Sobol sampler with the given seed.
+    ///
+    /// Different seeds produce statistically independent Sobol sequences.
+    /// Using the same seed will produce the same sequence of sampled values.
+    #[must_use]
+    #[allow(clippy::cast_possible_truncation)]
+    pub fn with_seed(seed: u64) -> Self {
+        Self {
+            seed: seed as u32,
+            state: Mutex::new(SobolState {
+                current_trial: u64::MAX,
+                next_dimension: 0,
+            }),
+        }
+    }
+}
+
+impl Default for SobolSampler {
+    fn default() -> Self {
+        Self::new()
+    }
+}
+
+impl Sampler for SobolSampler {
+    #[allow(clippy::cast_possible_truncation, clippy::cast_precision_loss)]
+    fn sample(
+        &self,
+        distribution: &Distribution,
+        trial_id: u64,
+        _history: &[CompletedTrial],
+    ) -> ParamValue {
+        let mut state = self.state.lock();
+
+        // Reset dimension counter when a new trial starts.
+        if state.current_trial != trial_id {
+            state.current_trial = trial_id;
+            state.next_dimension = 0;
+        }
+
+        let dimension = state.next_dimension;
+        state.next_dimension = dimension + 1;
+
+        // Use trial_id as the Sobol sequence index.
+        let index = trial_id as u32;
+
+        // Generate a quasi-random point in [0, 1).
+        let point = f64::from(sample(index, dimension, self.seed));
+
+        map_point_to_distribution(point, distribution)
+    }
+}
+
+/// Maps a uniform [0, 1) point to a value within the given distribution.
+#[allow(
+    clippy::cast_possible_truncation,
+    clippy::cast_precision_loss,
+    clippy::cast_sign_loss
+)]
+fn map_point_to_distribution(point: f64, distribution: &Distribution) -> ParamValue {
+    match distribution {
+        Distribution::Float(d) => {
+            let value = if d.log_scale {
+                let log_low = d.low.ln();
+                let log_high = d.high.ln();
+                (log_low + point * (log_high - log_low)).exp()
+            } else if let Some(step) = d.step {
+                let n_steps = ((d.high - d.low) / step).floor() as i64;
+                let k = (point * (n_steps + 1) as f64).floor() as i64;
+                let k = k.min(n_steps);
+                d.low + (k as f64) * step
+            } else {
+                d.low + point * (d.high - d.low)
+            };
+            ParamValue::Float(value)
+        }
+        Distribution::Int(d) => {
+            let value = if d.log_scale {
+                let log_low = (d.low as f64).ln();
+                let log_high = (d.high as f64).ln();
+                let raw = (log_low + point * (log_high - log_low)).exp().round() as i64;
+                raw.clamp(d.low, d.high)
+            } else if let Some(step) = d.step {
+                let n_steps = (d.high - d.low) / step;
+                let k = (point * (n_steps + 1) as f64).floor() as i64;
+                let k = k.min(n_steps);
+                d.low + k * step
+            } else {
+                let range = d.high - d.low + 1;
+                let k = (point * range as f64).floor() as i64;
+                (d.low + k).min(d.high)
+            };
+            ParamValue::Int(value)
+        }
+        Distribution::Categorical(d) => {
+            let index = (point * d.n_choices as f64).floor() as usize;
+            let index = index.min(d.n_choices - 1);
+            ParamValue::Categorical(index)
+        }
+    }
+}
+
+#[cfg(test)]
+#[allow(
+    clippy::cast_possible_truncation,
+    clippy::cast_precision_loss,
+    clippy::cast_sign_loss
+)]
+mod tests {
+    use super::*;
+    use crate::distribution::{CategoricalDistribution, FloatDistribution, IntDistribution};
+
+    #[test]
+    fn float_within_bounds() {
+        let sampler = SobolSampler::with_seed(42);
+        let dist = Distribution::Float(FloatDistribution {
+            low: -5.0,
+            high: 5.0,
+            log_scale: false,
+            step: None,
+        });
+
+        for i in 0..100 {
+            let value = sampler.sample(&dist, i, &[]);
+            if let ParamValue::Float(v) = value {
+                assert!(
+                    (-5.0..=5.0).contains(&v),
+                    "value {v} out of bounds at trial {i}"
+                );
+            } else {
+                panic!("Expected Float value");
+            }
+        }
+    }
+
+    #[test]
+    fn float_log_scale_within_bounds() {
+        let sampler = SobolSampler::with_seed(42);
+        let dist = Distribution::Float(FloatDistribution {
+            low: 1e-5,
+            high: 1.0,
+            log_scale: true,
+            step: None,
+        });
+
+        for i in 0..100 {
+            let value = sampler.sample(&dist, i, &[]);
+            if let ParamValue::Float(v) = value {
+                assert!(
+                    (1e-5..=1.0).contains(&v),
+                    "value {v} out of bounds at trial {i}"
+                );
+            } else {
+                panic!("Expected Float value");
+            }
+        }
+    }
+
+    #[test]
+    fn float_step_respects_grid() {
+        let sampler = SobolSampler::with_seed(42);
+        let dist = Distribution::Float(FloatDistribution {
+            low: 0.0,
+            high: 1.0,
+            log_scale: false,
+            step: Some(0.25),
+        });
+
+        for i in 0..100 {
+            let value = sampler.sample(&dist, i, &[]);
+            if let ParamValue::Float(v) = value {
+                assert!((0.0..=1.0).contains(&v), "value {v} out of bounds");
+                let k = (v / 0.25).round() as i64;
+                let expected = k as f64 * 0.25;
+                assert!((v - expected).abs() < 1e-10, "value {v} not on step grid");
+            } else {
+                panic!("Expected Float value");
+            }
+        }
+    }
+
+    #[test]
+    fn int_within_bounds() {
+        let sampler = SobolSampler::with_seed(42);
+        let dist = Distribution::Int(IntDistribution {
+            low: 0,
+            high: 10,
+            log_scale: false,
+            step: None,
+        });
+
+        for i in 0..100 {
+            let value = sampler.sample(&dist, i, &[]);
+            if let ParamValue::Int(v) = value {
+                assert!(
+                    (0..=10).contains(&v),
+                    "value {v} out of bounds at trial {i}"
+                );
+            } else {
+                panic!("Expected Int value");
+            }
+        }
+    }
+
+    #[test]
+    fn int_log_scale_within_bounds() {
+        let sampler = SobolSampler::with_seed(42);
+        let dist = Distribution::Int(IntDistribution {
+            low: 1,
+            high: 1000,
+            log_scale: true,
+            step: None,
+        });
+
+        for i in 0..100 {
+            let value = sampler.sample(&dist, i, &[]);
+            if let ParamValue::Int(v) = value {
+                assert!(
+                    (1..=1000).contains(&v),
+                    "value {v} out of bounds at trial {i}"
+                );
+            } else {
+                panic!("Expected Int value");
+            }
+        }
+    }
+
+    #[test]
+    fn int_step_respects_grid() {
+        let sampler = SobolSampler::with_seed(42);
+        let dist = Distribution::Int(IntDistribution {
+            low: 0,
+            high: 10,
+            log_scale: false,
+            step: Some(2),
+        });
+
+        for i in 0..100 {
+            let value = sampler.sample(&dist, i, &[]);
+            if let ParamValue::Int(v) = value {
+                assert!((0..=10).contains(&v), "value {v} out of bounds");
+                assert!(v % 2 == 0, "value {v} not on step grid");
+            } else {
+                panic!("Expected Int value");
+            }
+        }
+    }
+
+    #[test]
+    fn categorical_within_bounds() {
+        let sampler = SobolSampler::with_seed(42);
+        let dist = Distribution::Categorical(CategoricalDistribution { n_choices: 5 });
+
+        for i in 0..100 {
+            let value = sampler.sample(&dist, i, &[]);
+            if let ParamValue::Categorical(idx) = value {
+                assert!(idx < 5, "index {idx} out of bounds at trial {i}");
+            } else {
+                panic!("Expected Categorical value");
+            }
+        }
+    }
+
+    #[test]
+    fn deterministic_with_same_seed() {
+        let dist = Distribution::Float(FloatDistribution {
+            low: 0.0,
+            high: 1.0,
+            log_scale: false,
+            step: None,
+        });
+
+        let sampler1 = SobolSampler::with_seed(42);
+        let sampler2 = SobolSampler::with_seed(42);
+
+        for i in 0..20 {
+            let v1 = sampler1.sample(&dist, i, &[]);
+            let v2 = sampler2.sample(&dist, i, &[]);
+            assert_eq!(v1, v2, "mismatch at trial {i}");
+        }
+    }
+
+    #[test]
+    fn different_seeds_produce_different_sequences() {
+        let dist = Distribution::Float(FloatDistribution {
+            low: 0.0,
+            high: 1.0,
+            log_scale: false,
+            step: None,
+        });
+
+        let sampler1 = SobolSampler::with_seed(0);
+        let sampler2 = SobolSampler::with_seed(12345);
+
+        let mut any_different = false;
+        for i in 0..20 {
+            let v1 = sampler1.sample(&dist, i, &[]);
+            let v2 = sampler2.sample(&dist, i, &[]);
+            if v1 != v2 {
+                any_different = true;
+                break;
+            }
+        }
+        assert!(
+            any_different,
+            "different seeds should produce different sequences"
+        );
+    }
+
+    #[test]
+    fn better_coverage_than_random() {
+        // Sobol sequence should cover [0,1] more uniformly than random.
+        // We measure this by checking that the Sobol samples fill all
+        // 10 equal-width bins with only 20 samples.
+        let sampler = SobolSampler::with_seed(0);
+        let dist = Distribution::Float(FloatDistribution {
+            low: 0.0,
+            high: 1.0,
+            log_scale: false,
+            step: None,
+        });
+
+        let n_bins = 10;
+        let n_samples = 20;
+        let mut bins = vec![0u32; n_bins];
+
+        for i in 0..n_samples {
+            let value = sampler.sample(&dist, i, &[]);
+            if let ParamValue::Float(v) = value {
+                let bin = ((v * n_bins as f64).floor() as usize).min(n_bins - 1);
+                bins[bin] += 1;
+            }
+        }
+
+        let filled_bins = bins.iter().filter(|&&c| c > 0).count();
+        assert!(
+            filled_bins >= 8,
+            "Expected at least 8/10 bins filled, got {filled_bins}: {bins:?}"
+        );
+    }
+
+    #[test]
+    fn multi_parameter_uses_different_dimensions() {
+        // When sampling multiple parameters per trial, each parameter
+        // should get a different Sobol dimension, producing different values.
+        let sampler = SobolSampler::with_seed(0);
+        let dist = Distribution::Float(FloatDistribution {
+            low: 0.0,
+            high: 1.0,
+            log_scale: false,
+            step: None,
+        });
+
+        // Sample two parameters for trial 0.
+        let v1 = sampler.sample(&dist, 0, &[]);
+        let v2 = sampler.sample(&dist, 0, &[]);
+
+        // They should differ (different Sobol dimensions for the same index).
+        assert_ne!(
+            v1, v2,
+            "multi-parameter samples should use different dimensions"
+        );
+    }
+}

src/sampler/tpe/multivariate.rs

@@ -116,14 +116,13 @@ use std::collections::HashMap;
 use std::sync::Arc;
 
 use parking_lot::Mutex;
-use rand::rngs::StdRng;
-use rand::{RngExt, SeedableRng};
 
 use super::{FixedGamma, GammaStrategy};
 use crate::distribution::Distribution;
 use crate::error::Result;
 use crate::param::ParamValue;
 use crate::parameter::ParamId;
+use crate::rng_util;
 use crate::sampler::{CompletedTrial, PendingTrial, Sampler};
 
 /// Strategy for imputing objective values for pending/running trials during parallel optimization.
@@ -189,7 +188,7 @@ pub struct MultivariateTpeSampler {
     /// Strategy for imputing objective values for pending trials in parallel optimization.
     constant_liar: ConstantLiarStrategy,
     /// Thread-safe RNG for sampling.
-    rng: Mutex<StdRng>,
+    rng: Mutex<fastrand::Rng>,
     /// Cache for joint samples to maintain consistency across parameters within the same trial.
     /// The tuple contains (`trial_id`, cached joint sample).
     joint_sample_cache: Mutex<Option<(u64, HashMap<ParamId, ParamValue>)>>,
@@ -219,7 +218,7 @@ impl MultivariateTpeSampler {
             n_ei_candidates: 24,
             group: false,
             constant_liar: ConstantLiarStrategy::None,
-            rng: Mutex::new(rand::make_rng()),
+            rng: Mutex::new(fastrand::Rng::new()),
             joint_sample_cache: Mutex::new(None),
         }
     }
@@ -695,7 +694,7 @@ impl MultivariateTpeSampler {
 
         // Generate candidates from the good distribution
         let candidates: Vec<Vec<f64>> = (0..self.n_ei_candidates)
-            .map(|_| good_kde.sample(&mut *rng))
+            .map(|_| good_kde.sample(&mut rng))
             .collect();
 
         // Compute log(l(x)) - log(g(x)) for each candidate
@@ -731,7 +730,7 @@ impl MultivariateTpeSampler {
         &self,
         good_kde: &crate::kde::MultivariateKDE,
         bad_kde: &crate::kde::MultivariateKDE,
-        rng: &mut StdRng,
+        rng: &mut fastrand::Rng,
     ) -> Vec<f64> {
         // Generate candidates from the good distribution
         let candidates: Vec<Vec<f64>> = (0..self.n_ei_candidates)
@@ -769,7 +768,7 @@ impl MultivariateTpeSampler {
     fn sample_all_uniform(
         &self,
         search_space: &HashMap<ParamId, Distribution>,
-        rng: &mut rand::rngs::StdRng,
+        rng: &mut fastrand::Rng,
     ) -> HashMap<ParamId, ParamValue> {
         search_space
             .iter()
@@ -778,25 +777,22 @@ impl MultivariateTpeSampler {
     }
 
     /// Samples a single parameter uniformly at random from its distribution.
-    fn sample_uniform_single(
-        distribution: &Distribution,
-        rng: &mut rand::rngs::StdRng,
-    ) -> ParamValue {
+    fn sample_uniform_single(distribution: &Distribution, rng: &mut fastrand::Rng) -> ParamValue {
         match distribution {
             Distribution::Float(d) => {
                 let value = if d.log_scale {
                     let log_low = d.low.ln();
                     let log_high = d.high.ln();
-                    rng.random_range(log_low..=log_high).exp()
+                    rng_util::f64_range(rng, log_low, log_high).exp()
                 } else if let Some(step) = d.step {
                     #[allow(clippy::cast_possible_truncation)]
                     let n_steps = ((d.high - d.low) / step).floor() as i64;
-                    let k = rng.random_range(0..=n_steps);
+                    let k = rng.i64(0..=n_steps);
                     #[allow(clippy::cast_precision_loss)]
                     let result = d.low + (k as f64) * step;
                     result
                 } else {
-                    rng.random_range(d.low..=d.high)
+                    rng_util::f64_range(rng, d.low, d.high)
                 };
                 ParamValue::Float(value)
             }
@@ -806,20 +802,20 @@ impl MultivariateTpeSampler {
                     let log_low = (d.low as f64).ln();
                     let log_high = (d.high as f64).ln();
                     #[allow(clippy::cast_possible_truncation)]
-                    let raw = rng.random_range(log_low..=log_high).exp().round() as i64;
+                    let raw = rng_util::f64_range(rng, log_low, log_high).exp().round() as i64;
                     raw.clamp(d.low, d.high)
                 } else if let Some(step) = d.step {
                     #[allow(clippy::cast_possible_truncation)]
                     let n_steps = (d.high - d.low) / step;
-                    let k = rng.random_range(0..=n_steps);
+                    let k = rng.i64(0..=n_steps);
                     d.low + k * step
                 } else {
-                    rng.random_range(d.low..=d.high)
+                    rng.i64(d.low..=d.high)
                 };
                 ParamValue::Int(value)
             }
             Distribution::Categorical(d) => {
-                let index = rng.random_range(0..d.n_choices);
+                let index = rng.usize(0..d.n_choices);
                 ParamValue::Categorical(index)
             }
         }
@@ -1018,7 +1014,7 @@ impl MultivariateTpeSampler {
         &self,
         search_space: &HashMap<ParamId, Distribution>,
         history: &[CompletedTrial],
-        rng: &mut StdRng,
+        rng: &mut fastrand::Rng,
     ) -> HashMap<ParamId, ParamValue> {
         use super::IntersectionSearchSpace;
         use crate::kde::MultivariateKDE;
@@ -1220,7 +1216,7 @@ impl MultivariateTpeSampler {
         _intersection: &HashMap<ParamId, Distribution>,
         history: &[CompletedTrial],
         result: &mut HashMap<ParamId, ParamValue>,
-        rng: &mut StdRng,
+        rng: &mut fastrand::Rng,
     ) {
         // Identify parameters not in result (and not in intersection)
         let missing_params: Vec<(&ParamId, &Distribution)> = search_space
@@ -1253,7 +1249,7 @@ impl MultivariateTpeSampler {
         distribution: &Distribution,
         good_trials: &[&CompletedTrial],
         bad_trials: &[&CompletedTrial],
-        rng: &mut StdRng,
+        rng: &mut fastrand::Rng,
     ) -> ParamValue {
         match distribution {
             Distribution::Float(d) => {
@@ -1370,7 +1366,7 @@ impl MultivariateTpeSampler {
         step: Option<f64>,
         good_values: Vec<f64>,
         bad_values: Vec<f64>,
-        rng: &mut StdRng,
+        rng: &mut fastrand::Rng,
     ) -> f64 {
         use crate::kde::KernelDensityEstimator;
 
@@ -1391,7 +1387,7 @@ impl MultivariateTpeSampler {
 
         // If KDE construction fails, fall back to uniform sampling
         let (Ok(l_kde), Ok(g_kde)) = (l_kde, g_kde) else {
-            return rng.random_range(low..=high);
+            return rng_util::f64_range(rng, low, high);
         };
 
         // Generate candidates from l(x) and select the one with best l(x)/g(x) ratio
@@ -1455,7 +1451,7 @@ impl MultivariateTpeSampler {
         step: Option<i64>,
         good_values: &[i64],
         bad_values: &[i64],
-        rng: &mut StdRng,
+        rng: &mut fastrand::Rng,
     ) -> i64 {
         // Convert to floats for KDE
         let good_floats: Vec<f64> = good_values.iter().map(|&v| v as f64).collect();
@@ -1518,7 +1514,7 @@ impl MultivariateTpeSampler {
         &self,
         search_space: &HashMap<ParamId, Distribution>,
         history: &[CompletedTrial],
-        rng: &mut StdRng,
+        rng: &mut fastrand::Rng,
     ) -> HashMap<ParamId, ParamValue> {
         // Split trials for independent sampling
         let (good_trials, bad_trials) = self.split_trials(&history.iter().collect::<Vec<_>>());
@@ -1562,7 +1558,7 @@ impl MultivariateTpeSampler {
         n_choices: usize,
         good_indices: &[usize],
         bad_indices: &[usize],
-        rng: &mut rand::rngs::StdRng,
+        rng: &mut fastrand::Rng,
     ) -> usize {
         // Count occurrences in good and bad groups
         let mut good_counts = vec![0usize; n_choices];
@@ -1593,7 +1589,7 @@ impl MultivariateTpeSampler {
 
         // Sample proportionally to weights
         let total_weight: f64 = weights.iter().sum();
-        let threshold = rng.random::<f64>() * total_weight;
+        let threshold = rng.f64() * total_weight;
 
         let mut cumulative = 0.0;
         for (i, &w) in weights.iter().enumerate() {
@@ -2069,8 +2065,8 @@ impl MultivariateTpeSamplerBuilder {
         };
 
         let rng = match self.seed {
-            Some(s) => StdRng::seed_from_u64(s),
-            None => rand::make_rng(),
+            Some(s) => fastrand::Rng::with_seed(s),
+            None => fastrand::Rng::new(),
         };
 
         Ok(MultivariateTpeSampler {
@@ -4854,8 +4850,7 @@ mod tests {
 
         #[test]
         fn test_sample_tpe_categorical_basic() {
-            use rand::SeedableRng;
-            let mut rng = rand::rngs::StdRng::seed_from_u64(42);
+            let mut rng = fastrand::Rng::with_seed(42);
 
             // Category 0 is good (appears more in good trials)
             let good_indices = vec![0, 0, 0, 1];
@@ -4890,8 +4885,7 @@ mod tests {
 
         #[test]
         fn test_sample_tpe_categorical_laplace_smoothing() {
-            use rand::SeedableRng;
-            let mut rng = rand::rngs::StdRng::seed_from_u64(42);
+            let mut rng = fastrand::Rng::with_seed(42);
 
             // Category 2 never appears, but should still be sampled due to Laplace smoothing
             let good_indices = vec![0, 0, 1];
@@ -4919,8 +4913,7 @@ mod tests {
 
         #[test]
         fn test_sample_tpe_categorical_empty_good() {
-            use rand::SeedableRng;
-            let mut rng = rand::rngs::StdRng::seed_from_u64(42);
+            let mut rng = fastrand::Rng::with_seed(42);
 
             // Empty good group - all categories should have equal probability
             let good_indices: Vec<usize> = vec![];
@@ -4945,8 +4938,7 @@ mod tests {
 
         #[test]
         fn test_sample_tpe_categorical_all_indices_valid() {
-            use rand::SeedableRng;
-            let mut rng = rand::rngs::StdRng::seed_from_u64(42);
+            let mut rng = fastrand::Rng::with_seed(42);
 
             let n_choices = 4;
             let good_indices = vec![0, 1, 2, 3];

src/sampler/tpe/sampler.rs

@@ -59,13 +59,12 @@ use core::fmt::Debug;
 use std::sync::Arc;
 
 use parking_lot::Mutex;
-use rand::rngs::StdRng;
-use rand::{RngExt, SeedableRng};
 
 use crate::distribution::Distribution;
 use crate::error::{Error, Result};
 use crate::kde::KernelDensityEstimator;
 use crate::param::ParamValue;
+use crate::rng_util;
 use crate::sampler::tpe::gamma::{FixedGamma, GammaStrategy};
 use crate::sampler::{CompletedTrial, Sampler};
 
@@ -131,7 +130,7 @@ pub struct TpeSampler {
     /// Optional fixed bandwidth for KDE. If None, uses Scott's rule.
     kde_bandwidth: Option<f64>,
     /// Thread-safe RNG for sampling.
-    rng: Mutex<StdRng>,
+    rng: Mutex<fastrand::Rng>,
 }
 
 impl TpeSampler {
@@ -149,7 +148,7 @@ impl TpeSampler {
             n_startup_trials: 10,
             n_ei_candidates: 24,
             kde_bandwidth: None,
-            rng: Mutex::new(rand::make_rng()),
+            rng: Mutex::new(fastrand::Rng::new()),
         }
     }
 
@@ -250,8 +249,8 @@ impl TpeSampler {
         }
 
         let rng = match seed {
-            Some(s) => StdRng::seed_from_u64(s),
-            None => rand::make_rng(),
+            Some(s) => fastrand::Rng::with_seed(s),
+            None => fastrand::Rng::new(),
         };
 
         Ok(Self {
@@ -328,19 +327,19 @@ impl TpeSampler {
         clippy::cast_precision_loss,
         clippy::unused_self
     )]
-    fn sample_uniform(&self, distribution: &Distribution, rng: &mut StdRng) -> ParamValue {
+    fn sample_uniform(&self, distribution: &Distribution, rng: &mut fastrand::Rng) -> ParamValue {
         match distribution {
             Distribution::Float(d) => {
                 let value = if d.log_scale {
                     let log_low = d.low.ln();
                     let log_high = d.high.ln();
-                    rng.random_range(log_low..=log_high).exp()
+                    rng_util::f64_range(rng, log_low, log_high).exp()
                 } else if let Some(step) = d.step {
                     let n_steps = ((d.high - d.low) / step).floor() as i64;
-                    let k = rng.random_range(0..=n_steps);
+                    let k = rng.i64(0..=n_steps);
                     d.low + (k as f64) * step
                 } else {
-                    rng.random_range(d.low..=d.high)
+                    rng_util::f64_range(rng, d.low, d.high)
                 };
                 ParamValue::Float(value)
             }
@@ -348,20 +347,18 @@ impl TpeSampler {
                 let value = if d.log_scale {
                     let log_low = (d.low as f64).ln();
                     let log_high = (d.high as f64).ln();
-                    let raw = rng.random_range(log_low..=log_high).exp().round() as i64;
+                    let raw = rng_util::f64_range(rng, log_low, log_high).exp().round() as i64;
                     raw.clamp(d.low, d.high)
                 } else if let Some(step) = d.step {
                     let n_steps = (d.high - d.low) / step;
-                    let k = rng.random_range(0..=n_steps);
+                    let k = rng.i64(0..=n_steps);
                     d.low + k * step
                 } else {
-                    rng.random_range(d.low..=d.high)
+                    rng.i64(d.low..=d.high)
                 };
                 ParamValue::Int(value)
             }
-            Distribution::Categorical(d) => {
-                ParamValue::Categorical(rng.random_range(0..d.n_choices))
-            }
+            Distribution::Categorical(d) => ParamValue::Categorical(rng.usize(0..d.n_choices)),
         }
     }
 
@@ -375,7 +372,7 @@ impl TpeSampler {
         step: Option<f64>,
         good_values: Vec<f64>,
         bad_values: Vec<f64>,
-        rng: &mut StdRng,
+        rng: &mut fastrand::Rng,
     ) -> f64 {
         // Transform to internal space (log space if needed)
         let (internal_low, internal_high, good_internal, bad_internal) = if log_scale {
@@ -400,7 +397,7 @@ impl TpeSampler {
 
         // If KDE construction fails, fall back to uniform sampling
         let (Ok(l_kde), Ok(g_kde)) = (l_kde, g_kde) else {
-            return rng.random_range(low..=high);
+            return rng_util::f64_range(rng, low, high);
         };
 
         // Generate candidates from l(x) and select the one with best l(x)/g(x) ratio
@@ -464,7 +461,7 @@ impl TpeSampler {
         step: Option<i64>,
         good_values: &[i64],
         bad_values: &[i64],
-        rng: &mut StdRng,
+        rng: &mut fastrand::Rng,
     ) -> i64 {
         // Convert to floats for KDE
         let good_floats: Vec<f64> = good_values.iter().map(|&v| v as f64).collect();
@@ -503,7 +500,7 @@ impl TpeSampler {
         n_choices: usize,
         good_indices: &[usize],
         bad_indices: &[usize],
-        rng: &mut StdRng,
+        rng: &mut fastrand::Rng,
     ) -> usize {
         // Count occurrences in good and bad groups
         let mut good_counts = vec![0usize; n_choices];
@@ -534,7 +531,7 @@ impl TpeSampler {
 
         // Sample proportionally to weights
         let total_weight: f64 = weights.iter().sum();
-        let threshold = rng.random::<f64>() * total_weight;
+        let threshold = rng.f64() * total_weight;
 
         let mut cumulative = 0.0;
         for (i, &w) in weights.iter().enumerate() {
@@ -856,8 +853,8 @@ impl TpeSamplerBuilder {
         }
 
         let rng = match self.seed {
-            Some(s) => StdRng::seed_from_u64(s),
-            None => rand::make_rng(),
+            Some(s) => fastrand::Rng::with_seed(s),
+            None => fastrand::Rng::new(),
         };
 
         Ok(TpeSampler {

src/storage/journal.rs

@@ -0,0 +1,171 @@
+//! JSONL-based journal storage backend.
+
+use core::marker::PhantomData;
+use std::fs::{File, OpenOptions};
+use std::io::{BufRead, BufReader, Write};
+use std::path::{Path, PathBuf};
+use std::sync::Arc;
+
+use fs2::FileExt;
+use parking_lot::{Mutex, RwLock};
+use serde::Serialize;
+use serde::de::DeserializeOwned;
+
+use super::{MemoryStorage, Storage};
+use crate::sampler::CompletedTrial;
+
+/// A storage backend that appends completed trials as JSON lines to a file.
+///
+/// Trials are kept in memory for fast read access and simultaneously
+/// persisted to a JSONL file.  Multiple processes can safely share
+/// the same file: writes use an exclusive file lock, reads use a
+/// shared file lock.
+///
+/// The type parameter `V` is the objective value type (typically `f64`).
+/// It must be serializable so that trials can be written to disk.
+///
+/// # Examples
+///
+/// ```no_run
+/// use optimizer::storage::JournalStorage;
+///
+/// let storage: JournalStorage<f64> = JournalStorage::new("trials.jsonl");
+/// ```
+pub struct JournalStorage<V = f64> {
+    memory: MemoryStorage<V>,
+    path: PathBuf,
+    /// Serialise in-process writes so we only hold the file lock briefly.
+    write_lock: Mutex<()>,
+    _marker: PhantomData<V>,
+}
+
+impl<V: Serialize + DeserializeOwned + Send + Sync> JournalStorage<V> {
+    /// Creates a new journal storage that writes to the given path.
+    ///
+    /// The file does not need to exist yet — it will be created on the
+    /// first write.  Existing trials in the file are **not** loaded
+    /// until [`refresh`](Storage::refresh) is called (which happens
+    /// automatically at the start of each trial via the [`Study`](crate::Study)).
+    ///
+    /// To pre-load existing trials, use [`JournalStorage::open`].
+    #[must_use]
+    pub fn new(path: impl AsRef<Path>) -> Self {
+        Self {
+            memory: MemoryStorage::new(),
+            path: path.as_ref().to_path_buf(),
+            write_lock: Mutex::new(()),
+            _marker: PhantomData,
+        }
+    }
+
+    /// Opens an existing journal file and loads all stored trials.
+    ///
+    /// If the file does not exist, returns an empty storage (no error).
+    ///
+    /// # Errors
+    ///
+    /// Returns a [`Storage`](crate::Error::Storage) error if the file
+    /// exists but cannot be read or parsed.
+    pub fn open(path: impl AsRef<Path>) -> crate::Result<Self> {
+        let path = path.as_ref().to_path_buf();
+        let trials = load_trials_from_file(&path)?;
+        Ok(Self {
+            memory: MemoryStorage::with_trials(trials),
+            path,
+            write_lock: Mutex::new(()),
+            _marker: PhantomData,
+        })
+    }
+
+    /// Append a single trial to the JSONL file (best-effort).
+    fn write_to_file(&self, trial: &CompletedTrial<V>) -> crate::Result<()> {
+        let _guard = self.write_lock.lock();
+
+        let mut file = OpenOptions::new()
+            .create(true)
+            .append(true)
+            .open(&self.path)
+            .map_err(|e| crate::Error::Storage(e.to_string()))?;
+
+        file.lock_exclusive()
+            .map_err(|e| crate::Error::Storage(e.to_string()))?;
+
+        let line =
+            serde_json::to_string(trial).map_err(|e| crate::Error::Storage(e.to_string()))?;
+
+        writeln!(file, "{line}").map_err(|e| crate::Error::Storage(e.to_string()))?;
+        file.flush()
+            .map_err(|e| crate::Error::Storage(e.to_string()))?;
+
+        file.unlock()
+            .map_err(|e| crate::Error::Storage(e.to_string()))?;
+
+        Ok(())
+    }
+}
+
+impl<V: Serialize + DeserializeOwned + Send + Sync> Storage<V> for JournalStorage<V> {
+    fn push(&self, trial: CompletedTrial<V>) {
+        // Best-effort persist; the trial stays in memory regardless.
+        let _ = self.write_to_file(&trial);
+        self.memory.push(trial);
+    }
+
+    fn trials_arc(&self) -> &Arc<RwLock<Vec<CompletedTrial<V>>>> {
+        self.memory.trials_arc()
+    }
+
+    fn next_trial_id(&self) -> u64 {
+        self.memory.next_trial_id()
+    }
+
+    fn refresh(&self) -> bool {
+        let Ok(loaded) = load_trials_from_file::<V>(&self.path) else {
+            return false;
+        };
+        let mut guard = self.memory.trials_arc().write();
+        if loaded.len() > guard.len() {
+            if let Some(max_id) = loaded.iter().map(|t| t.id).max() {
+                self.memory.bump_next_id(max_id + 1);
+            }
+            *guard = loaded;
+            true
+        } else {
+            false
+        }
+    }
+}
+
+/// Read all trials from a JSONL file.  Returns an empty vec if the
+/// file does not exist.
+fn load_trials_from_file<V: DeserializeOwned>(
+    path: &Path,
+) -> crate::Result<Vec<CompletedTrial<V>>> {
+    let file = match File::open(path) {
+        Ok(f) => f,
+        Err(e) if e.kind() == std::io::ErrorKind::NotFound => return Ok(Vec::new()),
+        Err(e) => return Err(crate::Error::Storage(e.to_string())),
+    };
+
+    file.lock_shared()
+        .map_err(|e| crate::Error::Storage(e.to_string()))?;
+
+    let reader = BufReader::new(&file);
+    let mut trials = Vec::new();
+
+    for line in reader.lines() {
+        let line = line.map_err(|e| crate::Error::Storage(e.to_string()))?;
+        let line = line.trim();
+        if line.is_empty() {
+            continue;
+        }
+        let trial: CompletedTrial<V> =
+            serde_json::from_str(line).map_err(|e| crate::Error::Storage(e.to_string()))?;
+        trials.push(trial);
+    }
+
+    file.unlock()
+        .map_err(|e| crate::Error::Storage(e.to_string()))?;
+
+    Ok(trials)
+}

src/storage/memory.rs

@@ -0,0 +1,61 @@
+use core::sync::atomic::{AtomicU64, Ordering};
+use std::sync::Arc;
+
+use parking_lot::RwLock;
+
+use super::Storage;
+use crate::sampler::CompletedTrial;
+
+/// In-memory trial storage (the default).
+///
+/// This is a thin wrapper around `Arc<RwLock<Vec<CompletedTrial<V>>>>`.
+pub struct MemoryStorage<V> {
+    trials: Arc<RwLock<Vec<CompletedTrial<V>>>>,
+    next_id: AtomicU64,
+}
+
+impl<V> MemoryStorage<V> {
+    /// Creates a new, empty in-memory store.
+    #[must_use]
+    pub fn new() -> Self {
+        Self {
+            trials: Arc::new(RwLock::new(Vec::new())),
+            next_id: AtomicU64::new(0),
+        }
+    }
+
+    /// Creates an in-memory store pre-populated with `trials`.
+    #[must_use]
+    pub fn with_trials(trials: Vec<CompletedTrial<V>>) -> Self {
+        let next_id = trials.iter().map(|t| t.id).max().map_or(0, |id| id + 1);
+        Self {
+            trials: Arc::new(RwLock::new(trials)),
+            next_id: AtomicU64::new(next_id),
+        }
+    }
+
+    /// Ensures the ID counter is at least `min_value`.
+    pub(crate) fn bump_next_id(&self, min_value: u64) {
+        self.next_id.fetch_max(min_value, Ordering::SeqCst);
+    }
+}
+
+impl<V> Default for MemoryStorage<V> {
+    fn default() -> Self {
+        Self::new()
+    }
+}
+
+impl<V: Send + Sync> Storage<V> for MemoryStorage<V> {
+    fn push(&self, trial: CompletedTrial<V>) {
+        self.trials.write().push(trial);
+    }
+
+    fn trials_arc(&self) -> &Arc<RwLock<Vec<CompletedTrial<V>>>> {
+        &self.trials
+    }
+
+    fn next_trial_id(&self) -> u64 {
+        self.next_id.fetch_add(1, Ordering::SeqCst)
+    }
+}

src/storage/mod.rs

@@ -0,0 +1,54 @@
+//! Trial storage backends.
+//!
+//! The [`Storage`] trait defines how completed trials are stored and
+//! accessed.  [`MemoryStorage`] keeps trials in memory (the default).
+//! With the `journal` feature enabled, [`JournalStorage`] appends
+//! trials to a JSONL file with file-level locking so multiple
+//! processes can safely share state.
+
+#[cfg(feature = "journal")]
+mod journal;
+
+use std::sync::Arc;
+
+#[cfg(feature = "journal")]
+pub use journal::JournalStorage;
+use parking_lot::RwLock;
+
+mod memory;
+pub use memory::MemoryStorage;
+
+use crate::sampler::CompletedTrial;
+
+/// Trait for storing and retrieving completed trials.
+///
+/// Every [`Study`](crate::Study) owns an `Arc<dyn Storage<V>>`.  The
+/// default implementation is [`MemoryStorage`], which keeps trials in
+/// a plain `Vec` behind a read-write lock.
+///
+/// Implementations must be safe to use from multiple threads.
+pub trait Storage<V>: Send + Sync {
+    /// Append a completed trial to the store.
+    fn push(&self, trial: CompletedTrial<V>);
+
+    /// Return a reference to the in-memory trial buffer.
+    ///
+    /// All implementations must maintain an `Arc<RwLock<Vec<…>>>` that
+    /// reflects the current set of trials.  Callers may acquire a read
+    /// lock for efficient, allocation-free access.
+    fn trials_arc(&self) -> &Arc<RwLock<Vec<CompletedTrial<V>>>>;
+
+    /// Atomically returns the next unique trial ID.
+    ///
+    /// Each call increments an internal counter so that consecutive
+    /// calls always produce distinct IDs.
+    fn next_trial_id(&self) -> u64;
+
+    /// Reload from an external source (e.g. a file written by another
+    /// process).  Returns `true` if the in-memory buffer was updated.
+    ///
+    /// The default implementation is a no-op that returns `false`.
+    fn refresh(&self) -> bool {
+        false
+    }
+}

src/trial.rs

@@ -9,9 +9,54 @@ use crate::distribution::Distribution;
 use crate::error::{Error, Result};
 use crate::param::ParamValue;
 use crate::parameter::{ParamId, Parameter};
+use crate::pruner::Pruner;
 use crate::sampler::{CompletedTrial, Sampler};
 use crate::types::TrialState;
 
+/// A user attribute value that can be stored on a trial.
+#[derive(Clone, Debug, PartialEq)]
+#[cfg_attr(feature = "serde", derive(serde::Serialize, serde::Deserialize))]
+pub enum AttrValue {
+    /// A floating-point attribute.
+    Float(f64),
+    /// An integer attribute.
+    Int(i64),
+    /// A string attribute.
+    String(String),
+    /// A boolean attribute.
+    Bool(bool),
+}
+
+impl From<f64> for AttrValue {
+    fn from(v: f64) -> Self {
+        Self::Float(v)
+    }
+}
+
+impl From<i64> for AttrValue {
+    fn from(v: i64) -> Self {
+        Self::Int(v)
+    }
+}
+
+impl From<String> for AttrValue {
+    fn from(v: String) -> Self {
+        Self::String(v)
+    }
+}
+
+impl From<&str> for AttrValue {
+    fn from(v: &str) -> Self {
+        Self::String(v.to_owned())
+    }
+}
+
+impl From<bool> for AttrValue {
+    fn from(v: bool) -> Self {
+        Self::Bool(v)
+    }
+}
+
 /// A trial represents a single evaluation of the objective function.
 ///
 /// Each trial has a unique ID and stores the sampled parameters along with
@@ -36,6 +81,16 @@ pub struct Trial {
     sampler: Option<Arc<dyn Sampler>>,
     /// Access to the history of completed trials (shared with Study).
     history: Option<Arc<RwLock<Vec<CompletedTrial<f64>>>>>,
+    /// Intermediate objective values reported at each step.
+    intermediate_values: Vec<(u64, f64)>,
+    /// The pruner used to decide whether to stop this trial early.
+    pruner: Option<Arc<dyn Pruner>>,
+    /// User-defined attributes for logging, debugging, and analysis.
+    user_attrs: HashMap<String, AttrValue>,
+    /// Pre-filled parameter values from enqueue (used instead of sampling).
+    fixed_params: HashMap<ParamId, ParamValue>,
+    /// Constraint values for this trial (<=0.0 means feasible).
+    constraint_values: Vec<f64>,
 }
 
 impl core::fmt::Debug for Trial {
@@ -48,6 +103,11 @@ impl core::fmt::Debug for Trial {
             .field("param_labels", &self.param_labels)
             .field("has_sampler", &self.sampler.is_some())
             .field("has_history", &self.history.is_some())
+            .field("intermediate_values", &self.intermediate_values)
+            .field("has_pruner", &self.pruner.is_some())
+            .field("user_attrs", &self.user_attrs)
+            .field("fixed_params", &self.fixed_params)
+            .field("constraint_values", &self.constraint_values)
             .finish()
     }
 }
@@ -83,6 +143,11 @@ impl Trial {
             param_labels: HashMap::new(),
             sampler: None,
             history: None,
+            intermediate_values: Vec::new(),
+            pruner: None,
+            user_attrs: HashMap::new(),
+            fixed_params: HashMap::new(),
+            constraint_values: Vec::new(),
         }
     }
 
@@ -100,6 +165,7 @@ impl Trial {
         id: u64,
         sampler: Arc<dyn Sampler>,
         history: Arc<RwLock<Vec<CompletedTrial<f64>>>>,
+        pruner: Arc<dyn Pruner>,
     ) -> Self {
         Self {
             id,
@@ -109,9 +175,22 @@ impl Trial {
             param_labels: HashMap::new(),
             sampler: Some(sampler),
             history: Some(history),
+            intermediate_values: Vec::new(),
+            pruner: Some(pruner),
+            user_attrs: HashMap::new(),
+            fixed_params: HashMap::new(),
+            constraint_values: Vec::new(),
         }
     }
 
+    /// Sets pre-filled parameters on this trial.
+    ///
+    /// When `suggest_param` is called for a parameter that has a fixed value,
+    /// the fixed value is used instead of sampling.
+    pub(crate) fn set_fixed_params(&mut self, params: HashMap<ParamId, ParamValue>) {
+        self.fixed_params = params;
+    }
+
     /// Samples a value from the given distribution using the sampler.
     ///
     /// If the trial has a sampler, it delegates to the sampler's sample method
@@ -159,6 +238,79 @@ impl Trial {
         &self.param_labels
     }
 
+    /// Reports an intermediate objective value at a given step.
+    ///
+    /// Steps should be monotonically increasing (e.g., epoch number).
+    /// Duplicate steps overwrite the previous value.
+    pub fn report(&mut self, step: u64, value: f64) {
+        if let Some(entry) = self
+            .intermediate_values
+            .iter_mut()
+            .find(|(s, _)| *s == step)
+        {
+            entry.1 = value;
+        } else {
+            self.intermediate_values.push((step, value));
+        }
+    }
+
+    /// Ask whether this trial should be pruned at the current step.
+    ///
+    /// Returns `true` if the pruner recommends stopping this trial.
+    /// The caller should return `Err(TrialPruned)` from the objective.
+    #[must_use]
+    pub fn should_prune(&self) -> bool {
+        let (Some(pruner), Some(history)) = (&self.pruner, &self.history) else {
+            return false;
+        };
+        let Some(&(step, _)) = self.intermediate_values.last() else {
+            return false;
+        };
+        let history_guard = history.read();
+        let prune = pruner.should_prune(self.id, step, &self.intermediate_values, &history_guard);
+        if prune {
+            trace_info!(trial_id = self.id, step, "pruner recommends stopping");
+        }
+        prune
+    }
+
+    /// Returns all intermediate values reported so far.
+    #[must_use]
+    pub fn intermediate_values(&self) -> &[(u64, f64)] {
+        &self.intermediate_values
+    }
+
+    /// Sets a user attribute on this trial.
+    pub fn set_user_attr(&mut self, key: impl Into<String>, value: impl Into<AttrValue>) {
+        self.user_attrs.insert(key.into(), value.into());
+    }
+
+    /// Gets a user attribute by key.
+    #[must_use]
+    pub fn user_attr(&self, key: &str) -> Option<&AttrValue> {
+        self.user_attrs.get(key)
+    }
+
+    /// Returns all user attributes.
+    #[must_use]
+    pub fn user_attrs(&self) -> &HashMap<String, AttrValue> {
+        &self.user_attrs
+    }
+
+    /// Sets constraint values for this trial.
+    ///
+    /// Each value represents a constraint; a value <= 0.0 means the constraint
+    /// is satisfied (feasible). A value > 0.0 means the constraint is violated.
+    pub fn set_constraints(&mut self, values: Vec<f64>) {
+        self.constraint_values = values;
+    }
+
+    /// Returns the constraint values for this trial.
+    #[must_use]
+    pub fn constraint_values(&self) -> &[f64] {
+        &self.constraint_values
+    }
+
     /// Sets the trial state to Complete.
     pub(crate) fn set_complete(&mut self) {
         self.state = TrialState::Complete;
@@ -169,6 +321,11 @@ impl Trial {
         self.state = TrialState::Failed;
     }
 
+    /// Sets the trial state to Pruned.
+    pub(crate) fn set_pruned(&mut self) {
+        self.state = TrialState::Pruned;
+    }
+
     /// Suggests a parameter value using a [`Parameter`] definition.
     ///
     /// This is the primary entry point for sampling parameters. It handles
@@ -223,10 +380,23 @@ impl Trial {
             });
         }
 
-        // Sample using the sampler
-        let value = self.sample_value(&distribution);
+        // Check for a pre-filled (enqueued) value for this parameter
+        let value = if let Some(fixed_value) = self.fixed_params.remove(&param_id) {
+            fixed_value
+        } else {
+            // Sample using the sampler
+            self.sample_value(&distribution)
+        };
+
         let result = param.cast_param_value(&value)?;
 
+        trace_debug!(
+            trial_id = self.id,
+            param = %param.label(),
+            value = %value,
+            "parameter sampled"
+        );
+
         // Store distribution, value, and label
         self.distributions.insert(param_id, distribution);
         self.params.insert(param_id, value);

src/types.rs

@@ -2,6 +2,7 @@
 
 /// The direction of optimization.
 #[derive(Clone, Copy, Debug, PartialEq, Eq)]
+#[cfg_attr(feature = "serde", derive(serde::Serialize, serde::Deserialize))]
 pub enum Direction {
     /// Minimize the objective value.
     Minimize,
@@ -11,6 +12,7 @@ pub enum Direction {
 
 /// The state of a trial in its lifecycle.
 #[derive(Clone, Copy, Debug, PartialEq, Eq)]
+#[cfg_attr(feature = "serde", derive(serde::Serialize, serde::Deserialize))]
 pub enum TrialState {
     /// The trial is currently running.
     Running,
@@ -18,4 +20,6 @@ pub enum TrialState {
     Complete,
     /// The trial failed with an error.
     Failed,
+    /// The trial was pruned (stopped early).
+    Pruned,
 }

src/visualization.rs

@@ -0,0 +1,458 @@
+//! HTML report generation for optimization visualization.
+//!
+//! Generates self-contained HTML files with embedded Plotly.js charts
+//! for offline visualization of optimization results.
+
+use core::fmt::Write as _;
+use std::collections::BTreeMap;
+use std::path::Path;
+
+use crate::distribution::Distribution;
+use crate::param::ParamValue;
+use crate::parameter::ParamId;
+use crate::sampler::CompletedTrial;
+use crate::types::{Direction, TrialState};
+
+/// Generate an HTML report with interactive Plotly.js charts.
+///
+/// Creates a self-contained HTML file at `path` containing:
+/// - **Optimization history**: Objective value vs trial number with best-so-far line
+/// - **Slice plots**: Objective value vs each parameter (1D scatter)
+/// - **Parallel coordinates**: Multi-parameter relationship view
+/// - **Trial timeline**: Duration index of each trial (horizontal bar)
+/// - **Intermediate values**: Learning curves per trial (if pruning data available)
+/// - **Parameter importance**: Bar chart (if enough completed trials)
+///
+/// # Errors
+///
+/// Returns an I/O error if the file cannot be created or written.
+pub fn generate_html_report(
+    study: &crate::Study<f64>,
+    path: impl AsRef<Path>,
+) -> std::io::Result<()> {
+    let trials = study.trials();
+    let direction = study.direction();
+    let importance = study.param_importance();
+
+    let html = build_html(&trials, direction, &importance);
+    std::fs::write(path, html)
+}
+
+fn build_html(
+    trials: &[CompletedTrial<f64>],
+    direction: Direction,
+    importance: &[(String, f64)],
+) -> String {
+    let mut html = String::with_capacity(8192);
+
+    let dir_label = match direction {
+        Direction::Minimize => "Minimize",
+        Direction::Maximize => "Maximize",
+    };
+
+    // Collect parameter metadata.
+    let param_info = collect_param_info(trials);
+    let has_intermediate = trials.iter().any(|t| !t.intermediate_values.is_empty());
+
+    let _ = write!(
+        html,
+        r#"<!DOCTYPE html>
+<html lang="en">
+<head>
+<meta charset="utf-8">
+<meta name="viewport" content="width=device-width, initial-scale=1">
+<title>Optimization Report</title>
+<script src="https://cdn.plot.ly/plotly-2.35.2.min.js"></script>
+<style>
+  * {{ margin: 0; padding: 0; box-sizing: border-box; }}
+  body {{ font-family: -apple-system, BlinkMacSystemFont, "Segoe UI", Roboto, sans-serif;
+         background: #f5f6fa; color: #2c3e50; padding: 24px; }}
+  h1 {{ text-align: center; margin-bottom: 8px; font-size: 1.8em; }}
+  .subtitle {{ text-align: center; color: #7f8c8d; margin-bottom: 24px; }}
+  .chart {{ background: #fff; border-radius: 8px; box-shadow: 0 2px 8px rgba(0,0,0,0.08);
+            margin-bottom: 24px; padding: 16px; }}
+  .chart-title {{ font-size: 1.1em; font-weight: 600; margin-bottom: 8px; }}
+</style>
+</head>
+<body>
+<h1>Optimization Report</h1>
+<p class="subtitle">{dir_label} &middot; {n} trials</p>
+"#,
+        n = trials.len(),
+    );
+
+    // Optimization history chart.
+    html.push_str("<div class=\"chart\"><div class=\"chart-title\">Optimization History</div><div id=\"history\"></div></div>\n");
+    write_history_chart(&mut html, trials, direction);
+
+    // Slice plots.
+    if !param_info.is_empty() {
+        html.push_str("<div class=\"chart\"><div class=\"chart-title\">Slice Plots</div><div id=\"slices\"></div></div>\n");
+        write_slice_charts(&mut html, trials, &param_info);
+    }
+
+    // Parallel coordinates.
+    if param_info.len() >= 2 {
+        html.push_str("<div class=\"chart\"><div class=\"chart-title\">Parallel Coordinates</div><div id=\"parcoords\"></div></div>\n");
+        write_parallel_coordinates(&mut html, trials, &param_info, direction);
+    }
+
+    // Parameter importance.
+    if !importance.is_empty() {
+        html.push_str("<div class=\"chart\"><div class=\"chart-title\">Parameter Importance</div><div id=\"importance\"></div></div>\n");
+        write_importance_chart(&mut html, importance);
+    }
+
+    // Trial timeline.
+    html.push_str("<div class=\"chart\"><div class=\"chart-title\">Trial Timeline</div><div id=\"timeline\"></div></div>\n");
+    write_timeline_chart(&mut html, trials);
+
+    // Intermediate values.
+    if has_intermediate {
+        html.push_str("<div class=\"chart\"><div class=\"chart-title\">Intermediate Values</div><div id=\"intermediate\"></div></div>\n");
+        write_intermediate_chart(&mut html, trials);
+    }
+
+    html.push_str("</body>\n</html>\n");
+    html
+}
+
+/// Metadata about each parameter seen across trials.
+struct ParamMeta {
+    label: String,
+    #[allow(dead_code)]
+    dist: Option<Distribution>,
+}
+
+/// Collect parameter labels and distributions across all trials.
+fn collect_param_info(trials: &[CompletedTrial<f64>]) -> BTreeMap<ParamId, ParamMeta> {
+    let mut info: BTreeMap<ParamId, ParamMeta> = BTreeMap::new();
+    for trial in trials {
+        for &id in trial.params.keys() {
+            info.entry(id).or_insert_with(|| {
+                let label = trial
+                    .param_labels
+                    .get(&id)
+                    .cloned()
+                    .unwrap_or_else(|| id.to_string());
+                let dist = trial.distributions.get(&id).cloned();
+                ParamMeta { label, dist }
+            });
+        }
+    }
+    info
+}
+
+// ---------------------------------------------------------------------------
+// Chart generators
+// ---------------------------------------------------------------------------
+
+fn write_history_chart(html: &mut String, trials: &[CompletedTrial<f64>], direction: Direction) {
+    let complete: Vec<_> = trials
+        .iter()
+        .filter(|t| t.state == TrialState::Complete)
+        .collect();
+    if complete.is_empty() {
+        return;
+    }
+
+    let mut ids = Vec::with_capacity(complete.len());
+    let mut vals = Vec::with_capacity(complete.len());
+    let mut best_vals = Vec::with_capacity(complete.len());
+    let mut best = complete[0].value;
+    for t in &complete {
+        ids.push(t.id);
+        vals.push(t.value);
+        best = match direction {
+            Direction::Minimize => best.min(t.value),
+            Direction::Maximize => best.max(t.value),
+        };
+        best_vals.push(best);
+    }
+
+    let _ = write!(
+        html,
+        r##"<script>
+Plotly.newPlot("history", [
+  {{ x: {ids:?}, y: {vals:?}, mode: "markers", name: "Objective", type: "scatter",
+     marker: {{ color: "#3498db", size: 6 }} }},
+  {{ x: {ids:?}, y: {best_vals:?}, mode: "lines", name: "Best so far", type: "scatter",
+     line: {{ color: "#e74c3c", width: 2 }} }}
+], {{ xaxis: {{ title: "Trial" }}, yaxis: {{ title: "Objective Value" }},
+     margin: {{ t: 10 }}, legend: {{ x: 1, xanchor: "right", y: 1 }} }},
+   {{ responsive: true }});
+</script>
+"##,
+    );
+}
+
+fn write_slice_charts(
+    html: &mut String,
+    trials: &[CompletedTrial<f64>],
+    param_info: &BTreeMap<ParamId, ParamMeta>,
+) {
+    let complete: Vec<_> = trials
+        .iter()
+        .filter(|t| t.state == TrialState::Complete)
+        .collect();
+    if complete.is_empty() {
+        return;
+    }
+
+    let n_params = param_info.len();
+    let cols = if n_params <= 2 { n_params } else { 2 };
+    let rows = n_params.div_ceil(cols);
+
+    // Build subplot titles and data.
+    let mut subplot_titles = Vec::new();
+    let mut traces = String::new();
+    for (i, (id, meta)) in param_info.iter().enumerate() {
+        subplot_titles.push(format!("\"{}\"", escape_js(&meta.label)));
+
+        let mut x_vals = Vec::new();
+        let mut y_vals = Vec::new();
+        for t in &complete {
+            if let Some(pv) = t.params.get(id) {
+                x_vals.push(param_value_to_f64(pv));
+                y_vals.push(t.value);
+            }
+        }
+
+        let subplot_idx = i + 1;
+        let xa = if subplot_idx == 1 {
+            "x".to_string()
+        } else {
+            format!("x{subplot_idx}")
+        };
+        let ya = if subplot_idx == 1 {
+            "y".to_string()
+        } else {
+            format!("y{subplot_idx}")
+        };
+        let _ = write!(
+            traces,
+            r##"{{ x: {x_vals:?}, y: {y_vals:?}, mode: "markers", type: "scatter",
+               xaxis: "{xa}", yaxis: "{ya}",
+               marker: {{ color: "#3498db", size: 5 }}, showlegend: false }},"##,
+        );
+    }
+
+    let _ = write!(
+        html,
+        r#"<script>
+Plotly.newPlot("slices", [{traces}],
+  {{ grid: {{ rows: {rows}, columns: {cols}, pattern: "independent" }},
+     annotations: [{annotations}],
+     margin: {{ t: 30 }}, showlegend: false }},
+  {{ responsive: true }});
+</script>
+"#,
+        annotations = build_subplot_annotations(&subplot_titles, rows, cols),
+    );
+}
+
+fn write_parallel_coordinates(
+    html: &mut String,
+    trials: &[CompletedTrial<f64>],
+    param_info: &BTreeMap<ParamId, ParamMeta>,
+    direction: Direction,
+) {
+    let complete: Vec<_> = trials
+        .iter()
+        .filter(|t| t.state == TrialState::Complete)
+        .collect();
+    if complete.is_empty() {
+        return;
+    }
+
+    let mut dimensions = String::new();
+
+    // Add objective value as the first dimension.
+    let obj_vals: Vec<f64> = complete.iter().map(|t| t.value).collect();
+    let _ = write!(
+        dimensions,
+        r#"{{ label: "Objective", values: {obj_vals:?} }},"#,
+    );
+
+    // Add each parameter as a dimension.
+    for (id, meta) in param_info {
+        let vals: Vec<f64> = complete
+            .iter()
+            .map(|t| t.params.get(id).map_or(f64::NAN, param_value_to_f64))
+            .collect();
+        let _ = write!(
+            dimensions,
+            r#"{{ label: "{label}", values: {vals:?} }},"#,
+            label = escape_js(&meta.label),
+        );
+    }
+
+    // Color by objective value: green = good, red = bad.
+    let (cmin, cmax) = min_max(&obj_vals);
+    let colorscale = match direction {
+        Direction::Minimize => r##"[[0,"#2ecc71"],[1,"#e74c3c"]]"##,
+        Direction::Maximize => r##"[[0,"#e74c3c"],[1,"#2ecc71"]]"##,
+    };
+
+    let _ = write!(
+        html,
+        r#"<script>
+Plotly.newPlot("parcoords", [{{
+  type: "parcoords",
+  line: {{ color: {obj_vals:?}, colorscale: {colorscale},
+           cmin: {cmin}, cmax: {cmax}, showscale: true }},
+  dimensions: [{dimensions}]
+}}], {{ margin: {{ t: 10 }} }}, {{ responsive: true }});
+</script>
+"#,
+    );
+}
+
+fn write_importance_chart(html: &mut String, importance: &[(String, f64)]) {
+    let names: Vec<_> = importance.iter().map(|(n, _)| format!("\"{n}\"")).collect();
+    let values: Vec<f64> = importance.iter().map(|(_, v)| *v).collect();
+
+    let _ = write!(
+        html,
+        r##"<script>
+Plotly.newPlot("importance", [{{
+  x: {values:?}, y: [{names}], type: "bar", orientation: "h",
+  marker: {{ color: "#9b59b6" }}
+}}], {{ xaxis: {{ title: "Importance (|Spearman correlation|)" }},
+       yaxis: {{ automargin: true }}, margin: {{ t: 10, l: 120 }} }},
+   {{ responsive: true }});
+</script>
+"##,
+        names = names.join(","),
+    );
+}
+
+fn write_timeline_chart(html: &mut String, trials: &[CompletedTrial<f64>]) {
+    let mut ids = Vec::with_capacity(trials.len());
+    let mut colors = Vec::with_capacity(trials.len());
+    let mut labels = Vec::with_capacity(trials.len());
+
+    for t in trials {
+        ids.push(format!("\"Trial {}\"", t.id));
+        let (color, label) = match t.state {
+            TrialState::Complete => ("#2ecc71", "Complete"),
+            TrialState::Pruned => ("#f39c12", "Pruned"),
+            TrialState::Failed => ("#e74c3c", "Failed"),
+            TrialState::Running => ("#3498db", "Running"),
+        };
+        colors.push(format!("\"{color}\""));
+        labels.push(format!("\"{label}\""));
+    }
+
+    // Use trial index as a proxy for duration (no wallclock data available).
+    let indices: Vec<usize> = (0..trials.len()).collect();
+
+    let _ = write!(
+        html,
+        r#"<script>
+Plotly.newPlot("timeline", [{{
+  y: [{ids}], x: {indices:?}, type: "bar", orientation: "h",
+  text: [{labels}], textposition: "auto",
+  marker: {{ color: [{colors}] }}
+}}], {{ xaxis: {{ title: "Trial Index" }}, yaxis: {{ automargin: true, autorange: "reversed" }},
+       margin: {{ t: 10, l: 80 }}, showlegend: false }},
+   {{ responsive: true }});
+</script>
+"#,
+        ids = ids.join(","),
+        colors = colors.join(","),
+        labels = labels.join(","),
+    );
+}
+
+fn write_intermediate_chart(html: &mut String, trials: &[CompletedTrial<f64>]) {
+    let trials_with_iv: Vec<_> = trials
+        .iter()
+        .filter(|t| !t.intermediate_values.is_empty())
+        .collect();
+    if trials_with_iv.is_empty() {
+        return;
+    }
+
+    let mut traces = String::new();
+    for t in &trials_with_iv {
+        let steps: Vec<u64> = t.intermediate_values.iter().map(|(s, _)| *s).collect();
+        let values: Vec<f64> = t.intermediate_values.iter().map(|(_, v)| *v).collect();
+        let color = match t.state {
+            TrialState::Pruned => "#f39c12",
+            _ => "#3498db",
+        };
+        let _ = write!(
+            traces,
+            r#"{{ x: {steps:?}, y: {values:?}, mode: "lines+markers", name: "Trial {id}",
+               line: {{ color: "{color}", width: 1 }}, marker: {{ size: 3 }} }},"#,
+            id = t.id,
+        );
+    }
+
+    let _ = write!(
+        html,
+        r#"<script>
+Plotly.newPlot("intermediate", [{traces}],
+  {{ xaxis: {{ title: "Step" }}, yaxis: {{ title: "Intermediate Value" }},
+     margin: {{ t: 10 }}, showlegend: true }},
+  {{ responsive: true }});
+</script>
+"#,
+    );
+}
+
+// ---------------------------------------------------------------------------
+// Helpers
+// ---------------------------------------------------------------------------
+
+#[allow(clippy::cast_precision_loss)]
+fn param_value_to_f64(pv: &ParamValue) -> f64 {
+    match *pv {
+        ParamValue::Float(v) => v,
+        ParamValue::Int(v) => v as f64,
+        ParamValue::Categorical(v) => v as f64,
+    }
+}
+
+fn escape_js(s: &str) -> String {
+    s.replace('\\', "\\\\")
+        .replace('"', "\\\"")
+        .replace('\n', "\\n")
+}
+
+fn min_max(vals: &[f64]) -> (f64, f64) {
+    let mut mn = f64::INFINITY;
+    let mut mx = f64::NEG_INFINITY;
+    for &v in vals {
+        if v < mn {
+            mn = v;
+        }
+        if v > mx {
+            mx = v;
+        }
+    }
+    (mn, mx)
+}
+
+/// Build Plotly annotation objects to act as subplot titles.
+#[allow(clippy::cast_precision_loss)]
+fn build_subplot_annotations(titles: &[String], rows: usize, cols: usize) -> String {
+    let mut anns = Vec::new();
+    for (i, title) in titles.iter().enumerate() {
+        let row = i / cols;
+        let col = i % cols;
+        // Compute x/y anchor in paper coordinates.
+        let x = if cols == 1 {
+            0.5
+        } else {
+            (f64::from(u32::try_from(col).unwrap_or(0))) / (cols as f64 - 1.0)
+        };
+        let y = 1.0 - (f64::from(u32::try_from(row).unwrap_or(0))) / (rows as f64).max(1.0) + 0.02;
+        anns.push(format!(
+            r#"{{ text: {title}, x: {x:.3}, y: {y:.3}, xref: "paper", yref: "paper",
+               showarrow: false, font: {{ size: 12 }} }}"#,
+        ));
+    }
+    anns.join(",")
+}

tests/bohb_integration.rs

@@ -0,0 +1,133 @@
+//! Integration tests for the BOHB sampler.
+
+#![allow(
+    clippy::cast_sign_loss,
+    clippy::cast_precision_loss,
+    clippy::cast_possible_truncation
+)]
+
+use optimizer::parameter::{FloatParam, Parameter};
+use optimizer::sampler::bohb::BohbSampler;
+use optimizer::{Direction, Error, Study, TrialPruned};
+
+#[test]
+fn bohb_converges_on_quadratic() {
+    let bohb = BohbSampler::builder()
+        .min_resource(1)
+        .max_resource(9)
+        .reduction_factor(3)
+        .min_points_in_model(5)
+        .seed(42)
+        .build()
+        .unwrap();
+
+    let pruner = bohb.matching_pruner(Direction::Minimize);
+    let study: Study<f64> = Study::with_sampler_and_pruner(Direction::Minimize, bohb, pruner);
+
+    let x_param = FloatParam::new(-10.0, 10.0);
+
+    study
+        .optimize(60, |trial| {
+            let x = x_param.suggest(trial)?;
+
+            // Report intermediate values at budget steps 1, 3, 9
+            let obj = (x - 3.0).powi(2);
+            // Simulate budget-based evaluation with noise decreasing at higher budgets
+            trial.report(1, obj + 5.0);
+            trial.report(3, obj + 1.0);
+            trial.report(9, obj);
+
+            Ok::<_, Error>(obj)
+        })
+        .expect("optimization should succeed");
+
+    let best = study.best_trial().expect("should have trials");
+    assert!(
+        best.value < 10.0,
+        "BOHB should find a reasonable solution, got {}",
+        best.value
+    );
+}
+
+#[test]
+fn bohb_with_pruning() {
+    let bohb = BohbSampler::builder()
+        .min_resource(1)
+        .max_resource(27)
+        .reduction_factor(3)
+        .min_points_in_model(3)
+        .seed(123)
+        .build()
+        .unwrap();
+
+    let pruner = bohb.matching_pruner(Direction::Minimize);
+    let study: Study<f64> = Study::with_sampler_and_pruner(Direction::Minimize, bohb, pruner);
+
+    let x_param = FloatParam::new(-5.0, 5.0);
+
+    study
+        .optimize(40, |trial| {
+            let x = x_param.suggest(trial)?;
+            let obj = x * x;
+
+            // Report at each rung step and check for pruning
+            for &step in &[1u64, 3, 9, 27] {
+                let noisy_obj = obj + 10.0 / step as f64;
+                trial.report(step, noisy_obj);
+
+                if trial.should_prune() {
+                    return Err(TrialPruned.into());
+                }
+            }
+
+            Ok::<_, Error>(obj)
+        })
+        .expect("optimization should succeed");
+
+    // Verify we have completed trials
+    let best = study.best_trial().expect("should have at least one trial");
+    assert!(
+        best.value < 25.0,
+        "best value {} should be reasonable",
+        best.value
+    );
+}
+
+#[test]
+fn bohb_uses_budget_conditioned_history() {
+    // Verify that BOHB conditions on budget level by testing that samples
+    // are influenced by intermediate values, not just final values.
+    let bohb = BohbSampler::builder()
+        .min_resource(1)
+        .max_resource(9)
+        .reduction_factor(3)
+        .min_points_in_model(3)
+        .seed(42)
+        .build()
+        .unwrap();
+
+    let pruner = bohb.matching_pruner(Direction::Minimize);
+    let study: Study<f64> = Study::with_sampler_and_pruner(Direction::Minimize, bohb, pruner);
+
+    let x_param = FloatParam::new(0.0, 10.0);
+
+    study
+        .optimize(30, |trial| {
+            let x = x_param.suggest(trial)?;
+            // Intermediate values that guide optimization toward x=2
+            trial.report(1, (x - 2.0).powi(2) + 1.0);
+            trial.report(3, (x - 2.0).powi(2) + 0.5);
+            trial.report(9, (x - 2.0).powi(2));
+
+            Ok::<_, Error>((x - 2.0).powi(2))
+        })
+        .expect("optimization should succeed");
+
+    let best = study.best_trial().unwrap();
+    let best_x: f64 = best.get(&x_param).unwrap();
+    // Should find x reasonably close to 2.0
+    assert!(
+        (best_x - 2.0).abs() < 5.0,
+        "BOHB should explore near x=2, got x={best_x}"
+    );
+}

tests/cma_es_tests.rs

@@ -0,0 +1,259 @@
+#![cfg(feature = "cma-es")]
+
+use optimizer::prelude::*;
+use optimizer::sampler::cma_es::CmaEsSampler;
+
+#[test]
+fn sphere_function() {
+    let sampler = CmaEsSampler::with_seed(42);
+    let study: Study<f64> = Study::with_sampler(Direction::Minimize, sampler);
+
+    let x = FloatParam::new(-5.0, 5.0).name("x");
+    let y = FloatParam::new(-5.0, 5.0).name("y");
+
+    study
+        .optimize(200, |trial| {
+            let xv = x.suggest(trial)?;
+            let yv = y.suggest(trial)?;
+            Ok::<_, Error>(xv * xv + yv * yv)
+        })
+        .unwrap();
+
+    let best = study.best_trial().unwrap();
+    assert!(
+        best.value < 1.0,
+        "sphere best value should be < 1.0, got {}",
+        best.value
+    );
+}
+
+#[test]
+fn rosenbrock_function() {
+    let sampler = CmaEsSampler::builder().population_size(20).seed(42).build();
+    let study: Study<f64> = Study::with_sampler(Direction::Minimize, sampler);
+
+    let x = FloatParam::new(-5.0, 5.0).name("x");
+    let y = FloatParam::new(-5.0, 5.0).name("y");
+
+    study
+        .optimize(300, |trial| {
+            let xv = x.suggest(trial)?;
+            let yv = y.suggest(trial)?;
+            let val = (1.0 - xv).powi(2) + 100.0 * (yv - xv * xv).powi(2);
+            Ok::<_, Error>(val)
+        })
+        .unwrap();
+
+    let best = study.best_trial().unwrap();
+    // Rosenbrock minimum is 0 at (1, 1); we just check reasonable convergence
+    assert!(
+        best.value < 50.0,
+        "rosenbrock best value should be < 50.0, got {}",
+        best.value
+    );
+}
+
+#[test]
+fn bounds_respected() {
+    let sampler = CmaEsSampler::with_seed(123);
+    let study: Study<f64> = Study::with_sampler(Direction::Minimize, sampler);
+
+    let x = FloatParam::new(-2.0, 3.0).name("x");
+    let y = FloatParam::new(0.0, 10.0).name("y");
+
+    study
+        .optimize(100, |trial| {
+            let xv = x.suggest(trial)?;
+            let yv = y.suggest(trial)?;
+            Ok::<_, Error>(xv + yv)
+        })
+        .unwrap();
+
+    for trial in study.trials() {
+        let xv: f64 = trial.get(&x).unwrap();
+        let yv: f64 = trial.get(&y).unwrap();
+        assert!((-2.0..=3.0).contains(&xv), "x = {xv} out of bounds [-2, 3]");
+        assert!((0.0..=10.0).contains(&yv), "y = {yv} out of bounds [0, 10]");
+    }
+}
+
+#[test]
+fn mixed_params_float_and_categorical() {
+    let sampler = CmaEsSampler::with_seed(42);
+    let study: Study<f64> = Study::with_sampler(Direction::Minimize, sampler);
+
+    let x = FloatParam::new(-5.0, 5.0).name("x");
+    let cat = CategoricalParam::new(vec!["a", "b", "c"]).name("cat");
+
+    study
+        .optimize(50, |trial| {
+            let xv = x.suggest(trial)?;
+            let cv = cat.suggest(trial)?;
+            let penalty = match cv {
+                "a" => 0.0,
+                "b" => 1.0,
+                _ => 2.0,
+            };
+            Ok::<_, Error>(xv * xv + penalty)
+        })
+        .unwrap();
+
+    let best = study.best_trial().unwrap();
+    // Should find a reasonable value
+    assert!(
+        best.value < 10.0,
+        "best value should be < 10.0, got {}",
+        best.value
+    );
+}
+
+#[test]
+fn seeded_reproducibility() {
+    let x = FloatParam::new(-5.0, 5.0).name("x");
+    let y = FloatParam::new(-5.0, 5.0).name("y");
+
+    let run = |seed: u64| {
+        let sampler = CmaEsSampler::with_seed(seed);
+        let study: Study<f64> = Study::with_sampler(Direction::Minimize, sampler);
+        study
+            .optimize(50, |trial| {
+                let xv = x.suggest(trial)?;
+                let yv = y.suggest(trial)?;
+                Ok::<_, Error>(xv * xv + yv * yv)
+            })
+            .unwrap();
+        study.trials().iter().map(|t| t.value).collect::<Vec<_>>()
+    };
+
+    let results1 = run(42);
+    let results2 = run(42);
+    assert_eq!(results1, results2, "same seed should produce same results");
+}
+
+#[test]
+fn different_seeds_different_results() {
+    let x = FloatParam::new(-5.0, 5.0).name("x");
+    let y = FloatParam::new(-5.0, 5.0).name("y");
+
+    let run = |seed: u64| {
+        let sampler = CmaEsSampler::with_seed(seed);
+        let study: Study<f64> = Study::with_sampler(Direction::Minimize, sampler);
+        study
+            .optimize(20, |trial| {
+                let xv = x.suggest(trial)?;
+                let yv = y.suggest(trial)?;
+                Ok::<_, Error>(xv * xv + yv * yv)
+            })
+            .unwrap();
+        study.trials().iter().map(|t| t.value).collect::<Vec<_>>()
+    };
+
+    let results1 = run(42);
+    let results2 = run(99);
+    assert_ne!(
+        results1, results2,
+        "different seeds should produce different results"
+    );
+}
+
+#[test]
+fn single_dimension() {
+    let sampler = CmaEsSampler::with_seed(42);
+    let study: Study<f64> = Study::with_sampler(Direction::Minimize, sampler);
+
+    let x = FloatParam::new(-10.0, 10.0).name("x");
+
+    study
+        .optimize(100, |trial| {
+            let xv = x.suggest(trial)?;
+            Ok::<_, Error>((xv - 3.0).powi(2))
+        })
+        .unwrap();
+
+    let best = study.best_trial().unwrap();
+    assert!(
+        best.value < 1.0,
+        "1-D optimization should converge, got {}",
+        best.value
+    );
+}
+
+#[test]
+fn integer_params() {
+    let sampler = CmaEsSampler::with_seed(42);
+    let study: Study<f64> = Study::with_sampler(Direction::Minimize, sampler);
+
+    let n = IntParam::new(1, 20).name("n");
+
+    study
+        .optimize(100, |trial| {
+            let nv = n.suggest(trial)?;
+            // Minimum at n = 10
+            Ok::<_, Error>(((nv - 10) * (nv - 10)) as f64)
+        })
+        .unwrap();
+
+    let best = study.best_trial().unwrap();
+    let best_n: i64 = best.get(&n).unwrap();
+    assert!(
+        (1..=20).contains(&best_n),
+        "integer value {best_n} out of bounds"
+    );
+    assert!(
+        best.value < 10.0,
+        "integer optimization should converge, got {}",
+        best.value
+    );
+}
+
+#[test]
+fn log_scale_params() {
+    let sampler = CmaEsSampler::with_seed(42);
+    let study: Study<f64> = Study::with_sampler(Direction::Minimize, sampler);
+
+    let lr = FloatParam::new(1e-5, 1.0).log_scale().name("lr");
+
+    study
+        .optimize(100, |trial| {
+            let lrv = lr.suggest(trial)?;
+            // Minimum at lr = 0.01
+            Ok::<_, Error>((lrv.ln() - 0.01_f64.ln()).powi(2))
+        })
+        .unwrap();
+
+    for trial in study.trials() {
+        let lrv: f64 = trial.get(&lr).unwrap();
+        assert!(
+            (1e-5..=1.0).contains(&lrv),
+            "log-scale value {lrv} out of bounds"
+        );
+    }
+}
+
+#[test]
+fn custom_population_size_and_sigma() {
+    let sampler = CmaEsSampler::builder()
+        .sigma0(1.0)
+        .population_size(10)
+        .seed(42)
+        .build();
+    let study: Study<f64> = Study::with_sampler(Direction::Minimize, sampler);
+
+    let x = FloatParam::new(-5.0, 5.0).name("x");
+    let y = FloatParam::new(-5.0, 5.0).name("y");
+
+    study
+        .optimize(100, |trial| {
+            let xv = x.suggest(trial)?;
+            let yv = y.suggest(trial)?;
+            Ok::<_, Error>(xv * xv + yv * yv)
+        })
+        .unwrap();
+
+    let best = study.best_trial().unwrap();
+    assert!(
+        best.value < 5.0,
+        "custom config optimization should work, got {}",
+        best.value
+    );
+}

tests/differential_evolution_tests.rs

@@ -0,0 +1,350 @@
+use optimizer::prelude::*;
+use optimizer::sampler::differential_evolution::{
+    DifferentialEvolutionSampler, DifferentialEvolutionStrategy,
+};
+
+#[test]
+fn sphere_function() {
+    let sampler = DifferentialEvolutionSampler::with_seed(42);
+    let study: Study<f64> = Study::with_sampler(Direction::Minimize, sampler);
+
+    let x = FloatParam::new(-5.0, 5.0).name("x");
+    let y = FloatParam::new(-5.0, 5.0).name("y");
+
+    study
+        .optimize(200, |trial| {
+            let xv = x.suggest(trial)?;
+            let yv = y.suggest(trial)?;
+            Ok::<_, Error>(xv * xv + yv * yv)
+        })
+        .unwrap();
+
+    let best = study.best_trial().unwrap();
+    assert!(
+        best.value < 1.0,
+        "sphere best value should be < 1.0, got {}",
+        best.value
+    );
+}
+
+#[test]
+fn rosenbrock_function() {
+    let sampler = DifferentialEvolutionSampler::builder()
+        .population_size(20)
+        .seed(42)
+        .build();
+    let study: Study<f64> = Study::with_sampler(Direction::Minimize, sampler);
+
+    let x = FloatParam::new(-5.0, 5.0).name("x");
+    let y = FloatParam::new(-5.0, 5.0).name("y");
+
+    study
+        .optimize(400, |trial| {
+            let xv = x.suggest(trial)?;
+            let yv = y.suggest(trial)?;
+            let val = (1.0 - xv).powi(2) + 100.0 * (yv - xv * xv).powi(2);
+            Ok::<_, Error>(val)
+        })
+        .unwrap();
+
+    let best = study.best_trial().unwrap();
+    assert!(
+        best.value < 50.0,
+        "rosenbrock best value should be < 50.0, got {}",
+        best.value
+    );
+}
+
+#[test]
+fn rastrigin_function() {
+    let sampler = DifferentialEvolutionSampler::builder()
+        .population_size(30)
+        .mutation_factor(0.7)
+        .crossover_rate(0.9)
+        .seed(42)
+        .build();
+    let study: Study<f64> = Study::with_sampler(Direction::Minimize, sampler);
+
+    let x = FloatParam::new(-5.12, 5.12).name("x");
+    let y = FloatParam::new(-5.12, 5.12).name("y");
+
+    study
+        .optimize(500, |trial| {
+            let xv = x.suggest(trial)?;
+            let yv = y.suggest(trial)?;
+            let val = 20.0
+                + (xv * xv - 10.0 * (2.0 * std::f64::consts::PI * xv).cos())
+                + (yv * yv - 10.0 * (2.0 * std::f64::consts::PI * yv).cos());
+            Ok::<_, Error>(val)
+        })
+        .unwrap();
+
+    let best = study.best_trial().unwrap();
+    // Rastrigin is multimodal; just check reasonable convergence
+    assert!(
+        best.value < 10.0,
+        "rastrigin best value should be < 10.0, got {}",
+        best.value
+    );
+}
+
+#[test]
+fn bounds_respected() {
+    let sampler = DifferentialEvolutionSampler::with_seed(123);
+    let study: Study<f64> = Study::with_sampler(Direction::Minimize, sampler);
+
+    let x = FloatParam::new(-2.0, 3.0).name("x");
+    let y = FloatParam::new(0.0, 10.0).name("y");
+
+    study
+        .optimize(100, |trial| {
+            let xv = x.suggest(trial)?;
+            let yv = y.suggest(trial)?;
+            Ok::<_, Error>(xv + yv)
+        })
+        .unwrap();
+
+    for trial in study.trials() {
+        let xv: f64 = trial.get(&x).unwrap();
+        let yv: f64 = trial.get(&y).unwrap();
+        assert!((-2.0..=3.0).contains(&xv), "x = {xv} out of bounds [-2, 3]");
+        assert!((0.0..=10.0).contains(&yv), "y = {yv} out of bounds [0, 10]");
+    }
+}
+
+#[test]
+fn strategy_best1() {
+    let sampler = DifferentialEvolutionSampler::builder()
+        .strategy(DifferentialEvolutionStrategy::Best1)
+        .population_size(15)
+        .seed(42)
+        .build();
+    let study: Study<f64> = Study::with_sampler(Direction::Minimize, sampler);
+
+    let x = FloatParam::new(-5.0, 5.0).name("x");
+    let y = FloatParam::new(-5.0, 5.0).name("y");
+
+    study
+        .optimize(200, |trial| {
+            let xv = x.suggest(trial)?;
+            let yv = y.suggest(trial)?;
+            Ok::<_, Error>(xv * xv + yv * yv)
+        })
+        .unwrap();
+
+    let best = study.best_trial().unwrap();
+    assert!(
+        best.value < 5.0,
+        "Best1 strategy should converge, got {}",
+        best.value
+    );
+}
+
+#[test]
+fn strategy_current_to_best1() {
+    let sampler = DifferentialEvolutionSampler::builder()
+        .strategy(DifferentialEvolutionStrategy::CurrentToBest1)
+        .population_size(15)
+        .seed(42)
+        .build();
+    let study: Study<f64> = Study::with_sampler(Direction::Minimize, sampler);
+
+    let x = FloatParam::new(-5.0, 5.0).name("x");
+    let y = FloatParam::new(-5.0, 5.0).name("y");
+
+    study
+        .optimize(200, |trial| {
+            let xv = x.suggest(trial)?;
+            let yv = y.suggest(trial)?;
+            Ok::<_, Error>(xv * xv + yv * yv)
+        })
+        .unwrap();
+
+    let best = study.best_trial().unwrap();
+    assert!(
+        best.value < 5.0,
+        "CurrentToBest1 strategy should converge, got {}",
+        best.value
+    );
+}
+
+#[test]
+fn mixed_params_float_and_categorical() {
+    let sampler = DifferentialEvolutionSampler::with_seed(42);
+    let study: Study<f64> = Study::with_sampler(Direction::Minimize, sampler);
+
+    let x = FloatParam::new(-5.0, 5.0).name("x");
+    let cat = CategoricalParam::new(vec!["a", "b", "c"]).name("cat");
+
+    study
+        .optimize(100, |trial| {
+            let xv = x.suggest(trial)?;
+            let cv = cat.suggest(trial)?;
+            let penalty = match cv {
+                "a" => 0.0,
+                "b" => 1.0,
+                _ => 2.0,
+            };
+            Ok::<_, Error>(xv * xv + penalty)
+        })
+        .unwrap();
+
+    let best = study.best_trial().unwrap();
+    assert!(
+        best.value < 10.0,
+        "best value should be < 10.0, got {}",
+        best.value
+    );
+}
+
+#[test]
+fn seeded_reproducibility() {
+    let x = FloatParam::new(-5.0, 5.0).name("x");
+    let y = FloatParam::new(-5.0, 5.0).name("y");
+
+    let run = |seed: u64| {
+        let sampler = DifferentialEvolutionSampler::with_seed(seed);
+        let study: Study<f64> = Study::with_sampler(Direction::Minimize, sampler);
+        study
+            .optimize(50, |trial| {
+                let xv = x.suggest(trial)?;
+                let yv = y.suggest(trial)?;
+                Ok::<_, Error>(xv * xv + yv * yv)
+            })
+            .unwrap();
+        study.trials().iter().map(|t| t.value).collect::<Vec<_>>()
+    };
+
+    let results1 = run(42);
+    let results2 = run(42);
+    assert_eq!(results1, results2, "same seed should produce same results");
+}
+
+#[test]
+fn different_seeds_different_results() {
+    let x = FloatParam::new(-5.0, 5.0).name("x");
+    let y = FloatParam::new(-5.0, 5.0).name("y");
+
+    let run = |seed: u64| {
+        let sampler = DifferentialEvolutionSampler::with_seed(seed);
+        let study: Study<f64> = Study::with_sampler(Direction::Minimize, sampler);
+        study
+            .optimize(20, |trial| {
+                let xv = x.suggest(trial)?;
+                let yv = y.suggest(trial)?;
+                Ok::<_, Error>(xv * xv + yv * yv)
+            })
+            .unwrap();
+        study.trials().iter().map(|t| t.value).collect::<Vec<_>>()
+    };
+
+    let results1 = run(42);
+    let results2 = run(99);
+    assert_ne!(
+        results1, results2,
+        "different seeds should produce different results"
+    );
+}
+
+#[test]
+fn single_dimension() {
+    let sampler = DifferentialEvolutionSampler::with_seed(42);
+    let study: Study<f64> = Study::with_sampler(Direction::Minimize, sampler);
+
+    let x = FloatParam::new(-10.0, 10.0).name("x");
+
+    study
+        .optimize(100, |trial| {
+            let xv = x.suggest(trial)?;
+            Ok::<_, Error>((xv - 3.0).powi(2))
+        })
+        .unwrap();
+
+    let best = study.best_trial().unwrap();
+    assert!(
+        best.value < 1.0,
+        "1-D optimization should converge, got {}",
+        best.value
+    );
+}
+
+#[test]
+fn integer_params() {
+    let sampler = DifferentialEvolutionSampler::with_seed(42);
+    let study: Study<f64> = Study::with_sampler(Direction::Minimize, sampler);
+
+    let n = IntParam::new(1, 20).name("n");
+
+    study
+        .optimize(100, |trial| {
+            let nv = n.suggest(trial)?;
+            // Minimum at n = 10
+            Ok::<_, Error>(((nv - 10) * (nv - 10)) as f64)
+        })
+        .unwrap();
+
+    let best = study.best_trial().unwrap();
+    let best_n: i64 = best.get(&n).unwrap();
+    assert!(
+        (1..=20).contains(&best_n),
+        "integer value {best_n} out of bounds"
+    );
+    assert!(
+        best.value < 10.0,
+        "integer optimization should converge, got {}",
+        best.value
+    );
+}
+
+#[test]
+fn log_scale_params() {
+    let sampler = DifferentialEvolutionSampler::with_seed(42);
+    let study: Study<f64> = Study::with_sampler(Direction::Minimize, sampler);
+
+    let lr = FloatParam::new(1e-5, 1.0).log_scale().name("lr");
+
+    study
+        .optimize(100, |trial| {
+            let lrv = lr.suggest(trial)?;
+            // Minimum at lr = 0.01
+            Ok::<_, Error>((lrv.ln() - 0.01_f64.ln()).powi(2))
+        })
+        .unwrap();
+
+    for trial in study.trials() {
+        let lrv: f64 = trial.get(&lr).unwrap();
+        assert!(
+            (1e-5..=1.0).contains(&lrv),
+            "log-scale value {lrv} out of bounds"
+        );
+    }
+}
+
+#[test]
+fn custom_mutation_and_crossover() {
+    let sampler = DifferentialEvolutionSampler::builder()
+        .mutation_factor(0.5)
+        .crossover_rate(0.7)
+        .population_size(10)
+        .seed(42)
+        .build();
+    let study: Study<f64> = Study::with_sampler(Direction::Minimize, sampler);
+
+    let x = FloatParam::new(-5.0, 5.0).name("x");
+    let y = FloatParam::new(-5.0, 5.0).name("y");
+
+    study
+        .optimize(100, |trial| {
+            let xv = x.suggest(trial)?;
+            let yv = y.suggest(trial)?;
+            Ok::<_, Error>(xv * xv + yv * yv)
+        })
+        .unwrap();
+
+    let best = study.best_trial().unwrap();
+    assert!(
+        best.value < 5.0,
+        "custom F/CR optimization should work, got {}",
+        best.value
+    );
+}

tests/export_tests.rs

@@ -0,0 +1,230 @@
+use optimizer::parameter::{FloatParam, IntParam, Parameter};
+use optimizer::sampler::random::RandomSampler;
+use optimizer::{Direction, Study};
+
+#[test]
+fn csv_empty_study_produces_header_only() {
+    let study: Study<f64> = Study::new(Direction::Minimize);
+    let mut buf = Vec::new();
+    study.to_csv(&mut buf).unwrap();
+    let csv = String::from_utf8(buf).unwrap();
+    assert_eq!(csv, "trial_id,value,state\n");
+}
+
+#[test]
+fn csv_includes_all_trial_data() {
+    let study: Study<f64> = Study::with_sampler(Direction::Minimize, RandomSampler::with_seed(42));
+    let x = FloatParam::new(0.0, 10.0).name("x");
+    let y = IntParam::new(1, 5).name("y");
+
+    study
+        .optimize(3, |trial| {
+            let xv = x.suggest(trial)?;
+            let yv = y.suggest(trial)?;
+            Ok::<_, optimizer::Error>(xv + yv as f64)
+        })
+        .unwrap();
+
+    let mut buf = Vec::new();
+    study.to_csv(&mut buf).unwrap();
+    let csv = String::from_utf8(buf).unwrap();
+
+    let lines: Vec<&str> = csv.lines().collect();
+    // Header + 3 data rows.
+    assert_eq!(lines.len(), 4);
+
+    // Header should contain our parameter names.
+    let header = lines[0];
+    assert!(header.starts_with("trial_id,value,state"));
+    assert!(header.contains("x"));
+    assert!(header.contains("y"));
+
+    // Each data row should have the right number of columns.
+    let n_cols = header.split(',').count();
+    for line in &lines[1..] {
+        assert_eq!(line.split(',').count(), n_cols);
+    }
+
+    // All rows should have "Complete" state.
+    for line in &lines[1..] {
+        assert!(line.contains("Complete"));
+    }
+}
+
+#[test]
+fn csv_handles_pruned_trials() {
+    let study: Study<f64> = Study::with_sampler(Direction::Minimize, RandomSampler::with_seed(42));
+    let x = FloatParam::new(0.0, 10.0).name("x");
+
+    // First trial: complete
+    let mut trial = study.create_trial();
+    let _ = x.suggest(&mut trial).unwrap();
+    study.complete_trial(trial, 1.0);
+
+    // Second trial: pruned
+    let mut trial = study.create_trial();
+    let _ = x.suggest(&mut trial).unwrap();
+    study.prune_trial(trial);
+
+    let mut buf = Vec::new();
+    study.to_csv(&mut buf).unwrap();
+    let csv = String::from_utf8(buf).unwrap();
+
+    let lines: Vec<&str> = csv.lines().collect();
+    assert_eq!(lines.len(), 3); // header + 2 data rows
+
+    // Pruned trial should have empty value.
+    let pruned_line = lines[2];
+    assert!(pruned_line.contains("Pruned"));
+    // The value field (second column) should be empty.
+    let cols: Vec<&str> = pruned_line.split(',').collect();
+    assert_eq!(cols[2], "Pruned");
+    assert_eq!(cols[1], ""); // empty value for pruned
+}
+
+#[test]
+fn csv_handles_different_parameter_sets() {
+    let study: Study<f64> = Study::with_sampler(Direction::Minimize, RandomSampler::with_seed(42));
+    let x = FloatParam::new(0.0, 10.0).name("x");
+    let y = FloatParam::new(0.0, 10.0).name("y");
+
+    // First trial: only x
+    let mut trial = study.create_trial();
+    let xv = x.suggest(&mut trial).unwrap();
+    study.complete_trial(trial, xv);
+
+    // Second trial: only y
+    let mut trial = study.create_trial();
+    let yv = y.suggest(&mut trial).unwrap();
+    study.complete_trial(trial, yv);
+
+    let mut buf = Vec::new();
+    study.to_csv(&mut buf).unwrap();
+    let csv = String::from_utf8(buf).unwrap();
+
+    let lines: Vec<&str> = csv.lines().collect();
+    assert_eq!(lines.len(), 3);
+
+    // Both x and y columns should exist.
+    let header = lines[0];
+    assert!(header.contains("x"));
+    assert!(header.contains("y"));
+
+    // Each row has the right column count (missing params are empty).
+    let n_cols = header.split(',').count();
+    for line in &lines[1..] {
+        assert_eq!(line.split(',').count(), n_cols);
+    }
+}
+
+#[test]
+fn csv_output_is_parseable() {
+    let study: Study<f64> = Study::with_sampler(Direction::Minimize, RandomSampler::with_seed(42));
+    let lr = FloatParam::new(0.001, 0.1).name("learning_rate");
+    let layers = IntParam::new(1, 5).name("n_layers");
+
+    study
+        .optimize(5, |trial| {
+            let l = lr.suggest(trial)?;
+            let n = layers.suggest(trial)?;
+            Ok::<_, optimizer::Error>(l * n as f64)
+        })
+        .unwrap();
+
+    let mut buf = Vec::new();
+    study.to_csv(&mut buf).unwrap();
+    let csv = String::from_utf8(buf).unwrap();
+
+    // Parse each row: every value field should be a valid f64 for complete trials.
+    let lines: Vec<&str> = csv.lines().collect();
+    for line in &lines[1..] {
+        let cols: Vec<&str> = line.split(',').collect();
+        // trial_id should be a number
+        cols[0].parse::<u64>().unwrap();
+        // value should be parseable as f64
+        cols[1].parse::<f64>().unwrap();
+        // state should be a known value
+        assert!(["Complete", "Pruned", "Failed", "Running"].contains(&cols[2]));
+    }
+}
+
+#[test]
+fn export_csv_writes_file() {
+    let study: Study<f64> = Study::with_sampler(Direction::Minimize, RandomSampler::with_seed(42));
+    let x = FloatParam::new(0.0, 10.0).name("x");
+    study
+        .optimize(3, |trial| {
+            let xv = x.suggest(trial)?;
+            Ok::<_, optimizer::Error>(xv * xv)
+        })
+        .unwrap();
+
+    let dir = std::env::temp_dir().join("optimizer_export_test");
+    std::fs::create_dir_all(&dir).unwrap();
+    let path = dir.join("test_export.csv");
+
+    study.export_csv(&path).unwrap();
+
+    let contents = std::fs::read_to_string(&path).unwrap();
+    assert!(contents.starts_with("trial_id,value,state"));
+    assert!(contents.lines().count() == 4); // header + 3 rows
+
+    // Clean up.
+    let _ = std::fs::remove_dir_all(&dir);
+}
+
+#[cfg(feature = "serde")]
+#[test]
+fn export_json_writes_file() {
+    let study: Study<f64> = Study::with_sampler(Direction::Minimize, RandomSampler::with_seed(42));
+    let x = FloatParam::new(0.0, 10.0).name("x");
+    study
+        .optimize(3, |trial| {
+            let xv = x.suggest(trial)?;
+            Ok::<_, optimizer::Error>(xv * xv)
+        })
+        .unwrap();
+
+    let dir = std::env::temp_dir().join("optimizer_json_export_test");
+    std::fs::create_dir_all(&dir).unwrap();
+    let path = dir.join("test_export.json");
+
+    study.export_json(&path).unwrap();
+
+    let contents = std::fs::read_to_string(&path).unwrap();
+    let parsed: serde_json::Value = serde_json::from_str(&contents).unwrap();
+    let arr = parsed.as_array().unwrap();
+    assert_eq!(arr.len(), 3);
+
+    // Each entry should have the expected fields.
+    for entry in arr {
+        assert!(entry.get("id").is_some());
+        assert!(entry.get("value").is_some());
+        assert!(entry.get("state").is_some());
+        assert!(entry.get("params").is_some());
+    }
+
+    // Clean up.
+    let _ = std::fs::remove_dir_all(&dir);
+}
+
+#[test]
+fn csv_includes_user_attributes() {
+    let study: Study<f64> = Study::with_sampler(Direction::Minimize, RandomSampler::with_seed(42));
+    let x = FloatParam::new(0.0, 10.0).name("x");
+
+    study
+        .optimize(2, |trial| {
+            let xv = x.suggest(trial)?;
+            trial.set_user_attr("training_time_secs", 45.2);
+            Ok::<_, optimizer::Error>(xv * xv)
+        })
+        .unwrap();
+
+    let mut buf = Vec::new();
+    study.to_csv(&mut buf).unwrap();
+    let csv = String::from_utf8(buf).unwrap();
+
+    let header = csv.lines().next().unwrap();
+    assert!(header.contains("training_time_secs"));
+}

tests/fanova_tests.rs

@@ -0,0 +1,86 @@
+//! Integration tests for fANOVA parameter importance.
+
+use optimizer::prelude::*;
+
+#[test]
+fn fanova_dominant_parameter() {
+    // f(x, y) = x^2 — x should dominate
+    let x = FloatParam::new(0.0, 10.0).name("x");
+    let y = FloatParam::new(0.0, 10.0).name("y");
+
+    let study: Study<f64> = Study::with_sampler(Direction::Minimize, RandomSampler::with_seed(42));
+    study
+        .optimize(50, |trial| {
+            let xv = x.suggest(trial)?;
+            let _yv = y.suggest(trial)?;
+            Ok::<_, Error>(xv * xv)
+        })
+        .unwrap();
+
+    let result = study.fanova().unwrap();
+    assert_eq!(result.main_effects[0].0, "x");
+    assert!(
+        result.main_effects[0].1 > 0.7,
+        "x importance = {}",
+        result.main_effects[0].1
+    );
+}
+
+#[test]
+fn fanova_interaction() {
+    // f(x, y) = x * y — both matter and interact
+    let x = FloatParam::new(0.0, 10.0).name("x");
+    let y = FloatParam::new(0.0, 10.0).name("y");
+
+    let study: Study<f64> = Study::with_sampler(Direction::Minimize, RandomSampler::with_seed(7));
+    study
+        .optimize(100, |trial| {
+            let xv = x.suggest(trial)?;
+            let yv = y.suggest(trial)?;
+            Ok::<_, Error>(xv * yv)
+        })
+        .unwrap();
+
+    let config = FanovaConfig {
+        n_trees: 128,
+        ..FanovaConfig::default()
+    };
+    let result = study.fanova_with_config(&config).unwrap();
+
+    // Should detect interaction
+    assert!(
+        !result.interactions.is_empty(),
+        "should detect x*y interaction"
+    );
+}
+
+#[test]
+fn fanova_consistent_with_correlation() {
+    // f(x, y) = 3*x + 0.5*y — x should rank higher in both methods
+    let x = FloatParam::new(0.0, 10.0).name("x");
+    let y = FloatParam::new(0.0, 10.0).name("y");
+
+    let study: Study<f64> = Study::with_sampler(Direction::Minimize, RandomSampler::with_seed(99));
+    study
+        .optimize(80, |trial| {
+            let xv = x.suggest(trial)?;
+            let yv = y.suggest(trial)?;
+            Ok::<_, Error>(3.0 * xv + 0.5 * yv)
+        })
+        .unwrap();
+
+    let corr = study.param_importance();
+    let fanova = study.fanova().unwrap();
+
+    // Both methods should rank x above y
+    assert_eq!(corr[0].0, "x", "correlation should rank x first");
+    assert_eq!(fanova.main_effects[0].0, "x", "fanova should rank x first");
+}
+
+#[test]
+fn fanova_too_few_trials() {
+    let study: Study<f64> = Study::new(Direction::Minimize);
+
+    let result = study.fanova();
+    assert!(result.is_err(), "should error with no trials");
+}

tests/gp_tests.rs

@@ -0,0 +1,256 @@
+#![cfg(feature = "gp")]
+
+use optimizer::prelude::*;
+use optimizer::sampler::gp::GpSampler;
+
+#[test]
+fn sphere_function() {
+    let sampler = GpSampler::with_seed(42);
+    let study: Study<f64> = Study::with_sampler(Direction::Minimize, sampler);
+
+    let x = FloatParam::new(-5.0, 5.0).name("x");
+    let y = FloatParam::new(-5.0, 5.0).name("y");
+
+    study
+        .optimize(80, |trial| {
+            let xv = x.suggest(trial)?;
+            let yv = y.suggest(trial)?;
+            Ok::<_, Error>(xv * xv + yv * yv)
+        })
+        .unwrap();
+
+    let best = study.best_trial().unwrap();
+    assert!(
+        best.value < 2.0,
+        "sphere best value should be < 2.0, got {}",
+        best.value
+    );
+}
+
+#[test]
+fn rosenbrock_function() {
+    let sampler = GpSampler::builder().n_startup_trials(15).seed(42).build();
+    let study: Study<f64> = Study::with_sampler(Direction::Minimize, sampler);
+
+    let x = FloatParam::new(-5.0, 5.0).name("x");
+    let y = FloatParam::new(-5.0, 5.0).name("y");
+
+    study
+        .optimize(100, |trial| {
+            let xv = x.suggest(trial)?;
+            let yv = y.suggest(trial)?;
+            let val = (1.0 - xv).powi(2) + 100.0 * (yv - xv * xv).powi(2);
+            Ok::<_, Error>(val)
+        })
+        .unwrap();
+
+    let best = study.best_trial().unwrap();
+    assert!(
+        best.value < 100.0,
+        "rosenbrock best value should be < 100.0, got {}",
+        best.value
+    );
+}
+
+#[test]
+fn bounds_respected() {
+    let sampler = GpSampler::with_seed(123);
+    let study: Study<f64> = Study::with_sampler(Direction::Minimize, sampler);
+
+    let x = FloatParam::new(-2.0, 3.0).name("x");
+    let y = FloatParam::new(0.0, 10.0).name("y");
+
+    study
+        .optimize(100, |trial| {
+            let xv = x.suggest(trial)?;
+            let yv = y.suggest(trial)?;
+            Ok::<_, Error>(xv + yv)
+        })
+        .unwrap();
+
+    for trial in study.trials() {
+        let xv: f64 = trial.get(&x).unwrap();
+        let yv: f64 = trial.get(&y).unwrap();
+        assert!((-2.0..=3.0).contains(&xv), "x = {xv} out of bounds [-2, 3]");
+        assert!((0.0..=10.0).contains(&yv), "y = {yv} out of bounds [0, 10]");
+    }
+}
+
+#[test]
+fn mixed_params_float_and_categorical() {
+    let sampler = GpSampler::with_seed(42);
+    let study: Study<f64> = Study::with_sampler(Direction::Minimize, sampler);
+
+    let x = FloatParam::new(-5.0, 5.0).name("x");
+    let cat = CategoricalParam::new(vec!["a", "b", "c"]).name("cat");
+
+    study
+        .optimize(50, |trial| {
+            let xv = x.suggest(trial)?;
+            let cv = cat.suggest(trial)?;
+            let penalty = match cv {
+                "a" => 0.0,
+                "b" => 1.0,
+                _ => 2.0,
+            };
+            Ok::<_, Error>(xv * xv + penalty)
+        })
+        .unwrap();
+
+    let best = study.best_trial().unwrap();
+    assert!(
+        best.value < 10.0,
+        "best value should be < 10.0, got {}",
+        best.value
+    );
+}
+
+#[test]
+fn seeded_reproducibility() {
+    let x = FloatParam::new(-5.0, 5.0).name("x");
+    let y = FloatParam::new(-5.0, 5.0).name("y");
+
+    let run = |seed: u64| {
+        let sampler = GpSampler::with_seed(seed);
+        let study: Study<f64> = Study::with_sampler(Direction::Minimize, sampler);
+        study
+            .optimize(50, |trial| {
+                let xv = x.suggest(trial)?;
+                let yv = y.suggest(trial)?;
+                Ok::<_, Error>(xv * xv + yv * yv)
+            })
+            .unwrap();
+        study.trials().iter().map(|t| t.value).collect::<Vec<_>>()
+    };
+
+    let results1 = run(42);
+    let results2 = run(42);
+    assert_eq!(results1, results2, "same seed should produce same results");
+}
+
+#[test]
+fn different_seeds_different_results() {
+    let x = FloatParam::new(-5.0, 5.0).name("x");
+    let y = FloatParam::new(-5.0, 5.0).name("y");
+
+    let run = |seed: u64| {
+        let sampler = GpSampler::with_seed(seed);
+        let study: Study<f64> = Study::with_sampler(Direction::Minimize, sampler);
+        study
+            .optimize(20, |trial| {
+                let xv = x.suggest(trial)?;
+                let yv = y.suggest(trial)?;
+                Ok::<_, Error>(xv * xv + yv * yv)
+            })
+            .unwrap();
+        study.trials().iter().map(|t| t.value).collect::<Vec<_>>()
+    };
+
+    let results1 = run(42);
+    let results2 = run(99);
+    assert_ne!(
+        results1, results2,
+        "different seeds should produce different results"
+    );
+}
+
+#[test]
+fn single_dimension() {
+    let sampler = GpSampler::with_seed(42);
+    let study: Study<f64> = Study::with_sampler(Direction::Minimize, sampler);
+
+    let x = FloatParam::new(-10.0, 10.0).name("x");
+
+    study
+        .optimize(100, |trial| {
+            let xv = x.suggest(trial)?;
+            Ok::<_, Error>((xv - 3.0).powi(2))
+        })
+        .unwrap();
+
+    let best = study.best_trial().unwrap();
+    assert!(
+        best.value < 1.0,
+        "1-D optimization should converge, got {}",
+        best.value
+    );
+}
+
+#[test]
+fn integer_params() {
+    let sampler = GpSampler::with_seed(42);
+    let study: Study<f64> = Study::with_sampler(Direction::Minimize, sampler);
+
+    let n = IntParam::new(1, 20).name("n");
+
+    study
+        .optimize(100, |trial| {
+            let nv = n.suggest(trial)?;
+            Ok::<_, Error>(((nv - 10) * (nv - 10)) as f64)
+        })
+        .unwrap();
+
+    let best = study.best_trial().unwrap();
+    let best_n: i64 = best.get(&n).unwrap();
+    assert!(
+        (1..=20).contains(&best_n),
+        "integer value {best_n} out of bounds"
+    );
+    assert!(
+        best.value < 10.0,
+        "integer optimization should converge, got {}",
+        best.value
+    );
+}
+
+#[test]
+fn log_scale_params() {
+    let sampler = GpSampler::with_seed(42);
+    let study: Study<f64> = Study::with_sampler(Direction::Minimize, sampler);
+
+    let lr = FloatParam::new(1e-5, 1.0).log_scale().name("lr");
+
+    study
+        .optimize(100, |trial| {
+            let lrv = lr.suggest(trial)?;
+            Ok::<_, Error>((lrv.ln() - 0.01_f64.ln()).powi(2))
+        })
+        .unwrap();
+
+    for trial in study.trials() {
+        let lrv: f64 = trial.get(&lr).unwrap();
+        assert!(
+            (1e-5..=1.0).contains(&lrv),
+            "log-scale value {lrv} out of bounds"
+        );
+    }
+}
+
+#[test]
+fn builder_configuration() {
+    let sampler = GpSampler::builder()
+        .n_startup_trials(5)
+        .n_candidates(500)
+        .noise_variance(1e-4)
+        .seed(42)
+        .build();
+    let study: Study<f64> = Study::with_sampler(Direction::Minimize, sampler);
+
+    let x = FloatParam::new(-5.0, 5.0).name("x");
+    let y = FloatParam::new(-5.0, 5.0).name("y");
+
+    study
+        .optimize(100, |trial| {
+            let xv = x.suggest(trial)?;
+            let yv = y.suggest(trial)?;
+            Ok::<_, Error>(xv * xv + yv * yv)
+        })
+        .unwrap();
+
+    let best = study.best_trial().unwrap();
+    assert!(
+        best.value < 5.0,
+        "custom config optimization should work, got {}",
+        best.value
+    );
+}

tests/importance_tests.rs

@@ -0,0 +1,167 @@
+use optimizer::parameter::{CategoricalParam, FloatParam, IntParam, Parameter};
+use optimizer::sampler::random::RandomSampler;
+use optimizer::{Direction, Study};
+
+#[test]
+fn known_perfect_correlation() {
+    let study: Study<f64> = Study::with_sampler(Direction::Minimize, RandomSampler::with_seed(42));
+    let x = FloatParam::new(0.0, 100.0).name("x");
+
+    // Objective = x, so perfect correlation.
+    for _ in 0..30 {
+        let mut trial = study.ask();
+        let xv = x.suggest(&mut trial).unwrap();
+        study.tell(trial, Ok::<_, &str>(xv));
+    }
+
+    let importance = study.param_importance();
+    assert_eq!(importance.len(), 1);
+    assert_eq!(importance[0].0, "x");
+    assert!(
+        (importance[0].1 - 1.0).abs() < 1e-10,
+        "single param should be 1.0"
+    );
+}
+
+#[test]
+fn no_effect_parameter() {
+    let study: Study<f64> = Study::with_sampler(Direction::Minimize, RandomSampler::with_seed(42));
+    let x = FloatParam::new(0.0, 100.0).name("x");
+    let noise = FloatParam::new(0.0, 100.0).name("noise");
+
+    // Objective depends only on x; noise is unused in objective.
+    for _ in 0..50 {
+        let mut trial = study.ask();
+        let xv = x.suggest(&mut trial).unwrap();
+        let _nv = noise.suggest(&mut trial).unwrap();
+        study.tell(trial, Ok::<_, &str>(xv));
+    }
+
+    let importance = study.param_importance();
+    assert_eq!(importance.len(), 2);
+    // x should have much higher importance than noise.
+    let x_score = importance.iter().find(|(l, _)| l == "x").unwrap().1;
+    let noise_score = importance.iter().find(|(l, _)| l == "noise").unwrap().1;
+    assert!(
+        x_score > noise_score,
+        "x ({x_score}) should outrank noise ({noise_score})"
+    );
+    // x should dominate
+    assert!(x_score > 0.7, "x importance {x_score} should be dominant");
+}
+
+#[test]
+fn multiple_parameters_varying_importance() {
+    let study: Study<f64> = Study::with_sampler(Direction::Minimize, RandomSampler::with_seed(42));
+    let x = FloatParam::new(0.0, 10.0).name("x");
+    let y = FloatParam::new(0.0, 10.0).name("y");
+
+    // Objective = 10*x + 0.01*y → x should be far more important.
+    for _ in 0..50 {
+        let mut trial = study.ask();
+        let xv = x.suggest(&mut trial).unwrap();
+        let yv = y.suggest(&mut trial).unwrap();
+        study.tell(trial, Ok::<_, &str>(10.0 * xv + 0.01 * yv));
+    }
+
+    let importance = study.param_importance();
+    assert_eq!(importance.len(), 2);
+    assert_eq!(importance[0].0, "x", "x should rank first");
+    assert!(importance[0].1 > importance[1].1);
+}
+
+#[test]
+fn fewer_than_two_trials_returns_empty() {
+    let study: Study<f64> = Study::new(Direction::Minimize);
+
+    // 0 trials
+    assert!(study.param_importance().is_empty());
+
+    // 1 trial
+    let x = FloatParam::new(0.0, 1.0).name("x");
+    let mut trial = study.ask();
+    let xv = x.suggest(&mut trial).unwrap();
+    study.tell(trial, Ok::<_, &str>(xv));
+    assert!(study.param_importance().is_empty());
+}
+
+#[test]
+fn int_parameter() {
+    let study: Study<f64> = Study::with_sampler(Direction::Minimize, RandomSampler::with_seed(42));
+    let n = IntParam::new(1, 100).name("n");
+
+    for _ in 0..30 {
+        let mut trial = study.ask();
+        let nv = n.suggest(&mut trial).unwrap();
+        study.tell(trial, Ok::<_, &str>(nv as f64));
+    }
+
+    let importance = study.param_importance();
+    assert_eq!(importance.len(), 1);
+    assert_eq!(importance[0].0, "n");
+    assert!(importance[0].1 > 0.9);
+}
+
+#[test]
+fn categorical_parameter() {
+    let study: Study<f64> = Study::with_sampler(Direction::Minimize, RandomSampler::with_seed(42));
+    let cat = CategoricalParam::new(vec!["a", "b", "c"]).name("cat");
+    let x = FloatParam::new(0.0, 100.0).name("x");
+
+    // Objective depends only on x; categorical is random noise.
+    for _ in 0..50 {
+        let mut trial = study.ask();
+        let _c = cat.suggest(&mut trial).unwrap();
+        let xv = x.suggest(&mut trial).unwrap();
+        study.tell(trial, Ok::<_, &str>(xv));
+    }
+
+    let importance = study.param_importance();
+    assert_eq!(importance.len(), 2);
+    let x_score = importance.iter().find(|(l, _)| l == "x").unwrap().1;
+    assert!(x_score > 0.5, "x should dominate over categorical noise");
+}
+
+#[test]
+fn normalization_sums_to_one() {
+    let study: Study<f64> = Study::with_sampler(Direction::Minimize, RandomSampler::with_seed(42));
+    let x = FloatParam::new(0.0, 10.0).name("x");
+    let y = FloatParam::new(0.0, 10.0).name("y");
+    let z = FloatParam::new(0.0, 10.0).name("z");
+
+    for _ in 0..50 {
+        let mut trial = study.ask();
+        let xv = x.suggest(&mut trial).unwrap();
+        let yv = y.suggest(&mut trial).unwrap();
+        let zv = z.suggest(&mut trial).unwrap();
+        study.tell(trial, Ok::<_, &str>(xv + 0.5 * yv + 0.1 * zv));
+    }
+
+    let importance = study.param_importance();
+    let sum: f64 = importance.iter().map(|(_, s)| *s).sum();
+    assert!(
+        (sum - 1.0).abs() < 1e-10,
+        "scores should sum to 1.0, got {sum}"
+    );
+}
+
+#[test]
+fn label_when_unnamed_uses_debug() {
+    let study: Study<f64> = Study::with_sampler(Direction::Minimize, RandomSampler::with_seed(42));
+    // No .name() call → label defaults to Debug representation.
+    let x = FloatParam::new(0.0, 10.0);
+
+    for _ in 0..10 {
+        let mut trial = study.ask();
+        let xv = x.suggest(&mut trial).unwrap();
+        study.tell(trial, Ok::<_, &str>(xv));
+    }
+
+    let importance = study.param_importance();
+    assert_eq!(importance.len(), 1);
+    assert!(
+        importance[0].0.starts_with("FloatParam"),
+        "expected Debug label, got {:?}",
+        importance[0].0
+    );
+}

tests/journal_tests.rs

@@ -0,0 +1,216 @@
+//! Integration tests for the journal storage backend.
+
+use std::collections::HashMap;
+use std::sync::Arc;
+
+use optimizer::parameter::{FloatParam, Parameter};
+use optimizer::sampler::CompletedTrial;
+use optimizer::sampler::random::RandomSampler;
+use optimizer::storage::{JournalStorage, Storage};
+use optimizer::{Direction, Study};
+
+fn temp_path() -> std::path::PathBuf {
+    use std::sync::atomic::{AtomicU64, Ordering};
+    static COUNTER: AtomicU64 = AtomicU64::new(0);
+
+    let mut path = std::env::temp_dir();
+    path.push(format!(
+        "optimizer_journal_test_{}_{}.jsonl",
+        std::process::id(),
+        COUNTER.fetch_add(1, Ordering::Relaxed)
+    ));
+    path
+}
+
+fn sample_trial(id: u64, value: f64) -> CompletedTrial<f64> {
+    CompletedTrial::new(id, HashMap::new(), HashMap::new(), HashMap::new(), value)
+}
+
+#[test]
+fn roundtrip_single_trial() {
+    let path = temp_path();
+    let storage = JournalStorage::new(&path);
+
+    storage.push(sample_trial(0, 42.0));
+
+    let loaded = storage.trials_arc().read().clone();
+    assert_eq!(loaded.len(), 1);
+    assert_eq!(loaded[0].id, 0);
+    assert_eq!(loaded[0].value, 42.0);
+
+    // Also verify via a fresh open from disk
+    let storage2 = JournalStorage::<f64>::open(&path).unwrap();
+    let loaded2 = storage2.trials_arc().read().clone();
+    assert_eq!(loaded2.len(), 1);
+    assert_eq!(loaded2[0].value, 42.0);
+
+    std::fs::remove_file(&path).ok();
+}
+
+#[test]
+fn append_multiple_trials() {
+    let path = temp_path();
+    let storage = JournalStorage::new(&path);
+
+    for i in 0..5 {
+        storage.push(sample_trial(i, i as f64));
+    }
+
+    // Reload from disk
+    let storage2 = JournalStorage::<f64>::open(&path).unwrap();
+    let loaded = storage2.trials_arc().read().clone();
+    assert_eq!(loaded.len(), 5);
+    for (i, trial) in loaded.iter().enumerate() {
+        assert_eq!(trial.id, i as u64);
+        assert_eq!(trial.value, i as f64);
+    }
+
+    std::fs::remove_file(&path).ok();
+}
+
+#[test]
+fn missing_file_returns_empty() {
+    let path = temp_path();
+    let storage = JournalStorage::<f64>::open(&path).unwrap();
+
+    let loaded = storage.trials_arc().read().clone();
+    assert!(loaded.is_empty());
+}
+
+#[test]
+fn concurrent_writes() {
+    let path = temp_path();
+    let storage = Arc::new(JournalStorage::new(&path));
+
+    let mut handles = Vec::new();
+    for thread_id in 0..4u64 {
+        let s = Arc::clone(&storage);
+        handles.push(std::thread::spawn(move || {
+            for i in 0..25u64 {
+                let id = thread_id * 25 + i;
+                s.push(sample_trial(id, id as f64));
+            }
+        }));
+    }
+    for h in handles {
+        h.join().unwrap();
+    }
+
+    // Reload from disk to verify persistence
+    let storage2 = JournalStorage::<f64>::open(&path).unwrap();
+    let loaded = storage2.trials_arc().read().clone();
+    assert_eq!(loaded.len(), 100);
+
+    // Verify all IDs are present (order may vary)
+    let mut ids: Vec<u64> = loaded.iter().map(|t| t.id).collect();
+    ids.sort();
+    assert_eq!(ids, (0..100).collect::<Vec<_>>());
+
+    std::fs::remove_file(&path).ok();
+}
+
+#[test]
+fn study_with_journal_integration() {
+    let path = temp_path();
+    let x = FloatParam::new(-10.0, 10.0);
+
+    // First "process": run some trials
+    {
+        let study =
+            Study::with_journal(Direction::Minimize, RandomSampler::with_seed(1), &path).unwrap();
+        study
+            .optimize(5, |trial| {
+                let val = x.suggest(trial)?;
+                Ok::<_, optimizer::Error>(val * val)
+            })
+            .unwrap();
+        assert_eq!(study.n_trials(), 5);
+    }
+
+    // Second "process": loads the same file, sees existing trials
+    let study2 =
+        Study::with_journal(Direction::Minimize, RandomSampler::with_seed(2), &path).unwrap();
+    assert_eq!(study2.n_trials(), 5);
+
+    // Continue optimizing
+    study2
+        .optimize(5, |trial| {
+            let val = x.suggest(trial)?;
+            Ok::<_, optimizer::Error>(val * val)
+        })
+        .unwrap();
+    assert_eq!(study2.n_trials(), 10);
+
+    // Verify all 10 written to disk
+    let storage3 = JournalStorage::<f64>::open(&path).unwrap();
+    let loaded = storage3.trials_arc().read().clone();
+    assert_eq!(loaded.len(), 10);
+
+    std::fs::remove_file(&path).ok();
+}
+
+#[test]
+fn ids_are_unique_after_reload() {
+    let path = temp_path();
+
+    // First batch
+    {
+        let study =
+            Study::with_journal(Direction::Minimize, RandomSampler::with_seed(1), &path).unwrap();
+        study
+            .optimize(3, |trial| {
+                let _ = FloatParam::new(0.0, 1.0).suggest(trial)?;
+                Ok::<_, optimizer::Error>(1.0)
+            })
+            .unwrap();
+    }
+
+    // Second batch — IDs should continue from 3
+    let study =
+        Study::with_journal(Direction::Minimize, RandomSampler::with_seed(2), &path).unwrap();
+    study
+        .optimize(3, |trial| {
+            let _ = FloatParam::new(0.0, 1.0).suggest(trial)?;
+            Ok::<_, optimizer::Error>(1.0)
+        })
+        .unwrap();
+
+    let all = study.trials();
+    let mut ids: Vec<u64> = all.iter().map(|t| t.id).collect();
+    ids.sort();
+    // All 6 IDs should be unique
+    ids.dedup();
+    assert_eq!(ids.len(), 6);
+
+    std::fs::remove_file(&path).ok();
+}
+
+#[test]
+fn pruned_trials_are_stored() {
+    let path = temp_path();
+    let study =
+        Study::with_journal(Direction::Minimize, RandomSampler::with_seed(1), &path).unwrap();
+
+    // Complete one, prune one
+    let x = FloatParam::new(0.0, 1.0);
+    study
+        .optimize(3, |trial| {
+            let _ = x.suggest(trial)?;
+            if trial.id() == 1 {
+                Err(optimizer::TrialPruned)?;
+            }
+            Ok::<_, optimizer::Error>(1.0)
+        })
+        .unwrap();
+
+    let storage2 = JournalStorage::<f64>::open(&path).unwrap();
+    let loaded = storage2.trials_arc().read().clone();
+    assert_eq!(loaded.len(), 3);
+    assert!(
+        loaded
+            .iter()
+            .any(|t| t.state == optimizer::TrialState::Pruned)
+    );
+
+    std::fs::remove_file(&path).ok();
+}

tests/median_pruner_tests.rs

@@ -0,0 +1,228 @@
+use std::collections::HashMap;
+
+use optimizer::Direction;
+use optimizer::pruner::{MedianPruner, Pruner};
+use optimizer::sampler::CompletedTrial;
+
+/// Helper to build a completed trial with given intermediate values.
+fn trial_with_values(id: u64, intermediate_values: Vec<(u64, f64)>) -> CompletedTrial {
+    CompletedTrial::with_intermediate_values(
+        id,
+        HashMap::new(),
+        HashMap::new(),
+        HashMap::new(),
+        0.0,
+        intermediate_values,
+        HashMap::new(),
+    )
+}
+
+// --- Minimize direction ---
+
+#[test]
+fn prune_when_worse_than_median_minimize() {
+    let pruner = MedianPruner::new(Direction::Minimize);
+    // 3 completed trials with values at step 2: [1.0, 2.0, 3.0] => median = 2.0
+    let completed = vec![
+        trial_with_values(0, vec![(0, 0.5), (1, 0.8), (2, 1.0)]),
+        trial_with_values(1, vec![(0, 0.6), (1, 1.5), (2, 2.0)]),
+        trial_with_values(2, vec![(0, 0.7), (1, 2.0), (2, 3.0)]),
+    ];
+    // Current trial value at step 2 is 2.5 > median 2.0 => prune
+    let current = vec![(0, 0.5), (1, 1.0), (2, 2.5)];
+    assert!(pruner.should_prune(3, 2, &current, &completed));
+}
+
+#[test]
+fn no_prune_when_better_than_median_minimize() {
+    let pruner = MedianPruner::new(Direction::Minimize);
+    let completed = vec![
+        trial_with_values(0, vec![(0, 0.5), (1, 0.8), (2, 1.0)]),
+        trial_with_values(1, vec![(0, 0.6), (1, 1.5), (2, 2.0)]),
+        trial_with_values(2, vec![(0, 0.7), (1, 2.0), (2, 3.0)]),
+    ];
+    // Current trial value at step 2 is 1.5 < median 2.0 => don't prune
+    let current = vec![(0, 0.5), (1, 1.0), (2, 1.5)];
+    assert!(!pruner.should_prune(3, 2, &current, &completed));
+}
+
+// --- Maximize direction ---
+
+#[test]
+fn prune_when_worse_than_median_maximize() {
+    let pruner = MedianPruner::new(Direction::Maximize);
+    // Values at step 1: [5.0, 7.0, 9.0] => median = 7.0
+    let completed = vec![
+        trial_with_values(0, vec![(0, 3.0), (1, 5.0)]),
+        trial_with_values(1, vec![(0, 4.0), (1, 7.0)]),
+        trial_with_values(2, vec![(0, 5.0), (1, 9.0)]),
+    ];
+    // Current value 6.0 < median 7.0 => prune (worse for maximize)
+    let current = vec![(0, 4.0), (1, 6.0)];
+    assert!(pruner.should_prune(3, 1, &current, &completed));
+}
+
+#[test]
+fn no_prune_when_better_than_median_maximize() {
+    let pruner = MedianPruner::new(Direction::Maximize);
+    let completed = vec![
+        trial_with_values(0, vec![(0, 3.0), (1, 5.0)]),
+        trial_with_values(1, vec![(0, 4.0), (1, 7.0)]),
+        trial_with_values(2, vec![(0, 5.0), (1, 9.0)]),
+    ];
+    // Current value 8.0 > median 7.0 => don't prune
+    let current = vec![(0, 4.0), (1, 8.0)];
+    assert!(!pruner.should_prune(3, 1, &current, &completed));
+}
+
+// --- Warmup steps ---
+
+#[test]
+fn no_prune_during_warmup() {
+    let pruner = MedianPruner::new(Direction::Minimize).n_warmup_steps(5);
+    let completed = vec![trial_with_values(0, vec![(0, 1.0), (1, 1.0), (2, 1.0)])];
+    // Step 2 < warmup 5 => never prune, even if value is terrible
+    let current = vec![(0, 100.0), (1, 100.0), (2, 100.0)];
+    assert!(!pruner.should_prune(1, 2, &current, &completed));
+}
+
+#[test]
+fn prune_after_warmup() {
+    let pruner = MedianPruner::new(Direction::Minimize).n_warmup_steps(2);
+    let completed = vec![trial_with_values(0, vec![(0, 1.0), (1, 1.0), (2, 1.0)])];
+    // Step 2 >= warmup 2 => pruning allowed; current 100.0 > median 1.0
+    let current = vec![(0, 100.0), (1, 100.0), (2, 100.0)];
+    assert!(pruner.should_prune(1, 2, &current, &completed));
+}
+
+// --- n_min_trials ---
+
+#[test]
+fn no_prune_when_fewer_than_n_min_trials() {
+    let pruner = MedianPruner::new(Direction::Minimize).n_min_trials(3);
+    // Only 2 completed trials — below threshold of 3
+    let completed = vec![
+        trial_with_values(0, vec![(0, 1.0)]),
+        trial_with_values(1, vec![(0, 2.0)]),
+    ];
+    let current = vec![(0, 100.0)];
+    assert!(!pruner.should_prune(2, 0, &current, &completed));
+}
+
+#[test]
+fn prune_when_at_least_n_min_trials() {
+    let pruner = MedianPruner::new(Direction::Minimize).n_min_trials(3);
+    // 3 completed trials with step 0: [1.0, 2.0, 3.0] => median 2.0
+    let completed = vec![
+        trial_with_values(0, vec![(0, 1.0)]),
+        trial_with_values(1, vec![(0, 2.0)]),
+        trial_with_values(2, vec![(0, 3.0)]),
+    ];
+    // 5.0 > median 2.0 => prune
+    let current = vec![(0, 5.0)];
+    assert!(pruner.should_prune(3, 0, &current, &completed));
+}
+
+// --- No completed trials with values at step ---
+
+#[test]
+fn no_prune_when_no_completed_trials_at_step() {
+    let pruner = MedianPruner::new(Direction::Minimize);
+    // Completed trials only have values at step 0, not step 5
+    let completed = vec![
+        trial_with_values(0, vec![(0, 1.0)]),
+        trial_with_values(1, vec![(0, 2.0)]),
+    ];
+    let current = vec![(0, 0.5), (5, 100.0)];
+    assert!(!pruner.should_prune(2, 5, &current, &completed));
+}
+
+// --- Median calculation edge cases ---
+
+#[test]
+fn correct_median_with_even_number_of_trials() {
+    let pruner = MedianPruner::new(Direction::Minimize);
+    // 4 trials at step 0: [1.0, 2.0, 3.0, 4.0] => median = 2.5
+    let completed = vec![
+        trial_with_values(0, vec![(0, 1.0)]),
+        trial_with_values(1, vec![(0, 2.0)]),
+        trial_with_values(2, vec![(0, 3.0)]),
+        trial_with_values(3, vec![(0, 4.0)]),
+    ];
+    // 2.6 > 2.5 => prune
+    let current = vec![(0, 2.6)];
+    assert!(pruner.should_prune(4, 0, &current, &completed));
+    // 2.4 < 2.5 => don't prune
+    let current = vec![(0, 2.4)];
+    assert!(!pruner.should_prune(4, 0, &current, &completed));
+}
+
+#[test]
+fn correct_median_with_odd_number_of_trials() {
+    let pruner = MedianPruner::new(Direction::Minimize);
+    // 5 trials at step 0: [1.0, 2.0, 3.0, 4.0, 5.0] => median = 3.0
+    let completed = vec![
+        trial_with_values(0, vec![(0, 1.0)]),
+        trial_with_values(1, vec![(0, 2.0)]),
+        trial_with_values(2, vec![(0, 3.0)]),
+        trial_with_values(3, vec![(0, 4.0)]),
+        trial_with_values(4, vec![(0, 5.0)]),
+    ];
+    // 3.5 > 3.0 => prune
+    let current = vec![(0, 3.5)];
+    assert!(pruner.should_prune(5, 0, &current, &completed));
+    // 2.5 < 3.0 => don't prune
+    let current = vec![(0, 2.5)];
+    assert!(!pruner.should_prune(5, 0, &current, &completed));
+}
+
+// --- Non-contiguous step numbers ---
+
+#[test]
+fn works_with_non_contiguous_steps() {
+    let pruner = MedianPruner::new(Direction::Minimize);
+    // Steps are 0, 10, 100 — non-contiguous
+    let completed = vec![
+        trial_with_values(0, vec![(0, 1.0), (10, 2.0), (100, 3.0)]),
+        trial_with_values(1, vec![(0, 1.5), (10, 2.5), (100, 4.0)]),
+        trial_with_values(2, vec![(0, 2.0), (10, 3.0), (100, 5.0)]),
+    ];
+    // At step 100: [3.0, 4.0, 5.0] => median = 4.0
+    let current = vec![(0, 1.0), (10, 2.0), (100, 4.5)];
+    assert!(pruner.should_prune(3, 100, &current, &completed));
+
+    let current = vec![(0, 1.0), (10, 2.0), (100, 3.5)];
+    assert!(!pruner.should_prune(3, 100, &current, &completed));
+}
+
+// --- No intermediate values for current trial ---
+
+#[test]
+fn no_prune_when_no_intermediate_values() {
+    let pruner = MedianPruner::new(Direction::Minimize);
+    let completed = vec![trial_with_values(0, vec![(0, 1.0)])];
+    assert!(!pruner.should_prune(1, 0, &[], &completed));
+}
+
+// --- Pruned trials are excluded from median calculation ---
+
+#[test]
+fn pruned_trials_excluded_from_median() {
+    use optimizer::TrialState;
+
+    let pruner = MedianPruner::new(Direction::Minimize);
+
+    let mut pruned = trial_with_values(0, vec![(0, 0.1)]);
+    pruned.state = TrialState::Pruned;
+
+    // Only the completed trial (value 5.0) counts. Pruned trial (0.1) is excluded.
+    let completed = vec![pruned, trial_with_values(1, vec![(0, 5.0)])];
+
+    // 3.0 < 5.0 => don't prune (only 1 completed trial with median 5.0)
+    let current = vec![(0, 3.0)];
+    assert!(!pruner.should_prune(2, 0, &current, &completed));
+
+    // 6.0 > 5.0 => prune
+    let current = vec![(0, 6.0)];
+    assert!(pruner.should_prune(2, 0, &current, &completed));
+}

tests/multi_objective_tests.rs

@@ -0,0 +1,738 @@
+//! Integration tests for multi-objective optimization.
+
+use optimizer::multi_objective::MultiObjectiveStudy;
+use optimizer::parameter::{CategoricalParam, FloatParam, Parameter};
+use optimizer::sampler::moead::MoeadSampler;
+use optimizer::sampler::nsga2::Nsga2Sampler;
+use optimizer::sampler::nsga3::Nsga3Sampler;
+use optimizer::{Decomposition, Direction};
+
+// ---------------------------------------------------------------------------
+// Pareto utility tests (via public MultiObjectiveStudy)
+// ---------------------------------------------------------------------------
+
+#[test]
+fn test_basic_two_objective_random() {
+    let study = MultiObjectiveStudy::new(vec![Direction::Minimize, Direction::Minimize]);
+    let x = FloatParam::new(0.0, 1.0);
+
+    study
+        .optimize(30, |trial| {
+            let xv = x.suggest(trial)?;
+            Ok::<_, optimizer::Error>(vec![xv, 1.0 - xv])
+        })
+        .unwrap();
+
+    let front = study.pareto_front();
+    assert!(!front.is_empty(), "Pareto front should be non-empty");
+
+    // Verify no solution in the front dominates another
+    for a in &front {
+        for b in &front {
+            if core::ptr::eq(a, b) {
+                continue;
+            }
+            let a_dom_b = a.values[0] <= b.values[0]
+                && a.values[1] <= b.values[1]
+                && (a.values[0] < b.values[0] || a.values[1] < b.values[1]);
+            assert!(
+                !a_dom_b,
+                "Front solution {:?} dominates {:?}",
+                a.values, b.values
+            );
+        }
+    }
+}
+
+#[test]
+fn test_dimension_mismatch_error() {
+    let study = MultiObjectiveStudy::new(vec![Direction::Minimize, Direction::Minimize]);
+    let x = FloatParam::new(0.0, 1.0);
+
+    let result = study.optimize(1, |trial| {
+        let xv = x.suggest(trial)?;
+        // Return wrong number of values
+        Ok::<_, optimizer::Error>(vec![xv])
+    });
+
+    assert!(result.is_err());
+    let err = result.unwrap_err();
+    assert!(
+        matches!(
+            err,
+            optimizer::Error::ObjectiveDimensionMismatch {
+                expected: 2,
+                got: 1
+            }
+        ),
+        "Expected ObjectiveDimensionMismatch, got: {err}"
+    );
+}
+
+#[test]
+fn test_ask_tell() {
+    let study = MultiObjectiveStudy::new(vec![Direction::Minimize, Direction::Maximize]);
+    let x = FloatParam::new(0.0, 10.0);
+
+    for _ in 0..10 {
+        let mut trial = study.ask();
+        let xv = x.suggest(&mut trial).unwrap();
+        study
+            .tell(trial, Ok::<_, &str>(vec![xv, 10.0 - xv]))
+            .unwrap();
+    }
+
+    assert_eq!(study.n_trials(), 10);
+    let front = study.pareto_front();
+    assert!(!front.is_empty());
+}
+
+#[test]
+fn test_ask_tell_dimension_mismatch() {
+    let study = MultiObjectiveStudy::new(vec![Direction::Minimize, Direction::Minimize]);
+    let trial = study.ask();
+    let result = study.tell(trial, Ok::<_, &str>(vec![1.0, 2.0, 3.0]));
+    assert!(result.is_err());
+}
+
+#[test]
+fn test_n_trials_counting() {
+    let study = MultiObjectiveStudy::new(vec![Direction::Minimize, Direction::Minimize]);
+    assert_eq!(study.n_trials(), 0);
+
+    let x = FloatParam::new(0.0, 1.0);
+
+    study
+        .optimize(5, |trial| {
+            let xv = x.suggest(trial)?;
+            Ok::<_, optimizer::Error>(vec![xv, 1.0 - xv])
+        })
+        .unwrap();
+
+    assert_eq!(study.n_trials(), 5);
+}
+
+#[test]
+fn test_three_objectives() {
+    let study = MultiObjectiveStudy::new(vec![
+        Direction::Minimize,
+        Direction::Minimize,
+        Direction::Maximize,
+    ]);
+    let x = FloatParam::new(0.0, 1.0);
+    let y = FloatParam::new(0.0, 1.0);
+
+    study
+        .optimize(30, |trial| {
+            let xv = x.suggest(trial)?;
+            let yv = y.suggest(trial)?;
+            Ok::<_, optimizer::Error>(vec![xv, yv, 1.0 - xv - yv])
+        })
+        .unwrap();
+
+    let front = study.pareto_front();
+    assert!(!front.is_empty());
+    assert_eq!(study.n_objectives(), 3);
+}
+
+#[test]
+fn test_directions_accessor() {
+    let dirs = vec![Direction::Minimize, Direction::Maximize];
+    let study = MultiObjectiveStudy::new(dirs.clone());
+    assert_eq!(study.directions(), &dirs);
+    assert_eq!(study.n_objectives(), 2);
+}
+
+#[test]
+fn test_trials_accessor() {
+    let study = MultiObjectiveStudy::new(vec![Direction::Minimize, Direction::Minimize]);
+    let x = FloatParam::new(0.0, 1.0);
+
+    study
+        .optimize(3, |trial| {
+            let xv = x.suggest(trial)?;
+            Ok::<_, optimizer::Error>(vec![xv, 1.0 - xv])
+        })
+        .unwrap();
+
+    let trials = study.trials();
+    assert_eq!(trials.len(), 3);
+    for t in &trials {
+        assert_eq!(t.values.len(), 2);
+    }
+}
+
+// ---------------------------------------------------------------------------
+// NSGA-II sampler tests
+// ---------------------------------------------------------------------------
+
+#[test]
+fn test_nsga2_zdt1() {
+    // ZDT1 benchmark: minimize both objectives
+    let n_vars = 5;
+    let params: Vec<FloatParam> = (0..n_vars).map(|_| FloatParam::new(0.0, 1.0)).collect();
+
+    let sampler = Nsga2Sampler::builder().population_size(20).seed(42).build();
+    let study =
+        MultiObjectiveStudy::with_sampler(vec![Direction::Minimize, Direction::Minimize], sampler);
+
+    study
+        .optimize(200, |trial| {
+            let xs: Vec<f64> = params
+                .iter()
+                .map(|p| p.suggest(trial))
+                .collect::<Result<_, _>>()?;
+
+            let f1 = xs[0];
+            let g = 1.0 + 9.0 * xs[1..].iter().sum::<f64>() / (n_vars - 1) as f64;
+            let f2 = g * (1.0 - (f1 / g).sqrt());
+            Ok::<_, optimizer::Error>(vec![f1, f2])
+        })
+        .unwrap();
+
+    let front = study.pareto_front();
+    assert!(!front.is_empty(), "Pareto front should be non-empty");
+
+    // Verify no dominated solutions in the front
+    for a in &front {
+        for b in &front {
+            if core::ptr::eq(a, b) {
+                continue;
+            }
+            let a_dom_b = a.values[0] <= b.values[0]
+                && a.values[1] <= b.values[1]
+                && (a.values[0] < b.values[0] || a.values[1] < b.values[1]);
+            assert!(
+                !a_dom_b,
+                "Front solution {:?} dominates {:?}",
+                a.values, b.values
+            );
+        }
+    }
+}
+
+#[test]
+fn test_nsga2_with_seed_reproducible() {
+    let x = FloatParam::new(0.0, 1.0);
+    let y = FloatParam::new(0.0, 1.0);
+
+    let run = |seed: u64| -> Vec<Vec<f64>> {
+        let sampler = Nsga2Sampler::with_seed(seed);
+        let study = MultiObjectiveStudy::with_sampler(
+            vec![Direction::Minimize, Direction::Minimize],
+            sampler,
+        );
+        study
+            .optimize(30, |trial| {
+                let xv = x.suggest(trial)?;
+                let yv = y.suggest(trial)?;
+                Ok::<_, optimizer::Error>(vec![xv, yv])
+            })
+            .unwrap();
+        study.trials().iter().map(|t| t.values.clone()).collect()
+    };
+
+    let r1 = run(123);
+    let r2 = run(123);
+    assert_eq!(r1, r2, "Same seed should produce same results");
+
+    let r3 = run(456);
+    assert_ne!(r1, r3, "Different seeds should produce different results");
+}
+
+#[test]
+fn test_nsga2_builder() {
+    let sampler = Nsga2Sampler::builder()
+        .population_size(10)
+        .crossover_prob(0.8)
+        .crossover_eta(15.0)
+        .mutation_eta(25.0)
+        .seed(42)
+        .build();
+
+    let study =
+        MultiObjectiveStudy::with_sampler(vec![Direction::Minimize, Direction::Minimize], sampler);
+    let x = FloatParam::new(0.0, 1.0);
+
+    study
+        .optimize(30, |trial| {
+            let xv = x.suggest(trial)?;
+            Ok::<_, optimizer::Error>(vec![xv, 1.0 - xv])
+        })
+        .unwrap();
+
+    assert_eq!(study.n_trials(), 30);
+}
+
+#[test]
+fn test_nsga2_categorical_params() {
+    let sampler = Nsga2Sampler::with_seed(42);
+    let study =
+        MultiObjectiveStudy::with_sampler(vec![Direction::Minimize, Direction::Minimize], sampler);
+
+    let x = FloatParam::new(0.0, 1.0);
+    let cat = CategoricalParam::new(vec!["a", "b", "c"]);
+
+    study
+        .optimize(30, |trial| {
+            let xv = x.suggest(trial)?;
+            let cv = cat.suggest(trial)?;
+            let bonus = match cv {
+                "a" => 0.0,
+                "b" => 0.5,
+                _ => 1.0,
+            };
+            Ok::<_, optimizer::Error>(vec![xv + bonus, 1.0 - xv])
+        })
+        .unwrap();
+
+    assert_eq!(study.n_trials(), 30);
+    let front = study.pareto_front();
+    assert!(!front.is_empty());
+}
+
+#[test]
+fn test_nsga2_constraints() {
+    let sampler = Nsga2Sampler::with_seed(42);
+    let study =
+        MultiObjectiveStudy::with_sampler(vec![Direction::Minimize, Direction::Minimize], sampler);
+
+    let x = FloatParam::new(0.0, 1.0);
+
+    study
+        .optimize(50, |trial| {
+            let xv = x.suggest(trial)?;
+            // Constraint: x >= 0.3 (i.e. 0.3 - x <= 0)
+            trial.set_constraints(vec![0.3 - xv]);
+            Ok::<_, optimizer::Error>(vec![xv, 1.0 - xv])
+        })
+        .unwrap();
+
+    let front = study.pareto_front();
+    assert!(!front.is_empty());
+
+    // Check that feasible solutions exist on the front
+    let feasible_count = front.iter().filter(|t| t.is_feasible()).count();
+    assert!(
+        feasible_count > 0,
+        "Should have feasible solutions on front"
+    );
+}
+
+#[test]
+fn test_multi_objective_trial_get() {
+    let study = MultiObjectiveStudy::new(vec![Direction::Minimize, Direction::Minimize]);
+    let x = FloatParam::new(0.0, 10.0).name("x");
+
+    study
+        .optimize(5, |trial| {
+            let xv = x.suggest(trial)?;
+            Ok::<_, optimizer::Error>(vec![xv, 10.0 - xv])
+        })
+        .unwrap();
+
+    let front = study.pareto_front();
+    for t in &front {
+        let xv: f64 = t.get(&x).unwrap();
+        assert!((0.0..=10.0).contains(&xv));
+    }
+}
+
+#[test]
+fn test_multi_objective_trial_is_feasible() {
+    let study = MultiObjectiveStudy::new(vec![Direction::Minimize, Direction::Minimize]);
+    let x = FloatParam::new(0.0, 1.0);
+
+    study
+        .optimize(10, |trial| {
+            let xv = x.suggest(trial)?;
+            trial.set_constraints(vec![0.5 - xv]); // feasible if x >= 0.5
+            Ok::<_, optimizer::Error>(vec![xv, 1.0 - xv])
+        })
+        .unwrap();
+
+    let trials = study.trials();
+    for t in &trials {
+        let xv = t.values[0];
+        if xv >= 0.5 {
+            assert!(t.is_feasible());
+        } else {
+            assert!(!t.is_feasible());
+        }
+    }
+}
+
+#[test]
+fn test_multi_objective_trial_user_attrs() {
+    let study = MultiObjectiveStudy::new(vec![Direction::Minimize, Direction::Minimize]);
+    let x = FloatParam::new(0.0, 1.0);
+
+    study
+        .optimize(3, |trial| {
+            let xv = x.suggest(trial)?;
+            trial.set_user_attr("iteration", 42_i64);
+            Ok::<_, optimizer::Error>(vec![xv, 1.0 - xv])
+        })
+        .unwrap();
+
+    let trials = study.trials();
+    for t in &trials {
+        assert!(t.user_attr("iteration").is_some());
+    }
+}
+
+#[test]
+fn test_tell_with_failure() {
+    let study = MultiObjectiveStudy::new(vec![Direction::Minimize, Direction::Minimize]);
+
+    let trial = study.ask();
+    study
+        .tell(trial, Err::<Vec<f64>, _>("evaluation failed"))
+        .unwrap();
+
+    // Failed trial not counted
+    assert_eq!(study.n_trials(), 0);
+}
+
+// ---------------------------------------------------------------------------
+// NSGA-III sampler tests
+// ---------------------------------------------------------------------------
+
+#[test]
+fn test_nsga3_zdt1() {
+    let n_vars = 5;
+    let params: Vec<FloatParam> = (0..n_vars).map(|_| FloatParam::new(0.0, 1.0)).collect();
+
+    let sampler = Nsga3Sampler::builder().population_size(20).seed(42).build();
+    let study =
+        MultiObjectiveStudy::with_sampler(vec![Direction::Minimize, Direction::Minimize], sampler);
+
+    study
+        .optimize(200, |trial| {
+            let xs: Vec<f64> = params
+                .iter()
+                .map(|p| p.suggest(trial))
+                .collect::<Result<_, _>>()?;
+
+            let f1 = xs[0];
+            let g = 1.0 + 9.0 * xs[1..].iter().sum::<f64>() / (n_vars - 1) as f64;
+            let f2 = g * (1.0 - (f1 / g).sqrt());
+            Ok::<_, optimizer::Error>(vec![f1, f2])
+        })
+        .unwrap();
+
+    let front = study.pareto_front();
+    assert!(
+        !front.is_empty(),
+        "NSGA-III Pareto front should be non-empty"
+    );
+
+    // Verify no dominated solutions in the front
+    for a in &front {
+        for b in &front {
+            if core::ptr::eq(a, b) {
+                continue;
+            }
+            let a_dom_b = a.values[0] <= b.values[0]
+                && a.values[1] <= b.values[1]
+                && (a.values[0] < b.values[0] || a.values[1] < b.values[1]);
+            assert!(
+                !a_dom_b,
+                "Front solution {:?} dominates {:?}",
+                a.values, b.values
+            );
+        }
+    }
+}
+
+#[test]
+fn test_nsga3_four_objectives() {
+    // DTLZ2 with 4 objectives
+    let n_obj = 4;
+    let n_vars = n_obj + 4; // k = 5 decision variables beyond the first (n_obj-1)
+    let params: Vec<FloatParam> = (0..n_vars).map(|_| FloatParam::new(0.0, 1.0)).collect();
+
+    let sampler = Nsga3Sampler::builder().population_size(50).seed(42).build();
+    let directions = vec![Direction::Minimize; n_obj];
+    let study = MultiObjectiveStudy::with_sampler(directions, sampler);
+
+    study
+        .optimize(500, |trial| {
+            let xs: Vec<f64> = params
+                .iter()
+                .map(|p| p.suggest(trial))
+                .collect::<Result<_, _>>()?;
+
+            // DTLZ2 formulation
+            let g: f64 = xs[n_obj - 1..]
+                .iter()
+                .map(|&xi| (xi - 0.5).powi(2))
+                .sum::<f64>();
+
+            let mut objectives = vec![0.0_f64; n_obj];
+            for i in 0..n_obj {
+                let mut f = 1.0 + g;
+                for xj in &xs[..(n_obj - 1 - i)] {
+                    f *= (xj * core::f64::consts::FRAC_PI_2).cos();
+                }
+                if i > 0 {
+                    f *= (xs[n_obj - 1 - i] * core::f64::consts::FRAC_PI_2).sin();
+                }
+                objectives[i] = f;
+            }
+
+            Ok::<_, optimizer::Error>(objectives)
+        })
+        .unwrap();
+
+    let front = study.pareto_front();
+    assert!(!front.is_empty(), "4-objective front should be non-empty");
+    // All front solutions should have 4 objectives
+    for t in &front {
+        assert_eq!(t.values.len(), 4);
+    }
+}
+
+#[test]
+fn test_nsga3_reproducible() {
+    let x = FloatParam::new(0.0, 1.0);
+    let y = FloatParam::new(0.0, 1.0);
+
+    let run = |seed: u64| -> Vec<Vec<f64>> {
+        let sampler = Nsga3Sampler::with_seed(seed);
+        let study = MultiObjectiveStudy::with_sampler(
+            vec![Direction::Minimize, Direction::Minimize],
+            sampler,
+        );
+        study
+            .optimize(30, |trial| {
+                let xv = x.suggest(trial)?;
+                let yv = y.suggest(trial)?;
+                Ok::<_, optimizer::Error>(vec![xv, yv])
+            })
+            .unwrap();
+        study.trials().iter().map(|t| t.values.clone()).collect()
+    };
+
+    let r1 = run(123);
+    let r2 = run(123);
+    assert_eq!(r1, r2, "Same seed should produce same results");
+
+    let r3 = run(456);
+    assert_ne!(r1, r3, "Different seeds should produce different results");
+}
+
+#[test]
+fn test_nsga3_builder() {
+    let sampler = Nsga3Sampler::builder()
+        .population_size(12)
+        .n_divisions(4)
+        .crossover_prob(0.9)
+        .crossover_eta(20.0)
+        .mutation_eta(20.0)
+        .seed(42)
+        .build();
+
+    let study =
+        MultiObjectiveStudy::with_sampler(vec![Direction::Minimize, Direction::Minimize], sampler);
+    let x = FloatParam::new(0.0, 1.0);
+
+    study
+        .optimize(30, |trial| {
+            let xv = x.suggest(trial)?;
+            Ok::<_, optimizer::Error>(vec![xv, 1.0 - xv])
+        })
+        .unwrap();
+
+    assert_eq!(study.n_trials(), 30);
+}
+
+#[test]
+fn test_nsga3_constraints() {
+    let sampler = Nsga3Sampler::with_seed(42);
+    let study =
+        MultiObjectiveStudy::with_sampler(vec![Direction::Minimize, Direction::Minimize], sampler);
+
+    let x = FloatParam::new(0.0, 1.0);
+
+    study
+        .optimize(50, |trial| {
+            let xv = x.suggest(trial)?;
+            trial.set_constraints(vec![0.3 - xv]);
+            Ok::<_, optimizer::Error>(vec![xv, 1.0 - xv])
+        })
+        .unwrap();
+
+    let front = study.pareto_front();
+    assert!(!front.is_empty());
+
+    let feasible_count = front.iter().filter(|t| t.is_feasible()).count();
+    assert!(
+        feasible_count > 0,
+        "Should have feasible solutions on front"
+    );
+}
+
+// ---------------------------------------------------------------------------
+// MOEA/D sampler tests
+// ---------------------------------------------------------------------------
+
+#[test]
+fn test_moead_zdt1_tchebycheff() {
+    let n_vars = 5;
+    let params: Vec<FloatParam> = (0..n_vars).map(|_| FloatParam::new(0.0, 1.0)).collect();
+
+    let sampler = MoeadSampler::builder().population_size(20).seed(42).build();
+    let study =
+        MultiObjectiveStudy::with_sampler(vec![Direction::Minimize, Direction::Minimize], sampler);
+
+    study
+        .optimize(200, |trial| {
+            let xs: Vec<f64> = params
+                .iter()
+                .map(|p| p.suggest(trial))
+                .collect::<Result<_, _>>()?;
+
+            let f1 = xs[0];
+            let g = 1.0 + 9.0 * xs[1..].iter().sum::<f64>() / (n_vars - 1) as f64;
+            let f2 = g * (1.0 - (f1 / g).sqrt());
+            Ok::<_, optimizer::Error>(vec![f1, f2])
+        })
+        .unwrap();
+
+    let front = study.pareto_front();
+    assert!(!front.is_empty(), "MOEA/D Pareto front should be non-empty");
+
+    for a in &front {
+        for b in &front {
+            if core::ptr::eq(a, b) {
+                continue;
+            }
+            let a_dom_b = a.values[0] <= b.values[0]
+                && a.values[1] <= b.values[1]
+                && (a.values[0] < b.values[0] || a.values[1] < b.values[1]);
+            assert!(
+                !a_dom_b,
+                "Front solution {:?} dominates {:?}",
+                a.values, b.values
+            );
+        }
+    }
+}
+
+#[test]
+fn test_moead_zdt1_weighted_sum() {
+    let n_vars = 3;
+    let params: Vec<FloatParam> = (0..n_vars).map(|_| FloatParam::new(0.0, 1.0)).collect();
+
+    let sampler = MoeadSampler::builder()
+        .population_size(20)
+        .decomposition(Decomposition::WeightedSum)
+        .seed(42)
+        .build();
+    let study =
+        MultiObjectiveStudy::with_sampler(vec![Direction::Minimize, Direction::Minimize], sampler);
+
+    study
+        .optimize(200, |trial| {
+            let xs: Vec<f64> = params
+                .iter()
+                .map(|p| p.suggest(trial))
+                .collect::<Result<_, _>>()?;
+
+            let f1 = xs[0];
+            let g = 1.0 + 9.0 * xs[1..].iter().sum::<f64>() / (n_vars - 1) as f64;
+            let f2 = g * (1.0 - (f1 / g).sqrt());
+            Ok::<_, optimizer::Error>(vec![f1, f2])
+        })
+        .unwrap();
+
+    let front = study.pareto_front();
+    assert!(!front.is_empty());
+}
+
+#[test]
+fn test_moead_zdt1_pbi() {
+    let n_vars = 3;
+    let params: Vec<FloatParam> = (0..n_vars).map(|_| FloatParam::new(0.0, 1.0)).collect();
+
+    let sampler = MoeadSampler::builder()
+        .population_size(20)
+        .decomposition(Decomposition::Pbi { theta: 5.0 })
+        .seed(42)
+        .build();
+    let study =
+        MultiObjectiveStudy::with_sampler(vec![Direction::Minimize, Direction::Minimize], sampler);
+
+    study
+        .optimize(200, |trial| {
+            let xs: Vec<f64> = params
+                .iter()
+                .map(|p| p.suggest(trial))
+                .collect::<Result<_, _>>()?;
+
+            let f1 = xs[0];
+            let g = 1.0 + 9.0 * xs[1..].iter().sum::<f64>() / (n_vars - 1) as f64;
+            let f2 = g * (1.0 - (f1 / g).sqrt());
+            Ok::<_, optimizer::Error>(vec![f1, f2])
+        })
+        .unwrap();
+
+    let front = study.pareto_front();
+    assert!(!front.is_empty());
+}
+
+#[test]
+fn test_moead_reproducible() {
+    let x = FloatParam::new(0.0, 1.0);
+    let y = FloatParam::new(0.0, 1.0);
+
+    let run = |seed: u64| -> Vec<Vec<f64>> {
+        let sampler = MoeadSampler::with_seed(seed);
+        let study = MultiObjectiveStudy::with_sampler(
+            vec![Direction::Minimize, Direction::Minimize],
+            sampler,
+        );
+        study
+            .optimize(30, |trial| {
+                let xv = x.suggest(trial)?;
+                let yv = y.suggest(trial)?;
+                Ok::<_, optimizer::Error>(vec![xv, yv])
+            })
+            .unwrap();
+        study.trials().iter().map(|t| t.values.clone()).collect()
+    };
+
+    let r1 = run(123);
+    let r2 = run(123);
+    assert_eq!(r1, r2, "Same seed should produce same results");
+
+    let r3 = run(456);
+    assert_ne!(r1, r3, "Different seeds should produce different results");
+}
+
+#[test]
+fn test_moead_builder() {
+    let sampler = MoeadSampler::builder()
+        .population_size(15)
+        .neighborhood_size(5)
+        .decomposition(Decomposition::Tchebycheff)
+        .crossover_prob(0.9)
+        .crossover_eta(20.0)
+        .mutation_eta(20.0)
+        .seed(42)
+        .build();
+
+    let study =
+        MultiObjectiveStudy::with_sampler(vec![Direction::Minimize, Direction::Minimize], sampler);
+    let x = FloatParam::new(0.0, 1.0);
+
+    study
+        .optimize(30, |trial| {
+            let xv = x.suggest(trial)?;
+            Ok::<_, optimizer::Error>(vec![xv, 1.0 - xv])
+        })
+        .unwrap();
+
+    assert_eq!(study.n_trials(), 30);
+}

tests/serde_tests.rs

@@ -0,0 +1,295 @@
+#![cfg(feature = "serde")]
+
+use std::collections::HashMap;
+
+use optimizer::parameter::{FloatParam, IntParam, Parameter};
+use optimizer::sampler::CompletedTrial;
+use optimizer::{Direction, ParamValue, Study, StudySnapshot, TrialState};
+
+#[test]
+fn round_trip_save_load() {
+    let study: Study<f64> = Study::new(Direction::Minimize);
+    let x = FloatParam::new(-10.0, 10.0).name("x");
+    let n = IntParam::new(1, 100).name("n");
+
+    study
+        .optimize(5, |trial| {
+            let x_val = x.suggest(trial)?;
+            let n_val = n.suggest(trial)?;
+            Ok::<_, optimizer::Error>(x_val * x_val + n_val as f64)
+        })
+        .unwrap();
+
+    let dir = tempdir();
+    let path = dir.join("study.json");
+
+    study.save(&path).unwrap();
+    let loaded: Study<f64> = Study::load(&path).unwrap();
+
+    assert_eq!(loaded.direction(), study.direction());
+    assert_eq!(loaded.n_trials(), study.n_trials());
+
+    let orig_trials = study.trials();
+    let loaded_trials = loaded.trials();
+
+    for (orig, loaded) in orig_trials.iter().zip(loaded_trials.iter()) {
+        assert_eq!(orig.id, loaded.id);
+        assert!((orig.value - loaded.value).abs() < 1e-10);
+        assert_eq!(orig.state, loaded.state);
+        assert_eq!(orig.params.len(), loaded.params.len());
+        assert_eq!(orig.distributions, loaded.distributions);
+        assert_eq!(orig.param_labels, loaded.param_labels);
+    }
+
+    // Clean up
+    std::fs::remove_dir_all(&dir).ok();
+}
+
+#[test]
+fn json_output_is_human_readable() {
+    let study: Study<f64> = Study::new(Direction::Maximize);
+    let x = FloatParam::new(0.0, 1.0).name("x");
+
+    study
+        .optimize(2, |trial| {
+            let v = x.suggest(trial)?;
+            Ok::<_, optimizer::Error>(v)
+        })
+        .unwrap();
+
+    let dir = tempdir();
+    let path = dir.join("study.json");
+    study.save(&path).unwrap();
+
+    let contents = std::fs::read_to_string(&path).unwrap();
+
+    // Verify it's pretty-printed JSON with recognizable fields
+    assert!(contents.contains("\"version\""));
+    assert!(contents.contains("\"direction\""));
+    assert!(contents.contains("\"trials\""));
+    assert!(contents.contains("\"next_trial_id\""));
+    assert!(contents.contains("\"Maximize\""));
+
+    std::fs::remove_dir_all(&dir).ok();
+}
+
+#[test]
+fn round_trip_empty_study() {
+    let study: Study<f64> = Study::new(Direction::Minimize);
+
+    let dir = tempdir();
+    let path = dir.join("empty.json");
+
+    study.save(&path).unwrap();
+    let loaded: Study<f64> = Study::load(&path).unwrap();
+
+    assert_eq!(loaded.direction(), Direction::Minimize);
+    assert_eq!(loaded.n_trials(), 0);
+
+    std::fs::remove_dir_all(&dir).ok();
+}
+
+#[test]
+fn snapshot_version_field_is_present() {
+    let study: Study<f64> = Study::new(Direction::Minimize);
+
+    let dir = tempdir();
+    let path = dir.join("version.json");
+    study.save(&path).unwrap();
+
+    let contents = std::fs::read_to_string(&path).unwrap();
+    let snapshot: StudySnapshot<f64> = serde_json::from_str(&contents).unwrap();
+
+    assert_eq!(snapshot.version, 1);
+
+    std::fs::remove_dir_all(&dir).ok();
+}
+
+#[test]
+fn completed_trial_serde_round_trip() {
+    let trial = CompletedTrial::new(42, HashMap::new(), HashMap::new(), HashMap::new(), 2.78);
+
+    let json = serde_json::to_string(&trial).unwrap();
+    let deserialized: CompletedTrial<f64> = serde_json::from_str(&json).unwrap();
+
+    assert_eq!(deserialized.id, 42);
+    assert_eq!(deserialized.value, 2.78);
+    assert_eq!(deserialized.state, TrialState::Complete);
+}
+
+#[test]
+fn param_value_serde_round_trip() {
+    let values = vec![
+        ParamValue::Float(1.23),
+        ParamValue::Int(42),
+        ParamValue::Categorical(2),
+    ];
+
+    for val in &values {
+        let json = serde_json::to_string(val).unwrap();
+        let deserialized: ParamValue = serde_json::from_str(&json).unwrap();
+        assert_eq!(&deserialized, val);
+    }
+}
+
+#[test]
+fn direction_serde_round_trip() {
+    let min_json = serde_json::to_string(&Direction::Minimize).unwrap();
+    let max_json = serde_json::to_string(&Direction::Maximize).unwrap();
+
+    assert_eq!(
+        serde_json::from_str::<Direction>(&min_json).unwrap(),
+        Direction::Minimize
+    );
+    assert_eq!(
+        serde_json::from_str::<Direction>(&max_json).unwrap(),
+        Direction::Maximize
+    );
+}
+
+#[test]
+fn round_trip_preserves_trial_id_counter() {
+    let study: Study<f64> = Study::new(Direction::Minimize);
+    let x = FloatParam::new(0.0, 1.0);
+
+    study
+        .optimize(10, |trial| {
+            let v = x.suggest(trial)?;
+            Ok::<_, optimizer::Error>(v)
+        })
+        .unwrap();
+
+    let dir = tempdir();
+    let path = dir.join("counter.json");
+    study.save(&path).unwrap();
+
+    let loaded: Study<f64> = Study::load(&path).unwrap();
+
+    // Creating a new trial should use an ID >= 10
+    let trial = loaded.create_trial();
+    assert!(trial.id() >= 10);
+
+    std::fs::remove_dir_all(&dir).ok();
+}
+
+#[test]
+fn checkpoint_file_created_at_interval() {
+    let study: Study<f64> = Study::new(Direction::Minimize);
+    let x = FloatParam::new(-10.0, 10.0).name("x");
+
+    let dir = tempdir();
+    let checkpoint = dir.join("checkpoint.json");
+
+    study
+        .optimize_with_checkpoint(10, 3, &checkpoint, |trial| {
+            let v = x.suggest(trial)?;
+            Ok::<_, optimizer::Error>(v * v)
+        })
+        .unwrap();
+
+    // Checkpoint should exist (written at trials 3, 6, 9)
+    assert!(checkpoint.exists(), "checkpoint file was not created");
+
+    // Load it and verify it's valid
+    let loaded: Study<f64> = Study::load(&checkpoint).unwrap();
+    // Last checkpoint was at trial 9, so it should have 9 trials
+    assert_eq!(loaded.n_trials(), 9);
+
+    std::fs::remove_dir_all(&dir).ok();
+}
+
+#[test]
+fn checkpoint_overwrites_previous() {
+    let study: Study<f64> = Study::new(Direction::Minimize);
+    let x = FloatParam::new(0.0, 1.0);
+
+    let dir = tempdir();
+    let checkpoint = dir.join("checkpoint.json");
+
+    study
+        .optimize_with_checkpoint(6, 3, &checkpoint, |trial| {
+            let v = x.suggest(trial)?;
+            Ok::<_, optimizer::Error>(v)
+        })
+        .unwrap();
+
+    // The checkpoint at trial 6 should overwrite the one from trial 3
+    let loaded: Study<f64> = Study::load(&checkpoint).unwrap();
+    assert_eq!(loaded.n_trials(), 6);
+
+    std::fs::remove_dir_all(&dir).ok();
+}
+
+#[test]
+fn resume_from_checkpoint_continues_trial_ids() {
+    let study: Study<f64> = Study::new(Direction::Minimize);
+    let x = FloatParam::new(-5.0, 5.0).name("x");
+
+    let dir = tempdir();
+    let checkpoint = dir.join("resume.json");
+
+    // Run 10 trials with checkpointing
+    study
+        .optimize_with_checkpoint(10, 5, &checkpoint, |trial| {
+            let v = x.suggest(trial)?;
+            Ok::<_, optimizer::Error>(v * v)
+        })
+        .unwrap();
+
+    // Load and continue
+    let loaded: Study<f64> = Study::load(&checkpoint).unwrap();
+    assert_eq!(loaded.n_trials(), 10);
+
+    let remaining = 15 - loaded.n_trials();
+    loaded
+        .optimize(remaining, |trial| {
+            let v = x.suggest(trial)?;
+            Ok::<_, optimizer::Error>(v * v)
+        })
+        .unwrap();
+
+    assert_eq!(loaded.n_trials(), 15);
+
+    // Verify no duplicate trial IDs
+    let trials = loaded.trials();
+    let mut ids: Vec<u64> = trials.iter().map(|t| t.id).collect();
+    ids.sort_unstable();
+    ids.dedup();
+    assert_eq!(ids.len(), 15, "duplicate trial IDs found");
+
+    std::fs::remove_dir_all(&dir).ok();
+}
+
+#[test]
+fn atomic_write_no_temp_file_left_behind() {
+    let study: Study<f64> = Study::new(Direction::Minimize);
+    let x = FloatParam::new(0.0, 1.0);
+
+    let dir = tempdir();
+    let checkpoint = dir.join("atomic.json");
+
+    study
+        .optimize_with_checkpoint(3, 3, &checkpoint, |trial| {
+            let v = x.suggest(trial)?;
+            Ok::<_, optimizer::Error>(v)
+        })
+        .unwrap();
+
+    // The temp file should have been renamed, not left behind
+    let tmp_path = dir.join(".atomic.json.tmp");
+    assert!(!tmp_path.exists(), "temp file was not cleaned up");
+    assert!(checkpoint.exists(), "checkpoint file was not created");
+
+    std::fs::remove_dir_all(&dir).ok();
+}
+
+/// Helper to create a unique temporary directory.
+fn tempdir() -> std::path::PathBuf {
+    use std::sync::atomic::{AtomicU64, Ordering};
+    static COUNTER: AtomicU64 = AtomicU64::new(0);
+    let id = COUNTER.fetch_add(1, Ordering::Relaxed);
+    let dir =
+        std::env::temp_dir().join(format!("optimizer_serde_test_{}_{id}", std::process::id()));
+    std::fs::create_dir_all(&dir).unwrap();
+    dir
+}

tests/test_functions.rs

@@ -0,0 +1,45 @@
+#[path = "../benches/test_functions.rs"]
+mod test_functions;
+
+use test_functions::*;
+
+const TOL: f64 = 1e-10;
+
+#[test]
+fn sphere_at_optimum() {
+    assert!(sphere(&[0.0, 0.0]).abs() < TOL);
+    assert!(sphere(&[0.0; 10]).abs() < TOL);
+}
+
+#[test]
+fn rosenbrock_at_optimum() {
+    assert!(rosenbrock(&[1.0, 1.0]).abs() < TOL);
+    assert!(rosenbrock(&[1.0; 5]).abs() < TOL);
+}
+
+#[test]
+fn rastrigin_at_optimum() {
+    assert!(rastrigin(&[0.0, 0.0]).abs() < TOL);
+    assert!(rastrigin(&[0.0; 10]).abs() < TOL);
+}
+
+#[test]
+fn ackley_at_optimum() {
+    assert!(ackley(&[0.0, 0.0]).abs() < 1e-8);
+    assert!(ackley(&[0.0; 10]).abs() < 1e-8);
+}
+
+#[test]
+fn branin_at_optimum() {
+    let target = 0.397_887_357_729_738_1;
+    let val = branin(&[std::f64::consts::PI, 2.275]);
+    assert!((val - target).abs() < 1e-3);
+}
+
+#[test]
+fn hartmann6_at_optimum() {
+    let target = -3.3224;
+    let x_opt = [0.20169, 0.150011, 0.476874, 0.275332, 0.311652, 0.6573];
+    let val = hartmann6(&x_opt);
+    assert!((val - target).abs() < 0.01);
+}

tests/threshold_pruner_tests.rs

@@ -0,0 +1,64 @@
+use optimizer::pruner::{Pruner, ThresholdPruner};
+
+#[test]
+fn prune_when_value_exceeds_upper_threshold() {
+    let pruner = ThresholdPruner::new().upper(10.0);
+    let values = vec![(0, 5.0), (1, 8.0), (2, 11.0)];
+    assert!(pruner.should_prune(0, 2, &values, &[]));
+}
+
+#[test]
+fn prune_when_value_falls_below_lower_threshold() {
+    let pruner = ThresholdPruner::new().lower(0.0);
+    let values = vec![(0, 5.0), (1, 2.0), (2, -1.0)];
+    assert!(pruner.should_prune(0, 2, &values, &[]));
+}
+
+#[test]
+fn no_prune_when_value_within_bounds() {
+    let pruner = ThresholdPruner::new().upper(10.0).lower(0.0);
+    let values = vec![(0, 3.0), (1, 5.0), (2, 7.0)];
+    assert!(!pruner.should_prune(0, 2, &values, &[]));
+}
+
+#[test]
+fn no_prune_when_no_intermediate_values() {
+    let pruner = ThresholdPruner::new().upper(10.0).lower(0.0);
+    assert!(!pruner.should_prune(0, 0, &[], &[]));
+}
+
+#[test]
+fn works_with_only_upper_set() {
+    let pruner = ThresholdPruner::new().upper(5.0);
+    let below = vec![(0, 3.0)];
+    let above = vec![(0, 6.0)];
+    assert!(!pruner.should_prune(0, 0, &below, &[]));
+    assert!(pruner.should_prune(0, 0, &above, &[]));
+}
+
+#[test]
+fn works_with_only_lower_set() {
+    let pruner = ThresholdPruner::new().lower(2.0);
+    let above = vec![(0, 5.0)];
+    let below = vec![(0, 1.0)];
+    assert!(!pruner.should_prune(0, 0, &above, &[]));
+    assert!(pruner.should_prune(0, 0, &below, &[]));
+}
+
+#[test]
+fn works_with_both_thresholds_set() {
+    let pruner = ThresholdPruner::new().upper(10.0).lower(0.0);
+
+    // Within bounds
+    assert!(!pruner.should_prune(0, 0, &[(0, 5.0)], &[]));
+
+    // Exceeds upper
+    assert!(pruner.should_prune(0, 0, &[(0, 15.0)], &[]));
+
+    // Below lower
+    assert!(pruner.should_prune(0, 0, &[(0, -3.0)], &[]));
+
+    // At exact boundary (not pruned — strictly greater/less)
+    assert!(!pruner.should_prune(0, 0, &[(0, 10.0)], &[]));
+    assert!(!pruner.should_prune(0, 0, &[(0, 0.0)], &[]));
+}

tests/user_attr_tests.rs

@@ -0,0 +1,148 @@
+use optimizer::parameter::{FloatParam, Parameter};
+use optimizer::{AttrValue, Direction, Study};
+
+#[test]
+fn set_and_get_float_attr() {
+    let study: Study<f64> = Study::new(Direction::Minimize);
+    let x = FloatParam::new(0.0, 1.0);
+
+    study
+        .optimize(1, |trial| {
+            let _ = x.suggest(trial)?;
+            trial.set_user_attr("score", 42.5);
+            assert_eq!(trial.user_attr("score"), Some(&AttrValue::Float(42.5)));
+            Ok::<_, optimizer::Error>(1.0)
+        })
+        .unwrap();
+}
+
+#[test]
+fn set_and_get_int_attr() {
+    let study: Study<f64> = Study::new(Direction::Minimize);
+    let x = FloatParam::new(0.0, 1.0);
+
+    study
+        .optimize(1, |trial| {
+            let _ = x.suggest(trial)?;
+            trial.set_user_attr("epoch", 42_i64);
+            assert_eq!(trial.user_attr("epoch"), Some(&AttrValue::Int(42)));
+            Ok::<_, optimizer::Error>(1.0)
+        })
+        .unwrap();
+}
+
+#[test]
+fn set_and_get_string_attr() {
+    let study: Study<f64> = Study::new(Direction::Minimize);
+    let x = FloatParam::new(0.0, 1.0);
+
+    study
+        .optimize(1, |trial| {
+            let _ = x.suggest(trial)?;
+            trial.set_user_attr("model", "resnet50");
+            assert_eq!(
+                trial.user_attr("model"),
+                Some(&AttrValue::String("resnet50".to_owned()))
+            );
+            Ok::<_, optimizer::Error>(1.0)
+        })
+        .unwrap();
+}
+
+#[test]
+fn set_and_get_bool_attr() {
+    let study: Study<f64> = Study::new(Direction::Minimize);
+    let x = FloatParam::new(0.0, 1.0);
+
+    study
+        .optimize(1, |trial| {
+            let _ = x.suggest(trial)?;
+            trial.set_user_attr("converged", true);
+            assert_eq!(trial.user_attr("converged"), Some(&AttrValue::Bool(true)));
+            Ok::<_, optimizer::Error>(1.0)
+        })
+        .unwrap();
+}
+
+#[test]
+fn attrs_propagate_to_completed_trial() {
+    let study: Study<f64> = Study::new(Direction::Minimize);
+    let x = FloatParam::new(0.0, 1.0);
+
+    study
+        .optimize(1, |trial| {
+            let _ = x.suggest(trial)?;
+            trial.set_user_attr("time_secs", 1.5);
+            trial.set_user_attr("tag", "baseline");
+            Ok::<_, optimizer::Error>(1.0)
+        })
+        .unwrap();
+
+    let best = study.best_trial().unwrap();
+    assert_eq!(best.user_attr("time_secs"), Some(&AttrValue::Float(1.5)));
+    assert_eq!(
+        best.user_attr("tag"),
+        Some(&AttrValue::String("baseline".to_owned()))
+    );
+}
+
+#[test]
+fn overwrite_attr_replaces_value() {
+    let study: Study<f64> = Study::new(Direction::Minimize);
+    let x = FloatParam::new(0.0, 1.0);
+
+    study
+        .optimize(1, |trial| {
+            let _ = x.suggest(trial)?;
+            trial.set_user_attr("key", "old");
+            trial.set_user_attr("key", "new");
+            assert_eq!(
+                trial.user_attr("key"),
+                Some(&AttrValue::String("new".to_owned()))
+            );
+            Ok::<_, optimizer::Error>(1.0)
+        })
+        .unwrap();
+
+    let best = study.best_trial().unwrap();
+    assert_eq!(
+        best.user_attr("key"),
+        Some(&AttrValue::String("new".to_owned()))
+    );
+}
+
+#[test]
+fn missing_attr_returns_none() {
+    let study: Study<f64> = Study::new(Direction::Minimize);
+    let x = FloatParam::new(0.0, 1.0);
+
+    study
+        .optimize(1, |trial| {
+            let _ = x.suggest(trial)?;
+            assert_eq!(trial.user_attr("nonexistent"), None);
+            Ok::<_, optimizer::Error>(1.0)
+        })
+        .unwrap();
+
+    let best = study.best_trial().unwrap();
+    assert_eq!(best.user_attr("nonexistent"), None);
+}
+
+#[test]
+fn user_attrs_map_returns_all() {
+    let study: Study<f64> = Study::new(Direction::Minimize);
+    let x = FloatParam::new(0.0, 1.0);
+
+    study
+        .optimize(1, |trial| {
+            let _ = x.suggest(trial)?;
+            trial.set_user_attr("a", 1.0);
+            trial.set_user_attr("b", true);
+            assert_eq!(trial.user_attrs().len(), 2);
+            Ok::<_, optimizer::Error>(1.0)
+        })
+        .unwrap();
+
+    let best = study.best_trial().unwrap();
+    assert_eq!(best.user_attrs().len(), 2);
+}

tests/visualization_tests.rs

@@ -0,0 +1,180 @@
+use optimizer::parameter::{FloatParam, IntParam, Parameter};
+use optimizer::sampler::random::RandomSampler;
+use optimizer::{Direction, Study, generate_html_report};
+
+#[test]
+fn html_report_creates_file() {
+    let study: Study<f64> = Study::with_sampler(Direction::Minimize, RandomSampler::with_seed(42));
+    let x = FloatParam::new(0.0, 10.0).name("x");
+    let y = IntParam::new(1, 5).name("y");
+
+    study
+        .optimize(10, |trial| {
+            let xv = x.suggest(trial)?;
+            let yv = y.suggest(trial)?;
+            Ok::<_, optimizer::Error>(xv + yv as f64)
+        })
+        .unwrap();
+
+    let path = std::env::temp_dir().join("test_report_creates_file.html");
+    generate_html_report(&study, &path).unwrap();
+
+    let content = std::fs::read_to_string(&path).unwrap();
+    assert!(content.contains("<!DOCTYPE html>"));
+    assert!(content.contains("plotly"));
+    std::fs::remove_file(&path).ok();
+}
+
+#[test]
+fn html_report_contains_all_chart_sections() {
+    let study: Study<f64> = Study::with_sampler(Direction::Minimize, RandomSampler::with_seed(42));
+    let x = FloatParam::new(0.0, 10.0).name("x");
+    let y = FloatParam::new(-5.0, 5.0).name("y");
+
+    study
+        .optimize(20, |trial| {
+            let xv = x.suggest(trial)?;
+            let yv = y.suggest(trial)?;
+            Ok::<_, optimizer::Error>(xv * xv + yv * yv)
+        })
+        .unwrap();
+
+    let path = std::env::temp_dir().join("test_report_all_charts.html");
+    generate_html_report(&study, &path).unwrap();
+
+    let content = std::fs::read_to_string(&path).unwrap();
+
+    // Should contain all chart divs.
+    assert!(content.contains("id=\"history\""));
+    assert!(content.contains("id=\"slices\""));
+    assert!(content.contains("id=\"parcoords\""));
+    assert!(content.contains("id=\"importance\""));
+    assert!(content.contains("id=\"timeline\""));
+
+    // Should contain chart titles.
+    assert!(content.contains("Optimization History"));
+    assert!(content.contains("Slice Plots"));
+    assert!(content.contains("Parallel Coordinates"));
+    assert!(content.contains("Parameter Importance"));
+    assert!(content.contains("Trial Timeline"));
+
+    // Should show direction and trial count.
+    assert!(content.contains("Minimize"));
+    assert!(content.contains("20 trials"));
+
+    std::fs::remove_file(&path).ok();
+}
+
+#[test]
+fn html_report_empty_study() {
+    let study: Study<f64> = Study::new(Direction::Minimize);
+
+    let path = std::env::temp_dir().join("test_report_empty.html");
+    generate_html_report(&study, &path).unwrap();
+
+    let content = std::fs::read_to_string(&path).unwrap();
+    assert!(content.contains("<!DOCTYPE html>"));
+    assert!(content.contains("0 trials"));
+
+    std::fs::remove_file(&path).ok();
+}
+
+#[test]
+fn html_report_single_param_no_parcoords() {
+    let study: Study<f64> = Study::with_sampler(Direction::Minimize, RandomSampler::with_seed(42));
+    let x = FloatParam::new(0.0, 10.0).name("x");
+
+    study
+        .optimize(5, |trial| {
+            let xv = x.suggest(trial)?;
+            Ok::<_, optimizer::Error>(xv * xv)
+        })
+        .unwrap();
+
+    let path = std::env::temp_dir().join("test_report_single_param.html");
+    generate_html_report(&study, &path).unwrap();
+
+    let content = std::fs::read_to_string(&path).unwrap();
+
+    // Should have slice plot but not parallel coordinates (needs >= 2 params).
+    assert!(content.contains("id=\"slices\""));
+    assert!(!content.contains("id=\"parcoords\""));
+
+    std::fs::remove_file(&path).ok();
+}
+
+#[test]
+fn html_report_maximize_direction() {
+    let study: Study<f64> = Study::with_sampler(Direction::Maximize, RandomSampler::with_seed(42));
+    let x = FloatParam::new(0.0, 10.0).name("x");
+
+    study
+        .optimize(5, |trial| {
+            let xv = x.suggest(trial)?;
+            Ok::<_, optimizer::Error>(xv)
+        })
+        .unwrap();
+
+    let path = std::env::temp_dir().join("test_report_maximize.html");
+    generate_html_report(&study, &path).unwrap();
+
+    let content = std::fs::read_to_string(&path).unwrap();
+    assert!(content.contains("Maximize"));
+
+    std::fs::remove_file(&path).ok();
+}
+
+#[test]
+fn export_html_convenience_method() {
+    let study: Study<f64> = Study::with_sampler(Direction::Minimize, RandomSampler::with_seed(42));
+    let x = FloatParam::new(0.0, 10.0).name("x");
+
+    study
+        .optimize(5, |trial| {
+            let xv = x.suggest(trial)?;
+            Ok::<_, optimizer::Error>(xv * xv)
+        })
+        .unwrap();
+
+    let path = std::env::temp_dir().join("test_export_html.html");
+    study.export_html(&path).unwrap();
+
+    let content = std::fs::read_to_string(&path).unwrap();
+    assert!(content.contains("<!DOCTYPE html>"));
+    assert!(content.contains("id=\"history\""));
+
+    std::fs::remove_file(&path).ok();
+}
+
+#[test]
+fn html_report_with_intermediate_values() {
+    use optimizer::pruner::MedianPruner;
+
+    let mut study: Study<f64> =
+        Study::with_sampler(Direction::Minimize, RandomSampler::with_seed(42));
+    study.set_pruner(MedianPruner::new(Direction::Minimize));
+    let x = FloatParam::new(0.0, 10.0).name("x");
+
+    study
+        .optimize(10, |trial| {
+            let xv = x.suggest(trial)?;
+            for step in 0..5 {
+                let val = xv * xv + step as f64;
+                trial.report(step, val);
+                if trial.should_prune() {
+                    return Err(optimizer::TrialPruned.into());
+                }
+            }
+            Ok::<_, optimizer::Error>(xv * xv)
+        })
+        .unwrap();
+
+    let path = std::env::temp_dir().join("test_report_intermediate.html");
+    generate_html_report(&study, &path).unwrap();
+
+    let content = std::fs::read_to_string(&path).unwrap();
+    assert!(content.contains("id=\"intermediate\""));
+    assert!(content.contains("Intermediate Values"));
+
+    std::fs::remove_file(&path).ok();
+}