jax.lax.rsqrt# jax.lax.rsqrt(x)[source]# Elementwise reciprocal square root: \(1 \over \sqrt{x}\). Parameters: x (jax.typing.ArrayLike) Return type: Array