code: def FeedForward(dim, mult=4):
      inner_dim = int(dim * mult)
      return nn.Sequential(
          LayerNorm(dim),
          nn.Linear(dim, inner_dim * 2, bias=False),
          SwiGLU(),
          nn.Linear(inner_dim, dim, bias=False),
      )
  
imports: import torch
  import torch.nn.functional
  from einops import rearrange
  from torch import matmul, nn
label_desc: Use image node_id 3 for calling a global function with example usage: FeedForward(dim, mult) and returns: nn