Back to pull request #790

Pass `train_state` to `update_params` #4443

Sign in to view logs

Run time

Learn about OS pricing on GitHub Actions

Job	Run time
criteo_pytorch	3m 5s
fastmri	4m 13s
imagenet_jax	5m 27s
criteo_jax	2m 50s
imagenet_pytorch	2m 57s
ogbg	3m 11s
pytest	7m 50s
pytest-baselines	4m 55s
speech_pytorch	3m 41s
speech_jax	6m 33s
wmt_jax	12m 29s
wmt_pytorch	13m 40s
	1h 10m 51s