orbax icon indicating copy to clipboard operation
orbax copied to clipboard

Propagate batch_padding_policy through BatchOptions in JSV

Open copybara-service[bot] opened this issue 3 months ago • 1 comments

Propagate batch_padding_policy through BatchOptions in JSV

  • Propagate batch_padding_policy from BatchOptions -> BatchRunnerOpOptions -> QueueOptions.
  • Temporarily convert batch_padding_policy to string for propagation. Add TODOs to use BatchPaddingPolicy enum when BATCH_DOWN and MINIMIZE_TPU_COST_PER_REQUEST are supported.

copybara-service[bot] avatar Sep 04 '25 18:09 copybara-service[bot]

Thanks for your pull request! It looks like this may be your first contribution to a Google open source project. Before we can look at your pull request, you'll need to sign a Contributor License Agreement (CLA).

View this failed invocation of the CLA check for more information.

For the most up to date status, view the checks section at the bottom of the pull request.

google-cla[bot] avatar Sep 04 '25 18:09 google-cla[bot]