We do not currently support custom autograd functions, but it is something on our radar that we would like to do in the future. You can find more context in this issue 182.
It is also possible to replicate most of the behavior in custom autograd functions now via custom C++ operators.