pytorch

c6b7ffab - [functorch] Custom Function prototype

[functorch] Custom Function prototype WIP, but works for reverse-mode autograd right now. H/T https://github.com/pytorch/pytorch/pull/62660 for half of the solution.