[WIP] Use the new grad interface.

I don't know if argsort should have a grad defined, so I mark it as not implemented.

Don't merge, I'll add other stuff in that PR.

NEWS.txt:

no new, make argsort use the new grad interface