code: def amax(
      x: tf.Tensor,
      /,
      *,
      axis: Optional[Union[int, Sequence[int]]] = None,
      keepdims: bool = False,
      out: Optional[tf.Tensor] = None,
  ) -> tf.Tensor:
      axis = tuple(axis) if isinstance(axis, list) else axis
      return tf.experimental.numpy.amax(x, axis=axis, keepdims=keepdims)
  
imports: import operator
  from typing import Union, Optional, Tuple, List, Sequence
  from numbers import Number
  import tensorflow
  from tensorflow.python.ops.numpy_ops import np_math_ops
  import ivy
  from ivy import promote_types_of_inputs
  from ivy.func_wrapper import with_unsupported_dtypes, with_supported_dtypes
  from ..None import backend_version
label_desc: Use image node_id 1 for calling a global function with example usage: amax() and returns: tf