build

Add benchmarks for P2P CheckpointManager. #42808

Job	Run time
multiprocess-unit-tests (Python 3.11, jax=newest)	9m 25s
multiprocess-checkpoint-benchmarks (Python 3.10, jax=0.6.0)	13m 2s
build-orbax-model (Python 3.11, jax=newest)	1m 0s
build-export (Python 3.10, jax=0.4.34)	2m 15s
build-export (Python 3.10, jax=newest)	2m 14s
build-checkpoint (Python 3.10, jax=0.6.0)	4m 16s
build-checkpoint (Python 3.11, jax=newest)	4m 3s
build-orbax-model (Python 3.10, jax=0.5.0)	53s
build-checkpoint (Python 3.10, jax=newest)	4m 16s
build-orbax-model (Python 3.12, jax=newest)	58s
build-export (Python 3.11, jax=newest)	2m 7s
build-export (Python 3.12, jax=newest)	2m 11s
build-orbax-model (Python 3.10, jax=newest)	1m 2s
build-checkpoint (Python 3.12, jax=newest)	4m 25s
	52m 7s

Provide feedback