code: def make_layer_stack(layers=gin.REQUIRED, num_layers=6):
    
    return LayerStack([cls() for cls in layers] * num_layers)
docs: Configurable layer stack.
  
    Args:
      layers: a list of subclasses of TransformerLayer
      num_layers: an integer
    Returns:
      a LayerStack