AMax

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

Abstract Signature:

AMax(a: Tensor, axis: int, keepdims: bool = False)

PyTorch

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

JAX (Core)

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

NumPy

API: numpy.amax
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 ↗

Flax NNX

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

PaxML / Praxis

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