Add `diag_ggn` - Githubissues

normal-computing / posteriors

Uncertainty quantification with PyTorch

https://normal-computing.github.io/posteriors/

Apache License 2.0

314 stars 12 forks source link

Closed SamDuffield closed 5 months ago

SamDuffield commented 5 months ago

Add diag_ggn function to calculate the diagonal of a Generalized Gauss-Newton matrix
Add tests (mimicking ggn)