CI

Pass `train_state` to `update_params` #4389

Sign in to view logs

Run time

Learn about OS pricing on GitHub Actions

Job	Run time
criteo_jax	2m 53s
criteo_pytorch	2m 46s
fastmri	4m 10s
imagenet_jax	5m 27s
imagenet_pytorch	2m 57s
speech_jax	6m 17s
wmt_jax	13m 15s
wmt_pytorch	14m 30s
ogbg	3m 30s
pytest	8m 6s
pytest-baselines	5m 12s
speech_pytorch	3m 41s
	1h 12m 44s