optimization test overhaul #456

Job	Run time
build-mpi (3.9)	4m 7s
build-jax (3.8)	1m 36s
build-mpi (3.10)	1m 32s
build-numpy (3.8, short)	1m 16s
build-jax (3.9)	47s
build-docs (3.8)	3m 43s
build-numpy (3.8, optimize)	42s
build-jax (3.10)	43s
build-numpy (3.8, multivariate)	1m 20s
build-torch (3.8)	1m 59s
build-numpy (3.8, optimize-experimental)	45s
build-torch (3.9)	2m 10s
build-numpy (3.9, short)	1m 17s
build-torch (3.10)	2m 24s
build-numpy (3.9, optimize)	42s
build-numpy (3.9, multivariate)	1m 19s
build-numpy (3.9, optimize-experimental)	56s
build-numpy (3.10, short)	1m 17s
build-numpy (3.10, optimize)	46s
build-numpy (3.10, multivariate)	1m 18s
build-numpy (3.10, optimize-experimental)	42s
	31m 21s

Provide feedback