Commit 9f0ba06

Orbax Authors

committed

Add multiprocess checkpoint benchmarks job.

PiperOrigin-RevId: 818718908

1 parent ff0b1f2 commit 9f0ba06Copy full SHA for 9f0ba06

5 files changed

+4324

-178

lines changed

.github/workflows
- build.yml
checkpoint/orbax/checkpoint
- _src
  - handlers
    - array_checkpoint_handler_test.py
  - multihost
    - multihost.py
  - testing
    - multiprocess_test.py
- checkpoint_manager_test.py

5 files changed

+4324

-178

lines changed

Comments

(0)