-
Notifications
You must be signed in to change notification settings - Fork 71
Pull requests: google/orbax
Author
Label
Projects
Milestones
Reviews
Assignee
Sort
Pull requests list
Add use_zarr3 to TreeMetadata and access it from OrbaxSourceCheckpoint.
#2722
opened Dec 17, 2025 by
copybara-service
bot
Loading…
Add VoxelProcessor updates and save test for VoxelModules
#2719
opened Dec 16, 2025 by
copybara-service
bot
Loading…
#v1 Add a fresh implementation of
StringLeafHandler to reduce unnecessary delegation complexity. Also introduce a typestr function for LeafHandler.
#2713
opened Dec 16, 2025 by
copybara-service
bot
Loading…
Fix the logic of oex.utils.get_lowering_platforms() as it didn't lower the passed native_serialization_platforms.
#2699
opened Dec 12, 2025 by
copybara-service
bot
Loading…
#v1 Modify background thread execution to allow the runner to accept multiple concurrent tasks for the background event loop. This allows scheduling multiple operations, like async directory creation and background commits, on the same event loop.
#2686
opened Dec 10, 2025 by
copybara-service
bot
Loading…
Enforce topology check in StandardCheckpointHandler.restore
#2663
opened Dec 5, 2025 by
copybara-service
bot
Loading…
Remove v0 logic out of
OrbaxLayout which now only represents v1 checkpoints/layouts
#2662
opened Dec 4, 2025 by
copybara-service
bot
Loading…
Split v0 checkpoint format/layout logic out from OrbaxLayout and into seperate V0Layout
#2661
opened Dec 4, 2025 by
copybara-service
bot
Loading…
Add testing to verify backwards compatibility when missing _CHECKPOINT_METADATA or saving pytree directly to checkpoint.
#2660
opened Dec 4, 2025 by
copybara-service
bot
Loading…
Fix composite handler's extraction logic for handler type strings when missing _CHECKPOINT_METADATA
#2659
opened Dec 4, 2025 by
copybara-service
bot
Loading…
Previous Next
ProTip!
no:milestone will show everything without a milestone.