code: def FeedForwardBlock(
      d_model, d_ff, dropout, dropout_shared_axes, mode, activation
  ):
      # We copy the ff block function because we cannot import it from models
      return [
          core.Dense(d_ff),
          activation(),
          core.Dropout(
              rate=dropout, shared_axes=dropout_shared_axes, mode=mode
          ),
          core.Dense(d_model),
      ]
  
file_path: trax/trax/layers/research/resampling.py
imports: from trax.fastmath import numpy
  from trax.layers import combinators
  from trax.layers import core
  from trax.layers.normalization import LayerNorm
  from trax.layers.pooling import AvgPool
  from trax.layers.research.rel_attention import RelativeAttentionLMLayer
label_desc: Use image node_id 6 for calling a global function with example usage: FeedForwardBlock(d_model, d_ff, dropout, dropout_shared_axes, mode, activation) and returns: list