Skip to content

Pull requests: google/orbax

New pull request New
241 Open 2,181 Closed
241 Open 2,181 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

Always write array metadata if self._array_metadata_store is not None.
#2580 opened Nov 13, 2025 by copybara-service bot Loading…
Use layout.Layout as Format before jax 0.6.2
#2579 opened Nov 13, 2025 by copybara-service bot Loading…
Fix prefix_step option to allow for arbitrary naming
#2578 opened Nov 12, 2025 by copybara-service bot Loading…
1
Add helper method to merge checkpoint metadata.
#2577 opened Nov 12, 2025 by copybara-service bot Loading…
Internal change.
#2576 opened Nov 12, 2025 by copybara-service bot Loading…
Update checkpoint manager benchmark to use PyTree args and add deletion.
#2575 opened Nov 12, 2025 by copybara-service bot Loading…
1
Internal Change
#2573 opened Nov 12, 2025 by copybara-service bot Loading…
[2/N] Move create_supplemental_info_closure to tf2obm.
#2569 opened Nov 11, 2025 by copybara-service bot Loading…
obm: Match TF output signature argument names.
#2565 opened Nov 10, 2025 by copybara-service bot Loading…
Remove remaining custom tree utility functions and use jax.tree_util instead.
#2563 opened Nov 10, 2025 by copybara-service bot Loading…
obm: Use original argument names for TF input signatures.
#2559 opened Nov 7, 2025 by copybara-service bot Loading…
Add checkpoint_manager_test.py to multiprocess tests.
#2547 opened Nov 5, 2025 by copybara-service bot Loading…
1
Disable multiprocess-checkpoint-benchmarks job due to limited chips
#2546 opened Nov 5, 2025 by copybara-service bot Loading…
1
Run pytree_checkpoint_handler_test in github actions
#2544 opened Nov 5, 2025 by copybara-service bot Loading…
1
Add PyTorchLayout for loading PyTorch checkpoints.
#2522 opened Oct 28, 2025 by copybara-service bot Loading…
Eliminate usages of ParamInfo.path in favor of parent_dir / name.
#2517 opened Oct 24, 2025 by copybara-service bot Loading…
Add utilities for working with local steps on Pathways.
#2516 opened Oct 24, 2025 by copybara-service bot Loading…
Add flag option to raise an error if Pathways client is initialized before _gemini_lost_callback is set.
#2512 opened Oct 24, 2025 by copybara-service bot Loading…
Add a multihost build job for Orbax Checkpoint.
#2480 opened Oct 15, 2025 by copybara-service bot Loading…
1
Update docstring for train_steps in EmergencyCheckpointManagerBenchmarkConfig.
#2474 opened Oct 14, 2025 by copybara-service bot Loading…
1
Add multiprocess checkpoint benchmarks job.
#2471 opened Oct 13, 2025 by copybara-service bot Loading…
1
Build jobs with a multiprocess checkpoint benchmark job.
#2458 opened Oct 9, 2025 by copybara-service bot Loading…
1
Add guide for Orbax Export for PyTorch Users.
#2427 opened Sep 25, 2025 by copybara-service bot Loading…
1
Refactor GCS Operations to Use TensorFlow's File System
#2426 opened Sep 25, 2025 by deepikarajani24 • Draft
Add additional support for TENSORSTORE_GCS_BACKEND environment variable
#2413 opened Sep 22, 2025 by copybara-service bot Loading…
1
ProTip! Type g p on any issue or pull request to go back to the pull request listing page.