Closed tingwl0122 closed 4 months ago
Will add regularization
to the function soon.
Hi @TheaperDeng , Regularization option added, should be good to merge.
@sleepymalc please take a look at the algorithm logic.
Just to give a bit more context, I follow the official implementation logic (it was written in JAX) and re-write it into PyTorch. (https://github.com/google-research/jax-influence/blob/main/jax_influence/arnoldi.py)
Description
Add one more ihvp method: Arnoldi that uses Arnoldi Iteration.
1. Motivation and Context
2. Summary of the change
3. What tests have been added/updated for the change?