This helped.  What I did was removed the unused parameters from the gru function, as well as the get_params function (not shown here, but it’s the function that initializes the parameters using the normal distribution and sets the gradient to True on all of them).
Thank you.