pytorch

03fbb542 - [functorch] Very experimental PyTorch forward-mode AD support

[functorch] Very experimental PyTorch forward-mode AD support PyTorch forward mode AD doesn't support a lot of operations yet. I've verified that `jvp` can compose with `vmap`, but unfortunately `jvp` doesn't compose with itself.