Skip to content

Pull requests: google/orbax

New pull request New
238 Open 2,125 Closed
238 Open 2,125 Closed
Author
Filter by author
Loading
Label
Filter by label
Loading
Use alt + click/return to exclude labels
or ⇧ + click/return for logical OR
Projects
Filter by project
Loading
Milestones
Filter by milestone
Loading
Reviews
Filter by reviews
No reviews Review required Approved review Changes requested
Assignee
Filter by who’s assigned
Assigned to nobody Loading
Sort
Sort by
Newest Oldest Most commented Least commented Recently updated Least recently updated Best match
Most reactions
👍 👎 😄 🎉 😕 ❤️ 🚀 👀

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…
1
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…
1
Allow unexpected keys in restore_item during partial restoration.
#2508 opened Oct 23, 2025 by copybara-service bot Loading…
Internal change.
#2504 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…
1
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…
OBM tree_util: minor readability changes
#2495 opened Oct 21, 2025 by copybara-service bot Loading…
1
Internal change
#2487 opened Oct 17, 2025 by copybara-service bot Loading…
Add a multihost build job for Orbax Checkpoint.
#2480 opened Oct 15, 2025 by copybara-service bot Loading…
1
Update docstring for train_steps in EmergencyCheckpointManagerBenchmarkConfig.
#2474 opened Oct 14, 2025 by copybara-service bot Loading…
1
Add multiprocess checkpoint benchmarks job.
#2471 opened Oct 13, 2025 by copybara-service bot Loading…
1
Internal changes.
#2470 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…
1
Add guide for Orbax Export for PyTorch Users.
#2427 opened Sep 25, 2025 by copybara-service bot Loading…
1
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…
1
update flag parsing logic and support --flag=value format
#2410 opened Sep 19, 2025 by copybara-service bot Loading…
1
Revert to apple's patch for testing.
#2401 opened Sep 18, 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…
ProTip! Add no:assignee to see everything that’s not assigned.