Add `laplace.dense_ggn` and `laplace.diag_ggn`

normal-computing / posteriors

Uncertainty quantification with PyTorch

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

Apache License 2.0

282 stars 12 forks source link

Closed SamDuffield closed 2 months ago

SamDuffield commented 2 months ago

Using #63

SamDuffield commented 2 months ago

~I think laplace.diag_ggn is not efficient to calculate~

Now I think laplace.diag_ggn is efficient to calculate again 😅, with the help of an einsum

SamDuffield commented 2 months ago

diag_ggn util added in #81

SamDuffield commented 2 months ago

Added in #87