-
Couldn't load subscription status.
- Fork 71
Pull requests: google/orbax
Author
Label
Projects
Milestones
Reviews
Assignee
Sort
Pull requests list
Add
PyTorchLayout for loading PyTorch checkpoints.
#2522
opened Oct 28, 2025 by
copybara-service
bot
Loading…
Add dispatcher support jax array handler.
#2519
opened Oct 27, 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 priority batch support - Add LowPriorityBatchOptions.
#2513
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…
Allow unexpected keys in
restore_item during partial restoration.
#2508
opened Oct 23, 2025 by
copybara-service
bot
Loading…
Add a GitHub workflow for multiprocess checkpoint benchmarks and tests.
#2503
opened Oct 23, 2025 by
copybara-service
bot
Loading…
Add a multihost build job for Orbax Checkpoint.
#2480
opened Oct 15, 2025 by
copybara-service
bot
Loading…
Update docstring for
train_steps in EmergencyCheckpointManagerBenchmarkConfig.
#2474
opened Oct 14, 2025 by
copybara-service
bot
Loading…
Add multiprocess checkpoint benchmarks job.
#2471
opened Oct 13, 2025 by
copybara-service
bot
Loading…
Build jobs with a multiprocess checkpoint benchmark job.
#2458
opened Oct 9, 2025 by
copybara-service
bot
Loading…
Add guide for Orbax Export for PyTorch Users.
#2427
opened Sep 25, 2025 by
copybara-service
bot
Loading…
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…
update flag parsing logic and support --flag=value format
#2410
opened Sep 19, 2025 by
copybara-service
bot
Loading…
#v1 Add
LeafHandler as a CheckpointableHandler, so that ordinary PyTree leaves can also be saved as individual checkpointables.
#2391
opened Sep 15, 2025 by
copybara-service
bot
Loading…
Internal adjustment to tree-verity reporting
#2386
opened Sep 15, 2025 by
copybara-service
bot
Loading…
Previous Next
ProTip!
Filter pull requests by the default branch with base:main.