Skip to content
Containerized Regression Tests

Migrate JAX workloads from pmap to jit #359

Sign in to view logs

Migrate JAX workloads from pmap to jit

Migrate JAX workloads from pmap to jit #359

Run time

Learn about OS pricing on GitHub Actions
Job Run time
build_and_push_jax_docker_image
20m 42s
build_and_push_pytorch_docker_image
19m 56s
imagenet_resnet_pytorch
8m 41s
librispeech_conformer_pytorch
52s
fastmri_pytorch
4m 21s
criteo_pytorch
34m 53s
librispeech_deepspeech_pytorch
55s
imagenet_vit_pytorch
5m 52s
ogbg_pytorch
53s
wmt_pytorch
29m 27s
ogbg_jax
9m 9s
fastmri_jax
5m 56s
imagenet_resnet_jax
2m 52s
imagenet_vit_jax
4m 51s
librispeech_conformer_jax
9m 57s
librispeech_deepspeech_jax
13m 2s
criteo_jax
38m 38s
wmt_jax
1m 50s
wmt_jax
1m 47s
librispeech_deepspeech_jax
13m 2s
imagenet_vit_pytorch
5m 52s
ogbg_pytorch
53s
criteo_pytorch
34m 53s
fastmri_jax
5m 56s
criteo_jax
38m 38s
ogbg_jax
9m 9s
imagenet_vit_jax
4m 51s
build_and_push_pytorch_docker_image
19m 56s
librispeech_conformer_jax
9m 57s
imagenet_resnet_jax
2m 52s
librispeech_conformer_pytorch
52s
librispeech_deepspeech_pytorch
55s
fastmri_pytorch
4m 21s
imagenet_resnet_pytorch
8m 41s
wmt_pytorch
29m 27s
build_and_push_jax_docker_image
20m 42s
7h 5m 31s