Add missing kwargs to optimize_scale #509

Job	Run time
build-docs (3.8)	3m 34s
build-mpi (3.9)	6h 0m 15s
build-jax (3.8)	1m 45s
build-mpi (3.10)	2m 51s
build-jax (3.9)	1m 1s
build-jax (3.10)	1m 4s
build-numpy (3.8, short)	1m 58s
build-torch (3.8)	2m 21s
build-numpy (3.8, optimize)	1m 52s
build-torch (3.9)	2m 11s
build-numpy (3.8, multivariate)	1m 33s
build-torch (3.10)	2m 7s
build-numpy (3.8, optimize-experimental)	2m 9s
build-numpy (3.9, short)	1m 27s
build-numpy (3.9, optimize)	50s
build-numpy (3.9, multivariate)	1m 22s
build-numpy (3.9, optimize-experimental)	50s
build-numpy (3.10, short)	1m 19s
build-numpy (3.10, optimize)	52s
build-numpy (3.10, multivariate)	1m 27s
build-numpy (3.10, optimize-experimental)	50s
	6h 33m 38s

Provide feedback