code: def __init__(self):
      self.x = Tensor(x_init.copy(), requires_grad=True)
      self.W = Tensor(W_init.copy(), requires_grad=True)
      self.m = Tensor(m_init.copy())
  
imports: import unittest
  import numpy
  import tensorflow
  import tensorflow_addons
  from tinygrad.tensor import Tensor
  from tinygrad.nn.optim import LAMB
label_desc: Use image node_id 1 to create a new TinyNet object  with example: obj = TinyNet()