Fast Fourier Transform (FFT) on GPU with Nx?

Questions & Help Questions
1

Hi!

Did anyone already implement FFT on Nx?

I came across various non-Nx solutions (including GitHub - brunosantanaa/fft: FFT - Fast Fourrier Transformer (Elixir Lang) · GitHub, stuff in Matrex etc).

Having implemented FFT in the past already, I could implement it myself on top of Nx but if someone already did the job somewhere, I am interested!

Thanks!

– Thibaut

2

Aaaand… I found the perfect link right after posting. Here it is in case someone needs this:

https://github.com/elixir-nx/nx_signal

With the neat companion Livebook:

https://github.com/elixir-nx/nx_signal/blob/main/guides/fft.livemd

Thanks @polvalente et al.!

– Thibaut