To resolve this, use fp32 or bf16 if your hardware supports it (TPU, Ampere GPUs or newer).