Solved by albanD in post #2 
              
              
                Hi, 
As mentioned in the note on the function definition, the function uses the “double backward trick” to compute the jvp and is thus expected to be slower. 
We are working on changing that for 1.7.