Skip to content

Pull requests: google/orbax

New pull request New
119 Open 1,140 Closed
119 Open 1,140 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 support to Layout using StandardCheckpointHandler
#1374 opened Nov 22, 2024 by copybara-service bot Loading…
Add shape property to ArrayRestoreArgs to be interchangeable with global_shape. This allows generic functions like to_shape_dtype_struct to reference it.
#1372 opened Nov 22, 2024 by copybara-service bot Loading…
ServingConfig.extra_trackable_resources is now supported in the Orbax Model path when support_tf_resources is True.
#1371 opened Nov 21, 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…
Add an option to disable saving root-level metadata in CheckpointManager.
#1368 opened Nov 21, 2024 by copybara-service bot Loading…
Internal
#1365 opened Nov 21, 2024 by copybara-service bot Loading…
[Experimental Feature] Support empty NamedTuple leaf when PyTreeMetadataOptions.support_rich_types=true.
#1362 opened Nov 20, 2024 by copybara-service bot Loading…
Test, this shoudl fail.
#1361 opened Nov 20, 2024 by copybara-service bot Loading…
[replica-parallel] Avoid jax.sharding.Sharding#devices_indices_map
#1358 opened Nov 20, 2024 by gspschmid Loading…
11
Internal
#1350 opened Nov 18, 2024 by copybara-service bot Loading…
[emergency checkpoint] Add ReplicatorCheckpointManager implementation for interoperating with replicator service provided by GKE (or theoretically, any other similar service).
#1339 opened Nov 15, 2024 by copybara-service bot Loading…
Add MultiProcessTest base.
#1328 opened Nov 13, 2024 by copybara-service bot Loading…
Resubmit single array error message
#1323 opened Nov 12, 2024 by cpgaffney1 Loading…
internal change
#1312 opened Nov 8, 2024 by copybara-service bot Loading…
Fix local restore by re-mapping device ids directly instead of inferring them from how process indexes changed across restarts with some false assumptions.
#1310 opened Nov 8, 2024 by copybara-service bot Loading…
1
Fix a bug in Orbax checkpointing where None values PyTree are not handled correctly. This is caused by a recent update in jax where None values are no longer considered as a leaf node: https://jax.readthedocs.io/en/latest/changelog.html#jax-0-4-34-october-4-2024.
#1286 opened Oct 31, 2024 by copybara-service bot Loading…
1
Internal change.
#1282 opened Oct 30, 2024 by copybara-service bot Loading…
Add user-facing TreeMetadata object returned by PyTreeCheckpointHandler.metadata. This object mimics an ordinary PyTree to make the change unnoticeable to most users, but also has additional accessible properties not included in any tree mapping operations.
#1276 opened Oct 28, 2024 by copybara-service bot Loading…
Add shutdown logging wrapper.
#1263 opened Oct 21, 2024 by copybara-service bot Loading…
Update fix:
#1260 opened Oct 21, 2024 by lamcodeofpwnosec Loading…
1
Internal.
#1233 opened Oct 8, 2024 by copybara-service bot Loading…
Internal change.
#1222 opened Oct 2, 2024 by copybara-service bot Loading…
1
No public description
#1206 opened Sep 25, 2024 by copybara-service bot Loading…
1
Bump ipython from 7.23.1 to 8.10.0 in /docs/requirements in the pip group across 1 directory dependencies Pull requests that update a dependency file
#1201 opened Sep 24, 2024 by dependabot bot Loading…
Add emergency checkpoint logging of arrays for debugging.
#1198 opened Sep 23, 2024 by copybara-service bot Loading…
ProTip! What’s not been updated in a month: updated:<2024-10-21.