Migrate JAX workloads from pmap to jit #358
priyakasimbegregression_tests.yml
on: pull_request
build_and_push_jax_docker_image
20m 16s
build_and_push_pytorch_docker_image
21m 18s
fastmri_jax
5m 28s
imagenet_resnet_jax
2m 57s
imagenet_vit_jax
4m 19s
ogbg_jax
8m 34s
criteo_jax
37m 16s
librispeech_conformer_jax
1m 16s
librispeech_deepspeech_jax
1m 51s
wmt_jax
40s
fastmri_pytorch
6m 3s
imagenet_resnet_pytorch
6m 6s
imagenet_vit_pytorch
3m 7s
ogbg_pytorch
51s
criteo_pytorch
33m 54s
librispeech_conformer_pytorch
50s
librispeech_deepspeech_pytorch
56s
wmt_pytorch
29m 36s