build

#v1 #safetensors Refactor `SafetensorsLayout` to eliminate unperforma… #43769

Job	Run time
multiprocess-checkpoint-benchmarks (Python 3.10, jax=0.6.0)	14m 3s
multiprocess-unit-tests (Python 3.11, jax=newest)	3m 15s
build-orbax-model (Python 3.10, jax=newest)	57s
build-orbax-model (Python 3.11, jax=newest)	51s
build-orbax-model (Python 3.10, jax=0.5.0)	53s
build-export (Python 3.10, jax=newest)	2m 11s
build-orbax-model (Python 3.12, jax=newest)	54s
build-export (Python 3.12, jax=newest)	2m 22s
build-checkpoint (Python 3.12, jax=newest)	4m 5s
build-export (Python 3.11, jax=newest)	2m 9s
build-checkpoint (Python 3.10, jax=newest)	4m 1s
build-checkpoint (Python 3.10, jax=0.6.0)	3m 51s
build-checkpoint (Python 3.11, jax=newest)	4m 14s
build-export (Python 3.10, jax=0.4.34)	2m 18s
	46m 4s

Provide feedback