jax.lax.max# jax.lax.max(x, y)[source]# Elementwise maximum: \(\mathrm{max}(x, y)\) For complex numbers, uses a lexicographic comparison on the (real, imaginary) pairs. Parameters: x (jax.typing.ArrayLike) y (jax.typing.ArrayLike) Return type: Array
