Migrate JAX workloads from pmap to jit #5104
priyakasimbegCI.yml
on: pull_request
fastmri
5m 20s
wmt_jax
11m 52s
wmt_pytorch
14m 6s
imagenet_jax
5m 10s
imagenet_pytorch
3m 52s
criteo_jax
3m 50s
criteo_pytorch
3m 54s
speech_jax
5m 47s
speech_pytorch
4m 28s
ogbg
4m 8s
pytest-params
7m 57s
pytest-baselines
4m 40s