code: def Min(axis=-1, keepdims=False):
      """Returns a layer that applies min along one tensor axis.
  
      Args:
        axis: Axis along which values are grouped for computing minimum.
        keepdims: If `True`, keep the resulting size 1 axis as a separate tensor
            axis; else, remove that axis.
      """
      return Fn("Min", lambda x: jnp.min(x, axis, keepdims=keepdims))
  
imports: from absl import logging
  import numpy
  import tensorflow
  from trax import fastmath
  from trax.fastmath import numpy
  from trax.layers import base
  from trax.layers import initializers
  from trax.layers.assert_shape import assert_shape
  from trax.layers.base import Fn
label_desc: Use image node_id 8 for calling a global function with example usage: Min(axis, keepdims) and returns: Fn