Migrate JAX workloads from pmap to jit #360
Triggered via pull request
August 18, 2025 18:50
priyakasimbeg
Status
Success
Total duration
4h 48m 22s
Artifacts
–
regression_tests.yml
on: pull_request
build_and_push_jax_docker_image
20m 34s
build_and_push_pytorch_docker_image
20m 1s
fastmri_jax
4m 14s
imagenet_resnet_jax
3m 45s
imagenet_vit_jax
4m 50s
ogbg_jax
8m 39s
criteo_jax
38m 47s
librispeech_conformer_jax
9m 16s
librispeech_deepspeech_jax
12m 13s
wmt_jax
1h 5m
fastmri_pytorch
6m 9s
imagenet_resnet_pytorch
10m 15s
imagenet_vit_pytorch
5m 34s
ogbg_pytorch
51s
criteo_pytorch
34m 12s
librispeech_conformer_pytorch
55s
librispeech_deepspeech_pytorch
1m 2s
wmt_pytorch
29m 31s