From 846ef62573270a77cba9a3fd7c6e28fd2d571425 Mon Sep 17 00:00:00 2001 From: rajasekharporeddy Date: Fri, 27 Sep 2024 20:18:05 +0530 Subject: [PATCH] remove implements decorator --- jax/_src/numpy/ufuncs.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/jax/_src/numpy/ufuncs.py b/jax/_src/numpy/ufuncs.py index dba22c4f0979..5c3f8be63f38 100644 --- a/jax/_src/numpy/ufuncs.py +++ b/jax/_src/numpy/ufuncs.py @@ -873,7 +873,7 @@ def arcsinh(x: ArrayLike, /) -> Array: """ return lax.asinh(*promote_args_inexact('arcsinh', x)) -@implements(np.arccosh, module='numpy') + @jit def arccosh(x: ArrayLike, /) -> Array: r"""Calculate element-wise inverse of hyperbolic cosine of input.