Not only that, but FFT support on TPU has always been best effort. Last I tried this, there were serious precision issues.