BatchNorm fixes for JAX and PyTorch workloads #4428

Sign in to view logs

Run time

Learn about OS pricing on GitHub Actions

Job	Run time
fastmri	4m 7s
imagenet_jax	5m 52s
wmt_jax	12m 29s
wmt_pytorch	14m 0s
criteo_jax	2m 50s
criteo_pytorch	2m 43s
imagenet_pytorch	2m 50s
ogbg	3m 12s
pytest	7m 28s
pytest-baselines	5m 40s
speech_jax	6m 8s
speech_pytorch	3m 19s
	1h 10m 38s