fix wmt jax #5105
priyakasimbegCI.yml
on: push
fastmri
4m 55s
wmt_jax
11m 48s
wmt_pytorch
13m 4s
imagenet_jax
5m 0s
imagenet_pytorch
3m 51s
criteo_jax
3m 40s
criteo_pytorch
3m 34s
speech_jax
5m 37s
speech_pytorch
4m 7s
ogbg
4m 13s
pytest-params
7m 34s
pytest-baselines
4m 26s