Everything is messed up with the gradients. I have no idea what is
happening.