Tutorials
Advanced JAX Tutorials
Notes
Developer documentation
API documentation
jit
vmap
pmap
Warning
This page was created from a pull request.
jax.lax.
max
Elementwise maximum: \(\mathrm{max}(x, y)\)
For complex numbers, uses a lexicographic comparison on the (real, imaginary) pairs.
x (Any) –
Any
y (Any) –