Issues

Issues

• 🚨 Continuous Failure: multiprocess-tests-schedule-run

  google/orbax#3690 · opened on Oct 22, 2026

• JaxRandomKeyCheckpointHandler loses the PRNG implementation of typed keys

  google/orbax#3680 · opened on Oct 20, 2026
  1

• Awaitable-signal keys are shared across processes: per-process checkpoints raise FileExistsError for their own step's tmp directory (and hang with primary_host=None)

  google/orbax#3621 · opened on Oct 5, 2026
  1

• 🚨 Continuous Failure: multiprocess-tests-schedule-run

  google/orbax#3528 · opened on Aug 10, 2026

• Support mixed SingleReplicaArrayRestoreArgs / ArrayRestoreArgs batches in SingleReplicaArrayHandler.deserialize

  google/orbax#3527 · opened on Aug 10, 2026
  1

• Dedup device-mesh serialization in NamedShardingMetadata.to_serialized_string

  google/orbax#3494 · opened on Jul 27, 2026
  3

• Batch per-slice device_put in transfer_arrays_to_host to reduce D2H dispatch overhead

  google/orbax#3493 · opened on Jul 27, 2026
  2

• Is it possible to forbid reading symlinks from the checkpoint?

  google/orbax#3375 · opened on Jun 16, 2026
  1

• broadcast_one_replica_to_all deadlocks on first resume without barrier before distributed JIT compilation

  google/orbax#3181 · opened on May 8, 2026

• Checkpoint save fails with ENOENT during OCDBT atomic rename on TPU v5litepod

  google/orbax#2837 · opened on Feb 1, 2026
  1

• Enable saving (and loading) zero size arrays

  google/orbax#2587 · opened on Nov 15, 2025

• orbax uses logging.info directly and not the supplied log function, hence impossible to suppress logging w/o silencing the whole INFO log

  google/orbax#1951 · opened on May 21, 2025
  1

• ocp.tree.serialize_tree filtering logic for sequences with empty leaves

  google/orbax#1356 · opened on Nov 19, 2024
  1

• Custom TypeHandler and "No per-process OCDBT checkpoint subdirs" warning

  google/orbax#1326 · opened on Nov 13, 2024
  4

• Tensorstore spec configuration

  google/orbax#872 · opened on May 10, 2024
  7

• Strange behavior of saving sharded trainstate in GCP.

  google/orbax#660 · opened on Jan 11, 2024
  3

• Could you please provide the py.typed file for type checking?

  google/orbax#561 · opened on Oct 19, 2023

• Package(s) vendoring issues

  google/orbax#397 · opened on Jul 10, 2023
  4

• Is it possible to store checkpoints in an external storage such as S3?

  google/orbax#359 · opened on Jun 18, 2023
  4