CI

Migrate JAX workloads from pmap to jit #5104

Sign in to view logs

Run time

Learn about OS pricing on GitHub Actions

Job	Run time
wmt_jax	11m 52s
fastmri	5m 20s
imagenet_jax	5m 10s
wmt_pytorch	14m 6s
speech_jax	5m 47s
ogbg	4m 8s
pytest-baselines	4m 40s
criteo_jax	3m 50s
speech_pytorch	4m 28s
pytest-params	7m 57s
criteo_pytorch	3m 54s
imagenet_pytorch	3m 52s
	1h 15m 4s