build

Separate Pathways tests in Orbax OSS build. #42755

Job	Run time
multiprocess-unit-tests (Python 3.11, jax=newest)	58s
multiprocess-checkpoint-benchmarks (Python 3.10, jax=0.6.0)	1m 3s
build-export (Python 3.11, jax=newest)	1m 7s
build-export (Python 3.10, jax=newest)	1m 7s
build-export (Python 3.12, jax=newest)	1m 7s
build-export (Python 3.10, jax=0.4.34)	1m 8s
build-checkpoint (Python 3.12, jax=newest)	1m 7s
build-orbax-model (Python 3.10, jax=0.5.0)	55s
build-orbax-model (Python 3.11, jax=newest)	53s
build-orbax-model (Python 3.12, jax=newest)	57s
build-orbax-model (Python 3.10, jax=newest)	57s
build-checkpoint (Python 3.10, jax=newest)	1m 8s
build-checkpoint (Python 3.11, jax=newest)	1m 8s
build-checkpoint (Python 3.10, jax=0.6.0)	1m 7s
	14m 42s

Provide feedback