Add Fft functions to Jax frontend

[fnhirwa]

Add Fft functions to Jax frontend

_

Please keep in mind that the proper way to link an issue to this list is to comment "- [ ] #issue_number" while the issue's title only includes the name of the function you've chosen.
_

• fft #15832
  fft2
  fftfreq #23662
  fftn #26302
  fftshift #20853
  hfft #26088
  ifft
  ifft #23141
  ifft2 #23585
  ifftn
  ifftshift #23801
  ifftshift #26672
  ihfft #26198
  irfft #26885
  irfft2
  irfftn
  irfftn #27289
  rfft #23800
  #228782
  rfftfreq #23671
  rfftn

The main file paths where these functions are likely to be added are:

ivy/functional/frontends/jax/numpy/fft.py
ivy_tests/test_ivy/test_frontends/test_jax/test_numpy/test_fft.py

[ManoBharathi93]

• fft
• fft2
  Started working on this two function.

[ManoBharathi93]

I have implemented this function but I don't understand why it's not passing the github workflow. can anyone help me with this ? I know the fft and ifft logic I can do it from scratch.

import jax.numpy.fft as jax_fft
import ivy.numpy as ivy_np

def fft(x):
    # convert input array to Jax array
    x_jax = ivy_np.asarray(x)
    # calculate FFT using Jax
    y_jax = jax_fft.fft(x_jax)
    # convert output array to Ivy array
    y = ivy_np.asarray(y_jax)
    return y

def ifft(x):
    # convert input array to Jax array
    x_jax = ivy_np.asarray(x)
    # calculate inverse FFT using Jax
    y_jax = jax_fft.ifft(x_jax)
    # convert output array to Ivy array
    y = ivy_np.asarray(y_jax)
    return y

[AbdullahSabry]

• fft2 #22162

[shovan777]

• rfftn #22262

[abdllahdev]

• ifftn #22389