add mixed precision training #5245
rka97CI.yml
on: push
fastmri
5m 43s
wmt_jax
12m 45s
wmt_pytorch
13m 22s
imagenet_jax
5m 39s
imagenet_pytorch
4m 37s
criteo_jax
4m 27s
criteo_pytorch
3m 54s
speech_jax
5m 53s
speech_pytorch
4m 28s
ogbg
4m 13s
pytest-params
7m 54s
pytest-baselines
4m 54s