aleximmer / Laplace

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

Add default Jacobians and individual gradients to curvature #105

Closed aleximmer closed 4 months ago

aleximmer commented 2 years ago

Jacobians can also be computed naively but for general models using either pytorch-functional or using loops. This should be an option if the layers cannot be extended using either backpack or asdfghjkl.

runame commented 4 months ago

Addressed in #146.