JaxBackend.max

JaxBackend.max(*args, **kwargs)

Compute the maximum of array elements.

Parameters:

arrBackendArray

Input data.

axisint or tuple of ints, optional

Axis or axes to perform the operation on. Default is all.

Returns:

Union[BackendArray, Scalar]

Maximum of arr.