Skip to content

Pull requests: google/orbax

New pull request New
209 Open 1,890 Closed
209 Open 1,890 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

Add SaveDecisionPolicy support to emergency checkpoint manager.
#2246 opened Aug 13, 2025 by copybara-service bot Loading…
1
Allow checkpoint loader to override use_ocdbt option
#2245 opened Aug 13, 2025 by copybara-service bot Loading…
1
removing XLA_COMPILE_OPTIONS from the supported jax2obm kwargs
#2244 opened Aug 12, 2025 by copybara-service bot Loading…
1
Emergency Checkpointer enable single slice restore from persistent
#2243 opened Aug 12, 2025 by copybara-service bot Loading…
1
Adjust module directives in checkpoint_manager.rst to use simpler namespaces.
#2242 opened Aug 12, 2025 by copybara-service bot Loading…
1
Update SafeTensors layout to support sharded loading, allow tensor placement from a Safetensors checkpoint onto a JAX mesh.
#2238 opened Aug 11, 2025 by copybara-service bot Loading…
Use SingleReplicaArrayHandler for restoring jax.Array in emergency checkpoint manager.
#2236 opened Aug 11, 2025 by copybara-service bot Loading…
1
Emergency Checkpointer enable single slice restore from persistent
#2235 opened Aug 9, 2025 by copybara-service bot Loading…
1
Tmp path detection via TemporaryPath
#2232 opened Aug 7, 2025 by copybara-service bot Loading…
Support Colocated Python Checkpointing
#2221 opened Aug 5, 2025 by copybara-service bot Loading…
Fix listing perf for gcs by completely avoiding the extra list calls
#2220 opened Aug 5, 2025 by copybara-service bot Loading…
1
Update Checkpointer.metadata to handle non-composite handlers.
#2219 opened Aug 5, 2025 by copybara-service bot Loading…
Internal change
#2209 opened Aug 1, 2025 by copybara-service bot Loading…
Cache serialized sharding string.
#2203 opened Jul 31, 2025 by copybara-service bot Loading…
1
#v1 Re-implement training.Checkpointer in terms of v1 free functions.
#2202 opened Jul 30, 2025 by copybara-service bot Loading…
Allow reading locally empty arrays in multi-controller jax
#2198 opened Jul 30, 2025 by copybara-service bot Loading…
Fix readthedocs build file. Add the oex_orchestration.proto to the proto compile function.
#2196 opened Jul 30, 2025 by copybara-service bot Loading…
1
Add partial_save parameter to CheckpointableHandler.save method.
#2182 opened Jul 28, 2025 by copybara-service bot Loading…
Adds a function to check the inputs have default values in orbax savedmodel.
#2174 opened Jul 25, 2025 by copybara-service bot Loading…
Internal.
#2169 opened Jul 25, 2025 by copybara-service bot Loading…
Adding selective e2e tests obm export tests
#2168 opened Jul 25, 2025 by copybara-service bot Loading…
1
[replica-parallel] Configurable limits for replica-parallel checkpoint saving
#2165 opened Jul 24, 2025 by rao-ashish Loading…
3
Typo in resume docsting in standard_checkpointer.py
#2161 opened Jul 24, 2025 by vfdev-5 Loading…
Internal change
#2160 opened Jul 24, 2025 by copybara-service bot Loading…
1
HNS checkpoint folder moved to _todelete_subdirectory for future deletion to avoid slow listing
#2156 opened Jul 23, 2025 by deepikarajani24 Loading…
ProTip! Type g i on any issue or pull request to go back to the issue listing page.