Skip to content

Pull requests: google/orbax

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

Adds a function to check the inputs have default values in orbax savedmodel.
#2174 opened Jul 25, 2025 by copybara-service bot Loading…
[OBM] Separating expected manifest from obm_export_test to testdata
#2173 opened Jul 25, 2025 by copybara-service bot Loading…
1
Add model surgery api preview to docs guide.
#2172 opened Jul 25, 2025 by copybara-service bot Loading…
Internal change.
#2171 opened Jul 25, 2025 by copybara-service bot Loading…
Modernize type annotations in atomicity modules.
#2170 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…
2
Internal change.
#2164 opened Jul 24, 2025 by copybara-service bot Loading…
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
Add custom LeafHandler documentation notebook.
#2158 opened Jul 23, 2025 by copybara-service bot Loading…
Refactor Snapshots to raise errors.
#2157 opened Jul 23, 2025 by copybara-service bot Loading…
HNS checkpoint folder moved to _todelete_subdirectory for future deletion to avoid slow listing
#2156 opened Jul 23, 2025 by deepikarajani24 Loading…
Internal change
#2155 opened Jul 23, 2025 by copybara-service bot Loading…
1
internal
#2154 opened Jul 23, 2025 by copybara-service bot Loading…
1
Internal change
#2151 opened Jul 22, 2025 by copybara-service bot Loading…
1
Copy Bf16Optimizaion implementation to Orbax.
#2149 opened Jul 22, 2025 by copybara-service bot Loading…
Small refactoring: Pass exactly what the function needs
#2146 opened Jul 21, 2025 by copybara-service bot Loading…
1
Move implementation details of Orbax V1 save_pytree and save_checkpointables into helper file.
#2127 opened Jul 17, 2025 by copybara-service bot Loading…
Fixes for safetensors import error from previous rollback.
#2125 opened Jul 17, 2025 by copybara-service bot Loading…
Add a new parameter background_thread_count to allow users to set the number of threads for ThreadedCheckpointDeleter.
#2116 opened Jul 16, 2025 by copybara-service bot Loading…
[REX] Enable ttl based deletion of step and substeps
#2114 opened Jul 16, 2025 by copybara-service bot Loading…
Add an unit test for oex_orchestration.create().
#2108 opened Jul 14, 2025 by copybara-service bot Loading…
1
Use Checkpoint Manager to benchmark in Orbax V0.
#2102 opened Jul 14, 2025 by copybara-service bot Loading…
ProTip! Updated in the last three days: updated:>2025-07-23.