Thanks, that seems to work pretty well. It seems like it’s possible to get a batch of jvps where there are multiple Jacobians, but only one vector, but when the vector is batched then the result is a batch of sums. Is that correct?
So, supposing that we have 3 vectors, this code will calculate a batch of sums of the 3 products?