aleximmer / Laplace

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

Make `CurvatureInterface.functorch_jacobians` simply an alias to `CurvatureInterface.jacobians` #182

Closed wiseodd closed 3 weeks ago

wiseodd commented 2 months ago

Before, they were duplicates. The rationale is to make functorch_jacobians accessible even when the subclasses overloaded jacobians. Useful for Bayesian optimization since not all backend's jacobians are backpropable (e.g. ASDL detaches everything).

wiseodd commented 3 weeks ago

Let's forget about this due to #187. The next plan is to decouple the Hessian (for fit) and Jacobian (for GLM) computations.