Skip to content

Pull requests: google/orbax

New pull request New
220 Open 1,984 Closed
220 Open 1,984 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

Simplify find_step_path logic by removing special casing for different filesystems - instead depending on TemporaryPath to identify uncommitted paths.
#2352 opened Sep 5, 2025 by copybara-service bot Loading…
Internal change
#2351 opened Sep 5, 2025 by copybara-service bot 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…
Use TfDataProcessor in OEX
#2337 opened Sep 3, 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
Remove old checkpoint restoration path in ArrayCheckpointHandler.
#2311 opened Aug 28, 2025 by copybara-service bot Loading…
1
Introduce DataProcessor base class and TfDataProcessor for Orbax Export.
#2309 opened Aug 27, 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
Create a function to convert FunctionInfo to ConcreteFunction.
#2294 opened Aug 23, 2025 by copybara-service bot Loading…
Stabilizing OBM proto -- consolidating device assignment and compile options
#2292 opened Aug 22, 2025 by copybara-service bot Loading…
1
Adding Orbax PreservationPolicy for checkpoint in GMAX PROD.
#2284 opened Aug 21, 2025 by copybara-service bot Loading…
Update CheckpointManagerOptions in API overview notebook.
#2271 opened Aug 18, 2025 by copybara-service bot Loading…
Add option to configure allow_split_physical_axes when setting up mesh using config.
#2269 opened Aug 18, 2025 by copybara-service bot Loading…
Enable shard_map for emergency checkpoint broadcast.
#2260 opened Aug 14, 2025 by copybara-service bot Loading…
1
Internal change
#2256 opened Aug 13, 2025 by copybara-service bot Loading…
Internal change
#2253 opened Aug 13, 2025 by copybara-service bot Loading…
Test multi-host runner
#2252 opened Aug 13, 2025 by quoctruong • Draft
Add Emergency Checkpoint Manager Benchmark
#2247 opened Aug 13, 2025 by copybara-service bot Loading…
Use SingleReplicaArrayHandler for restoring jax.Array in emergency checkpoint manager.
#2236 opened Aug 11, 2025 by copybara-service bot Loading…
1
Support Colocated Python Checkpointing
#2221 opened Aug 5, 2025 by copybara-service bot Loading…
ProTip! no:milestone will show everything without a milestone.