prompt: We are currently solving the following issue within our repository. Here is the issue text:
  --- BEGIN ISSUE ---
  beta
  
  
  --- 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/random.py`
  Content:
  ```
  1 # local
  2 import ivy
  3 from ivy.functional.frontends.jax.func_wrapper import (
  4     to_ivy_arrays_and_back,
  5     handle_jax_dtype,
  6 )
  7 
  8 
  9 @to_ivy_arrays_and_back
  10 def PRNGKey(seed):
  11     return ivy.array([0, seed % 4294967295 - (seed // 4294967295)], dtype=ivy.int64)
  12 
  13 
  14 @handle_jax_dtype
  15 @to_ivy_arrays_and_back
  16 def uniform(key, shape=(), dtype=None, minval=0.0, maxval=1.0):
  17     return ivy.random_uniform(
  18         low=minval, high=maxval, shape=shape, dtype=dtype, seed=ivy.to_scalar(key[1])
  19     )
  20 
  21 
  22 @handle_jax_dtype
  23 @to_ivy_arrays_and_back
  24 def normal(key, shape=(), dtype=None):
  25     return ivy.random_normal(shape=shape, dtype=dtype, seed=ivy.to_scalar(key[1]))
  26
  ```
  --- 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/random.py b/ivy/functional/frontends/jax/random.py
  --- a/ivy/functional/frontends/jax/random.py
  +++ b/ivy/functional/frontends/jax/random.py
  @@ -1,5 +1,6 @@
   # local
   import ivy
  +from ivy.func_wrapper import with_unsupported_dtypes
   from ivy.functional.frontends.jax.func_wrapper import (
       to_ivy_arrays_and_back,
       handle_jax_dtype,
  @@ -23,3 +24,24 @@
   @to_ivy_arrays_and_back
   def normal(key, shape=(), dtype=None):
       return ivy.random_normal(shape=shape, dtype=dtype, seed=ivy.to_scalar(key[1]))
  +
  +
  +def _get_seed(key):
  +    key1, key2 = int(key[0]), int(key[1])
  +    return ivy.to_scalar(int("".join(map(str, [key1, key2]))))
  +
  +
  +@handle_jax_dtype
  +@to_ivy_arrays_and_back
  +@with_unsupported_dtypes(
  +    {
  +        "0.3.14 and below": (
  +            "float16",
  +            "bfloat16",
  +        )
  +    },
  +    "jax",
  +)
  +def beta(key, a, b, shape=None, dtype=None):
  +    seed = _get_seed(key)
  +    return ivy.beta(a, b, shape=shape, dtype=dtype, seed=seed)
  
verification_info: {"golden_diff": "diff --git a/ivy/functional/frontends/jax/random.py b/ivy/functional/frontends/jax/random.py\n--- a/ivy/functional/frontends/jax/random.py\n+++ b/ivy/functional/frontends/jax/random.py\n@@ -1,5 +1,6 @@\n # local\n import ivy\n+from ivy.func_wrapper import with_unsupported_dtypes\n from ivy.functional.frontends.jax.func_wrapper import (\n     to_ivy_arrays_and_back,\n     handle_jax_dtype,\n@@ -23,3 +24,24 @@\n @to_ivy_arrays_and_back\n def normal(key, shape=(), dtype=None):\n     return ivy.random_normal(shape=shape, dtype=dtype, seed=ivy.to_scalar(key[1]))\n+\n+\n+def _get_seed(key):\n+    key1, key2 = int(key[0]), int(key[1])\n+    return ivy.to_scalar(int(\"\".join(map(str, [key1, key2]))))\n+\n+\n+@handle_jax_dtype\n+@to_ivy_arrays_and_back\n+@with_unsupported_dtypes(\n+    {\n+        \"0.3.14 and below\": (\n+            \"float16\",\n+            \"bfloat16\",\n+        )\n+    },\n+    \"jax\",\n+)\n+def beta(key, a, b, shape=None, dtype=None):\n+    seed = _get_seed(key)\n+    return ivy.beta(a, b, shape=shape, dtype=dtype, seed=seed)\n", "issue": "beta\n\n", "before_files": [{"content": "# local\nimport ivy\nfrom ivy.functional.frontends.jax.func_wrapper import (\n    to_ivy_arrays_and_back,\n    handle_jax_dtype,\n)\n\n\n@to_ivy_arrays_and_back\ndef PRNGKey(seed):\n    return ivy.array([0, seed % 4294967295 - (seed // 4294967295)], dtype=ivy.int64)\n\n\n@handle_jax_dtype\n@to_ivy_arrays_and_back\ndef uniform(key, shape=(), dtype=None, minval=0.0, maxval=1.0):\n    return ivy.random_uniform(\n        low=minval, high=maxval, shape=shape, dtype=dtype, seed=ivy.to_scalar(key[1])\n    )\n\n\n@handle_jax_dtype\n@to_ivy_arrays_and_back\ndef normal(key, shape=(), dtype=None):\n    return ivy.random_normal(shape=shape, dtype=dtype, seed=ivy.to_scalar(key[1]))\n", "path": "ivy/functional/frontends/jax/random.py"}], "after_files": [{"content": "# local\nimport ivy\nfrom ivy.func_wrapper import with_unsupported_dtypes\nfrom ivy.functional.frontends.jax.func_wrapper import (\n    to_ivy_arrays_and_back,\n    handle_jax_dtype,\n)\n\n\n@to_ivy_arrays_and_back\ndef PRNGKey(seed):\n    return ivy.array([0, seed % 4294967295 - (seed // 4294967295)], dtype=ivy.int64)\n\n\n@handle_jax_dtype\n@to_ivy_arrays_and_back\ndef uniform(key, shape=(), dtype=None, minval=0.0, maxval=1.0):\n    return ivy.random_uniform(\n        low=minval, high=maxval, shape=shape, dtype=dtype, seed=ivy.to_scalar(key[1])\n    )\n\n\n@handle_jax_dtype\n@to_ivy_arrays_and_back\ndef normal(key, shape=(), dtype=None):\n    return ivy.random_normal(shape=shape, dtype=dtype, seed=ivy.to_scalar(key[1]))\n\n\ndef _get_seed(key):\n    key1, key2 = int(key[0]), int(key[1])\n    return ivy.to_scalar(int(\"\".join(map(str, [key1, key2]))))\n\n\n@handle_jax_dtype\n@to_ivy_arrays_and_back\n@with_unsupported_dtypes(\n    {\n        \"0.3.14 and below\": (\n            \"float16\",\n            \"bfloat16\",\n        )\n    },\n    \"jax\",\n)\ndef beta(key, a, b, shape=None, dtype=None):\n    seed = _get_seed(key)\n    return ivy.beta(a, b, shape=shape, dtype=dtype, seed=seed)\n", "path": "ivy/functional/frontends/jax/random.py"}]}