Use tf32 in pytorch #5266
CI.yml
on: push
fastmri
4m 53s
wmt_jax
12m 13s
wmt_pytorch
13m 3s
imagenet_jax
5m 53s
imagenet_pytorch
7m 0s
criteo_jax
4m 0s
criteo_pytorch
3m 58s
speech_jax
5m 47s
speech_pytorch
4m 29s
ogbg
4m 25s
pytest-params
8m 26s
pytest-baselines
5m 2s