CI

Pass `train_state` to `update_params` #4441

Sign in to view logs

Run time

Learn about OS pricing on GitHub Actions

Job	Run time
criteo_jax	2m 59s
criteo_pytorch	2m 55s
fastmri	4m 7s
imagenet_jax	5m 21s
ogbg	3m 23s
imagenet_pytorch	2m 58s
pytest	7m 31s
pytest-baselines	5m 5s
speech_jax	6m 53s
speech_pytorch	3m 24s
wmt_jax	12m 35s
wmt_pytorch	14m 1s
	1h 11m 12s