[functorch] jvp x vjp testing (pytorch/functorch#343)
Failures are either:
- lack of PyTorch forward-mode AD support (mostly)
- efficient zero tensors errors
- CUDA asserts (really need to be investigated).
All of the problems should be reproducible on the pytorch/pytorch side.
