code: def __init__(
      self,
      vocab_size,
      pred_hidden_size,
      enc_hidden_size,
      joint_hidden_size,
      dropout,
  ):
      self.dropout = dropout
  
      self.l1 = Linear(
          pred_hidden_size + enc_hidden_size, joint_hidden_size
      )
      self.l2 = Linear(joint_hidden_size, vocab_size)
  
imports: from tinygrad.tensor import Tensor
  from tinygrad.jit import TinyJit
  from tinygrad.nn import Linear, Embedding
  from tinygrad.helpers import fetch
  import numpy
  from pathlib import Path
label_desc: Use image node_id 1 to create a new Joint object  with example: obj = Joint(vocab_size, pred_hidden_size, enc_hidden_size, joint_hidden_size, dropout)