Capture measurements with `jax.make_jaxpr`. #46588

Job	Run time
tests / setup-ci-load	1s
tests / jax-tests (1, 3.9) / jax-tests (1, 3.9)	1m 31s
tests / jax-tests (2, 3.9) / jax-tests (2, 3.9)	1m 31s
tests / jax-tests (3, 3.9) / jax-tests (3, 3.9)	1m 32s
tests / jax-tests (4, 3.9) / jax-tests (4, 3.9)	1m 31s
tests / jax-tests (5, 3.9) / jax-tests (5, 3.9)	1m 32s
tests / tf-tests (1, 3.9) / tf-tests (1, 3.9)	1m 19s
tests / tf-tests (2, 3.9) / tf-tests (2, 3.9)	1m 30s
tests / tf-tests (3, 3.9) / tf-tests (3, 3.9)	1m 31s
tests / autograd-tests (3.9) / autograd-tests (3.9)	1m 31s
tests / core-tests (1, 3.9) / core-tests (1, 3.9)	1m 31s
tests / core-tests (2, 3.9) / core-tests (2, 3.9)	1m 31s
tests / core-tests (3, 3.9) / core-tests (3, 3.9)	1m 30s
tests / core-tests (4, 3.9) / core-tests (4, 3.9)	1m 30s
tests / core-tests (5, 3.9) / core-tests (5, 3.9)	1m 30s
tests / all-interfaces-tests (3.9) / all-interfaces-tests (3.9)	1m 30s
tests / torch-tests (3.9) / torch-tests (3.9)	1m 31s
tests / data-tests (3.9) / data-tests (3.9)	1m 29s
tests / external-libraries-tests (3.9) / external-libraries-tests (3.9)	1m 30s
tests / gradients-tests (finite-diff, 3.9) / gradients-tests (finite-diff, 3.9)	1m 28s
tests / gradients-tests (param-shift, 3.9) / gradients-tests (param-shift, 3.9)	1m 30s
tests / qchem-tests (3.9) / qchem-tests (3.9)	1m 30s
tests / device-tests (default.qubit.legacy, None, 3.9) / device-tests (default.qubit.legacy, None, 3.9)	1m 30s
tests / qcut-tests (3.9) / qcut-tests (3.9)	1m 31s
tests / upload-to-codecov	1s
upload-stable-deps	1s
	34m 32s

Provide feedback