Commit 8cd17dd

Orbax Authors

committed

Add multiprocess checkpoint benchmarks job.

PiperOrigin-RevId: 818718908

1 parent aa9a3d8 commit 8cd17ddCopy full SHA for 8cd17dd

5 files changed

+642

-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

+642

-178

lines changed

Comments

(0)