code: def update_gradients(self, grads):
          
          self.sigma2.gradient = grads[0]
          self.v.gradient = grads[1]
docs: Pull out the gradients, be careful as the order must match the order
          in which the parameters are added