-
Notifications
You must be signed in to change notification settings - Fork 71
Pull requests: google/orbax
Author
Label
Projects
Milestones
Reviews
Assignee
Sort
Pull requests list
Eliminate usages of
ParamInfo.path in favor of parent_dir / name.
#2517
opened Oct 24, 2025 by
copybara-service
bot
Loading…
Add utilities for working with local steps on Pathways.
#2516
opened Oct 24, 2025 by
copybara-service
bot
Loading…
Add enable_replica_parallel_separate_folder V1 API and benchmark
#2515
opened Oct 24, 2025 by
copybara-service
bot
Loading…
Add ability to convert ArrayRestoreArgs to CPU shardings and mesh for colocated python.
#2514
opened Oct 24, 2025 by
copybara-service
bot
Loading…
Add priority batch support - Add LowPriorityBatchOptions.
#2513
opened Oct 24, 2025 by
copybara-service
bot
Loading…
Add flag option to raise an error if Pathways client is initialized before _gemini_lost_callback is set.
#2512
opened Oct 24, 2025 by
copybara-service
bot
Loading…
Add priority batch support - create a validation method for allowed batch sizes
#2510
opened Oct 24, 2025 by
copybara-service
bot
Loading…
Allow unexpected keys in
restore_item during partial restoration.
#2508
opened Oct 23, 2025 by
copybara-service
bot
Loading…
Add a GitHub workflow for multiprocess checkpoint benchmarks and tests.
#2503
opened Oct 23, 2025 by
copybara-service
bot
Loading…
Get the pipeline_named_signature from obm functions.
#2501
opened Oct 22, 2025 by
copybara-service
bot
Loading…
Add a new field "name" to Obm.ShloTensorSpec
#2496
opened Oct 21, 2025 by
copybara-service
bot
Loading…
Add a multihost build job for Orbax Checkpoint.
#2480
opened Oct 15, 2025 by
copybara-service
bot
Loading…
Update docstring for
train_steps in EmergencyCheckpointManagerBenchmarkConfig.
#2474
opened Oct 14, 2025 by
copybara-service
bot
Loading…
Add multiprocess checkpoint benchmarks job.
#2471
opened Oct 13, 2025 by
copybara-service
bot
Loading…
Build jobs with a multiprocess checkpoint benchmark job.
#2458
opened Oct 9, 2025 by
copybara-service
bot
Loading…
Add guide for Orbax Export for PyTorch Users.
#2427
opened Sep 25, 2025 by
copybara-service
bot
Loading…
Refactor GCS Operations to Use TensorFlow's File System
#2426
opened Sep 25, 2025 by
deepikarajani24
•
Draft
Add additional support for TENSORSTORE_GCS_BACKEND environment variable
#2413
opened Sep 22, 2025 by
copybara-service
bot
Loading…
update flag parsing logic and support --flag=value format
#2410
opened Sep 19, 2025 by
copybara-service
bot
Loading…
#v1 Add
LeafHandler as a CheckpointableHandler, so that ordinary PyTree leaves can also be saved as individual checkpointables.
#2391
opened Sep 15, 2025 by
copybara-service
bot
Loading…
Previous Next
ProTip!
Add no:assignee to see everything that’s not assigned.