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