jax.lax.cos# jax.lax.cos(x)[source]# Elementwise cosine: \(\mathrm{cos}(x)\). Parameters: x (Union[Array, ndarray, bool_, number, bool, int, float, complex]) Return type: Array