Enable torch.compile for loss_fn #3430
CI.yml
on: pull_request
fastmri
4m 29s
wmt_jax
12m 35s
wmt_pytorch
13m 45s
imagenet_jax
5m 27s
imagenet_pytorch
2m 55s
criteo_jax
3m 8s
criteo_pytorch
3m 7s
speech_jax
6m 17s
speech_pytorch
3m 19s
ogbg
3m 26s
pytest
6m 49s
pytest-baselines
4m 13s