Commit a596173

Orbax Authors

committed

Add multiprocess checkpoint benchmarks job.

PiperOrigin-RevId: 818718908

1 parent ff0b1f2 commit a596173Copy full SHA for a596173

5 files changed

+4322

-179

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

+4322

-179

lines changed

Comments

(0)