Support for structured Kronecker factors

aleximmer / Laplace

Laplace approximations for Deep Learning.

https://aleximmer.github.io/Laplace

MIT License

458 stars 71 forks source link

Open wiseodd opened 5 months ago

wiseodd commented 5 months ago

[ ] Add support for SINGD baseline. This enables structured Kronecker factors, e.g. diag & full, sparse, etc.
[ ] Update KronLaplace and matrix.py to accommodate the above.

wiseodd commented 5 months ago