oops, this commit has the whole gradient