Skip to content

Pull requests: google/orbax

New pull request New
227 Open 2,079 Closed
227 Open 2,079 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

Build jobs with a multiprocess checkpoint benchmark job.
#2458 opened Oct 9, 2025 by copybara-service bot Loading…
1
should not cause any public changes. the old API and new API have no difference besides naming for this case.
#2456 opened Oct 8, 2025 by copybara-service bot Loading…
1
Internal change
#2452 opened Oct 6, 2025 by copybara-service bot Loading…
#v1 Allow PyTree checkpointables to be automatically identified in both pytree_metadata and load_pytree.
#2451 opened Oct 6, 2025 by copybara-service bot Loading…
Add CheckpointManager performance benchmarks.
#2450 opened Oct 6, 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…
Internal change.
#2388 opened Sep 15, 2025 by copybara-service bot Loading…
Internal adjustment to tree-verity reporting
#2386 opened Sep 15, 2025 by copybara-service bot Loading…
1
Fix _glob_step_paths if HNS is enabled
#2384 opened Sep 13, 2025 by dvruette Loading…
1
Migrate Tunelab to Orbax PreservationPolicy for checkpoint management.
#2376 opened Sep 12, 2025 by copybara-service bot Loading…
Make PyTreeCheckpointHandlerTest a single test file.
#2375 opened Sep 12, 2025 by copybara-service bot Loading…
access state from restored composite args after restore
#2353 opened Sep 8, 2025 by keshavb96 Loading…
Internal change
#2350 opened Sep 5, 2025 by copybara-service bot Loading…
Propagate batch_padding_policy through BatchOptions in JSV
#2346 opened Sep 4, 2025 by copybara-service bot Loading…
1
Internal change
#2343 opened Sep 4, 2025 by copybara-service bot Loading…
Update sidv2, yt_prod_v3 test data and re-enable the test.
#2335 opened Sep 3, 2025 by copybara-service bot Loading…
Add workflow to test running multihost benchmark workflow
#2328 opened Sep 2, 2025 by copybara-service bot Loading…
Adding Orbax PreservationPolicy for checkpoint in GMAX PROD.
#2327 opened Sep 1, 2025 by copybara-service bot Loading…
small changes
#2326 opened Sep 1, 2025 by copybara-service bot Loading…
1
Refactor some test utils to test save and restore operations in more scenarios
#2307 opened Aug 27, 2025 by copybara-service bot Loading…
1
ProTip! Updated in the last three days: updated:>2025-10-05.