prompt: We are currently solving the following issue within our repository. Here is the issue text:
  --- BEGIN ISSUE ---
   ifft2
  
  
  --- END ISSUE ---
  
  Below are some code segments, each from a relevant file. One or more of these files may contain bugs.
  --- BEGIN FILES ---
  Path: `ivy/functional/frontends/jax/numpy/fft.py`
  Content:
  ```
  1 # local
  2 import ivy
  3 from ivy.functional.frontends.jax.func_wrapper import to_ivy_arrays_and_back
  4 from ivy.func_wrapper import with_unsupported_dtypes
  5 
  6 
  7 @to_ivy_arrays_and_back
  8 def fft(a, n=None, axis=-1, norm=None):
  9     if norm is None:
  10         norm = "backward"
  11     return ivy.fft(a, axis, norm=norm, n=n)
  12 
  13 
  14 @to_ivy_arrays_and_back
  15 def fft2(a, s=None, axes=(-2, -1), norm=None):
  16     if norm is None:
  17         norm = "backward"
  18     return ivy.array(ivy.fft2(a, s=s, dim=axes, norm=norm), dtype=ivy.dtype(a))
  19 
  20 
  21 @to_ivy_arrays_and_back
  22 @with_unsupported_dtypes({"2.4.2 and below": ("float16", "bfloat16")}, "paddle")
  23 def fftshift(x, axes=None, name=None):
  24     shape = x.shape
  25 
  26     if axes is None:
  27         axes = tuple(range(x.ndim))
  28         shifts = [(dim // 2) for dim in shape]
  29     elif isinstance(axes, int):
  30         shifts = shape[axes] // 2
  31     else:
  32         shifts = [shape[ax] // 2 for ax in axes]
  33 
  34     roll = ivy.roll(x, shifts, axis=axes)
  35 
  36     return roll
  37 
  38 
  39 @to_ivy_arrays_and_back
  40 def ifft(a, n=None, axis=-1, norm=None):
  41     if norm is None:
  42         norm = "backward"
  43     return ivy.ifft(a, axis, norm=norm, n=n)
  44
  ```
  --- END FILES ---
  
  Please first localize the bug based on the issue statement, and then generate a patch according to the `git diff` format fenced by three backticks.
  
  Here is an example:
  
  ```diff
  diff --git a/examples/server_async.py b/examples/server_async.py
  --- a/examples/server_async.py
  +++ b/examples/server_async.py
  @@ -313,4 +313,4 @@
   
   
   if __name__ == "__main__":
  -    asyncio.run(run_async_server("."), debug=True)
  +    asyncio.run(run_async_server(), debug=True)
  diff --git a/examples/server_sync.py b/examples/server_sync.py
  --- a/examples/server_sync.py
  +++ b/examples/server_sync.py
  @@ -313,5 +313,5 @@
   
   
   if __name__ == "__main__":
  -    server = run_sync_server(".")
  +    server = run_sync_server()
       server.shutdown()
  
  ```
  
golden_diff: diff --git a/ivy/functional/frontends/jax/numpy/fft.py b/ivy/functional/frontends/jax/numpy/fft.py
  --- a/ivy/functional/frontends/jax/numpy/fft.py
  +++ b/ivy/functional/frontends/jax/numpy/fft.py
  @@ -41,3 +41,10 @@
       if norm is None:
           norm = "backward"
       return ivy.ifft(a, axis, norm=norm, n=n)
  +
  +
  +@to_ivy_arrays_and_back
  +def ifft2(a, s=None, axes=(-2, -1), norm=None):
  +    if norm is None:
  +        norm = "backward"
  +    return ivy.array(ivy.ifft2(a, s=s, dim=axes, norm=norm), dtype=ivy.dtype(a))
  
