Skip to content

Pull requests: google/orbax

New pull request New
132 Open 1,281 Closed
132 Open 1,281 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
Sort
Sort by
Newest Oldest Most commented Least commented Recently updated Least recently updated Best match
Most reactions
👍 👎 😄 🎉 😕 ❤️ 🚀 👀

Pull requests list

Add random key support
#1534 opened Jan 23, 2025 by copybara-service bot Loading…
Add checkpoint/metadata/BUILD, checkpoint/multihost/BUILD, checkpoint/proto/testing/BUILD and checkpoint/serialization/BUILD.
#1532 opened Jan 23, 2025 by copybara-service bot Loading…
Allow users to enable continuous checkpointing in Gemax Prod. Note that this is an experimental feature and should not be adopted by users without explicit approval.
#1530 opened Jan 23, 2025 by copybara-service bot Loading…
Add step utility to get item name from item directory.
#1529 opened Jan 22, 2025 by copybara-service bot Loading…
Add CompositeCheckpointHandler._get_metadata_from_temporary_paths() for metadata access during saving (before finalize).
#1528 opened Jan 22, 2025 by copybara-service bot Loading…
Add ability to create directories asynchronously via signaling in Async Checkpointer.
#1526 opened Jan 22, 2025 by copybara-service bot Loading…
Support Async Directory Creation in Composite Checkpoint Handler
#1525 opened Jan 22, 2025 by copybara-service bot Loading…
Remove ThreadPoolExecutors from CheckpointHandlers and use futures.CommitFuture and futures.CommitFutureAwaitingContractedSignals that also support asynchronous directory creation.
#1523 opened Jan 22, 2025 by copybara-service bot Loading…
Add support for merging array groups across multiple devices, with uneven sharding between devices.
#1514 opened Jan 17, 2025 by copybara-service bot Loading…
Remove items_and_handlers-related CompositeCheckpointHandler code.
#1512 opened Jan 16, 2025 by copybara-service bot Loading…
Use RootMetadata and StepMetadata in CheckpointManager.
#1511 opened Jan 16, 2025 by copybara-service bot Loading…
Internal change.
#1505 opened Jan 15, 2025 by copybara-service bot Loading…
Refactor metadata de/serialization.
#1493 opened Jan 14, 2025 by copybara-service bot Loading…
Internal
#1464 opened Jan 6, 2025 by copybara-service bot Loading…
Add .gitignore
#1460 opened Jan 3, 2025 by hartikainen Loading…
Exclude BUILD files from flit builds
#1458 opened Jan 2, 2025 by hartikainen Loading…
1
my public commit msg
#1446 opened Dec 17, 2024 by copybara-service bot Loading…
1
Internal change.
#1433 opened Dec 11, 2024 by copybara-service bot Loading…
Add logs in consistent_restore_mesh to examine the invariant.
#1427 opened Dec 5, 2024 by copybara-service bot Loading…
1
Internal
#1417 opened Dec 2, 2024 by copybara-service bot Loading…
1
internal change.
#1412 opened Nov 30, 2024 by copybara-service bot Loading…
Internal
#1406 opened Nov 26, 2024 by copybara-service bot Loading…
Internal
#1393 opened Nov 25, 2024 by copybara-service bot Loading…
Internal
#1390 opened Nov 25, 2024 by copybara-service bot Loading…
Need to add because parent CL updates Orbax codebase.
#1369 opened Nov 21, 2024 by copybara-service bot Loading…
ProTip! no:milestone will show everything without a milestone.