All pull requests

Pull requests

• Restore function-local NamedTuple metadata

  #3717·AryanRG opened Oct 30, 2026·5 of 8 tasks·

  ·

• Add multi-process tests for ScheduledContinuousCheckpointingPolicy.

  #3715·copybara-service[bot] opened Oct 30, 2026Bot·

  ·
  1

• Add ScheduledContinuousCheckpointingPolicy, which agrees on the next save step once per save.

  #3713·copybara-service[bot] opened Oct 30, 2026Bot·

  ·
  1

• Add checkpoint_managers.continuous_checkpointing_options() for continuous checkpointing.

  #3712·copybara-service[bot] opened Oct 30, 2026Bot·

  ·
  1

• Preserve checkpoint history after failed asynchronous saves

  #3711·salma2vec opened Oct 30, 2026·

  ·
  1

• Internal.

  #3709·copybara-service[bot] opened Oct 29, 2026Bot·

  ·

• Auto-select gcs_grpc driver for GCS Zonal Rapid buckets

  #3707·copybara-service[bot] opened Oct 29, 2026Bot·

  ·

• Prototype Mode 2 raw tiled checkpoint save and restore as-is

  #3706·copybara-service[bot] opened Oct 28, 2026Bot·

  ·

• Support async host untiling and barrier in pinned-host transfer

  #3705·copybara-service[bot] opened Oct 28, 2026Bot·

  ·

• Persist step metrics in CheckpointManager when best_fn is unset

  #3703·copybara-service[bot] opened Oct 28, 2026Bot·

  ·

• [Orbax benchmarks] Read per-host metric sidecars in parallel when writing the summary.

  #3697·copybara-service[bot] opened Oct 27, 2026Bot·

  ·

• For graph-based pipelines, include inputs read only by later nodes (e.g. the model) in the exported pipeline signature.

  #3695·copybara-service[bot] opened Oct 26, 2026Bot·

  ·
  1

• Honor the bound contracting dimension when reading quantized rhs sub-channels

  #3693·copybara-service[bot] opened Oct 25, 2026Bot·

  ·

• Internal Change

  #3692·copybara-service[bot] opened Oct 23, 2026Bot·

  ·

• Internal.

  #3687·copybara-service[bot] opened Oct 21, 2026Bot·

  ·

• Migrate weights_name from orchestration to manifest Function proto.

  #3686·copybara-service[bot] opened Oct 21, 2026Bot·

  ·

• In orbax.checkpoint.v1.training.Checkpointer, evaluate the save decision policy before deleting an existing checkpoint when overwrite=True and force=False, so skipped saves do not delete the existing checkpoint.

  #3685·copybara-service[bot] opened Oct 21, 2026Bot·

  ·
  1

• Isolate awaitable-signal keys per process or active-process subgroup when saving checkpoints with primary_host=None or active_processes in V0 and V1.

  #3683·copybara-service[bot] opened Oct 21, 2026Bot·

  ·
  1

• Preserve checkpoints when overwrite is skipped by the save policy

  #3682·salma2vec opened Oct 20, 2026·5 of 8 tasks·

  ·
  1

  1

• Preserve typed JAX key implementations in random-key checkpoints

  #3681·salma2vec opened Oct 20, 2026·5 of 8 tasks·

  ·
  1

  1

• Internal change

  #3670·copybara-service[bot] opened Oct 16, 2026Bot·

  ·

• Internal change

  #3669·copybara-service[bot] opened Oct 16, 2026Bot·

  ·

• Internal.

  #3663·copybara-service[bot] opened Oct 16, 2026Bot·

  ·

• Internal.

  #3662·copybara-service[bot] opened Oct 16, 2026Bot·

  ·

• Add Pathways and ROC dependency check for V0.

  #3656·copybara-service[bot] opened Oct 14, 2026Bot·

  ·