code: def __init__(self, dim: int, hidden_dim: int, linear=nn.Linear):
      self.w1 = linear(dim, hidden_dim, bias=False)
      self.w2 = linear(hidden_dim, dim, bias=False)
      self.w3 = linear(dim, hidden_dim, bias=False)
  
imports: from typing import Tuple, Union, Optional, Dict
  from tinygrad import Tensor, Variable, TinyJit, dtypes, nn, Device
  from tinygrad.helpers import getenv
label_desc: Use image node_id 1 to create a new FeedForward object  with example: obj = FeedForward(dim, hidden_dim, linear)