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.