verification_info: {"golden_diff": "diff --git a/ivy/functional/frontends/jax/numpy/fft.py b/ivy/functional/frontends/jax/numpy/fft.py\n--- a/ivy/functional/frontends/jax/numpy/fft.py\n+++ b/ivy/functional/frontends/jax/numpy/fft.py\n@@ -41,3 +41,10 @@\n     if norm is None:\n         norm = \"backward\"\n     return ivy.ifft(a, axis, norm=norm, n=n)\n+\n+\n+@to_ivy_arrays_and_back\n+def ifft2(a, s=None, axes=(-2, -1), norm=None):\n+    if norm is None:\n+        norm = \"backward\"\n+    return ivy.array(ivy.ifft2(a, s=s, dim=axes, norm=norm), dtype=ivy.dtype(a))\n", "issue": " ifft2\n\n", "before_files": [{"content": "# local\nimport ivy\nfrom ivy.functional.frontends.jax.func_wrapper import to_ivy_arrays_and_back\nfrom ivy.func_wrapper import with_unsupported_dtypes\n\n\n@to_ivy_arrays_and_back\ndef fft(a, n=None, axis=-1, norm=None):\n    if norm is None:\n        norm = \"backward\"\n    return ivy.fft(a, axis, norm=norm, n=n)\n\n\n@to_ivy_arrays_and_back\ndef fft2(a, s=None, axes=(-2, -1), norm=None):\n    if norm is None:\n        norm = \"backward\"\n    return ivy.array(ivy.fft2(a, s=s, dim=axes, norm=norm), dtype=ivy.dtype(a))\n\n\n@to_ivy_arrays_and_back\n@with_unsupported_dtypes({\"2.4.2 and below\": (\"float16\", \"bfloat16\")}, \"paddle\")\ndef fftshift(x, axes=None, name=None):\n    shape = x.shape\n\n    if axes is None:\n        axes = tuple(range(x.ndim))\n        shifts = [(dim // 2) for dim in shape]\n    elif isinstance(axes, int):\n        shifts = shape[axes] // 2\n    else:\n        shifts = [shape[ax] // 2 for ax in axes]\n\n    roll = ivy.roll(x, shifts, axis=axes)\n\n    return roll\n\n\n@to_ivy_arrays_and_back\ndef ifft(a, n=None, axis=-1, norm=None):\n    if norm is None:\n        norm = \"backward\"\n    return ivy.ifft(a, axis, norm=norm, n=n)\n", "path": "ivy/functional/frontends/jax/numpy/fft.py"}], "after_files": [{"content": "# local\nimport ivy\nfrom ivy.functional.frontends.jax.func_wrapper import to_ivy_arrays_and_back\nfrom ivy.func_wrapper import with_unsupported_dtypes\n\n\n@to_ivy_arrays_and_back\ndef fft(a, n=None, axis=-1, norm=None):\n    if norm is None:\n        norm = \"backward\"\n    return ivy.fft(a, axis, norm=norm, n=n)\n\n\n@to_ivy_arrays_and_back\ndef fft2(a, s=None, axes=(-2, -1), norm=None):\n    if norm is None:\n        norm = \"backward\"\n    return ivy.array(ivy.fft2(a, s=s, dim=axes, norm=norm), dtype=ivy.dtype(a))\n\n\n@to_ivy_arrays_and_back\n@with_unsupported_dtypes({\"2.4.2 and below\": (\"float16\", \"bfloat16\")}, \"paddle\")\ndef fftshift(x, axes=None, name=None):\n    shape = x.shape\n\n    if axes is None:\n        axes = tuple(range(x.ndim))\n        shifts = [(dim // 2) for dim in shape]\n    elif isinstance(axes, int):\n        shifts = shape[axes] // 2\n    else:\n        shifts = [shape[ax] // 2 for ax in axes]\n\n    roll = ivy.roll(x, shifts, axis=axes)\n\n    return roll\n\n\n@to_ivy_arrays_and_back\ndef ifft(a, n=None, axis=-1, norm=None):\n    if norm is None:\n        norm = \"backward\"\n    return ivy.ifft(a, axis, norm=norm, n=n)\n\n\n@to_ivy_arrays_and_back\ndef ifft2(a, s=None, axes=(-2, -1), norm=None):\n    if norm is None:\n        norm = \"backward\"\n    return ivy.array(ivy.ifft2(a, s=s, dim=axes, norm=norm), dtype=ivy.dtype(a))\n", "path": "ivy/functional/frontends/jax/numpy/fft.py"}]}