Migrate JAX workloads from pmap to jit #359
priyakasimbegregression_tests.yml
on: pull_request
build_and_push_jax_docker_image
20m 42s
build_and_push_pytorch_docker_image
19m 56s
fastmri_jax
5m 56s
imagenet_resnet_jax
2m 52s
imagenet_vit_jax
4m 51s
ogbg_jax
9m 9s
criteo_jax
38m 38s
librispeech_conformer_jax
9m 57s
librispeech_deepspeech_jax
13m 2s
wmt_jax
1m 47s
fastmri_pytorch
4m 21s
imagenet_resnet_pytorch
8m 41s
imagenet_vit_pytorch
5m 52s
ogbg_pytorch
53s
criteo_pytorch
34m 53s
librispeech_conformer_pytorch
52s
librispeech_deepspeech_pytorch
55s
wmt_pytorch
29m 27s