Skip to content

Pull requests: google/orbax

New pull request New
218 Open 1,944 Closed
218 Open 1,944 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

Introduce DataProcessor base class and TfDataProcessor for Orbax Export.
#2309 opened Aug 27, 2025 by copybara-service bot Loading…
1
Refactor some test utils to test save and restore operations in more scenarios
#2307 opened Aug 27, 2025 by copybara-service bot Loading…
1
Move temporary path detection into TemporaryPath class. This class serves as a natural place for encapsulating core temporary path detection logic.
#2305 opened Aug 26, 2025 by copybara-service bot Loading…
Add support for partial saving in CompositeHandler.
#2303 opened Aug 26, 2025 by copybara-service bot Loading…
Add support for partial saving in BasePyTreeCheckpointHandler.
#2302 opened Aug 26, 2025 by copybara-service bot Loading…
Internal
#2300 opened Aug 26, 2025 by copybara-service bot Loading…
1
Create a function to convert FunctionInfo to ConcreteFunction.
#2294 opened Aug 23, 2025 by copybara-service bot Loading…
Stabilizing OBM proto -- consolidating device assignment and compile options
#2292 opened Aug 22, 2025 by copybara-service bot Loading…
1
Expose TENSORSTORE_GCS_BACKEND with gcs as default and gcs_grpc as a high performance option
#2290 opened Aug 21, 2025 by copybara-service bot Loading…
Refactor ArrayCheckpointHandler to use BasePyTreeCheckpointHandler.
#2289 opened Aug 21, 2025 by copybara-service bot Loading…
1
Adding Orbax PreservationPolicy for checkpoint in GMAX PROD.
#2284 opened Aug 21, 2025 by copybara-service bot Loading…
Internal
#2281 opened Aug 20, 2025 by copybara-service bot Loading…
1
Update CheckpointManagerOptions in API overview notebook.
#2271 opened Aug 18, 2025 by copybara-service bot Loading…
Add option to configure allow_split_physical_axes when setting up mesh using config.
#2269 opened Aug 18, 2025 by copybara-service bot Loading…
Enable shard_map for emergency checkpoint broadcast.
#2260 opened Aug 14, 2025 by copybara-service bot Loading…
1
Update orbax MIME types.
#2259 opened Aug 13, 2025 by copybara-service bot Loading…
1
Internal change
#2256 opened Aug 13, 2025 by copybara-service bot Loading…
Internal change
#2253 opened Aug 13, 2025 by copybara-service bot Loading…
Test multi-host runner
#2252 opened Aug 13, 2025 by quoctruong • Draft
Add Emergency Checkpoint Manager Benchmark
#2247 opened Aug 13, 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
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
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
ProTip! no:milestone will show everything without a milestone.