[do not merge] Running regression tests for python upgrades #6

Job	Run time
build_and_push_jax_docker_image	12m 34s
build_and_push_pytorch_docker_image	1s
fastmri_pytorch	1s
criteo_pytorch	1s
imagenet_vit_pytorch	1s
imagenet_resnet_pytorch	1s
ogbg_pytorch	1s
librispeech_conformer_pytorch	1s
librispeech_deepspeech_pytorch	1s
wmt_pytorch	1s
fastmri_jax	1s
imagenet_vit_jax	1s
librispeech_deepspeech_jax	1s
imagenet_resnet_jax	1s
ogbg_jax	1s
wmt_jax	1s
criteo_jax	1s
librispeech_conformer_jax	1s
	12m 51s

Provide feedback