-
Notifications
You must be signed in to change notification settings - Fork 59
Pull requests: google/orbax
Author
Label
Projects
Milestones
Reviews
Assignee
Sort
Pull requests list
Add CheckpointLayout protocol for various checkpoint formats to inherit from.
#2082
opened Jul 7, 2025 by
copybara-service
bot
Loading…
Add CheckpointLayout protocol for various checkpoint formats to inherit from.
#2081
opened Jul 7, 2025 by
copybara-service
bot
Loading…
V1 Make Serialization Compatibility support custom leaf types
#2079
opened Jul 7, 2025 by
copybara-service
bot
Loading…
V1: remove unnecessary await for folder creation.
#2078
opened Jul 7, 2025 by
copybara-service
bot
Loading…
Simplify is saving in progress by only tracking save status at primary host.
#2077
opened Jul 7, 2025 by
copybara-service
bot
Loading…
Remove
multiprocessing_options from TemporaryPath.
#2076
opened Jul 4, 2025 by
copybara-service
bot
Loading…
Make
TemporaryPath.create not multiprocessing-aware.
#2075
opened Jul 4, 2025 by
copybara-service
bot
Loading…
Add DescriptorHandler test. Also fix is_abstract_handleable method.
#2073
opened Jul 2, 2025 by
copybara-service
bot
Loading…
[OBM-Juno] Capture jax mesh in Genmedia exports
#2066
opened Jul 1, 2025 by
copybara-service
bot
Loading…
Add verbose logging to the saving and preservation policies.
#2055
opened Jun 29, 2025 by
copybara-service
bot
Loading…
[OBM] Adding device assignment by coords to OBM
#2045
opened Jun 25, 2025 by
copybara-service
bot
Loading…
Use
force=True when saving from CheckpointManager, since existence checks are already performed.
#2044
opened Jun 24, 2025 by
copybara-service
bot
Loading…
[#v1](https://critique.corp.google.com/search/tag:v1) Add guide for
Checkpointing & Exporting JAX Models package.
#2027
opened Jun 17, 2025 by
copybara-service
bot
Loading…
Previous Next
ProTip!
Exclude everything labeled
bug with -label:bug.