jax.lax.abs# jax.lax.abs(x)[source]# Elementwise absolute value: \(|x|\). Parameters: x (jax.typing.ArrayLike) Return type: Array