I have a MWE below that computes a Fast Fourier Transform of an array and its inverse with tensorflow-gpu, just for me to get an idea of computation times. I am
tensorflow-gpu