jax.lax.fft

jax.lax.fft(x, fft_type, fft_lengths)[source]