Segment matmul from jax-md, Pallas TPU

Recorded result

Pass, 65 of 65 shapes. Port median 161.4 us against 208.4 us for the fallback

Recorded 2026-10-02

Execution conditions

• TPU v5 lite, 8 chips, JAX 0.11.2, libtpu 0.0.48
• 5 warm up runs and 30 timed runs
• Reduced precision on the production shape is caught and fails, the fallback passes

Historical result; a new workload requires compatibility checking and fresh Verify.

Use this as your starting point

Save the reference, describe your workload and set the metric that matters. Your project keeps the selection and its recorded conditions together.

A historical result is not a compatibility guarantee. Your configuration needs a fresh Verify run.

Evidence summary imported from the local registry on October 2, 2026. A summary is not a newly issued proof pack.