code: def LayerNormSquash(mode, width=128):  # pylint: disable=invalid-name
      """Dense-LayerNorm-Tanh normalizer inspired by ACME."""
      # https://github.com/deepmind/acme/blob/master/acme/jax/networks/continuous.py#L34
      del mode
      return tl.Serial(
          [
              tl.Dense(width),
              tl.LayerNorm(),
              tl.Tanh(),
          ]
      )
  
file_path: trax/trax/rl/normalization.py
imports: import gin
  import numpy
  from trax import fastmath
  from trax import layers
label_desc: Use image node_id 10 for calling a global function with example usage: LayerNormSquash(mode, width) and returns: tl