Make `*Laplace` respect the `dtype` of the base model

aleximmer / Laplace

Laplace approximations for Deep Learning.

https://aleximmer.github.io/Laplace

MIT License

472 stars 73 forks source link

Closed wiseodd closed 2 hours ago

wiseodd commented 2 months ago

Closes #246

wiseodd commented 2 months ago

This seems to be a larger endeavor than expected: not only *Laplace classes but also all util functions and backends need to be checked.

Marking it as a draft for now.

wiseodd commented 2 months ago

Should be good now!

wiseodd commented 2 days ago

@aleximmer could you please double check and merge?

aleximmer commented 2 hours ago

Thanks @wiseodd, LGTM!