normal-computing / posteriors

Uncertainty quantification with PyTorch
https://normal-computing.github.io/posteriors/
Apache License 2.0
282 stars 12 forks source link

Add KFAC #55

Open SamDuffield opened 2 months ago

SamDuffield commented 2 months ago

Would be awesome to add KFAC support within posteriors, would take some planning though as we'd like to implement it in a general purpose way and KFAC (unlike the other posteriors methods explicitly uses the layerwise structure of neural networks).