f-dangel / curvlinops

scipy linear operators for the Hessian, Fisher/GGN, and more in PyTorch
https://curvlinops.readthedocs.io/en/latest/
MIT License
17 stars 5 forks source link

Implement EKFAC #116

Open runame opened 4 months ago

runame commented 4 months ago

Implement Eigenvalue-corrected Kronecker Factorization [George et al. (2018)].