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 (ArrayLike) y (ArrayLike) Return type: Array