Migrate JAX workloads from pmap to jit #5106
priyakasimbegCI.yml
on: pull_request
fastmri
5m 5s
wmt_jax
11m 58s
wmt_pytorch
13m 15s
imagenet_jax
5m 8s
imagenet_pytorch
3m 44s
criteo_jax
3m 48s
criteo_pytorch
3m 39s
speech_jax
5m 25s
speech_pytorch
4m 8s
ogbg
3m 59s
pytest-params
7m 25s
pytest-baselines
4m 18s