So far I've added ggnvp for vector products. It should be fairly straightforward to add ggn too now.
Might want to consider setting normalize=False and mention that normalize=True assumes that the first dimension of the output of the jvp is batchsize (or dimension or desired normalisation).
Discussed in #63
So far I've added
ggnvpfor vector products. It should be fairly straightforward to addggntoo now.Might want to consider setting
normalize=Falseand mention thatnormalize=Trueassumes that the first dimension of the output of thejvpis batchsize (or dimension or desired normalisation).