fix: fix failing test case at jax backend (#27815)

Co-authored-by: NripeshN <[email protected]>
ivy-llc · Dec 31, 2023 · 4e2905c · 4e2905c
1 parent 980f9c6
commit 4e2905c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/ivy/functional/backends/jax/activations.py b/ivy/functional/backends/jax/activations.py
@@ -37,7 +37,7 @@ def leaky_relu(
 def relu(
     x: JaxArray, /, *, complex_mode="jax", out: Optional[JaxArray] = None
 ) -> JaxArray:
-    return jnp.maximum(x, 0)
+    return jax.nn.relu(x)
 
 
 def sigmoid(