Max

Return the maximum of an array or maximum along an axis.

Abstract Signature:

Max(a: Tensor, axis: int | tuple | None, keepdims: bool = False)

PyTorch

API: torch.max
Strategy: Direct Mapping
Official Docs ↗

JAX (Core)

API: jax.numpy.max
Strategy: Direct Mapping
Official Docs ↗

NumPy

API: numpy.max
Strategy: Direct Mapping
Official Docs ↗

Keras

API: keras.ops.max
Strategy: Direct Mapping
Official Docs ↗

TensorFlow

API: tf.math.reduce_max
Strategy: Direct Mapping
Official Docs ↗

Apple MLX

API: mlx.core.max
Strategy: Direct Mapping
Official Docs ↗

AMD RDNA

API: v_max_f32
Strategy: Direct Mapping
Official Docs ↗

Flax NNX

API: optax.tree.max
Strategy: Direct Mapping
Official Docs ↗