Skip to content

Pull requests: google/orbax

New pull request New
226 Open 1,997 Closed
226 Open 1,997 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

Fix partial saving nested keys bug. Add relevant tests.
#2372 opened Sep 10, 2025 by copybara-service bot Loading…
Bump minimum JAX version to v0.6.0
#2371 opened Sep 10, 2025 by copybara-service bot Loading…
Branch the tf related functions from oex to tf_data_processor module.
#2370 opened Sep 10, 2025 by copybara-service bot Loading…
[JAX] Add a literal array type distinct from NumPy arrays.
#2369 opened Sep 10, 2025 by copybara-service bot Loading…
Internal
#2367 opened Sep 10, 2025 by copybara-service bot Loading…
1
Update EveryNSteps preservation policy.
#2366 opened Sep 10, 2025 by copybara-service bot Loading…
#v1 Adjust leaf and abstract leaf definitions.
#2365 opened Sep 9, 2025 by copybara-service bot Loading…
#v1 Add guide for Orbax Checkpointing for PyTorch Users.
#2363 opened Sep 9, 2025 by copybara-service bot Loading…
1
Add BATCH_DOWN to BatchPaddingPolicy enum
#2362 opened Sep 9, 2025 by copybara-service bot Loading…
1
Migrate Tunelab to Orbax PreservationPolicy for checkpoint management.
#2359 opened Sep 9, 2025 by copybara-service bot Loading…
Add PreservationPolicy support to EmergencyCheckpointManager.
#2355 opened Sep 8, 2025 by copybara-service bot Loading…
1
access state from restored composite args after restore
#2353 opened Sep 8, 2025 by keshavb96 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
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
ProTip! Filter pull requests by the default branch with base:main.