Skip to content

Pull requests: google/orbax

New pull request New
224 Open 1,974 Closed
224 Open 1,974 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

Propagate batch_padding_policy through BatchOptions in JSV
#2346 opened Sep 4, 2025 by copybara-service bot Loading…
1
internal
#2345 opened Sep 4, 2025 by copybara-service bot Loading…
1
Add batch padding policy in obm export.
#2344 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 partial saving addition support for Orbax v1 free functions.
#2332 opened Sep 2, 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
Change all numberings to be sequential with no 10-spaced gaps. This is to prevent accidental use of previously allocated and then de-allocated numbers. Going forward only use max+1 for numbering.
#2322 opened Aug 29, 2025 by copybara-service bot Loading…
1
changing Orbax Model version file uniquely identifiable
#2320 opened Aug 29, 2025 by copybara-service bot Loading…
1
support multiple model_functions
#2314 opened Aug 28, 2025 by copybara-service bot Loading…
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
Update orbax MIME types.
#2259 opened Aug 13, 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…
ProTip! What’s not been updated in a month: updated:<2025-08-04.