added cos function in jax frontend

.gitignore

@@ -22,4 +22,4 @@ with_time_logs/
 .hypothesis
 .array_api_tests_k_flag*
 internal_automation_tools/
-.vscode/*
+.vscode/*

ivy/functional/frontends/jax/lax/operators.py

@@ -95,6 +95,13 @@ def atan(x):
 atan.unsupported_dtypes = {"torch": ("float16",)}
 
 
+def cos(x):
+    return ivy.cos(x)
+
+
+cos.unsupported_dtypes = {"torch": ("float16",)}
+
+
 def ceil(x):
     return ivy.ceil(x)
 

ivy_tests/test_ivy/test_frontends/test_jax/test_jax_lax_operators.py

@@ -1360,7 +1360,7 @@ def test_jax_lax_cosh(
     fw,
 ):
     input_dtype, x = dtype_and_x
-
+    
     helpers.test_frontend_function(
         input_dtypes=input_dtype,
         as_variable_flags=as_variable,
@@ -1503,3 +1503,34 @@ def test_jax_lax_gt(
         x=xs[0],
         y=xs[1],
     )
+
+
+# cos
+@handle_cmd_line_args
+@given(
+    dtype_and_x=helpers.dtype_and_values(available_dtypes=ivy_jax.valid_float_dtypes),
+    as_variable=st.booleans(),
+    num_positional_args=helpers.num_positional_args(
+        fn_name="ivy.functional.frontends.jax.lax.cos"
+    ),
+    native_array=st.booleans(),
+)
+def test_jax_lax_cos(
+    dtype_and_x,
+    as_variable,
+    num_positional_args,
+    native_array,
+    fw,
+):
+    input_dtype, x = dtype_and_x
+    helpers.test_frontend_function(
+        input_dtypes=input_dtype,
+        as_variable_flags=as_variable,
+        with_out=False,
+        num_positional_args=num_positional_args,
+        native_array_flags=native_array,
+        fw=fw,
+        frontend="jax",
+        fn_tree="lax.cos",
+        x=np.asarray(x, dtype=input_dtype),
+    )