search

aleximmer / Laplace

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

Add fast computation of functional_variance for DiagLLLaplace and KronLLLaplace #145

Closed wiseodd closed 1 week ago

wiseodd commented 4 months ago

PR for #138. Very useful for LLMs or diffusion models or any models with many outputs.

TODO: Implementation for KronLLLaplace. I'd like input from @aleximmer, who's the author of matrix.py.