build

Add `ArrayStorageOptions` to customize per-leaf saving behavior for a… #18268

Job	Run time
build-checkpoint (Python 3.10, jax=newest)	32s
build-checkpoint (Python 3.11, jax=newest)	37s
build-export (Python 3.10, jax=newest)	1m 34s
build-checkpoint (Python 3.12, jax=newest)	31s
build-export (Python 3.11, jax=newest)	1m 36s
build-checkpoint (Python 3.10, jax=0.5.0)	35s
build-export (Python 3.12, jax=newest)	1m 41s
build-checkpoint (Python 3.13, jax=nightly)	37s
build-export (Python 3.10, jax=0.4.34)	1m 37s
build-export (Python 3.12, jax=nightly)	1m 39s
build-orbax-model (Python 3.12, jax=newest)	1m 25s
	12m 24s

Provide feedback