search

aleximmer / Laplace

Laplace approximations for Deep Learning.
https://aleximmer.github.io/Laplace
MIT License
436 stars 63 forks source link

Add `KronDecomposed.diag()` feature #121

Closed aleximmer closed 1 year ago

aleximmer commented 1 year ago

Small utility feature to get per-weight diagonal posterior precision when using KronLaplace.

aleximmer commented 1 year ago

Thanks, replaced all ger() in the file, there were a few more.