Migrate JAX workloads from pmap to jit #361
priyakasimbegregression_tests.yml
on: pull_request
build_and_push_jax_docker_image
20m 8s
build_and_push_pytorch_docker_image
19m 43s
fastmri_jax
5m 54s
imagenet_resnet_jax
4m 46s
imagenet_vit_jax
9m 25s
ogbg_jax
8m 43s
criteo_jax
40m 51s
librispeech_conformer_jax
9m 10s
librispeech_deepspeech_jax
11m 59s
wmt_jax
1h 5m
fastmri_pytorch
6m 3s
imagenet_resnet_pytorch
8m 34s
imagenet_vit_pytorch
14m 59s
ogbg_pytorch
3m 2s
criteo_pytorch
35m 56s
librispeech_conformer_pytorch
2m 21s
librispeech_deepspeech_pytorch
3m 56s
wmt_pytorch
29m 38s