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